三端用户手机号联通，增加临时用户

app/main/api/internal/handler/agent/getagentpromotionqrcodehandler.go

@@ -0,0 +1,36 @@
+package agent
+
+import (
+	"net/http"
+
+	"qnc-server/app/main/api/internal/logic/agent"
+	"qnc-server/app/main/api/internal/svc"
+	"qnc-server/app/main/api/internal/types"
+	"qnc-server/common/result"
+	"qnc-server/pkg/lzkit/validator"
+
+	"github.com/zeromicro/go-zero/rest/httpx"
+)
+
+func GetAgentPromotionQrcodeHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		var req types.GetAgentPromotionQrcodeReq
+		if err := httpx.Parse(r, &req); err != nil {
+			result.ParamErrorResult(r, w, err)
+			return
+		}
+		if err := validator.Validate(req); err != nil {
+			result.ParamValidateErrorResult(r, w, err)
+			return
+		}
+
+		// 注意：这里传入了 ResponseWriter，用于直接写入图片数据
+		l := agent.NewGetAgentPromotionQrcodeLogic(r.Context(), svcCtx, w)
+		err := l.GetAgentPromotionQrcode(&req)
+		if err != nil {
+			// 如果处理过程中出错，返回JSON错误响应
+			result.HttpResult(r, w, nil, err)
+		}
+		// 成功时，图片数据已经通过logic直接写入ResponseWriter，不需要额外处理
+	}
+}

app/main/api/internal/handler/routes.go

