315 lines
10 KiB
Go
315 lines
10 KiB
Go
package misc
|
||
|
||
import (
|
||
"fmt"
|
||
"strings"
|
||
"sync"
|
||
"time"
|
||
|
||
"git.rosy.net.cn/jx-callback/business/jxstore/report"
|
||
|
||
"git.rosy.net.cn/baseapi/utils"
|
||
"git.rosy.net.cn/baseapi/utils/errlist"
|
||
"git.rosy.net.cn/jx-callback/business/jxcallback/orderman"
|
||
"git.rosy.net.cn/jx-callback/business/jxcallback/scheduler/defsch"
|
||
"git.rosy.net.cn/jx-callback/business/jxstore/act"
|
||
"git.rosy.net.cn/jx-callback/business/jxstore/cms"
|
||
"git.rosy.net.cn/jx-callback/business/jxutils"
|
||
"git.rosy.net.cn/jx-callback/business/jxutils/jxcontext"
|
||
"git.rosy.net.cn/jx-callback/business/jxutils/netprinter"
|
||
"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/netspider"
|
||
"git.rosy.net.cn/jx-callback/business/partner"
|
||
"git.rosy.net.cn/jx-callback/globals"
|
||
"github.com/astaxie/beego"
|
||
)
|
||
|
||
const (
|
||
SpecialTaskID = "Running"
|
||
TaskNameSyncStoreSku = "SyncStoreSku"
|
||
)
|
||
|
||
var (
|
||
dailyWorkTimeList = []string{
|
||
"20:30:00",
|
||
}
|
||
priceReferTimeList = []string{
|
||
"01:00:00",
|
||
}
|
||
createStorePriceTimeList = []string{
|
||
"02:00:00",
|
||
}
|
||
refreshPageActTimeList = []string{
|
||
"7:00:00",
|
||
"8:00:00",
|
||
"9:00:00",
|
||
"10:00:00",
|
||
"11:00:00",
|
||
"12:00:00",
|
||
"13:00:00",
|
||
"14:00:00",
|
||
"15:00:00",
|
||
"16:00:00",
|
||
"17:00:00",
|
||
"18:00:00",
|
||
"19:00:00",
|
||
"22:00:00",
|
||
}
|
||
ChangeStoreSkuSaleStatusList = []string{
|
||
"7:00:00",
|
||
"8:00:00",
|
||
"9:00:00",
|
||
"10:00:00",
|
||
"11:00:00",
|
||
"12:00:00",
|
||
"13:00:00",
|
||
"14:00:00",
|
||
"15:00:00",
|
||
"16:00:00",
|
||
"17:00:00",
|
||
"18:00:00",
|
||
"19:00:00",
|
||
"20:00:00",
|
||
}
|
||
openRemoteStoreTimeList = []string{
|
||
"23:30:00",
|
||
}
|
||
updateActStatusTimeList = []string{
|
||
"00:01:00",
|
||
}
|
||
|
||
autoSaleStoreSkuTimeList = []string{
|
||
cms.AutoSaleAtStr,
|
||
}
|
||
|
||
importantTaskMap = &sync.Map{}
|
||
)
|
||
|
||
func GetImportantTaskID(taskName string) string {
|
||
if value, ok := importantTaskMap.Load(taskName); ok {
|
||
return value.(string)
|
||
}
|
||
return ""
|
||
}
|
||
|
||
func SaveImportantTaskID(taskName, taskID string) {
|
||
importantTaskMap.Store(taskName, taskID)
|
||
}
|
||
|
||
func IsImportantTaskRunning(taskName string) bool {
|
||
taskID := GetImportantTaskID(taskName)
|
||
if taskID == "" {
|
||
return false
|
||
} else if taskID == SpecialTaskID {
|
||
return true
|
||
}
|
||
return tasksch.IsTaskRunning(taskID)
|
||
}
|
||
|
||
func Init() {
|
||
if globals.IsProductEnv() {
|
||
ScheduleTimerFunc("doDailyWork", doDailyWork, dailyWorkTimeList)
|
||
|
||
ScheduleTimerFuncByInterval(func() {
|
||
RefreshRealMobile(jxcontext.AdminCtx, model.VendorIDEBAI, time.Now().Add(-24*time.Hour), utils.DefaultTimeValue, false, true)
|
||
}, 5*time.Second, 1*time.Hour)
|
||
|
||
ScheduleTimerFuncByInterval(func() {
|
||
defsch.FixedScheduler.ConfirmSelfTakeOrders(jxcontext.AdminCtx, []int{model.VendorIDJD}, time.Now().Add(-48*time.Hour), time.Now().Add(-30*time.Minute), true, true)
|
||
}, 5*time.Second, 10*time.Minute)
|
||
|
||
ScheduleTimerFunc("auto enable remote store", func() {
|
||
cms.EnableHaveRestStores(jxcontext.AdminCtx, false, true)
|
||
cms.OpenRemoteStoreByJxStatus(jxcontext.AdminCtx, nil, nil, false, false, true)
|
||
}, openRemoteStoreTimeList)
|
||
|
||
ScheduleTimerFunc("SaveAndSendAlarmVendorSnapshot", func() {
|
||
cms.SaveAndSendAlarmVendorSnapshot(jxcontext.AdminCtx, nil, nil, false)
|
||
}, cms.WatchVendorStoreTimeList)
|
||
|
||
ScheduleTimerFunc("RefreshPageActs", func() {
|
||
act.RefreshPageActs(jxcontext.AdminCtx, []int{model.VendorIDEBAI}, time.Now().Add(-30*24*time.Hour), false)
|
||
}, refreshPageActTimeList)
|
||
ScheduleTimerFunc("UpdateActStatusByTime", func() {
|
||
dao.UpdateActStatusByTime(dao.GetDB(), time.Now())
|
||
}, updateActStatusTimeList)
|
||
ScheduleScoreStore()
|
||
ScheduleCheckStoreAlert()
|
||
ScheduleTimerFunc("ChangeStoreSkuSaleStatus", func() {
|
||
cms.CurVendorSync.ChangeStoreSkuSaleStatus(jxcontext.AdminCtx, 0, true, true)
|
||
}, ChangeStoreSkuSaleStatusList)
|
||
ScheduleTimerFunc("BeginSavePriceRefer", func() {
|
||
report.BeginSavePriceRefer(jxcontext.AdminCtx, nil, nil, true, true)
|
||
}, priceReferTimeList)
|
||
ScheduleTimerFunc("CreateStorePriceScore", func() {
|
||
cms.CreateStorePriceScore(jxcontext.AdminCtx)
|
||
}, createStorePriceTimeList)
|
||
}
|
||
ScheduleTimerFunc("AutoSaleStoreSku", func() {
|
||
cms.AutoSaleStoreSku(jxcontext.AdminCtx, nil, false)
|
||
}, autoSaleStoreSkuTimeList)
|
||
|
||
if beego.BConfig.RunMode == "beta" {
|
||
ScheduleTimerFunc("GetAndStoreCitiesShops", func() {
|
||
netspider.GetAndStoreCitiesShops(jxcontext.AdminCtx, nil, nil, 0, 0, false, false)
|
||
}, []string{
|
||
"04:05:06",
|
||
})
|
||
}
|
||
}
|
||
|
||
func syncStoreSku() {
|
||
syncFlag := model.SyncFlagPriceMask
|
||
if (time.Now().Unix()/24*3600)%10 == 0 {
|
||
syncFlag |= model.SyncFlagSaleMask
|
||
}
|
||
task := tasksch.NewParallelTask("同步京西与平台数据", nil, jxcontext.AdminCtx,
|
||
func(task *tasksch.ParallelTask, batchItemList []interface{}, params ...interface{}) (retVal interface{}, err error) {
|
||
step := batchItemList[0].(int)
|
||
errList := errlist.New()
|
||
db := dao.GetDB()
|
||
switch step {
|
||
case 0:
|
||
errList.AddErr(cms.DeleteSkuNameExPrefixOverdue(db))
|
||
errList.AddErr(cms.SetMultiStoreSkuSyncModifyStatus(db, partner.GetMultiStoreVendorIDs()))
|
||
|
||
_, err = cms.CurVendorSync.LoopMultiStoresVendors(jxcontext.AdminCtx, db, "同步多门店平台商品库", false, true,
|
||
func(task *tasksch.ParallelTask, batchItemList []interface{}, params ...interface{}) (retVal interface{}, err error) {
|
||
vendorInfo := batchItemList[0].(*cms.MultiStoreVendorInfo)
|
||
_, err = cms.FullSyncVendorStuff(jxcontext.AdminCtx, task, vendorInfo.VendorID, vendorInfo.OrgCode, false, true)
|
||
return retVal, err
|
||
})
|
||
errList.AddErr(err)
|
||
|
||
// cms.CurVendorSync.SyncStoresSkus2(jxcontext.AdminCtx, db, partner.GetMultiStoreVendorIDs(), nil, false, nil, []int{27379}, syncFlag, true, true)
|
||
_, err = cms.CurVendorSync.FullSyncStoresSkus(jxcontext.AdminCtx, db, partner.GetMultiStoreVendorIDs(), nil, false, []int{27379}, true, true)
|
||
errList.AddErr(err)
|
||
case 1:
|
||
errList.AddErr(cms.SetSingleStoreSkuSyncModifyStatus(db, partner.GetSingleStoreVendorIDs()))
|
||
|
||
_, err = cms.CurVendorSync.AmendAndPruneStoreStuff(jxcontext.AdminCtx, partner.GetSingleStoreVendorIDs(), nil, false, true, cms.AmendPruneAll, false)
|
||
errList.AddErr(err)
|
||
|
||
SaveImportantTaskID(TaskNameSyncStoreSku, SpecialTaskID)
|
||
taskID, err2 := cms.CurVendorSync.SyncStoresSkus2(jxcontext.AdminCtx, db, partner.GetSingleStoreVendorIDs(), nil, false, nil, nil, syncFlag, true, true)
|
||
errList.AddErr(err2)
|
||
SaveImportantTaskID(TaskNameSyncStoreSku, taskID)
|
||
}
|
||
err = errList.GetErrListAsOne()
|
||
return retVal, err
|
||
}, []int{0, 1})
|
||
tasksch.HandleTask(task, nil, true).Run()
|
||
}
|
||
|
||
func doDailyWork() {
|
||
globals.SugarLogger.Debug("doDailyWork")
|
||
cms.SyncStoresCourierInfo(jxcontext.AdminCtx, nil, false, true)
|
||
netprinter.RebindAllPrinters(jxcontext.AdminCtx, false, true)
|
||
|
||
cms.CurVendorSync.SyncStore2(jxcontext.AdminCtx, dao.GetDB(), nil, nil, true, true)
|
||
|
||
syncStoreSku()
|
||
|
||
InitEx()
|
||
|
||
// 每天补全前一天与当天的订单
|
||
curDate := utils.Time2Date(time.Now())
|
||
orderman.FixedOrderManager.AmendMissingOrders(jxcontext.AdminCtx, nil, 0, curDate.Add(-72*time.Hour), curDate, true, true)
|
||
//订单门店归属补漏
|
||
//fromDate, toDate都不传默认刷新当前天5天以前的订单，只传fromDate默认刷新fromDate到当天的订单
|
||
//只传toDate默认刷新toDate到5天以前的订单
|
||
orderman.RefreshOrdersWithoutJxStoreID(jxcontext.AdminCtx, "", "", true, true)
|
||
//刷新京东门店的等级
|
||
cms.RefreshJdLevel(jxcontext.AdminCtx)
|
||
}
|
||
|
||
func RefreshRealMobile(ctx *jxcontext.Context, vendorID int, fromTime, toTime time.Time, isAsync, isContinueWhenError bool) (hint string, err error) {
|
||
globals.SugarLogger.Debug("RefreshRealMobile")
|
||
|
||
handler := partner.GetPurchasePlatformFromVendorID(vendorID)
|
||
if handler == nil {
|
||
return "", fmt.Errorf("不合法的vendorID:%d", vendorID)
|
||
}
|
||
sql := `
|
||
SELECT *
|
||
FROM goods_order
|
||
WHERE vendor_id = ? AND consignee_mobile2 = ''
|
||
`
|
||
sqlParams := []interface{}{
|
||
vendorID,
|
||
}
|
||
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 {
|
||
mobile = jxutils.FormalizeMobile(mobile)
|
||
if jxutils.IsStringLikeMobile(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
|
||
}
|
||
|
||
// 按时间序列循环
|
||
func ScheduleTimerFunc(name string, handler func(), timeList []string) {
|
||
now := time.Now()
|
||
nextTime := jxutils.GetNextTimeFromList(now, timeList)
|
||
duration := nextTime.Sub(now) + 1*time.Second
|
||
globals.SugarLogger.Debugf("ScheduleTimerFunc, func:%s, duration:%v", name, duration)
|
||
utils.AfterFuncWithRecover(duration, func() {
|
||
globals.SugarLogger.Debugf("ScheduleTimerFunc func:%s", name)
|
||
handler()
|
||
ScheduleTimerFunc(name, handler, timeList)
|
||
})
|
||
}
|
||
|
||
// 按时间调度一次
|
||
func ScheduleTimerFuncOnce(name string, handler func(interface{}), timeStr string, param interface{}) {
|
||
now := time.Now()
|
||
nextTime := jxutils.GetNextTimeFromList(now, []string{timeStr})
|
||
duration := nextTime.Sub(now) + 1*time.Second
|
||
globals.SugarLogger.Debugf("ScheduleTimerFuncOnce, func:%s, duration:%v", name, duration)
|
||
utils.AfterFuncWithRecover(duration, func() {
|
||
globals.SugarLogger.Debugf("ScheduleTimerFuncOnce func:%s", name)
|
||
handler(param)
|
||
})
|
||
}
|
||
|
||
func ScheduleTimerFuncByInterval(handler func(), delay, inerval time.Duration) {
|
||
globals.SugarLogger.Debugf("ScheduleTimerFuncByInterval, delay:%v, inerval:%v", delay, inerval)
|
||
utils.AfterFuncWithRecover(delay, func() {
|
||
beginTime := time.Now()
|
||
handler()
|
||
delay = inerval - time.Now().Sub(beginTime)
|
||
if delay < time.Second {
|
||
delay = time.Second
|
||
}
|
||
ScheduleTimerFuncByInterval(handler, delay, inerval)
|
||
})
|
||
}
|