jx-callback/business/jxstore/cms/system_store_sku.go

package cms

import (
	"fmt"
	"git.rosy.net.cn/baseapi/platformapi/mtwmapi"
	"git.rosy.net.cn/baseapi/utils"
	"git.rosy.net.cn/jx-callback/business/jxutils/jxcontext"
	"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/business/partner/putils"
	"time"
)

// CopyOnStoreSkuToOther 将一个美团门店分类和商品复制到另一个门店
func CopyOnStoreSkuToOther(ctx *jxcontext.Context, fromVendorStoreId, toStoreId int, isAsync bool) (hint string, err error) {
	var (
		db = dao.GetDB()
	)

	// 门店api加载
	// fromStore, err := dao.GetStoreDetail(db, fromStoreId, model.VendorIDMTWM, "")
	if err != nil {
		return "", err
	}
	toStore, err := dao.GetStoreDetail(db, toStoreId, model.VendorIDMTWM, "")
	if err != nil {
		return "", err
	}
	//fromApi := partner.CurAPIManager.GetAPI(model.VendorIDMTWM, toStore.VendorOrgCode).(*mtwmapi.API)
	toApi := partner.CurAPIManager.GetAPI(model.VendorIDMTWM, toStore.VendorOrgCode).(*mtwmapi.API)

	taskName := fmt.Sprintf("将平台门店[%d],分类和商品复制到[%d]", fromVendorStoreId, toStoreId)
	config := tasksch.NewParallelConfig().SetParallelCount(1).SetIsContinueWhenError(false)
	work := func(task *tasksch.ParallelTask, batchItemList []interface{}, params ...interface{}) (retVal interface{}, err error) {
		step := batchItemList[0].(int)
		switch step {
		case 1:
			// 1.加载门店商品,删除商品.当分类下没有商品时.删除分类
			errs := LoadingStoreSkuList(ctx, toApi, toStore.VendorStoreID)
			if errs != nil && len(errs) > 0 {
				return nil, errs[0]
			}
		case 2:
			// 同步分类
			fromCategoryList, err := toApi.RetailCatList(utils.Int2Str(fromVendorStoreId))
			if err != nil {
				return nil, err
			}
			for _, v := range fromCategoryList {
				if err := toApi.RetailCatUpdate(toStore.VendorStoreID, v.Name, &mtwmapi.Param4UpdateCat{
					CategoryCode: v.Code,
					Sequence:     v.Sequence,
				}); err != nil {
					return nil, err
				}
				if v.Children != nil && len(v.Children) != 0 {
					for _, c := range v.Children {
						if err := toApi.RetailCatUpdate(toStore.VendorStoreID, v.Name, &mtwmapi.Param4UpdateCat{
							CategoryCode:          v.Code,
							SecondaryCategoryCode: c.Code,
							SecondaryCategoryName: c.Name,
							Sequence:              c.Sequence,
						}); err != nil {
							return nil, err
						}
					}
				}
			}

		case 3:
			// 同步商品
			fromFoodList, err1 := toApi.RetailListAll(utils.Int2Str(fromVendorStoreId))
			if err1 != nil {
				return nil, err1
			}
			if err := BatchInitData(ctx, fromFoodList, toApi, toStore.VendorStoreID); err != nil {
				return nil, err
			}
		}
		return
	}
	task := tasksch.NewParallelTask(taskName, config, ctx, work, []int{1, 2, 3})
	tasksch.HandleTask(task, nil, true).Run()
	if !isAsync {
		_, err = task.GetResult(0)
		hint = "1"
	} else {
		hint = task.ID
	}
	return hint, err
}

func LoadingStoreSkuList(ctx *jxcontext.Context, api *mtwmapi.API, poiCode string) (err []error) {
	foodList, err1 := api.RetailListAll(poiCode)
	if err1 != nil {
		return append(err, err1)
	}

	i := 0
	for _, v := range foodList {
		err1 := api.RetailDelete(ctx.GetTrackInfo(), poiCode, v.AppFoodCode)
		if err1 != nil {
			err = append(err, err1)
		}
		if i%40 == 0 {
			time.Sleep(200 * time.Millisecond)
		}
		i++
	}

	return
}

// BatchInitData 批量创建商品
func BatchInitData(ctx *jxcontext.Context, fromSku []*mtwmapi.AppFood, toApi *mtwmapi.API, vendorStoreID string) error {
	foodDataList := make([]map[string]interface{}, len(fromSku))
	for i, storeSku := range fromSku {
		foodData := make(map[string]interface{})
		foodDataList[i] = foodData
		foodData[mtwmapi.KeyAppFoodCode] = storeSku.AppFoodCode
		foodData["skus"] = storeSku.Skus
		foodData["name"] = utils.LimitUTF8StringLen(storeSku.Name, mtwmapi.MaxSkuNameCharCount)
		foodData["description"] = storeSku.Description
		foodData["price"] = storeSku.Price
		if storeSku.MinOrderCount != 0 {
			foodData["min_order_count"] = storeSku.MinOrderCount
		} else {
			foodData["min_order_count"] = 1
		}
		foodData["unit"] = storeSku.Unit
		attr := mtwm.SwitchAttr(int64(storeSku.TagID))
		if attr != "" {
			foodData["common_attr_value"] = attr
		}
		foodData["category_code"] = storeSku.CategoryCode
		foodData["category_name"] = storeSku.CategoryName
		foodData["is_sold_out"] = storeSku.IsSoldOut
		foodData["picture"] = storeSku.Picture
		foodData["picture_contents"] = storeSku.PictureContents
		foodData["sequence"] = storeSku.Sequence
		foodData["tag_id"] = storeSku.TagID
	}

	failedFoodList, err2 := toApi.RetailBatchInitData(ctx.GetTrackInfo(), vendorStoreID, foodDataList)
	if err := err2; err == nil {
		if err = putils.GenPartialFailedErr(failedFoodList, len(failedFoodList)); err != nil {
			return err
		}
	} else if err2 != nil && len(failedFoodList) == 0 {
		if errExt, ok := err2.(*utils.ErrorWithCode); ok {
			return utils.UnmarshalUseNumber([]byte(errExt.ErrMsg()), &failedFoodList)
		}
	}
	return nil
}