- InitVendorCategory可对三个主流平台进行操作

business/jxstore/initdata/initdata.go

@@ -19,7 +19,6 @@ import (
 	"git.rosy.net.cn/jx-callback/business/model/dao"
 	"git.rosy.net.cn/jx-callback/business/partner"
 	"git.rosy.net.cn/jx-callback/business/partner/purchase/jd"
-	"git.rosy.net.cn/jx-callback/business/partner/purchase/mtwm"
 	"git.rosy.net.cn/jx-callback/globals"
 	"git.rosy.net.cn/jx-callback/globals/api"
 	"github.com/qiniu/api.v7/storage"
@@ -174,14 +173,21 @@ func InitSkuName(ctx *jxcontext.Context, isForce, isAsync, isContinueWhenError b
 }
 
 func InitVendorCategory(ctx *jxcontext.Context, vendorID int) (num int64, err error) {
-	if handler, ok := partner.PurchasePlatformHandlers[vendorID].(*mtwm.PurchaseHandler); ok {
-		cats, err2 := handler.GetVendorCategories()
+	if handler := partner.GetPurchasePlatformFromVendorID(vendorID); handler != nil {
+		cats, err2 := handler.GetVendorCategories(ctx)
 		if err2 != nil {
 			return num, err2
 		}
 		db := dao.GetDB()
 		dao.Begin(db)
-		defer dao.Rollback(db)
+		defer func() {
+			if r := recover(); r != nil || err != nil {
+				dao.Rollback(db)
+				if r != nil {
+					panic(r)
+				}
+			}
+		}()
 		sql := `
 		DELETE
 		FROM sku_vendor_category
@@ -198,6 +204,8 @@ func InitVendorCategory(ctx *jxcontext.Context, vendorID int) (num int64, err er
 		}
 		dao.Commit(db)
 		num = int64(len(cats))
+	} else {
+		err = fmt.Errorf("找不到平台:%d", vendorID)
 	}
 	return num, err
 }