wire 依赖注入

Author: arunfung

app/http/controllers/api/v1/auth/injector.go

@@ -0,0 +1,23 @@
+package auth
+
+import (
+	v1 "goapi/app/http/controllers/api/v1"
+
+	"github.com/google/wire"
+)
+
+var InjectorSet = wire.NewSet(
+	wire.Struct(new(v1.BaseAPIController), "*"),
+	wire.Struct(new(LoginController), "*"),
+	wire.Struct(new(SignupController), "*"),
+	wire.Struct(new(PasswordController), "*"),
+	wire.Struct(new(VerifyCodeController), "*"),
+	wire.Struct(new(Injector), "*"),
+)
+
+type Injector struct {
+	LoginController      *LoginController
+	SignupController     *SignupController
+	PasswordController   *PasswordController
+	VerifyCodeController *VerifyCodeController
+}

app/http/controllers/api/v1/auth/wire.go

@@ -0,0 +1,17 @@
+//go:build wireinject
+// +build wireinject
+
+package auth
+
+import (
+	"github.com/google/wire"
+)
+
+func BuildInjector() *Injector {
+	wire.Build(
+		// injector
+		InjectorSet,
+	)
+
+	return new(Injector)
+}

app/http/controllers/api/v1/auth/wire_gen.go

@@ -31,8 +31,8 @@ func SetupDB() {
 		})
 	case "sqlite":
 		// 初始化 sqlite
-		database := config.Get("database.sqlite.database")
-		dbConfig = sqlite.Open(database)
+		db := config.Get("database.sqlite.database")
+		dbConfig = sqlite.Open(db)
 	default:
 		panic(errors.New("database connection not supported"))
 	}

bootstrap/database.go

@@ -10,6 +10,7 @@ require (
 	github.com/getsentry/sentry-go v0.13.0
 	github.com/gin-gonic/gin v1.9.0
 	github.com/golang-jwt/jwt/v4 v4.4.2
+	github.com/google/wire v0.5.0
 	github.com/iancoleman/strcase v0.2.0
 	github.com/jordan-wright/email v4.0.1-0.20210109023952-943e75fe5223+incompatible
 	github.com/mgutz/ansi v0.0.0-20200706080929-d51e80ef957d

go.mod

@@ -290,9 +290,12 @@ github.com/google/pprof v0.0.0-20210601050228-01bbb1931b22/go.mod h1:kpwsk12EmLe
 github.com/google/pprof v0.0.0-20210609004039-a478d1d731e9/go.mod h1:kpwsk12EmLew5upagYY7GY0pfYCcupk39gWOCRROcvE=
 github.com/google/pprof v0.0.0-20210720184732-4bb14d4b1be1/go.mod h1:kpwsk12EmLew5upagYY7GY0pfYCcupk39gWOCRROcvE=
 github.com/google/renameio v0.1.0/go.mod h1:KWCgfxg9yswjAJkECMjeO8J8rahYeXnNhOm40UhjYkI=
+github.com/google/subcommands v1.0.1/go.mod h1:ZjhPrFU+Olkh9WazFPsl27BQ4UPiG37m3yTrtFlrHVk=
 github.com/google/uuid v1.0.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
 github.com/google/uuid v1.1.2 h1:EVhdT+1Kseyi1/pUmXKaFxYsDNy9RQYkMWRH68J/W7Y=
 github.com/google/uuid v1.1.2/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
+github.com/google/wire v0.5.0 h1:I7ELFeVBr3yfPIcc8+MWvrjk+3VjbcSzoXm3JVa+jD8=
+github.com/google/wire v0.5.0/go.mod h1:ngWDr9Qvq3yZA10YrxfyGELY/AFWGVpy9c1LTRi1EoU=
 github.com/googleapis/gax-go/v2 v2.0.4/go.mod h1:0Wqv26UfaUD9n4G6kQubkQ+KchISgw+vpHVxEJEs9eg=
 github.com/googleapis/gax-go/v2 v2.0.5/go.mod h1:DWXyrwAJ9X0FpwwEdw+IPEYBICEFu5mhpdKc/us6bOk=
 github.com/googleapis/gax-go/v2 v2.1.0/go.mod h1:Q3nei7sK6ybPYH7twZdmQpAd1MKb7pfu6SK+H1/DsU0=
@@ -886,6 +889,7 @@ golang.org/x/tools v0.0.0-20190312151545-0bb0c0a6e846/go.mod h1:LCzVGOaR6xXOjkQ3
 golang.org/x/tools v0.0.0-20190312170243-e65039ee4138/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs=
 golang.org/x/tools v0.0.0-20190327201419-c70d86f8b7cf/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs=
 golang.org/x/tools v0.0.0-20190328211700-ab21143f2384/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs=
+golang.org/x/tools v0.0.0-20190422233926-fe54fb35175b/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs=
 golang.org/x/tools v0.0.0-20190425150028-36563e24a262/go.mod h1:RgjU9mgBXZiqYHBnxXauZ1Gv1EHHAz9KjViQ78xBX0Q=
 golang.org/x/tools v0.0.0-20190506145303-2d16b83fe98c/go.mod h1:RgjU9mgBXZiqYHBnxXauZ1Gv1EHHAz9KjViQ78xBX0Q=
 golang.org/x/tools v0.0.0-20190524140312-2c0ae7006135/go.mod h1:RgjU9mgBXZiqYHBnxXauZ1Gv1EHHAz9KjViQ78xBX0Q=

go.sum

@@ -36,12 +36,14 @@ func RegisterAPIRoutes(r *gin.Engine) {
 		})
 		authGroup := v1.Group("/auth")
 