@@ -518,111 +518,141 @@ func RegisterHandlers(server *rest.Server, serverCtx *svc.ServiceContext) {
 		[]rest.Route{
 			{
 				Method:  http.MethodGet,
-				Path:    "/audit/status",
-				Handler: agent.GetAgentAuditStatusHandler(serverCtx),
-			},
-			{
-				Method:  http.MethodPost,
-				Path:    "/generating_link",
-				Handler: agent.GeneratingLinkHandler(serverCtx),
+				Path:    "/promotion/qrcode",
+				Handler: agent.GetAgentPromotionQrcodeHandler(serverCtx),
 			},
+		},
+		rest.WithPrefix("/api/v1/agent"),
+	)
+
+	server.AddRoutes(
+		[]rest.Route{
 			{
 				Method:  http.MethodGet,
 				Path:    "/info",
 				Handler: agent.GetAgentInfoHandler(serverCtx),
 			},
-			{
-				Method:  http.MethodGet,
-				Path:    "/product_config",
-				Handler: agent.GetAgentProductConfigHandler(serverCtx),
-			},
-			{
-				Method:  http.MethodPost,
-				Path:    "/real_name",
-				Handler: agent.AgentRealNameHandler(serverCtx),
-			},
-			{
-				Method:  http.MethodGet,
-				Path:    "/subordinate/contribution/detail",
-				Handler: agent.GetAgentSubordinateContributionDetailHandler(serverCtx),
-			},
-			{
-				Method:  http.MethodGet,
-				Path:    "/subordinate/list",
-				Handler: agent.GetAgentSubordinateListHandler(serverCtx),
-			},
-		},
-		rest.WithJwt(serverCtx.Config.JwtAuth.AccessSecret),
-		rest.WithPrefix("/api/v1/agent"),
-	)
-
-	server.AddRoutes(
-		[]rest.Route{
-			{
-				Method:  http.MethodPost,
-				Path:    "/membership/save_user_config",
-				Handler: agent.SaveAgentMembershipUserConfigHandler(serverCtx),
-			},
-			{
-				Method:  http.MethodGet,
-				Path:    "/membership/user_config",
-				Handler: agent.GetAgentMembershipProductConfigHandler(serverCtx),
-			},
-		},
-		rest.WithJwt(serverCtx.Config.JwtAuth.AccessSecret),
-		rest.WithPrefix("/api/v1/agent"),
-	)
-
-	server.AddRoutes(
-		[]rest.Route{
-			{
-				Method:  http.MethodGet,
-				Path:    "/commission",
-				Handler: agent.GetAgentCommissionHandler(serverCtx),
-			},
-			{
-				Method:  http.MethodPost,
-				Path:    "/membership/activate",
-				Handler: agent.ActivateAgentMembershipHandler(serverCtx),
-			},
 			{
 				Method:  http.MethodGet,
 				Path:    "/revenue",
 				Handler: agent.GetAgentRevenueInfoHandler(serverCtx),
 			},
-			{
-				Method:  http.MethodGet,
-				Path:    "/rewards",
-				Handler: agent.GetAgentRewardsHandler(serverCtx),
-			},
-			{
-				Method:  http.MethodGet,
-				Path:    "/withdrawal",
-				Handler: agent.GetAgentWithdrawalHandler(serverCtx),
-			},
-			{
-				Method:  http.MethodPost,
-				Path:    "/withdrawal",
-				Handler: agent.AgentWithdrawalHandler(serverCtx),
-			},
 		},
 		rest.WithJwt(serverCtx.Config.JwtAuth.AccessSecret),
 		rest.WithPrefix("/api/v1/agent"),
 	)
 
 	server.AddRoutes(
-		[]rest.Route{
-			{
-				Method:  http.MethodPost,
-				Path:    "/apply",
-				Handler: agent.ApplyForAgentHandler(serverCtx),
-			},
-			{
-				Method:  http.MethodGet,
-				Path:    "/link",
-				Handler: agent.GetLinkDataHandler(serverCtx),
-			},
-		},
+		rest.WithMiddlewares(
+			[]rest.Middleware{serverCtx.UserAuthInterceptor},
+			[]rest.Route{
+				{
+					Method:  http.MethodGet,
+					Path:    "/audit/status",
+					Handler: agent.GetAgentAuditStatusHandler(serverCtx),
+				},
+				{
+					Method:  http.MethodPost,
+					Path:    "/generating_link",
+					Handler: agent.GeneratingLinkHandler(serverCtx),
+				},
+				{
+					Method:  http.MethodGet,
+					Path:    "/product_config",
+					Handler: agent.GetAgentProductConfigHandler(serverCtx),
+				},
+				{
+					Method:  http.MethodPost,
+					Path:    "/real_name",
+					Handler: agent.AgentRealNameHandler(serverCtx),
+				},
+				{
+					Method:  http.MethodGet,
+					Path:    "/subordinate/contribution/detail",
+					Handler: agent.GetAgentSubordinateContributionDetailHandler(serverCtx),
+				},
+				{
+					Method:  http.MethodGet,
+					Path:    "/subordinate/list",
+					Handler: agent.GetAgentSubordinateListHandler(serverCtx),
+				},
+			}...,
+		),
+		rest.WithJwt(serverCtx.Config.JwtAuth.AccessSecret),
+		rest.WithPrefix("/api/v1/agent"),
+	)
+
+	server.AddRoutes(
+		rest.WithMiddlewares(
+			[]rest.Middleware{serverCtx.UserAuthInterceptor},
+			[]rest.Route{
+				{
+					Method:  http.MethodPost,
+					Path:    "/membership/save_user_config",
+					Handler: agent.SaveAgentMembershipUserConfigHandler(serverCtx),
+				},
+				{
+					Method:  http.MethodGet,
+					Path:    "/membership/user_config",
+					Handler: agent.GetAgentMembershipProductConfigHandler(serverCtx),
+				},
+			}...,
+		),
+		rest.WithJwt(serverCtx.Config.JwtAuth.AccessSecret),
+		rest.WithPrefix("/api/v1/agent"),
+	)
+
+	server.AddRoutes(
+		rest.WithMiddlewares(
+			[]rest.Middleware{serverCtx.UserAuthInterceptor},
+			[]rest.Route{
+				{
+					Method:  http.MethodGet,
+					Path:    "/commission",
+					Handler: agent.GetAgentCommissionHandler(serverCtx),
+				},
+				{
+					Method:  http.MethodPost,
+					Path:    "/membership/activate",
+					Handler: agent.ActivateAgentMembershipHandler(serverCtx),
+				},
+				{
+					Method:  http.MethodGet,
+					Path:    "/rewards",
+					Handler: agent.GetAgentRewardsHandler(serverCtx),
+				},
+				{
+					Method:  http.MethodGet,
+					Path:    "/withdrawal",
+					Handler: agent.GetAgentWithdrawalHandler(serverCtx),
+				},
+				{
+					Method:  http.MethodPost,
+					Path:    "/withdrawal",
+					Handler: agent.AgentWithdrawalHandler(serverCtx),
+				},
+			}...,
+		),
+		rest.WithJwt(serverCtx.Config.JwtAuth.AccessSecret),
+		rest.WithPrefix("/api/v1/agent"),
+	)
+
+	server.AddRoutes(
+		rest.WithMiddlewares(
+			[]rest.Middleware{serverCtx.AuthInterceptor},
+			[]rest.Route{
+				{
+					Method:  http.MethodPost,
+					Path:    "/apply",
+					Handler: agent.ApplyForAgentHandler(serverCtx),
+				},
+				{
+					Method:  http.MethodGet,
+					Path:    "/link",
+					Handler: agent.GetLinkDataHandler(serverCtx),
+				},
+			}...,
+		),
 		rest.WithPrefix("/api/v1/agent"),
 	)
 
