k2safe/go-omni
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: add dependency shutdown and configurable service endpoints

Browse Source
This commit is contained in:
k2safe
2026-09-24 18:33:24 +08:00
parent 8a348951eb
commit f0991d3ff2
51 changed files with 1433 additions and 73 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
examples/managedService/internal/handler/appGen.go
View File

@@ -6,12 +6,14 @@ import (
omni "git.yuntuo.hk/k2safe/go-omni"
"example.com/managed/internal/svc"
"git.yuntuo.hk/k2safe/go-omni/config"
"git.yuntuo.hk/k2safe/go-omni/health"
)
// NewApp is generated HTTP composition. The caller owns deps and its logger.
func NewApp(deps *svc.ServiceContext) (*omni.App, error) {
if deps == nil || deps.Health == nil {
return nil, fmt.Errorf("service context and health checker required")
if deps == nil || deps.Lifecycle == nil {
return nil, fmt.Errorf("service context and lifecycle controller required; see PROJECT.md upgrade instructions")
}
cfg := deps.Config.Server
for _, option := range deps.AppOptions() {
@@ -24,13 +26,42 @@ func NewApp(deps *svc.ServiceContext) (*omni.App, error) {
if err := cfg.Validate(); err != nil {
return nil, err
}
if err := cfg.Endpoints.ValidateRoutes([]config.EndpointRoute{
{Method: "get", Path: "/check"},
{Method: "get", Path: "/other"},
{Method: "get", Path: "/raw"},
{Method: "post", Path: "/upload"},
{Method: "post", Path: "/callback"},
{Method: "get", Path: "/download"},
{Method: "get", Path: "/slow"},
}...); err != nil {
return nil, err
}
if (cfg.Endpoints.Liveness.Enabled || cfg.Endpoints.Readiness.Enabled) && deps.Health == nil {
return nil, fmt.Errorf("enabled health endpoints require a checker")
}
app := omni.New(omni.WithConfig(cfg))
app.Use(deps.GlobalMiddleware()...)
if err := RegisterRoutes(app, deps); err != nil {
_ = app.Close()
return nil, err
}
app.GET("/openapi.json", OpenAPI)
app.GET("/health/live", deps.Health.LivenessHandler())
app.GET("/health/ready", deps.Health.ReadinessHandler())
if cfg.Endpoints.OpenAPI.Enabled {
app.GET(cfg.Endpoints.OpenAPI.Path, OpenAPI)
}
if cfg.Endpoints.Liveness.Enabled {
var options []health.HandlerOption
if cfg.Endpoints.Liveness.ResponseMode == "raw" {
options = append(options, health.WithRawResponse())
}
app.GET(cfg.Endpoints.Liveness.Path, deps.Health.LivenessHandler(options...))
}
if cfg.Endpoints.Readiness.Enabled {
options := []health.HandlerOption{health.WithShutdownContext(deps.Lifecycle.Context())}
if cfg.Endpoints.Readiness.ResponseMode == "raw" {
options = append(options, health.WithRawResponse())
}
app.GET(cfg.Endpoints.Readiness.Path, deps.Health.ReadinessHandler(options...))
}
return app, nil
}