generated from rmcguire/go-server-with-otel
add ui, new config opts
This commit is contained in:
+10
-1
@@ -22,6 +22,7 @@ import (
|
||||
// Hand-written tools (switch back by using this import instead):
|
||||
// econetmcp "gitea.libretechconsulting.com/rmcguire/econet-exporter/pkg/econet/econetmcp"
|
||||
"gitea.libretechconsulting.com/rmcguire/econet-exporter/pkg/econet/econetmetrics"
|
||||
"gitea.libretechconsulting.com/rmcguire/econet-exporter/pkg/econet/econetui"
|
||||
"gitea.libretechconsulting.com/rmcguire/econet-exporter/pkg/service"
|
||||
)
|
||||
|
||||
@@ -32,6 +33,7 @@ type EconetService struct {
|
||||
client *econetclient.Client
|
||||
grpc *econetgrpc.EconetGRPCServer
|
||||
mcp *econetmcp.EconetMCPServer
|
||||
ui *econetui.EconetUIServer
|
||||
metrics *econetmetrics.Collector
|
||||
}
|
||||
|
||||
@@ -48,6 +50,9 @@ func (e *EconetService) Init(ctx context.Context, cfg *config.ServiceConfig) (se
|
||||
e.client = econetclient.New(cfg, e.log)
|
||||
e.grpc = econetgrpc.NewEconetGRPCServer(ctx, cfg, e.client)
|
||||
e.mcp = econetmcp.NewEconetMCPServer(ctx, cfg, e.grpc)
|
||||
if cfg.WebUIEnabled() {
|
||||
e.ui = econetui.NewEconetUIServer(ctx, cfg, e.grpc)
|
||||
}
|
||||
e.metrics = econetmetrics.NewCollector(ctx, cfg, e.client)
|
||||
if err := e.metrics.RegisterGauges(); err != nil {
|
||||
return nil, err
|
||||
@@ -100,9 +105,13 @@ func (e *EconetService) GetGRPC() *optsgrpc.AppGRPC {
|
||||
}
|
||||
|
||||
func (e *EconetService) GetHTTP() *optshttp.AppHTTP {
|
||||
handlers := e.mcp.GetHandlers()
|
||||
if e.ui != nil {
|
||||
handlers = append(handlers, e.ui.GetHandlers()...)
|
||||
}
|
||||
return &optshttp.AppHTTP{
|
||||
Ctx: e.ctx,
|
||||
Handlers: e.mcp.GetHandlers(),
|
||||
Handlers: handlers,
|
||||
HealthChecks: e.healthChecks(),
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user