//go:build wireinject // +build wireinject package wire import ( "github.com/go-nunu/nunu-layout-advanced/internal/handler" admin3 "github.com/go-nunu/nunu-layout-advanced/internal/handler/admin" waf3 "github.com/go-nunu/nunu-layout-advanced/internal/handler/api/waf" "github.com/go-nunu/nunu-layout-advanced/internal/middleware" "github.com/go-nunu/nunu-layout-advanced/internal/repository" "github.com/go-nunu/nunu-layout-advanced/internal/repository/admin" waf2 "github.com/go-nunu/nunu-layout-advanced/internal/repository/api/waf" "github.com/go-nunu/nunu-layout-advanced/internal/server" "github.com/go-nunu/nunu-layout-advanced/internal/service" admin2 "github.com/go-nunu/nunu-layout-advanced/internal/service/admin" "github.com/go-nunu/nunu-layout-advanced/internal/service/api/gameShield" "github.com/go-nunu/nunu-layout-advanced/internal/service/api/waf" "github.com/go-nunu/nunu-layout-advanced/pkg/app" "github.com/go-nunu/nunu-layout-advanced/pkg/jwt" "github.com/go-nunu/nunu-layout-advanced/pkg/limiter" "github.com/go-nunu/nunu-layout-advanced/pkg/log" "github.com/go-nunu/nunu-layout-advanced/pkg/server/http" "github.com/go-nunu/nunu-layout-advanced/pkg/sid" "github.com/google/wire" "github.com/spf13/viper" ) var repositorySet = wire.NewSet( repository.NewDB, repository.NewRedis, repository.NewCasbinEnforcer, repository.NewMongoClient, repository.NewMongoDB, repository.NewRabbitMQ, repository.NewRepository, repository.NewTransaction, admin.NewAdminRepository, admin.NewUserRepository, repository.NewGameShieldRepository, repository.NewGameShieldPublicIpRepository, waf2.NewWebForwardingRepository, waf2.NewTcpforwardingRepository, waf2.NewUdpForWardingRepository, repository.NewGameShieldUserIpRepository, repository.NewGameShieldBackendRepository, repository.NewGameShieldSdkIpRepository, repository.NewHostRepository, waf2.NewGlobalLimitRepository, repository.NewGatewayGroupRepository, repository.NewGateWayGroupIpRepository, repository.NewCdnRepository, waf2.NewAllowAndDenyIpRepository, repository.NewProxyRepository, repository.NewCcRepository, repository.NewExpiredRepository, repository.NewLogRepository, waf2.NewGatewayipRepository, ) var serviceSet = wire.NewSet( service.NewService, admin2.NewUserService, admin2.NewAdminService, gameShield.NewGameShieldService, service.NewAoDunService, service.NewGameShieldPublicIpService, service.NewDuedateService, service.NewFormatterService, service.NewParserService, service.NewRequiredService, service.NewCrawlerService, waf.NewWebForwardingService, waf.NewTcpforwardingService, waf.NewUdpForWardingService, service.NewGameShieldUserIpService, gameShield.NewGameShieldBackendService, service.NewGameShieldSdkIpService, service.NewHostService, waf.NewGlobalLimitService, service.NewGatewayGroupService, waf.NewWafFormatterService, service.NewGateWayGroupIpService, service.NewRequestService, service.NewCdnService, waf.NewAllowAndDenyIpService, service.NewProxyService, service.NewSslCertService, service.NewWebsocketService, service.NewCcService, service.NewLogService, waf.NewGatewayipService, ) var handlerSet = wire.NewSet( handler.NewHandler, admin3.NewUserHandler, admin3.NewAdminHandler, handler.NewGameShieldHandler, handler.NewGameShieldPublicIpHandler, waf3.NewWebForwardingHandler, waf3.NewTcpforwardingHandler, waf3.NewUdpForWardingHandler, handler.NewGameShieldUserIpHandler, handler.NewGameShieldBackendHandler, handler.NewGameShieldSdkIpHandler, handler.NewHostHandler, waf3.NewGlobalLimitHandler, handler.NewGatewayGroupHandler, handler.NewGateWayGroupIpHandler, waf3.NewAllowAndDenyIpHandler, handler.NewCcHandler, waf3.NewGatewayipHandler, ) // 限流器依赖集 var limiterSet = wire.NewSet( limiter.NewLimiter, middleware.NewRateLimitMiddleware, ) var serverSet = wire.NewSet( server.NewHTTPServer, ) // build App func newApp( httpServer *http.Server, ) *app.App { return app.NewApp( app.WithServer(httpServer), app.WithName("demo-server"), ) } func NewWire(*viper.Viper, *log.Logger) (*app.App, func(), error) { panic(wire.Build( repositorySet, serviceSet, handlerSet, serverSet, limiterSet, sid.NewSid, jwt.NewJwt, newApp, )) }