From 375654659ce828de4c24035c023c9ab551a37410 Mon Sep 17 00:00:00 2001 From: Mike Crute Date: Wed, 17 Nov 2021 15:47:51 -0800 Subject: Add version to headers --- cmd/web/server.go | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) (limited to 'cmd') diff --git a/cmd/web/server.go b/cmd/web/server.go index e76b6b2..d13cd58 100644 --- a/cmd/web/server.go +++ b/cmd/web/server.go @@ -25,12 +25,12 @@ import ( "golang.org/x/time/rate" ) -func Register(root *cobra.Command, embeddedTemplates fs.FS) { +func Register(root *cobra.Command, embeddedTemplates fs.FS, version string) { webCmd := &cobra.Command{ Use: "web [options]", Short: "Run web server", Run: func(c *cobra.Command, args []string) { - webMain(app.NewConfigFromCmd(c), embeddedTemplates) + webMain(app.NewConfigFromCmd(c), embeddedTemplates, version) }, } @@ -53,7 +53,7 @@ func Register(root *cobra.Command, embeddedTemplates fs.FS) { root.AddCommand(webCmd) } -func webMain(cfg app.Config, embeddedTemplates fs.FS) { +func webMain(cfg app.Config, embeddedTemplates fs.FS, version string) { ctx := context.Background() s, err := glecho.NewDefaultEchoWithConfig(glecho.EchoConfig{ @@ -86,6 +86,7 @@ func webMain(cfg app.Config, embeddedTemplates fs.FS) { if err = s.Init(); err != nil { log.Fatalf("Error initializing echo: %w", err) } + s.Use(middleware.AddServerHeader(version)) mongo, err := mongodb.Connect(ctx, cfg.MongoDbUri, cfg.MongodbVaultPath) if err != nil { -- cgit v1.2.3