diff --git a/business/jxstore/initdata/initdata.go b/business/jxstore/initdata/initdata.go index b374eddf2..2d43df39c 100644 --- a/business/jxstore/initdata/initdata.go +++ b/business/jxstore/initdata/initdata.go @@ -12,6 +12,8 @@ import ( "git.rosy.net.cn/jx-callback/business/jxutils/tasksch" "git.rosy.net.cn/jx-callback/business/model" "git.rosy.net.cn/jx-callback/business/model/dao" + "git.rosy.net.cn/jx-callback/business/partner" + "git.rosy.net.cn/jx-callback/business/partner/purchase/mtwm" "git.rosy.net.cn/jx-callback/globals/api" ) @@ -161,3 +163,32 @@ func InitSkuName(ctx *jxcontext.Context, isForce, isAsync, isContinueWhenError b } return hint, err } + +func InitVendorCategory(ctx *jxcontext.Context) (err error) { + if handler, ok := partner.PurchasePlatformHandlers[model.VendorIDMTWM].(*mtwm.PurchaseHandler); ok { + cats, err2 := handler.GetVendorCategories() + if err2 != nil { + return err2 + } + + db := dao.GetDB() + dao.Begin(db) + defer dao.Rollback(db) + sql := ` + DELETE + FROM sku_vendor_category + WHERE vendor_id = ? + ` + if _, err = dao.ExecuteSQL(db, sql, model.VendorIDMTWM); err != nil { + return err + } + for _, cat := range cats { + dao.WrapAddIDCULEntity(cat, ctx.GetUserName()) + if err = dao.CreateEntity(db, cat); err != nil { + return err + } + } + dao.Commit(db) + } + return err +} diff --git a/business/partner/purchase/mtwm/mtwm.go b/business/partner/purchase/mtwm/mtwm.go index 34e86d5af..0089519c5 100644 --- a/business/partner/purchase/mtwm/mtwm.go +++ b/business/partner/purchase/mtwm/mtwm.go @@ -49,7 +49,7 @@ func (p *PurchaseHandler) GetVendorCategories() (vendorCats []*model.SkuVendorCa cat := &model.SkuVendorCategory{ VendorID: model.VendorIDMTWM, Name: utils.Interface2String(v["name"]), - Level: int(utils.MustInterface2Int64(v["level"])), + Level: i + 1, //int(utils.MustInterface2Int64(v["level"])), } cat.LastOperator = "builder" vendorCats = append(vendorCats, cat) diff --git a/controllers/init_data.go b/controllers/init_data.go index f45609485..b3ea70484 100644 --- a/controllers/init_data.go +++ b/controllers/init_data.go @@ -37,3 +37,16 @@ func (c *InitDataController) InitSkuName() { return retVal, "", err }) } + +// @Title 初始化vendor category +// @Description 初始化vendor category(当前只有美团外卖的通过这个设置) +// @Param token header string true "认证token" +// @Success 200 {object} controllers.CallResult +// @Failure 200 {object} controllers.CallResult +// @router /InitVendorCategory [post] +func (c *InitDataController) InitVendorCategory() { + c.callInitVendorCategory(func(params *tInitdataInitVendorCategoryParams) (retVal interface{}, errCode string, err error) { + err = initdata.InitVendorCategory(params.Ctx) + return retVal, "", err + }) +} diff --git a/routers/commentsRouter_controllers.go b/routers/commentsRouter_controllers.go index 31dec2d24..e9429833d 100644 --- a/routers/commentsRouter_controllers.go +++ b/routers/commentsRouter_controllers.go @@ -167,6 +167,14 @@ func init() { MethodParams: param.Make(), Params: nil}) + beego.GlobalControllerRouter["git.rosy.net.cn/jx-callback/controllers:InitDataController"] = append(beego.GlobalControllerRouter["git.rosy.net.cn/jx-callback/controllers:InitDataController"], + beego.ControllerComments{ + Method: "InitVendorCategory", + Router: `/InitVendorCategory`, + AllowHTTPMethods: []string{"post"}, + MethodParams: param.Make(), + Params: nil}) + beego.GlobalControllerRouter["git.rosy.net.cn/jx-callback/controllers:OrderController"] = append(beego.GlobalControllerRouter["git.rosy.net.cn/jx-callback/controllers:OrderController"], beego.ControllerComments{ Method: "CreateWaybillOnProviders",