Merge branch 'jdshop' of https://e.coding.net/rosydev/jx-callback into jdshop

controllers/cms_sku.go

@@ -11,6 +11,7 @@ import (
 	"git.rosy.net.cn/jx-callback/business/partner/purchase/tiktok_store"
 	"git.rosy.net.cn/jx-callback/globals"
 	"github.com/astaxie/beego/server/web"
+	"time"
 )
 
 type SkuController struct {
@@ -831,6 +832,13 @@ func (c *SkuController) BatchSetRestockingPrice() {
 	})
 }
 
+type CategoryList struct {
+	DdId    string `json:"dd_id"`
+	Id      int    `json:"id"`
+	NameId  int    `json:"name_id"`
+	StoreId int    `json:"store_id"`
+}
+
 // @Title 抖店商品类目修改
 // @Description 抖店商品类目修改
 // @Success 200 {object} controllers.CallResult
@@ -838,18 +846,69 @@ func (c *SkuController) BatchSetRestockingPrice() {
 // @router /UpdateTiktokCategory [post]
 func (c *SkuController) UpdateTiktokCategory() {
 	c.callUpdateTiktokCategory(func(params *tSkuUpdateTiktokCategoryParams) (interface{}, string, error) {
-		data := tiktok_store.GetProductAuditList("57939570")
-		globals.SugarLogger.Debugf("==========111 %s", utils.Format4Output(data, false))
-		for k, v := range data {
-			vendorCatgory, err := cms.GetVendorCategoriesByName(14, v)
-			if err != nil || vendorCatgory == nil {
-				data[k] = v
-				globals.SugarLogger.Debugf("err := %s", err)
-				continue
+		var page int64 = 1
+		var pageSize int64 = 100
+		var count int64 = 0
+		for {
+			data1, total := tiktok_store.GetProductAuditList("57939570", page, pageSize)
+			ddIDList := make([]string, 0, 0)
+			skuList := make([]CategoryList, 0, 0)
+			data2 := make(map[string]string, 0)
+			db := dao.GetDB()
+
+			for k, v := range data1 {
+				vendorCatgory, _ := cms.GetVendorCategoriesByName(14, v)
+				if vendorCatgory == nil || vendorCatgory.VendorCategoryID == "" {
+					continue
+				}
+				ddIDList = append(ddIDList, k)
+				data2[k] = vendorCatgory.VendorCategoryID
+			}
+
+			sql := " SELECT b.dd_id,b.store_id,s.id,s.name_id FROM store_sku_bind b INNER JOIN sku s ON b.sku_id = s.id AND s.deleted_at = ? WHERE b.dd_id IN (" + dao.GenQuestionMarks(len(ddIDList)) + ") AND b.deleted_at = ?"
+			param := []interface{}{utils.DefaultTimeValue, ddIDList, utils.DefaultTimeValue}
+			if err := dao.GetRows(db, &skuList, sql, param...); err != nil {
+				return nil, "", err
+			}
+			for _, v := range skuList {
+				count++
+				item := &model.SkuVendorCategoryMap{
+					ModelIDCULD: model.ModelIDCULD{
+						CreatedAt:    time.Now(),
+						UpdatedAt:    time.Now(),
+						DeletedAt:    utils.DefaultTimeValue,
+						LastOperator: "18981810340",
+					},
+					NameID:   v.NameId,
+					VendorID: model.VendorIDDD,
+				}
+
+				item.VendorCategoryID = data2[v.DdId]
+				skuVendorCategoryMap := &model.SkuVendorCategoryMap{}
+				if err := dao.GetRow(db, skuVendorCategoryMap, `SELECT * FROM sku_vendor_category_map WHERE name_id = ? AND vendor_id = ? AND deleted_at = ?`, []interface{}{v.NameId, model.VendorIDDD, utils.DefaultTimeValue}); err != nil {
+					globals.SugarLogger.Debugf("====================err3 %s", err)
+				}
+				if skuVendorCategoryMap != nil && skuVendorCategoryMap.ID > 0 {
+					if _, err := dao.UpdateEntity(db, skuVendorCategoryMap, "VendorCategoryID"); err != nil {
+						globals.SugarLogger.Debugf("====================err4 %s", err)
+					}
+				} else {
+					if err := dao.CreateEntity(db, item); err != nil {
+						globals.SugarLogger.Debugf("====================err5 %s", err)
+					}
+				}
+				go func() {
+					time.Sleep(500 * time.Millisecond)
+					hit, err := cms.CurVendorSync.SyncStoresSkus(params.Ctx, nil, 0, db, []int{14}, []int{v.StoreId}, []int{v.Id}, true, false, true)
+					globals.SugarLogger.Debugf("===============hit:%s err: %s", hit, err)
+				}()
+			}
+			if page*pageSize < total {
+				page++
+			} else {
+				break
 			}
-			data[k] = vendorCatgory.VendorCategoryID
 		}
-		globals.SugarLogger.Debugf("==========222 %s", utils.Format4Output(data, false))
-		return data, "", nil
+		return count, "", nil
 	})
 }