toki
620ca92e2e
- parser_map.go 업데이트하여 market status 파싱 로직 통일 - protobuf market.proto 변경사항 적용 (market.pb.go, market.pb.dart) - socket handlers, market, backtest 관련 테스트 및 런타임 코드 개선 - workerclient와 alt-worker main.go 변경사항 반영 - agent-task archive 이동 (01_import_contract_worker_api → archive/2026/06/)
87 lines
3.2 KiB
Go
87 lines
3.2 KiB
Go
package socket
|
|
|
|
import (
|
|
protoSocket "git.toki-labs.com/toki/proto-socket/go"
|
|
|
|
altv1 "git.toki-labs.com/toki/alt/packages/contracts/gen/go/alt/v1"
|
|
"git.toki-labs.com/toki/alt/services/api/internal/workerclient"
|
|
)
|
|
|
|
// sessionHandler is one request-response unit attached to a freshly connected
|
|
// client communicator. Splitting registration into independent units keeps the
|
|
// api hub control-plane surface thin: later command/query handlers attach
|
|
// through the same registry instead of growing a single registration function.
|
|
type sessionHandler struct {
|
|
// requestType is the ALT protobuf request type name the handler answers.
|
|
// It lets the registry detect duplicate or missing handler coverage without
|
|
// a live socket connection.
|
|
requestType string
|
|
// register attaches the handler onto the client communicator.
|
|
register func(*protoSocket.WsClient)
|
|
}
|
|
|
|
// sessionHandlers returns the ordered set of handlers registered on every API
|
|
// client session. Hello is registered through this registry so its behaviour
|
|
// stays identical while gaining the shared registration path.
|
|
func sessionHandlers(worker workerclient.WorkerClient) []sessionHandler {
|
|
handlers := []sessionHandler{
|
|
helloHandler(worker),
|
|
}
|
|
handlers = append(handlers, marketHandlers(worker)...)
|
|
handlers = append(handlers, backtestHandlers(worker)...)
|
|
return handlers
|
|
}
|
|
|
|
// registerSessionHandlers wires every session handler onto a newly connected
|
|
// client. It is the OnClientConnected entrypoint for the API socket server.
|
|
func registerSessionHandlers(worker workerclient.WorkerClient) func(*protoSocket.WsClient) {
|
|
return func(client *protoSocket.WsClient) {
|
|
registerHandlers(client, sessionHandlers(worker))
|
|
}
|
|
}
|
|
|
|
// registerHandlers attaches the given handlers onto a client. nil registrars
|
|
// are skipped so a malformed registry entry cannot panic the connection setup
|
|
// path.
|
|
func registerHandlers(client *protoSocket.WsClient, handlers []sessionHandler) {
|
|
for _, handler := range handlers {
|
|
if handler.register == nil {
|
|
continue
|
|
}
|
|
handler.register(client)
|
|
}
|
|
}
|
|
|
|
func helloHandler(worker workerclient.WorkerClient) sessionHandler {
|
|
return sessionHandler{
|
|
requestType: protoSocket.TypeNameOf(&altv1.HelloRequest{}),
|
|
register: func(client *protoSocket.WsClient) {
|
|
protoSocket.AddRequestListenerTyped[*altv1.HelloRequest, *altv1.HelloResponse](&client.Communicator, func(req *altv1.HelloRequest) (*altv1.HelloResponse, error) {
|
|
return handleHello(worker, req)
|
|
})
|
|
},
|
|
}
|
|
}
|
|
|
|
func handleHello(worker workerclient.WorkerClient, req *altv1.HelloRequest) (*altv1.HelloResponse, error) {
|
|
protocolVersion := req.GetAltProtocolVersion()
|
|
if protocolVersion == "" {
|
|
protocolVersion = defaultAltProtocolVersion
|
|
}
|
|
return &altv1.HelloResponse{
|
|
ServerName: serverName,
|
|
ServerVersion: serverVersion,
|
|
AltProtocolVersion: protocolVersion,
|
|
Capabilities: capabilitiesForSession(worker),
|
|
}, nil
|
|
}
|
|
|
|
func capabilitiesForSession(worker workerclient.WorkerClient) []string {
|
|
caps := []string{"hello", "request-response", "market-read", "market-import", "backtest-read", "backtest-start", "worker-execution"}
|
|
if worker != nil && worker.IsConnected() {
|
|
caps = append(caps, "worker-available")
|
|
} else {
|
|
caps = append(caps, "worker-unavailable")
|
|
}
|
|
return caps
|
|
}
|