- store courier map

business/jxstore/cms/store.go

@@ -6,6 +6,7 @@ import (
 	"strings"
 
 	"git.rosy.net.cn/baseapi/utils"
+	"git.rosy.net.cn/jx-callback/business/jxcallback/scheduler/basesch"
 	"git.rosy.net.cn/jx-callback/business/jxutils"
 	"git.rosy.net.cn/jx-callback/business/model"
 	"git.rosy.net.cn/jx-callback/business/model/dao"
@@ -368,3 +369,73 @@ func TmpGetJxBadCommentsByStoreId(storeID, page, size, commentType int) (retVal
 	}
 	return retVal, err
 }
+
+func GetStoreCourierMaps(db *dao.DaoDB, storeID int, vendorID int) (storeCourierMaps []*model.StoreCourierMap, err error) {
+	cond := map[string]interface{}{
+		model.FieldStoreID: storeID,
+	}
+	if vendorID != -1 {
+		cond[model.FieldVendorID] = vendorID
+	}
+	return storeCourierMaps, dao.GetEntitiesByKV(db, &storeCourierMaps, cond, false)
+}
+
+func AddStoreCourierMap(db *dao.DaoDB, storeID, vendorID int, storeCourierMap *model.StoreCourierMap, userName string) (outStoreCourierMap *model.StoreCourierMap, err error) {
+	if handler := basesch.FixedBaseScheduler.GetDeliveryPlatformFromVendorID(vendorID); handler != nil {
+		dao.WrapAddIDCULDEntity(storeCourierMap, userName)
+		storeCourierMap.StoreID = storeID
+		storeCourierMap.VendorID = vendorID
+		if db == nil {
+			db = dao.GetDB()
+		}
+		dao.Begin(db)
+		defer func() {
+			if r := recover(); r != nil {
+				dao.Rollback(db)
+				panic(r)
+			}
+		}()
+		if err = dao.CreateEntity(db, storeCourierMap); err == nil {
+			dao.Commit(db)
+			outStoreCourierMap = storeCourierMap
+			err = CurVendorSync.SyncStore(db, storeCourierMap.VendorID, storeID, true, userName)
+		}
+		if err != nil {
+			dao.Rollback(db)
+		}
+	} else {
+		err = ErrCanNotFindVendor
+	}
+	return outStoreCourierMap, err
+}
+
+func DeleteStoreCourierMap(db *dao.DaoDB, storeID, vendorID int, userName string) (num int64, err error) {
+	if db == nil {
+		db = dao.GetDB()
+	}
+	storeCourierMap := &model.StoreCourierMap{}
+	if num, err = dao.DeleteEntityLogically(db, storeCourierMap, map[string]interface{}{
+		model.FieldStatus: model.StoreStatusDisabled,
+	}, userName, map[string]interface{}{
+		model.FieldStoreID:  storeID,
+		model.FieldVendorID: vendorID,
+	}); err == nil && num > 0 {
+		err = CurVendorSync.SyncStore(db, vendorID, storeID, true, userName)
+	}
+	return num, err
+}
+
+func UpdateStoreCourierMap(db *dao.DaoDB, storeID, vendorID int, payload map[string]interface{}, userName string) (num int64, err error) {
+	if db == nil {
+		db = dao.GetDB()
+	}
+	dummyStoreCourierMap := &model.StoreCourierMap{}
+	valid := dao.NormalMakeMapByStructObject(payload, dummyStoreCourierMap, userName)
+	if len(valid) > 0 {
+		num, err = dao.UpdateEntityLogically(db, dummyStoreCourierMap, valid, userName, map[string]interface{}{
+			model.FieldStoreID:  storeID,
+			model.FieldVendorID: vendorID,
+		})
+	}
+	return num, err
+}

business/jxstore/cms/sync.go

@@ -98,11 +98,17 @@ func (v *VendorSync) GetStoreHandler(vendorID int) partner.IPurchasePlatformHand
 }
 
 func (v *VendorSync) GetMultiStoreHandler(vendorID int) partner.IMultipleStoresHandler {
-	return v.PurchaseHandlers[vendorID].(partner.IMultipleStoresHandler)
+	if v.PurchaseHandlers[vendorID] != nil {
+		return v.PurchaseHandlers[vendorID].(partner.IMultipleStoresHandler)
+	}
+	return nil
 }
 
 func (v *VendorSync) GetSingleStoreHandler(vendorID int) partner.ISingleStoreHandler {
-	return v.PurchaseHandlers[vendorID].(partner.ISingleStoreHandler)
+	if v.PurchaseHandlers[vendorID] != nil {
+		return v.PurchaseHandlers[vendorID].(partner.ISingleStoreHandler)
+	}
+	return nil
 }
 
 func (v *VendorSync) SyncCategory(db *dao.DaoDB, categoryID int, isForce bool, userName string) (err error) {