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

business/partner/purchase/jd/sku.go

@@ -515,3 +515,42 @@ func jxStatus2jdStatus(jxStatus int) (jdStatus int) {
 	}
 	return jdStatus
 }
+
+func (p *PurchaseHandler) getVendorCategories(level int, pid int64) (vendorCats []*model.SkuVendorCategory, err error) {
+	cats, err := api.JdAPI.QueryChildCategoriesForOP(pid)
+	if err != nil {
+		return nil, err
+	}
+	for _, v := range cats {
+		if v.Status == 1 {
+			cat := &model.SkuVendorCategory{
+				VendorID:         model.VendorIDJD,
+				Name:             v.Name,
+				Level:            level,
+				VendorCategoryID: utils.Int64ToStr(v.Id),
+			}
+			cat.LastOperator = "builder"
+			if level > 1 {
+				cat.ParentID = utils.Int64ToStr(v.ParentId)
+				if level == 3 {
+					cat.IsLeaf = 1
+				}
+			}
+			vendorCats = append(vendorCats, cat)
+			if level < 3 {
+				childVendorCats, err2 := p.getVendorCategories(level+1, v.Id)
+				if err2 == nil && len(childVendorCats) > 0 {
+					vendorCats = append(vendorCats, childVendorCats...)
+				} else {
+					cat.IsLeaf = 1
+				}
+			}
+		}
+	}
+	return vendorCats, nil
+}
+
+func (p *PurchaseHandler) GetVendorCategories(ctx *jxcontext.Context) (vendorCats []*model.SkuVendorCategory, err error) {
+	vendorCats, err = p.getVendorCategories(1, 0)
+	return vendorCats, err
+}

business/partner/purchase/jd/sku_test.go

@@ -3,6 +3,8 @@ package jd
 import (
 	"testing"
 
+	"git.rosy.net.cn/jx-callback/business/jxutils/jxcontext"
+
 	"git.rosy.net.cn/baseapi/utils"
 	"git.rosy.net.cn/jx-callback/business/model"
 	"git.rosy.net.cn/jx-callback/business/model/dao"
@@ -53,3 +55,11 @@ func TestReadSku(t *testing.T) {
 		t.Log(string(utils.MustMarshal(skuName)))
 	}
 }
+
+func TestGetVendorCategories(t *testing.T) {
+	catList, err := new(PurchaseHandler).GetVendorCategories(jxcontext.AdminCtx)
+	if err != nil {
+		t.Fatal(err.Error())
+	}
+	t.Log(utils.Format4Output(catList, false))
+}