lethdat
4de6baee61
- Implement tests for player configuration creation, update, and deletion, ensuring proper handling of free and paid user scenarios. - Add tests for referral registration, including valid and invalid referrer cases. - Create tests for referral reward flow, verifying correct reward distribution and eligibility. - Establish a test database setup with necessary schema for user, plan, payment, and notification models. - Introduce helper functions for seeding test data and loading entities from the database.
71 lines
2.4 KiB
Go
71 lines
2.4 KiB
Go
package grpc
|
|
|
|
import (
|
|
"context"
|
|
"net"
|
|
|
|
grpcpkg "google.golang.org/grpc"
|
|
"gorm.io/gorm"
|
|
redisadapter "stream.api/internal/adapters/redis"
|
|
"stream.api/internal/config"
|
|
"stream.api/internal/service"
|
|
"stream.api/internal/video"
|
|
runtime "stream.api/internal/video/runtime"
|
|
runtimegrpc "stream.api/internal/video/runtime/grpc"
|
|
"stream.api/internal/video/runtime/services"
|
|
"stream.api/pkg/logger"
|
|
)
|
|
|
|
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, rds *redisadapter.RedisAdapter, appLogger logger.Logger) (*GRPCModule, error) {
|
|
jobService := services.NewJobService(rds, rds)
|
|
healthService := services.NewHealthService(db, rds.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)
|
|
service.Register(grpcServer, service.NewServices(rds, 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()
|
|
}
|
|
}
|