toki
2d6fde0876
apps/node 중심 구현 — TCP+JSON transport, Hexagonal Architecture, mock/cli adapter, fx DI, SQLite 실행 이력 저장. edge/control-plane/worker는 cobra placeholder. 유닛 테스트 및 통합 테스트 클라이언트 계획서 추가.
53 lines
1.3 KiB
Go
53 lines
1.3 KiB
Go
package router
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"go.uber.org/zap"
|
|
|
|
"iop/apps/node/internal/adapters"
|
|
"iop/apps/node/internal/runtime"
|
|
)
|
|
|
|
type defaultRouter struct {
|
|
registry *adapters.Registry
|
|
logger *zap.Logger
|
|
}
|
|
|
|
// New returns a Router that resolves requests to registered adapters.
|
|
func New(reg *adapters.Registry, logger *zap.Logger) runtime.Router {
|
|
return &defaultRouter{registry: reg, logger: logger}
|
|
}
|
|
|
|
func (r *defaultRouter) Resolve(_ context.Context, req runtime.RunRequest) (runtime.ExecutionSpec, error) {
|
|
adapterName := req.Adapter
|
|
if adapterName == "" {
|
|
adapterName = r.registry.Default()
|
|
}
|
|
if adapterName == "" {
|
|
return runtime.ExecutionSpec{}, fmt.Errorf("router: no adapters registered")
|
|
}
|
|
|
|
if _, ok := r.registry.Get(adapterName); !ok {
|
|
return runtime.ExecutionSpec{}, fmt.Errorf("router: adapter %q not found", adapterName)
|
|
}
|
|
|
|
spec := runtime.ExecutionSpec{
|
|
RunID: req.RunID,
|
|
Adapter: adapterName,
|
|
Model: req.Model,
|
|
Workspace: req.Workspace,
|
|
Policy: req.Policy,
|
|
Input: req.Input,
|
|
TimeoutSec: req.TimeoutSec,
|
|
Metadata: req.Metadata,
|
|
}
|
|
|
|
r.logger.Debug("resolved execution spec",
|
|
zap.String("run_id", req.RunID),
|
|
zap.String("adapter", spec.Adapter),
|
|
zap.String("model", spec.Model),
|
|
)
|
|
return spec, nil
|
|
}
|