server: shutdown endpoint (#466)

fixes #390
This commit is contained in:
C Cirello
2016-12-29 20:19:42 +01:00
committed by GitHub
parent c2c2a0fb06
commit e3c85d3e7e
5 changed files with 42 additions and 3 deletions

View File

@@ -33,7 +33,7 @@ type Server struct {
singleflight singleflight // singleflight assists Datastore
}
func New(ctx context.Context, ds models.Datastore, mq models.MessageQueue, r *runner.Runner, tasks chan task.Request, enqueue models.Enqueue) *Server {
func New(ctx context.Context, ds models.Datastore, mq models.MessageQueue, r *runner.Runner, tasks chan task.Request, enqueue models.Enqueue, opts ...ServerOption) *Server {
s := &Server{
Runner: r,
Router: gin.New(),
@@ -45,6 +45,10 @@ func New(ctx context.Context, ds models.Datastore, mq models.MessageQueue, r *ru
s.Router.Use(prepareMiddleware(ctx))
for _, opt := range opts {
opt(s)
}
return s
}