+		authControllers := auth.BuildInjector()
+
 		// 限流中间件：每小时限流，作为参考 Github API 每小时最多 60 个请求（根据 IP）
 		// 测试时，可以调高一点
 		authGroup.Use(middlewares.LimitIP("1000-H"))
 		{
 
-			suc := new(auth.SignupController)
+			suc := authControllers.SignupController
 			// 判断手机是否已注册
 			authGroup.POST("/signup/phone/exist", middlewares.GuestJWT(), middlewares.LimitPerRoute("60-H"), suc.IsPhoneExist)
 			// 判断 Email 是否已注册
@@ -52,15 +54,15 @@ func RegisterAPIRoutes(r *gin.Engine) {
 			authGroup.POST("/signup/using-email", middlewares.GuestJWT(), suc.SignupUsingEmail)
 
 			// 发送验证码
-			vcc := new(auth.VerifyCodeController)
+			vcc := authControllers.VerifyCodeController
 			// 图片验证码，需要加限流
 			authGroup.POST("/verify-codes/captcha", middlewares.LimitPerRoute("50-H"), vcc.ShowCaptcha)
 			// 发送短信验证码
 			authGroup.POST("/verify-codes/phone", middlewares.LimitPerRoute("20-H"), vcc.SendUsingPhone)
 			// 发送邮箱验证码
 			authGroup.POST("/verify-codes/email", middlewares.LimitPerRoute("20-H"), vcc.SendUsingEmail)
 
-			lgc := new(auth.LoginController)
+			lgc := authControllers.LoginController
 			// 使用手机号，短信验证码进行登录
 			authGroup.POST("/login/using-phone", middlewares.GuestJWT(), middlewares.LimitPerRoute("60-H"), lgc.LoginByPhone)
 			// 支持手机号，Email 和 用户名
@@ -69,7 +71,7 @@ func RegisterAPIRoutes(r *gin.Engine) {
 			authGroup.POST("/login/refresh-token", middlewares.AuthJWT(), lgc.RefreshToken)
 
 			// 重置密码
-			pwc := new(auth.PasswordController)
+			pwc := authControllers.PasswordController
 			authGroup.POST("/password-reset/using-phone", middlewares.LimitPerRoute("60-H"), middlewares.GuestJWT(), pwc.ResetByPhone)
 			authGroup.POST("/password-reset/using-email", middlewares.LimitPerRoute("60-H"), middlewares.GuestJWT(), pwc.ResetByEmail)
 		}