- jd.FullSyncStoreSkus

- refactor mtwm.SyncLocalStoreCategory
2018-12-07 11:34:49 +08:00
parent 2210e1020b
commit bf5d9ed219
9 changed files with 213 additions and 70 deletions
--- a/business/partner/purchase/jd/store_sku.go
+++ b/business/partner/purchase/jd/store_sku.go
@@ -132,3 +132,111 @@ func (p *PurchaseHandler) SyncStoreSkus(ctx *jxcontext.Context, parentTask tasks
 	}
 	return task.ID, err
 }
+
+func (p *PurchaseHandler) FullSyncStoreSkus(ctx *jxcontext.Context, parentTask tasksch.ITask, storeID int, isAsync, isContinueWhenError bool) (hint string, err error) {
+	globals.SugarLogger.Debugf("jd FullSyncStoreSkus, storeID:%d", storeID)
+	db := dao.GetDB()
+	_, err = dao.SetStoreSkuSyncStatus(ctx, db, model.VendorIDJD, storeID, nil, model.SyncFlagModifiedMask|model.SyncFlagPriceMask|model.SyncFlagSaleMask)
+	if err != nil {
+		return "", err
+	}
+	skus, err := dao.GetFullStoreSkus(db, model.VendorIDJD, storeID)
+	if err != nil {
+		return "", err
+	}
+	return p.syncStoreSkus(ctx, parentTask, db, storeID, skus, isAsync, isContinueWhenError)
+}
+
+// todo 之后应该与SyncStoreSkus合并
+func (p *PurchaseHandler) syncStoreSkus(ctx *jxcontext.Context, parentTask tasksch.ITask, db *dao.DaoDB, storeID int, skus []*dao.StoreSkuSyncInfo, isAsync, isContinueWhenError bool) (hint string, err error) {
+	globals.SugarLogger.Debugf("jd syncStoreSkus, storeID:%d, len(skus):%d", storeID, len(skus))
+	if len(skus) == 0 {
+		return "", nil
+	}
+	storeDetail, err := dao.GetStoreDetail(db, storeID, model.VendorIDJD)
+	if err != nil {
+		return "", err
+	}
+	stationNo := storeDetail.VendorStoreID
+	task := tasksch.NewParallelTask("SyncStoresSkus京东", tasksch.NewParallelConfig().SetBatchSize(jdapi.MaxStoreSkuBatchSize).SetIsContinueWhenError(isContinueWhenError), ctx.GetUserName(), func(t *tasksch.ParallelTask, batchItemList []interface{}, params ...interface{}) (interface{}, error) {
+		var skuPriceInfoList []*jdapi.SkuPriceInfo
+		var skuVendibilityList []*jdapi.StockVendibility
+		var skuStockList []*jdapi.SkuStock
+		var batchSkuIDs []int
+		for _, v := range batchItemList {
+			storeSku := v.(*dao.StoreSkuSyncInfo)
+			alreadyAddStock := false
+			if storeSku.SkuSyncStatus&model.SyncFlagChangedMask != 0 || storeSku.BindID == 0 {
+				if storeSku.BindID != 0 {
+					batchSkuIDs = append(batchSkuIDs, storeSku.BindID)
+				}
+				if storeSku.SkuSyncStatus&(model.SyncFlagDeletedMask|model.SyncFlagNewMask) != 0 || storeSku.BindID == 0 { // 关注或取消关注
+					stock := &jdapi.SkuStock{
+						OutSkuId: utils.Int2Str(storeSku.ID),
+						StockQty: model.MaxStoreSkuStockQty,
+					}
+					if storeSku.DeletedAt != utils.DefaultTimeValue || storeSku.BindID == 0 {
+						stock.StockQty = 0
+					} else {
+						alreadyAddStock = true
+					}
+					if stock.StockQty != 0 || !storeskulock.IsJdStoreSkuLocked(stationNo, storeSku.JdID) {
+						skuStockList = append(skuStockList, stock)
+					}
+				}
+				if storeSku.SkuSyncStatus&(model.SyncFlagPriceMask|model.SyncFlagNewMask) != 0 {
+					skuPriceInfoList = append(skuPriceInfoList, &jdapi.SkuPriceInfo{
+						OutSkuId: utils.Int2Str(storeSku.ID),
+						Price:    jxutils.CaculateSkuVendorPrice(int(storeSku.Price), int(storeDetail.PricePercentage)),
+					})
+				}
+				if storeSku.SkuSyncStatus&(model.SyncFlagSaleMask|model.SyncFlagNewMask) != 0 {
+					vendibility := &jdapi.StockVendibility{
+						OutSkuId: utils.Int2Str(storeSku.ID),
+						DoSale:   true,
+					}
+					if storeSku.Status != model.StoreSkuBindStatusNormal {
+						vendibility.DoSale = false
+					} else if !alreadyAddStock { // 如果是设置可售则自动将库存加满
+						stock := &jdapi.SkuStock{
+							OutSkuId: utils.Int2Str(storeSku.ID),
+							StockQty: model.MaxStoreSkuStockQty,
+						}
+						skuStockList = append(skuStockList, stock)
+					}
+					if vendibility.DoSale || !storeskulock.IsJdStoreSkuLocked(stationNo, storeSku.JdID) {
+						skuVendibilityList = append(skuVendibilityList, vendibility)
+					}
+				}
+			}
+		}
+		// globals.SugarLogger.Debug(utils.Format4Output(skuVendibilityList, false), utils.Format4Output(skuPriceInfoList, false), utils.Format4Output(skuStockList, false))
+		if globals.EnableStoreWrite {
+			// todo 以下可以优化为并行操作
+			if len(skuVendibilityList) > 0 {
+				_, err = api.JdAPI.BatchUpdateVendibility("", stationNo, skuVendibilityList, ctx.GetUserName())
+			}
+			if err == nil && len(skuPriceInfoList) > 0 {
+				_, err = api.JdAPI.UpdateVendorStationPrice("", stationNo, skuPriceInfoList)
+			}
+			if err == nil && len(skuStockList) > 0 {
+				_, err = api.JdAPI.BatchUpdateCurrentQtys("", stationNo, skuStockList, ctx.GetUserName())
+			}
+		}
+		if err == nil && len(batchSkuIDs) > 0 {
+			db := dao.GetDB() // 多线程问题
+			sql := `
+			UPDATE store_sku_bind t1
+			SET t1.jd_sync_status = 0
+			WHERE t1.id IN (` + dao.GenQuestionMarks(len(batchSkuIDs)) + ")"
+			_, err = dao.ExecuteSQL(db, sql, batchSkuIDs)
+		}
+		return nil, err
+	}, skus)
+	ctx.SetTaskOrAddChild(task, parentTask)
+	task.Run()
+	if !isAsync {
+		_, err = task.GetResult(0)
+	}
+	return task.ID, err
+}