- 重构RefreshOrderRealMobile

business/jxstore/misc/misc.go

@@ -1,11 +1,17 @@
 package misc
 
 import (
+	"fmt"
+	"strings"
 	"time"
 
 	"git.rosy.net.cn/baseapi/utils"
+	"git.rosy.net.cn/jx-callback/business/jxutils"
 	"git.rosy.net.cn/jx-callback/business/jxutils/jxcontext"
-	"git.rosy.net.cn/jx-callback/business/partner/purchase/ebai"
+	"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/globals"
 )
 
@@ -18,8 +24,56 @@ func StartRefreshEbaiRealMobile() {
 }
 
 func RefreshEbaiRealMobile() {
-	ebai.CurPurchaseHandler.RefreshRealMobile(jxcontext.AdminCtx, time.Now().Add(-6*time.Hour), utils.DefaultTimeValue, true, true)
+	RefreshRealMobile(jxcontext.AdminCtx, model.VendorIDEBAI, time.Now().Add(-24*time.Hour), utils.DefaultTimeValue, true, true)
 	utils.AfterFuncWithRecover(1*time.Hour, func() {
 		RefreshEbaiRealMobile()
 	})
 }
+
+func RefreshRealMobile(ctx *jxcontext.Context, vendorID int, fromTime, toTime time.Time, isAsync, isContinueWhenError bool) (hint string, err error) {
+	handler := partner.GetPurchasePlatformFromVendorID(vendorID)
+	if handler == nil {
+		return "", fmt.Errorf("不合法的vendorID:%d", vendorID)
+	}
+	sql := `
+	SELECT *
+	FROM goods_order
+	WHERE vendor_id = ? AND consignee_mobile2 = '' AND order_created_at <= ?
+	`
+	sqlParams := []interface{}{
+		vendorID,
+		time.Now().Add(-4 * time.Hour), // 最近的刷新意义不大
+	}
+	if !utils.IsTimeZero(fromTime) {
+		sql += " AND order_created_at >= ?"
+		sqlParams = append(sqlParams, fromTime)
+	}
+	if !utils.IsTimeZero(toTime) {
+		sql += " AND order_created_at <= ?"
+		sqlParams = append(sqlParams, toTime)
+	}
+	var orderList []*model.GoodsOrder
+	db := dao.GetDB()
+	if err = dao.GetRows(db, &orderList, sql, sqlParams...); err == nil && len(orderList) > 0 {
+		task := tasksch.NewParallelTask("misc RefreshRealMobile", tasksch.NewParallelConfig().SetIsContinueWhenError(isContinueWhenError), ctx,
+			func(task *tasksch.ParallelTask, batchItemList []interface{}, params ...interface{}) (retVal interface{}, err error) {
+				order := batchItemList[0].(*model.GoodsOrder)
+				mobile, err2 := handler.GetOrderRealMobile(ctx, order)
+				if err = err2; err == nil {
+					if !jxutils.IsMobileFake(mobile) && strings.Index(order.ConsigneeMobile, mobile) == -1 {
+						order.ConsigneeMobile2 = mobile
+						_, err = dao.UpdateEntity(db, order, "ConsigneeMobile2")
+					}
+				} else {
+					globals.SugarLogger.Infof("RefreshRealMobile orderID:%s failed with error:%v", order.VendorOrderID, err)
+				}
+				return nil, err
+			}, orderList)
+		tasksch.HandleTask(task, nil, true).Run()
+		hint = task.ID
+		if !isAsync {
+			_, err = task.GetResult(0)
+		}
+	}
+	return hint, err
+}