@@ -714,7 +744,7 @@ func RegisterHandlers(server *rest.Server, serverCtx *svc.ServiceContext) {
 
 	server.AddRoutes(
 		rest.WithMiddlewares(
-			[]rest.Middleware{serverCtx.SourceInterceptor},
+			[]rest.Middleware{serverCtx.UserAuthInterceptor},
 			[]rest.Route{
 				{
 					Method:  http.MethodPost,
@@ -743,18 +773,21 @@ func RegisterHandlers(server *rest.Server, serverCtx *svc.ServiceContext) {
 	)
 
 	server.AddRoutes(
-		[]rest.Route{
-			{
-				Method:  http.MethodGet,
-				Path:    "/:id",
-				Handler: product.GetProductByIDHandler(serverCtx),
-			},
-			{
-				Method:  http.MethodGet,
-				Path:    "/en/:product_en",
-				Handler: product.GetProductByEnHandler(serverCtx),
-			},
-		},
+		rest.WithMiddlewares(
+			[]rest.Middleware{serverCtx.UserAuthInterceptor},
+			[]rest.Route{
+				{
+					Method:  http.MethodGet,
+					Path:    "/:id",
+					Handler: product.GetProductByIDHandler(serverCtx),
+				},
+				{
+					Method:  http.MethodGet,
+					Path:    "/en/:product_en",
+					Handler: product.GetProductByEnHandler(serverCtx),
+				},
+			}...,
+		),
 		rest.WithJwt(serverCtx.Config.JwtAuth.AccessSecret),
 		rest.WithPrefix("/api/v1/product"),
 	)
@@ -791,75 +824,81 @@ func RegisterHandlers(server *rest.Server, serverCtx *svc.ServiceContext) {
 	)
 
 	server.AddRoutes(
-		[]rest.Route{
-			{
-				// query service
-				Method:  http.MethodPost,
-				Path:    "/query/service/:product",
-				Handler: query.QueryServiceHandler(serverCtx),
-			},
-		},
+		rest.WithMiddlewares(
+			[]rest.Middleware{serverCtx.UserAuthInterceptor},
+			[]rest.Route{
+				{
+					// query service
+					Method:  http.MethodPost,
+					Path:    "/query/service/:product",
+					Handler: query.QueryServiceHandler(serverCtx),
+				},
+			}...,
+		),
 		rest.WithJwt(serverCtx.Config.JwtAuth.AccessSecret),
 		rest.WithPrefix("/api/v1"),
 	)
 
 	server.AddRoutes(
-		[]rest.Route{
-			{
-				// 确认查询状态
-				Method:  http.MethodPost,
-				Path:    "/query/confirm_state",
-				Handler: query.ConfirmQueryStateHandler(serverCtx),
-			},
-			{
-				// 生成分享链接
-				Method:  http.MethodPost,
-				Path:    "/query/generate_share_link",
-				Handler: query.QueryGenerateShareLinkHandler(serverCtx),
-			},
-			{
-				// 查询列表
-				Method:  http.MethodGet,
-				Path:    "/query/list",
-				Handler: query.QueryListHandler(serverCtx),
-			},
-			{
-				// 查询详情 按订单号 付款查询时
-				Method:  http.MethodGet,
-				Path:    "/query/orderId/:order_id",
-				Handler: query.QueryDetailByOrderIdHandler(serverCtx),
-			},
-			{
-				// 查询详情 按订单号
-				Method:  http.MethodGet,
-				Path:    "/query/orderNo/:order_no",
-				Handler: query.QueryDetailByOrderNoHandler(serverCtx),
-			},
-			{
-				// 获取查询临时订单
-				Method:  http.MethodGet,
-				Path:    "/query/provisional_order/:id",
-				Handler: query.QueryProvisionalOrderHandler(serverCtx),
-			},
-			{
-				// 补查
-				Method:  http.MethodPost,
-				Path:    "/query/recheck/:query_id",
-				Handler: query.QueryRecheckHandler(serverCtx),
-			},
-			{
-				// 重试查询
-				Method:  http.MethodPost,
-				Path:    "/query/retry/:id",
-				Handler: query.QueryRetryHandler(serverCtx),
-			},
-			{
-				// 更新查询数据
-				Method:  http.MethodPost,
-				Path:    "/query/update_data",
-				Handler: query.UpdateQueryDataHandler(serverCtx),
-			},
-		},
+		rest.WithMiddlewares(
+			[]rest.Middleware{serverCtx.UserAuthInterceptor},
+			[]rest.Route{
+				{
+					// 确认查询状态
+					Method:  http.MethodPost,
+					Path:    "/query/confirm_state",
+					Handler: query.ConfirmQueryStateHandler(serverCtx),
+				},
+				{
+					// 生成分享链接
+					Method:  http.MethodPost,
+					Path:    "/query/generate_share_link",
+					Handler: query.QueryGenerateShareLinkHandler(serverCtx),
+				},
+				{
+					// 查询列表
+					Method:  http.MethodGet,
+					Path:    "/query/list",
+					Handler: query.QueryListHandler(serverCtx),
+				},
+				{
+					// 查询详情 按订单号 付款查询时
+					Method:  http.MethodGet,
+					Path:    "/query/orderId/:order_id",
+					Handler: query.QueryDetailByOrderIdHandler(serverCtx),
+				},
+				{
+					// 查询详情 按订单号
+					Method:  http.MethodGet,
+					Path:    "/query/orderNo/:order_no",
+					Handler: query.QueryDetailByOrderNoHandler(serverCtx),
+				},
+				{
+					// 获取查询临时订单
+					Method:  http.MethodGet,
+					Path:    "/query/provisional_order/:id",
+					Handler: query.QueryProvisionalOrderHandler(serverCtx),
+				},
+				{
+					// 补查
+					Method:  http.MethodPost,
+					Path:    "/query/recheck/:query_id",
+					Handler: query.QueryRecheckHandler(serverCtx),
+				},
+				{
+					// 重试查询
+					Method:  http.MethodPost,
+					Path:    "/query/retry/:id",
+					Handler: query.QueryRetryHandler(serverCtx),
+				},
+				{
+					// 更新查询数据
+					Method:  http.MethodPost,
+					Path:    "/query/update_data",
+					Handler: query.UpdateQueryDataHandler(serverCtx),
+				},
+			}...,
+		),
 		rest.WithJwt(serverCtx.Config.JwtAuth.AccessSecret),
 		rest.WithPrefix("/api/v1"),
 	)
@@ -889,30 +928,12 @@ func RegisterHandlers(server *rest.Server, serverCtx *svc.ServiceContext) {
 
 	server.AddRoutes(
 		[]rest.Route{
-			{
-				// agent mobile code login
-				Method:  http.MethodPost,
-				Path:    "/user/agent_mobile_code_login",
-				Handler: user.AgentMobileCodeLoginHandler(serverCtx),
-			},
 			{
 				// mobile code login
 				Method:  http.MethodPost,
 				Path:    "/user/mobileCodeLogin",
 				Handler: user.MobileCodeLoginHandler(serverCtx),
 			},
-			{
-				// mobile login
-				Method:  http.MethodPost,
-				Path:    "/user/mobileLogin",
-				Handler: user.MobileLoginHandler(serverCtx),
-			},
-			{
-				// register
-				Method:  http.MethodPost,
-				Path:    "/user/register",
-				Handler: user.RegisterHandler(serverCtx),
-			},
 			{
 				// wechat mini auth
 				Method:  http.MethodPost,
@@ -929,14 +950,23 @@ func RegisterHandlers(server *rest.Server, serverCtx *svc.ServiceContext) {
 		rest.WithPrefix("/api/v1"),
 	)
 
+	server.AddRoutes(
+		rest.WithMiddlewares(
+			[]rest.Middleware{serverCtx.AuthInterceptor},
+			[]rest.Route{
+				{
+					// 绑定手机号
+					Method:  http.MethodPost,
+					Path:    "/user/bindMobile",
+					Handler: user.BindMobileHandler(serverCtx),
+				},
+			}...,
+		),
+		rest.WithPrefix("/api/v1"),
+	)
+
 	server.AddRoutes(
 		[]rest.Route{
-			{
-				// 绑定手机号
-				Method:  http.MethodPost,
-				Path:    "/user/bindMobile",
-				Handler: user.BindMobileHandler(serverCtx),
-			},
 			{
 				Method:  http.MethodPost,
 				Path:    "/user/cancelOut",

app/main/api/internal/handler/user/agentmobilecodeloginhandler.go

@@ -1,30 +0,0 @@
-package user
-
-import (
-	"net/http"
-
-	"qnc-server/app/main/api/internal/logic/user"
-	"qnc-server/app/main/api/internal/svc"
-	"qnc-server/app/main/api/internal/types"
-	"qnc-server/common/result"
-	"qnc-server/pkg/lzkit/validator"
-
-	"github.com/zeromicro/go-zero/rest/httpx"
-)
-
-func AgentMobileCodeLoginHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
-	return func(w http.ResponseWriter, r *http.Request) {
-		var req types.MobileCodeLoginReq
-		if err := httpx.Parse(r, &req); err != nil {
-			result.ParamErrorResult(r, w, err)
-			return
-		}
-		if err := validator.Validate(req); err != nil {
-			result.ParamValidateErrorResult(r, w, err)
-			return
-		}
-		l := user.NewAgentMobileCodeLoginLogic(r.Context(), svcCtx)
-		resp, err := l.AgentMobileCodeLogin(&req)
-		result.HttpResult(r, w, resp, err)
-	}
-}

app/main/api/internal/handler/user/mobileloginhandler.go

@@ -1,30 +0,0 @@
-package user
-
-import (
-	"net/http"
-
-	"qnc-server/app/main/api/internal/logic/user"
-	"qnc-server/app/main/api/internal/svc"
-	"qnc-server/app/main/api/internal/types"
-	"qnc-server/common/result"
-	"qnc-server/pkg/lzkit/validator"
-
-	"github.com/zeromicro/go-zero/rest/httpx"
-)
-
-func MobileLoginHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
-	return func(w http.ResponseWriter, r *http.Request) {
-		var req types.MobileLoginReq
-		if err := httpx.Parse(r, &req); err != nil {
-			result.ParamErrorResult(r, w, err)
-			return
-		}
-		if err := validator.Validate(req); err != nil {
-			result.ParamValidateErrorResult(r, w, err)
-			return
-		}
-		l := user.NewMobileLoginLogic(r.Context(), svcCtx)
-		resp, err := l.MobileLogin(&req)
-		result.HttpResult(r, w, resp, err)
-	}
-}

app/main/api/internal/handler/user/registerhandler.go

@@ -1,30 +0,0 @@
-package user
-
-import (
-	"net/http"
-
-	"qnc-server/app/main/api/internal/logic/user"
-	"qnc-server/app/main/api/internal/svc"
-	"qnc-server/app/main/api/internal/types"
-	"qnc-server/common/result"
-	"qnc-server/pkg/lzkit/validator"
-
-	"github.com/zeromicro/go-zero/rest/httpx"
-)
-
-func RegisterHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
-	return func(w http.ResponseWriter, r *http.Request) {
-		var req types.RegisterReq
-		if err := httpx.Parse(r, &req); err != nil {
-			result.ParamErrorResult(r, w, err)
-			return
-		}
-		if err := validator.Validate(req); err != nil {
-			result.ParamValidateErrorResult(r, w, err)
-			return
-		}
-		l := user.NewRegisterLogic(r.Context(), svcCtx)
-		resp, err := l.Register(&req)
-		result.HttpResult(r, w, resp, err)
-	}
-}