internal/server/server.go (view raw)
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 | package server
import (
"context"
"net"
"net/http"
"strconv"
"time"
"alin.ovh/searchix/internal/config"
"alin.ovh/searchix/internal/index"
"alin.ovh/searchix/internal/manpages"
"alin.ovh/x/log"
"github.com/Southclaws/fault"
"github.com/Southclaws/fault/fmsg"
"golang.org/x/net/http2"
"golang.org/x/net/http2/h2c"
)
type Server struct {
cfg *config.Config
log *log.Logger
server *http.Server
listener net.Listener
}
type Options struct {
ReadIndex *index.ReadIndex
ManpagesURLMap *manpages.URLMap
}
func New(
conf *config.Config,
options *Options,
log *log.Logger,
liveReload bool,
) (*Server, error) {
mux, err := NewMux(conf, options, log, liveReload)
if err != nil {
return nil, err
}
return &Server{
cfg: conf,
log: log,
server: &http.Server{
Handler: http.MaxBytesHandler(
h2c.NewHandler(mux, &http2.Server{
IdleTimeout: 5 * time.Minute,
}),
1024*1024,
),
ReadHeaderTimeout: 20 * time.Second,
},
}, nil
}
func (s *Server) Start(ctx context.Context) error {
listenAddress := net.JoinHostPort(s.cfg.Web.ListenAddress, strconv.Itoa(s.cfg.Web.Port))
lc := net.ListenConfig{
KeepAlive: 5 * time.Minute,
}
l, err := lc.Listen(ctx, "tcp", listenAddress)
if err != nil {
return fault.Wrap(
err, fmsg.Withf("could not listen on address %s and port %d",
s.cfg.Web.ListenAddress,
s.cfg.Web.Port,
))
}
s.listener = l
if s.cfg.Web.Environment == "development" {
s.log.Info(
"server listening on",
"url",
s.cfg.Web.BaseURL.String(),
)
}
if err := s.server.Serve(l); err != nil && err != http.ErrServerClosed {
return fault.Wrap(err, fmsg.With("could not start server"))
}
return nil
}
func (s *Server) Stop() chan struct{} {
s.log.Debug("stop called")
idleConnsClosed := make(chan struct{})
go func() {
s.log.Debug("shutting down server")
ctx, cancel := context.WithTimeout(
context.Background(),
s.cfg.Web.GracefulShutdownTimeout.Duration,
)
err := s.server.Shutdown(ctx)
cancel()
s.log.Debug("server shut down")
if err != nil {
// Error from closing listeners, or context timeout:
s.log.Error("error shutting down server", "error", err)
}
close(idleConnsClosed)
}()
return idleConnsClosed
}
|