- add delete cats or skus task to rootTask

This commit is contained in:
gazebo
2018-12-19 23:46:57 +08:00
parent 056b9109a1
commit 55f2445eb4
4 changed files with 12 additions and 12 deletions

View File

@@ -148,11 +148,11 @@ func (p *PurchaseHandler) FullSyncStoreSkus(ctx *jxcontext.Context, parentTask t
rootTask := tasksch.NewSeqTask("FullSyncStoreSkus", userName, func(rootTask *tasksch.SeqTask, step int, params ...interface{}) (result interface{}, err error) { rootTask := tasksch.NewSeqTask("FullSyncStoreSkus", userName, func(rootTask *tasksch.SeqTask, step int, params ...interface{}) (result interface{}, err error) {
switch step { switch step {
case 0: case 0:
err = p.DeleteRemoteSkus(storeID, nil) err = p.DeleteRemoteSkus(ctx, rootTask, storeID, nil)
case 1: case 1:
_, err = p.setStoreSkuSyncStatus(ctx, db, storeID, nil, model.SyncFlagNewMask) _, err = p.setStoreSkuSyncStatus(ctx, db, storeID, nil, model.SyncFlagNewMask)
case 2: case 2:
err = p.DeleteRemoteCategories(storeID, nil) err = p.DeleteRemoteCategories(ctx, rootTask, storeID, nil)
case 3: case 3:
err = p.SyncLocalStoreCategory(db, storeID, userName) err = p.SyncLocalStoreCategory(db, storeID, userName)
case 4: case 4:
@@ -277,7 +277,7 @@ func (p *PurchaseHandler) GetAllRemoteSkus(storeID int) (skus []map[string]inter
return skus, err return skus, err
} }
func (p *PurchaseHandler) DeleteRemoteSkus(storeID int, vendorSkuIDs []string) (err error) { func (p *PurchaseHandler) DeleteRemoteSkus(ctx *jxcontext.Context, parentTask tasksch.ITask, storeID int, vendorSkuIDs []string) (err error) {
globals.SugarLogger.Debugf("DeleteRemoteSkus饿百 storeID:%d", storeID) globals.SugarLogger.Debugf("DeleteRemoteSkus饿百 storeID:%d", storeID)
if vendorSkuIDs == nil { if vendorSkuIDs == nil {
@@ -299,12 +299,12 @@ func (p *PurchaseHandler) DeleteRemoteSkus(storeID int, vendorSkuIDs []string) (
} }
return nil, err return nil, err
}, vendorSkuIDs) }, vendorSkuIDs)
task.Run() tasksch.AddChild(parentTask, task).Run()
_, err = task.GetResult(0) _, err = task.GetResult(0)
return err return err
} }
func (p *PurchaseHandler) DeleteRemoteCategories(storeID int, vendorCatIDs []int64) (err error) { func (p *PurchaseHandler) DeleteRemoteCategories(ctx *jxcontext.Context, parentTask tasksch.ITask, storeID int, vendorCatIDs []int64) (err error) {
globals.SugarLogger.Debugf("DeleteRemoteCategories饿百 storeID:%d", storeID) globals.SugarLogger.Debugf("DeleteRemoteCategories饿百 storeID:%d", storeID)
strStoreID := utils.Int2Str(storeID) strStoreID := utils.Int2Str(storeID)
@@ -323,7 +323,7 @@ func (p *PurchaseHandler) DeleteRemoteCategories(storeID int, vendorCatIDs []int
} }
return nil, err return nil, err
}, vendorCatIDs) }, vendorCatIDs)
task.Run() tasksch.AddChild(parentTask, task).Run()
_, err = task.GetResult(0) _, err = task.GetResult(0)
return err return err
} }

View File

@@ -45,14 +45,14 @@ func TestGetAllRemoteSkus(t *testing.T) {
} }
func TestDeleteRemoteSkus(t *testing.T) { func TestDeleteRemoteSkus(t *testing.T) {
err := new(PurchaseHandler).DeleteRemoteSkus(testShopID, nil) err := new(PurchaseHandler).DeleteRemoteSkus(jxcontext.AdminCtx, nil, testShopID, nil)
if err != nil { if err != nil {
t.Fatal(err.Error()) t.Fatal(err.Error())
} }
} }
func TestDeleteRemoteCategories(t *testing.T) { func TestDeleteRemoteCategories(t *testing.T) {
err := new(PurchaseHandler).DeleteRemoteCategories(testShopID, nil) err := new(PurchaseHandler).DeleteRemoteCategories(jxcontext.AdminCtx, nil, testShopID, nil)
if err != nil { if err != nil {
t.Fatal(err.Error()) t.Fatal(err.Error())
} }

View File

@@ -302,7 +302,7 @@ func (p *PurchaseHandler) FullSyncStoreSkus(ctx *jxcontext.Context, parentTask t
rootTask := tasksch.NewSeqTask("美团外卖FullSyncStoreSkus", userName, func(rootTask *tasksch.SeqTask, step int, params ...interface{}) (result interface{}, err error) { rootTask := tasksch.NewSeqTask("美团外卖FullSyncStoreSkus", userName, func(rootTask *tasksch.SeqTask, step int, params ...interface{}) (result interface{}, err error) {
switch step { switch step {
case 0: case 0:
err = p.DeleteRemoteSkus(storeID, nil) err = p.DeleteRemoteSkus(ctx, rootTask, storeID, nil)
case 1: case 1:
_, err = dao.SetStoreSkuSyncStatus(ctx, db, model.VendorIDMTWM, storeID, nil, model.SyncFlagNewMask) _, err = dao.SetStoreSkuSyncStatus(ctx, db, model.VendorIDMTWM, storeID, nil, model.SyncFlagNewMask)
case 2: case 2:
@@ -321,7 +321,7 @@ func (p *PurchaseHandler) FullSyncStoreSkus(ctx *jxcontext.Context, parentTask t
return rootTask.ID, err return rootTask.ID, err
} }
func (p *PurchaseHandler) DeleteRemoteSkus(storeID int, vendorSkuIDs []string) (err error) { func (p *PurchaseHandler) DeleteRemoteSkus(ctx *jxcontext.Context, parentTask tasksch.ITask, storeID int, vendorSkuIDs []string) (err error) {
if vendorSkuIDs == nil { if vendorSkuIDs == nil {
result, err2 := p.GetAllRemoteSkus(storeID) result, err2 := p.GetAllRemoteSkus(storeID)
if err = err2; err == nil { if err = err2; err == nil {
@@ -339,7 +339,7 @@ func (p *PurchaseHandler) DeleteRemoteSkus(storeID int, vendorSkuIDs []string) (
} }
return nil, err return nil, err
}, vendorSkuIDs) }, vendorSkuIDs)
task.Run() tasksch.AddChild(parentTask, task).Run()
_, err = task.GetResult(0) _, err = task.GetResult(0)
return err return err
} }

View File

@@ -32,7 +32,7 @@ func TestSyncStoreSkus(t *testing.T) {
} }
func TestDeleteRemoteSkus(t *testing.T) { func TestDeleteRemoteSkus(t *testing.T) {
err := new(PurchaseHandler).DeleteRemoteSkus(100077, nil) err := new(PurchaseHandler).DeleteRemoteSkus(jxcontext.AdminCtx, nil, 100077, nil)
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }