test push

db/enterprise.go

@@ -23,7 +23,7 @@ func CreateEnterprise(ctx context.Context, newEnterprise gorm_model.Enterprise)
 	return &newEnterprise.EnterpriseID, nil
 }
 
-//用户ID查找
+// 用户ID查找
 func GetEnterpriseByUID(ctx context.Context, userID int64) (*gorm_model.Enterprise, error) {
 	db := GetReadDB(ctx)
 	enterprise := gorm_model.Enterprise{}
@@ -215,3 +215,15 @@ func UpdateEnterprise(ctx context.Context, EnterpriseID string, BusinessName str
 	}
 	return nil
 }
+
+// CreateEnterpriseSubAccount ToDo
+// CreateEnterpriseSubAccount 创建商家子账号
+func CreateEnterpriseSubAccount(ctx context.Context, EnterpriseID string, BusinessName string) error {
+	db := GetReadDB(ctx)
+	err := db.Model(gorm_model.Enterprise{}).Where("enterprise_id=?", EnterpriseID).Update("business_name", BusinessName).Error
+	if err != nil {
+		fmt.Println("Update Enterprise Failed!")
+		return err
+	}
+	return nil
+}

db/user.go

@@ -17,7 +17,7 @@ func CreateUser(ctx context.Context, user gorm_model.YounggeeUser) (*int64, erro
 	return &user.ID, nil
 }
 
-//GetUserByPhone 查不到返回空
+// GetUserByPhone 查不到返回空
 func GetUserByPhone(ctx context.Context, phone string) (*gorm_model.YounggeeUser, error) {
 	db := GetReadDB(ctx)
 	user := &gorm_model.YounggeeUser{}

handler/addNewSubAccount.go

@@ -0,0 +1,43 @@
+package handler
+
+import (
+	"fmt"
+	"github.com/gin-gonic/gin"
+	"youngee_b_api/model/http_model"
+)
+
+func WrapAddNewSubAccountHandler(ctx *gin.Context) {
+	handler := newAddNewSubAccountHandler(ctx)
+	baseRun(handler)
+}
+
+func newAddNewSubAccountHandler(ctx *gin.Context) *AddNewSubAccountHandler {
+	return &AddNewSubAccountHandler{
+		req:  http_model.NewRegisterRequest(),
+		resp: http_model.NewRegisterResponse(),
+		ctx:  ctx,
+	}
+}
+
+type AddNewSubAccountHandler struct {
+	req  *http_model.RegisterRequest
+	resp *http_model.CommonResponse
+	ctx  *gin.Context
+}
+
+func (h *AddNewSubAccountHandler) getRequest() interface{} {
+	return h.req
+}
+func (h *AddNewSubAccountHandler) getContext() *gin.Context {
+	return h.ctx
+}
+func (h *AddNewSubAccountHandler) getResponse() interface{} {
+	return h.resp
+}
+func (h *AddNewSubAccountHandler) run() {
+	fmt.Println("AddNewSubAccountHandler Running")
+}
+
+func (h *AddNewSubAccountHandler) checkParam() error {
+	return nil
+}

route/init.go

@@ -13,8 +13,8 @@ import (
 
 func InitRoute(r *gin.Engine) {
 	r.GET("/swagger/*any", ginSwagger.WrapHandler(swaggerFiles.Handler))
-
 	r.POST("/register", handler.WrapRegisterHandler)
+	r.POST("/addNewSubAccount", handler.WrapAddNewSubAccountHandler)
 	r.POST("/sendCode", handler.WrapSendCodeHandler)
 	r.POST("/login", handler.WrapCodeLoginHandler)
 	r.GET("/test/ping", func(c *gin.Context) {

service/enterprise.go

@@ -116,3 +116,30 @@ func (*enterprise) CreateEnterprise(ctx context.Context, phone string) (*http_mo
 		return res, nil
 	}
 }
+
+// CreateEnterpriseSubUser 创建商家子账号
+func (*enterprise) CreateEnterpriseSubUser(ctx context.Context, newEnterprise http_model.RegisterRequest) (*http_model.RegisterData, error) {
+	// 1. 新建User
+	user := gorm_model.YounggeeUser{
+		Phone:         newEnterprise.Phone,
+		User:          "1001",
+		Username:      newEnterprise.BusinessName,
+		Password:      "1001",
+		RealName:      newEnterprise.RealName,
+		Role:          "3",
+		Email:         newEnterprise.Email,
+		LastLoginTime: time.Now().UTC().Local(),
+	}
+	userId, err := db.CreateUser(ctx, user)
+
+	// 2. 新建商家子账号
+	if err != nil {
+		log.Infof("[CreateEnterpriseUser] fail,err:%+v", err)
+		fmt.Println("User创建失败")
+		return nil, err
+	} else {
+		fmt.Println("User创建成功，开始创建商家子账号")
+		fmt.Println(userId)
+	}
+	return nil, err
+}