go_gameshield/cmd/server/wire/wire.go

//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,
	))
}