toki
a52ecb8efe
필드 테스트를 위해 edge OpenAI-compatible 경로와 node adapter 설정을 확장하고, Jenkins 바이너리 빌드 및 control-plane/web compose 배포 구성을 함께 정리한다.
99 lines
2.4 KiB
Go
99 lines
2.4 KiB
Go
package bootstrap
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"go.uber.org/zap"
|
|
|
|
edgeevents "iop/apps/edge/internal/events"
|
|
edgenode "iop/apps/edge/internal/node"
|
|
edgeopenai "iop/apps/edge/internal/openai"
|
|
edgeservice "iop/apps/edge/internal/service"
|
|
"iop/apps/edge/internal/transport"
|
|
"iop/packages/config"
|
|
"iop/packages/observability"
|
|
)
|
|
|
|
// Runtime is the shared edge runtime assembly used by both `serve` and
|
|
// `console` entrypoints. NewRuntime wires logger, registry, node store,
|
|
// event bus, service, and transport server; Start performs handler wiring
|
|
// and binds the server and metrics endpoint.
|
|
type Runtime struct {
|
|
Cfg *config.EdgeConfig
|
|
Logger *zap.Logger
|
|
Registry *edgenode.Registry
|
|
NodeStore *edgenode.NodeStore
|
|
EventBus *edgeevents.Bus
|
|
Service *edgeservice.Service
|
|
Server *transport.Server
|
|
OpenAI *edgeopenai.Server
|
|
}
|
|
|
|
func NewRuntime(cfg *config.EdgeConfig) (*Runtime, error) {
|
|
logger, err := observability.NewLogger(cfg.Logging.Level, cfg.Logging.Pretty)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
registry := edgenode.NewRegistry()
|
|
nodeStore, err := edgenode.LoadFromConfig(cfg.Nodes)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("edge: seed node store: %w", err)
|
|
}
|
|
|
|
bus := edgeevents.NewBus()
|
|
svc := edgeservice.New(registry, bus)
|
|
openaiServer := edgeopenai.NewServer(cfg.OpenAI, svc, logger.Named("openai"))
|
|
|
|
server, err := transport.NewServer(cfg.Server.Listen, registry, nodeStore, logger)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &Runtime{
|
|
Cfg: cfg,
|
|
Logger: logger,
|
|
Registry: registry,
|
|
NodeStore: nodeStore,
|
|
EventBus: bus,
|
|
Service: svc,
|
|
Server: server,
|
|
OpenAI: openaiServer,
|
|
}, nil
|
|
}
|
|
|
|
func (r *Runtime) wireHandlers() {
|
|
r.Server.SetRunEventHandler(r.EventBus.PublishRun)
|
|
r.Server.SetNodeEventHandler(r.EventBus.PublishNode)
|
|
}
|
|
|
|
func (r *Runtime) Start(ctx context.Context) error {
|
|
r.wireHandlers()
|
|
if err := r.Server.Start(ctx); err != nil {
|
|
return err
|
|
}
|
|
if err := r.OpenAI.Start(ctx); err != nil {
|
|
_ = r.Server.Stop()
|
|
return err
|
|
}
|
|
go func() {
|
|
if err := observability.ServeMetrics(r.Cfg.Metrics.Port); err != nil {
|
|
r.Logger.Warn("metrics server exited", zap.Error(err))
|
|
}
|
|
}()
|
|
return nil
|
|
}
|
|
|
|
func (r *Runtime) Stop() error {
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancel()
|
|
openAIErr := r.OpenAI.Stop(ctx)
|
|
err := r.Server.Stop()
|
|
_ = r.Logger.Sync()
|
|
if err == nil {
|
|
err = openAIErr
|
|
}
|
|
return err
|
|
}
|