- 美团外卖生成不同的售后单号，支持同一订单多个售后单

business/jxcallback/orderman/order_afs.go

@@ -2,6 +2,7 @@ package orderman
 
 import (
 	"fmt"
+	"strings"
 
 	"git.rosy.net.cn/baseapi/utils"
 	"git.rosy.net.cn/jx-callback/business/jxcallback/scheduler"
@@ -36,13 +37,17 @@ func (c *OrderManager) OnAfsOrderNew(afsOrder *model.AfsOrder, orderStatus *mode
 }
 
 func (c *OrderManager) onAfsOrderNew(afsOrder *model.AfsOrder, orderStatus *model.OrderStatus, isAdjust bool) (err error) {
+	db := dao.GetDB()
+	c.setAfsOrderID(db, orderStatus)
+	if afsOrder.AfsOrderID == "" {
+		afsOrder.AfsOrderID = orderStatus.VendorOrderID
+	}
 	if afsOrder.VendorStatus == "" {
 		afsOrder.VendorStatus = orderStatus.VendorStatus
 	}
 	if afsOrder.Status == model.OrderStatusUnknown {
 		afsOrder.Status = orderStatus.Status
 	}
-	db := dao.GetDB()
 	dao.Begin(db)
 	defer func() {
 		if r := recover(); r != nil || err != nil {
@@ -150,6 +155,7 @@ func (c *OrderManager) SaveAfsOrder(db *dao.DaoDB, afsOrder *model.AfsOrder, isD
 
 func (c *OrderManager) OnAfsOrderStatusChanged(orderStatus *model.OrderStatus) (err error) {
 	db := dao.GetDB()
+	c.setAfsOrderID(db, orderStatus)
 	dao.Begin(db)
 	defer func() {
 		if r := recover(); r != nil || err != nil {
@@ -299,3 +305,28 @@ func (c *OrderManager) UpdateAfsOrderFields(afsOrder *model.AfsOrder, fieldList
 	}, "UpdateAfsOrderFields orderID:%s failed with error:%v", afsOrder.VendorOrderID, err)
 	return err
 }
+
+func (c *OrderManager) setAfsOrderID(db *dao.DaoDB, orderStatus *model.OrderStatus) {
+	if orderStatus.VendorOrderID == "" {
+		index := 1
+		if afsOrderList, err2 := dao.GetAfsOrders(db, orderStatus.RefVendorID, orderStatus.RefVendorOrderID, ""); err2 == nil {
+			if len(afsOrderList) > 0 {
+				list := strings.Split(afsOrderList[0].AfsOrderID, "-")
+				if len(list) > 1 {
+					index = int(utils.Str2Int64WithDefault(list[1], 0))
+					if afsOrderList[1].Status >= model.AfsOrderStatusFinished {
+						index++
+					}
+				}
+			}
+		}
+		orderStatus.VendorOrderID = composeAfsOrderID(orderStatus.RefVendorOrderID, index)
+	}
+}
+
+func composeAfsOrderID(vendorOrderID string, index int) (afsOrderID string) {
+	return strings.Join([]string{
+		vendorOrderID,
+		utils.Int2Str(index),
+	}, "-")
+}