ちょっと工事

cmd/app/main.go

@@ -45,19 +45,15 @@ func run() error {
 		return err
 	}
 
-	// handler をinvoke
-	var handler http.Handler
-	if err := container.Invoke(func(h http.Handler) {
-		handler = h
-	}); err != nil {
+	handler, err := container.Invoke[http.Handler]()
+	if err != nil {
 		return err
 	}
 
 	server.New(
 		handler,
 		server.WithHost(config.Config.Server.Host),
 		server.WithPort(config.Config.Server.Port),
-		server.WithShutdownTimeout(config.Config.Server.ShutdownTimeout),
 	).RunWithGracefulShutdown(context.Background())
 
 	return nil

internal/container/dig.go

@@ -30,6 +30,13 @@ func NewContainer() error {
 }
 
 // Invoke は、 *dig.ContainerのInvokeをwrapしてる関数
-func Invoke(f any, opts ...dig.InvokeOption) error {
-	return container.Invoke(f, opts...)
+func Invoke[T any](opts ...dig.InvokeOption) (T, error) {
+	var r T
+	if err := container.Invoke(func(t T) error {
+		r = t
+		return nil
+	}, opts...); err != nil {
+		return r, err
+	}
+	return r, nil
 }

internal/framework/contexts/context.go

@@ -0,0 +1,40 @@
+package contexts
+
+import (
+	"context"
+
+	"github.com/labstack/echo/v4"
+)
+
+type ContextKey string
+
+func (c ContextKey) String() string {
+	return string(c)
+}
+
+const (
+	RequestID ContextKey = "x-request-id"
+)
+
+var contextkeys = []ContextKey{
+	RequestID,
+}
+
+func GetRequestID(ctx context.Context) string {
+	v, ok := ctx.Value(RequestID.String()).(string)
+	if !ok {
+		return ""
+	}
+	return v
+}
+
+// ConvertContext はecho.Contextのkeyをcopyしてcontext.Contextに変換
+func ConvertContext(c echo.Context) context.Context {
+	ctx := context.Background()
+	for _, key := range contextkeys {
+		v := c.Get(key.String())
+		ctx = context.WithValue(ctx, key, v)
+	}
+
+	return ctx
+}

internal/router/router.go

@@ -6,18 +6,13 @@ import (
 	"github.com/labstack/echo/v4"
 )
 
-type router struct {
-	engine *echo.Echo
-}
-
 // NewEcho は、echo/v4 を利用した http.Handlerを返す関数です。
 func NewEcho() http.Handler {
-	router := &router{
-		engine: echo.New(),
-	}
+	engine := echo.New()
 
-	router.health()
-	router.info()
+	engine.GET("/healthz", func(c echo.Context) error {
+		return c.String(200, "OK")
+	})
 
-	return router.engine
+	return engine
 }