claude
e7fdd0e1ab
- Implemented player_configs table to store multiple player configurations per user. - Migrated existing player settings from user_preferences to player_configs. - Removed player-related columns from user_preferences. - Added referral state fields to user for tracking referral rewards. - Created migration scripts for database changes and data migration. - Added test cases for app services and usage helpers. - Introduced video job service interfaces and implementations.
77 lines
2.6 KiB
Go
77 lines
2.6 KiB
Go
package app
|
|
|
|
import (
|
|
"context"
|
|
"net"
|
|
|
|
grpcpkg "google.golang.org/grpc"
|
|
"gorm.io/gorm"
|
|
"stream.api/internal/config"
|
|
apprpc "stream.api/internal/rpc/app"
|
|
"stream.api/internal/video"
|
|
runtime "stream.api/internal/video/runtime"
|
|
redisadapter "stream.api/internal/video/runtime/adapters/queue/redis"
|
|
runtimegrpc "stream.api/internal/video/runtime/grpc"
|
|
"stream.api/internal/video/runtime/services"
|
|
"stream.api/pkg/cache"
|
|
"stream.api/pkg/logger"
|
|
"stream.api/pkg/token"
|
|
)
|
|
|
|
type GRPCModule struct {
|
|
jobService *services.JobService
|
|
healthService *services.HealthService
|
|
agentRuntime *runtimegrpc.Server
|
|
mqttPublisher *runtime.MQTTBootstrap
|
|
grpcServer *grpcpkg.Server
|
|
cfg *config.Config
|
|
}
|
|
|
|
func NewGRPCModule(ctx context.Context, cfg *config.Config, db *gorm.DB, cacheClient cache.Cache, tokenProvider token.Provider, appLogger logger.Logger) (*GRPCModule, error) {
|
|
adapter, err := redisadapter.NewAdapter(cfg.Redis.Addr, cfg.Redis.Password, cfg.Redis.DB)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
jobService := services.NewJobService(adapter, adapter)
|
|
healthService := services.NewHealthService(db, adapter.Client(), cfg.Render.ServiceName)
|
|
agentRuntime := runtimegrpc.NewServer(jobService, cfg.Render.AgentSecret)
|
|
videoService := video.NewService(db, jobService)
|
|
grpcServer := grpcpkg.NewServer()
|
|
|
|
module := &GRPCModule{
|
|
jobService: jobService,
|
|
healthService: healthService,
|
|
agentRuntime: agentRuntime,
|
|
grpcServer: grpcServer,
|
|
cfg: cfg,
|
|
}
|
|
|
|
if publisher, err := runtime.NewMQTTBootstrap(jobService, agentRuntime, appLogger); err != nil {
|
|
appLogger.Error("Failed to initialize MQTT publisher", "error", err)
|
|
} else {
|
|
module.mqttPublisher = publisher
|
|
agentRuntime.SetAgentEventHandler(func(eventType string, agent *services.AgentWithStats) {
|
|
runtime.PublishAgentMQTTEvent(publisher.Client(), appLogger, eventType, agent)
|
|
})
|
|
}
|
|
|
|
agentRuntime.Register(grpcServer)
|
|
apprpc.Register(grpcServer, apprpc.NewServices(cacheClient, tokenProvider, db, appLogger, cfg, videoService, agentRuntime))
|
|
if module.mqttPublisher != nil {
|
|
module.mqttPublisher.Start(ctx)
|
|
}
|
|
|
|
return module, nil
|
|
}
|
|
|
|
func (m *GRPCModule) JobService() *services.JobService { return m.jobService }
|
|
func (m *GRPCModule) AgentRuntime() *runtimegrpc.Server { return m.agentRuntime }
|
|
func (m *GRPCModule) GRPCServer() *grpcpkg.Server { return m.grpcServer }
|
|
func (m *GRPCModule) GRPCAddress() string { return ":" + m.cfg.Server.GRPCPort }
|
|
func (m *GRPCModule) ServeGRPC(listener net.Listener) error { return m.grpcServer.Serve(listener) }
|
|
func (m *GRPCModule) Shutdown() {
|
|
if m.grpcServer != nil {
|
|
m.grpcServer.GracefulStop()
|
|
}
|
|
}
|