lan
9a1851f023
This commit performs a significant cleanup of the codebase by removing unused functions, methods, and entire files across various internal modules. This reduces technical debt and simplifies the project structure. Key changes include: - **cache**: Removed unused cache key generators and metrics snapshots. - **dto**: Removed redundant converter functions and segment creation helpers. - **middleware**: Deleted the unused `logger.go` middleware and simplified `ratelimit.go` and `casbin.go`. - **model**: Removed unused ID helpers, database closing functions, and batch decryption logic. - **pkg**: Cleaned up unused utility functions in `circuitbreaker`, `crypto`, `cursor`, `hook`, and `utils`. - **service**: Deleted `account_cleanup_service.go` and removed unused helper functions in `log_cleanup_service.go` and `sensitive_service.go`. - **repository**: Removed unused private loading methods in `comment_repo.go`.
99 lines
2.1 KiB
Go
99 lines
2.1 KiB
Go
package runner
|
||
|
||
import (
|
||
"fmt"
|
||
"net"
|
||
"time"
|
||
|
||
"with_you/internal/config"
|
||
"with_you/proto/runner"
|
||
|
||
"go.uber.org/zap"
|
||
"google.golang.org/grpc"
|
||
"google.golang.org/grpc/credentials"
|
||
)
|
||
|
||
// Server gRPC 服务器
|
||
type Server struct {
|
||
cfg *config.GRPCConfig
|
||
logger *zap.Logger
|
||
hub *RunnerHub
|
||
grpcSrv *grpc.Server
|
||
}
|
||
|
||
// NewServerWithDeps 使用外部依赖创建 gRPC 服务器（Wire DI 使用）
|
||
func NewServerWithDeps(cfg *config.GRPCConfig, logger *zap.Logger, hub *RunnerHub) *Server {
|
||
return &Server{
|
||
cfg: cfg,
|
||
logger: logger,
|
||
hub: hub,
|
||
}
|
||
}
|
||
|
||
// Start 启动 gRPC 服务器
|
||
func (s *Server) Start() error {
|
||
if !s.cfg.Enabled {
|
||
s.logger.Info("gRPC server is disabled")
|
||
return nil
|
||
}
|
||
|
||
lis, err := net.Listen("tcp", s.cfg.Address())
|
||
if err != nil {
|
||
return fmt.Errorf("failed to listen on %s: %w", s.cfg.Address(), err)
|
||
}
|
||
|
||
// 配置 gRPC 服务器选项
|
||
var opts []grpc.ServerOption
|
||
|
||
// 配置 TLS
|
||
if s.cfg.TLSEnabled {
|
||
if s.cfg.TLSCertFile == "" || s.cfg.TLSKeyFile == "" {
|
||
return fmt.Errorf("TLS enabled but cert/key files not specified")
|
||
}
|
||
creds, err := credentials.NewServerTLSFromFile(s.cfg.TLSCertFile, s.cfg.TLSKeyFile)
|
||
if err != nil {
|
||
return fmt.Errorf("failed to load TLS credentials: %w", err)
|
||
}
|
||
opts = append(opts, grpc.Creds(creds))
|
||
s.logger.Info("gRPC server TLS enabled")
|
||
} else {
|
||
s.logger.Warn("gRPC server running without TLS")
|
||
}
|
||
|
||
s.grpcSrv = grpc.NewServer(opts...)
|
||
runner.RegisterRunnerHubServer(s.grpcSrv, s.hub)
|
||
|
||
s.logger.Info("gRPC server starting", zap.String("address", s.cfg.Address()))
|
||
|
||
go func() {
|
||
if err := s.grpcSrv.Serve(lis); err != nil {
|
||
s.logger.Error("gRPC server error", zap.Error(err))
|
||
}
|
||
}()
|
||
|
||
// 启动后台任务清理协程
|
||
go s.backgroundCleanup()
|
||
|
||
return nil
|
||
}
|
||
|
||
// backgroundCleanup 后台清理任务
|
||
func (s *Server) backgroundCleanup() {
|
||
ticker := time.NewTicker(30 * time.Second)
|
||
defer ticker.Stop()
|
||
|
||
for range ticker.C {
|
||
// 清理超时的 Runner 连接
|
||
s.hub.CheckStaleRunners(90 * time.Second)
|
||
}
|
||
}
|
||
|
||
// Stop 停止 gRPC 服务器
|
||
func (s *Server) Stop() {
|
||
if s.grpcSrv != nil {
|
||
s.logger.Info("Stopping gRPC server")
|
||
s.grpcSrv.GracefulStop()
|
||
}
|
||
}
|
||
|