3455f491ca
This commit adds and passes context around to (almost) every function. This allows cancelling (almost) every function call globally.
68 lines
1.7 KiB
Go
68 lines
1.7 KiB
Go
package cmd
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
wisski_distillery "github.com/FAU-CDI/wisski-distillery"
|
|
"github.com/FAU-CDI/wisski-distillery/internal/cli"
|
|
"github.com/FAU-CDI/wisski-distillery/pkg/cancel"
|
|
"github.com/tkw1536/goprogram/exit"
|
|
)
|
|
|
|
// Server is the 'server' command
|
|
var Server wisski_distillery.Command = server{}
|
|
|
|
type server struct {
|
|
Prefix string `short:"p" long:"prefix" description:"prefix to listen under"`
|
|
Bind string `short:"b" long:"bind" description:"address to listen on" default:"127.0.0.1:8888"`
|
|
}
|
|
|
|
func (s server) Description() wisski_distillery.Description {
|
|
return wisski_distillery.Description{
|
|
Requirements: cli.Requirements{
|
|
NeedsDistillery: true,
|
|
},
|
|
Command: "server",
|
|
Description: "Starts a server with information about this distillery",
|
|
}
|
|
}
|
|
|
|
var errServerListen = exit.Error{
|
|
ExitCode: exit.ExitGeneric,
|
|
Message: "Unable to listen",
|
|
}
|
|
|
|
func (s server) Run(context wisski_distillery.Context) error {
|
|
dis := context.Environment
|
|
handler, err := dis.Control().Server(context.Context, context.IOStream)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
context.Printf("Listening on %s\n", s.Bind)
|
|
|
|
// make a new listener
|
|
listener, err := dis.Still.Environment.Listen("tcp", s.Bind)
|
|
if err != nil {
|
|
return errServerListen.Wrap(err)
|
|
}
|
|
|
|
go func() {
|
|
<-context.Context.Done()
|
|
listener.Close()
|
|
}()
|
|
|
|
server := http.Server{
|
|
Handler: http.StripPrefix(s.Prefix, handler),
|
|
}
|
|
|
|
err, _ = cancel.WithContext(context.Context, func(start func()) error {
|
|
start()
|
|
return server.Serve(listener)
|
|
}, func() {
|
|
// gracefully shutdown server
|
|
context.Printf("shutting down server")
|
|
server.Shutdown(context.Context)
|
|
})
|
|
return errServerListen.Wrap(err)
|
|
}
|