city_chips/cmd/server/wire/wire.go

//go:build wireinject
// +build wireinject

package wire

import (
	"city_chips/internal/handler"
	"city_chips/internal/repository"
	"city_chips/internal/server"
	"city_chips/internal/service"
	"city_chips/pkg/log"
	"github.com/gin-gonic/gin"
	"github.com/google/wire"
	"github.com/spf13/viper"
	"net/http"
)

var ServerSet = wire.NewSet(server.NewServerHTTP)

var RepositorySet = wire.NewSet(
	repository.NewDb,
	repository.NewRepository,
	repository.NewUserRepository,
	repository.NewAccessControlRepository,
	repository.NewHikvisionRepository,
	repository.NewConferenceRepository,
	repository.NewHomeRepository,
	repository.NewElevatorRepository,
	repository.NewBroadcastRepository,
	repository.NewPropertyRepository,
	repository.NewInformationRepository,
	repository.NewIlluminatingRepository,
	repository.NewEnergyRepository,
	repository.NewIntelligentBuildingControlRepository,
)

var ServiceSet = wire.NewSet(
	service.NewService,
	service.NewUserService,
	service.NewAccessControlService,
	service.NewHikvisionService,
	service.NewConferenceService,
	service.NewHomeService,
	service.NewElevatorService,
	service.NewBroadcastService,
	service.NewPropertyService,
	service.NewInformationService,
	service.NewIlluminatingService,
	service.NewEnergyService,
	service.NewIntelligentBuildingControlService,
	provideHTTPClient,
)

var HandlerSet = wire.NewSet(
	handler.NewHandler,
	handler.NewUserHandler,
	handler.NewAccessControlHandler,
	handler.NewHikvisionHandler,
	handler.NewConferenceHandler,
	handler.NewHomeHandler,
	handler.NewElevatorHandler,
	handler.NewBroadcastHandler,
	handler.NewPropertyHandler,
	handler.NewInformationHandler,
	handler.NewIlluminatingHandler,
	handler.NewEnergyHandler,
	handler.NewIntelligentBuildingControlHandler,
)

func provideHTTPClient() *http.Client {
	return http.DefaultClient
}
func NewWire(*viper.Viper, *log.Logger) (*gin.Engine, func(), error) {
	panic(wire.Build(
		ServerSet,
		RepositorySet,
		ServiceSet,
		HandlerSet,
	))
}