toki
57000c8cb6
- Archive completed subtask documents (02+01_api_lifecycle_capabilities, 03_client_bootstrap) - Update roadmap and PHASE.md for operator-surface progress - Refactor apps/client/lib/src/app/bootstrap.dart for client initialization - Update API socket layer: backtest, handlers, market, server, workerclient - Add bootstrap_test.dart for client app
69 lines
1.8 KiB
Go
69 lines
1.8 KiB
Go
package socket
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"log/slog"
|
|
|
|
protoSocket "git.toki-labs.com/toki/proto-socket/go"
|
|
"nhooyr.io/websocket"
|
|
|
|
"git.toki-labs.com/toki/alt/services/api/internal/config"
|
|
apiContracts "git.toki-labs.com/toki/alt/services/api/internal/contracts"
|
|
"git.toki-labs.com/toki/alt/services/api/internal/workerclient"
|
|
)
|
|
|
|
const (
|
|
serverName = "alt-api"
|
|
serverVersion = "dev"
|
|
defaultAltProtocolVersion = "alt.v1"
|
|
)
|
|
|
|
type Server struct {
|
|
wsServer *protoSocket.WsServer
|
|
worker workerclient.WorkerClient
|
|
}
|
|
|
|
func NewServer(cfg config.Config) *Server {
|
|
return NewServerWithWorker(cfg, workerclient.New(cfg.WorkerSocketURL))
|
|
}
|
|
|
|
func NewServerWithWorker(cfg config.Config, worker workerclient.WorkerClient) *Server {
|
|
options := protoSocket.WsServerOptions{}
|
|
if len(cfg.WSOriginPatterns) > 0 {
|
|
options.AcceptOptions = &websocket.AcceptOptions{
|
|
OriginPatterns: cfg.WSOriginPatterns,
|
|
}
|
|
}
|
|
wsServer := protoSocket.NewWsServerWithOptions(cfg.Host, cfg.Port, cfg.SocketPath, options, func(conn *websocket.Conn) *protoSocket.WsClient {
|
|
return protoSocket.NewWsClient(conn, cfg.HeartbeatIntervalSec, cfg.HeartbeatWaitSec, apiContracts.ParserMap())
|
|
})
|
|
wsServer.OnClientConnected = registerSessionHandlers(worker)
|
|
return &Server{
|
|
wsServer: wsServer,
|
|
worker: worker,
|
|
}
|
|
}
|
|
|
|
func (s *Server) Start(ctx context.Context) error {
|
|
go func() {
|
|
if err := s.worker.Connect(ctx); err != nil {
|
|
slog.Warn("eager worker connection failed on startup", "error", err)
|
|
}
|
|
}()
|
|
return s.wsServer.Start(ctx)
|
|
}
|
|
|
|
func (s *Server) Stop() error {
|
|
var errs []error
|
|
if err := s.wsServer.Stop(); err != nil {
|
|
errs = append(errs, err)
|
|
}
|
|
if err := s.worker.Close(); err != nil {
|
|
errs = append(errs, err)
|
|
}
|
|
if len(errs) > 0 {
|
|
return errors.Join(errs...)
|
|
}
|
|
return nil
|
|
}
|