wire.go 1.9 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576
  1. //go:build wireinject
  2. // +build wireinject
  3. package wire
  4. import (
  5. "city_chips/internal/handler"
  6. "city_chips/internal/repository"
  7. "city_chips/internal/server"
  8. "city_chips/internal/service"
  9. "city_chips/pkg/log"
  10. "github.com/gin-gonic/gin"
  11. "github.com/google/wire"
  12. "github.com/spf13/viper"
  13. )
  14. var ServerSet = wire.NewSet(server.NewServerHTTP)
  15. var RepositorySet = wire.NewSet(
  16. repository.NewDb,
  17. repository.NewRedis,
  18. repository.NewRepository,
  19. repository.NewUserRepository,
  20. repository.NewAccessControlRepository,
  21. repository.NewHikvisionRepository,
  22. repository.NewConferenceRepository,
  23. repository.NewHomeRepository,
  24. repository.NewElevatorRepository,
  25. repository.NewBroadcastRepository,
  26. repository.NewPropertyRepository,
  27. repository.NewInformationRepository,
  28. repository.NewIlluminatingRepository,
  29. repository.NewEnergyRepository,
  30. repository.NewIntelligentBuildingControlRepository,
  31. )
  32. var ServiceSet = wire.NewSet(
  33. service.NewService,
  34. service.NewUserService,
  35. service.NewAccessControlService,
  36. service.NewHikvisionService,
  37. service.NewConferenceService,
  38. service.NewHomeService,
  39. service.NewElevatorService,
  40. service.NewBroadcastService,
  41. service.NewPropertyService,
  42. service.NewInformationService,
  43. service.NewIlluminatingService,
  44. service.NewEnergyService,
  45. service.NewIntelligentBuildingControlService,
  46. )
  47. var HandlerSet = wire.NewSet(
  48. handler.NewHandler,
  49. handler.NewUserHandler,
  50. handler.NewAccessControlHandler,
  51. handler.NewHikvisionHandler,
  52. handler.NewConferenceHandler,
  53. handler.NewHomeHandler,
  54. handler.NewElevatorHandler,
  55. handler.NewBroadcastHandler,
  56. handler.NewPropertyHandler,
  57. handler.NewInformationHandler,
  58. handler.NewIlluminatingHandler,
  59. handler.NewEnergyHandler,
  60. handler.NewIntelligentBuildingControlHandler,
  61. )
  62. func NewWire(*viper.Viper, *log.Logger) (*gin.Engine, func(), error) {
  63. panic(wire.Build(
  64. ServerSet,
  65. RepositorySet,
  66. ServiceSet,
  67. HandlerSet,
  68. ))
  69. }