修复支付宝支付

2026-09-16 16:27:56 +08:00
parent 605c2a569f
commit 685dbc030d
9 changed files with 265 additions and 133 deletions
@@ -76,7 +76,7 @@ func (c *BackendModulesController) GetTenantList() {
 		return
 	}

-	// 查询当前租户套餐包含的模块 code 集合（含默认套餐回退）
+	// 查询当前租户已开通的模块 code 集合（套餐模块 + 已开通产品订单关联模块，编码已规范化）
 	tid := uint64(claims.TenantId)
 	allowedCodes := services.GetTenantModuleCodes(tid)

@@ -84,7 +84,7 @@ func (c *BackendModulesController) GetTenantList() {
 	for _, m := range rows {
 		enabled := true
 		if len(allowedCodes) > 0 {
-			enabled = allowedCodes[m.Code]
+			enabled = allowedCodes[services.NormalizeModuleCode(m.Code)]
 		}
 		enhanced = append(enhanced, moduleWithEnabled{SystemModules: m, Enabled: enabled})
 	}
@@ -0,0 +1,62 @@
+package controllers
+
+import (
+	"context"
+	"fmt"
+	"strings"
+	"time"
+
+	"server/models"
+	"server/services/payment"
+
+	"github.com/beego/beego/v2/client/orm"
+)
+
+// 支付成功 -> 产品/套餐购买订单同步：
+// 渠道异步回调（或主动查询）把支付单置为 paid 时，支付服务会调用此处注册的钩子，
+// 自动把购买订单置为已开通；否则订单只能等前端轮询兜底或平台手动开通。
+func init() {
+	payment.RegisterOrderSyncHook(syncProductOrderFromPayment)
+}
+
+// syncProductOrderFromPayment 支付单成功后的钩子：按业务订单号同步购买订单状态并自动开通。
+// 返回错误时支付服务会保留 order_synced=0，由支付补偿任务（5 分钟一轮）重试。
+func syncProductOrderFromPayment(_ context.Context, payOrder *models.PlatformPaymentOrder) error {
+	if payOrder == nil {
+		return nil
+	}
+	orderNo := strings.TrimSpace(payOrder.OutTradeNo)
+	if orderNo == "" {
+		return nil
+	}
+
+	var order models.PlatformProductOrder
+	err := models.Orm.QueryTable(new(models.PlatformProductOrder)).
+		Filter("order_no", orderNo).
+		Filter("delete_time__isnull", true).
+		One(&order)
+	if err == orm.ErrNoRows {
+		// 下单链路先建支付单、后落业务订单，渠道回调可能抢先到达：
+		// 短时间内保留重试；超过窗口仍不存在说明不是本模块的支付单，视为无需同步。
+		if time.Since(payOrder.CreateTime) < 10*time.Minute {
+			return fmt.Errorf("产品订单 %s 尚未入库，等待重试", orderNo)
+		}
+		return nil
+	}
+	if err != nil {
+		return err
+	}
+	// 已开通/已关闭等终态无需处理
+	if order.Status != models.ProductOrderStatusPending && order.Status != models.ProductOrderStatusPaid {
+		return nil
+	}
+	if order.Status == models.ProductOrderStatusPending {
+		now := time.Now()
+		order.Status = models.ProductOrderStatusPaid
+		order.UpdateTime = &now
+		if _, err := models.Orm.Update(&order, "Status", "UpdateTime"); err != nil {
+			return err
+		}
+	}
+	return activateProductOrder(&order, "支付成功自动开通")
+}
@@ -260,8 +260,17 @@ func (c *BackendProductPurchaseController) GetOrderStatus() {
 		Filter("tid", uint64(claims.TenantId)).
 		Filter("delete_time__isnull", true).
 		One(&order); err != nil {
-		c.jsonErr(404, 404, "订单不存在")
-		return
+		// 支付宝同步回跳地址携带的 out_trade_no 是支付单号（pay_no），此处兜底按支付单号查询，
+		// 保证「支付完成自动跳回」的场景也能直接触发状态同步与自动开通。
+		if err2 := models.Orm.QueryTable(new(models.PlatformProductOrder)).
+			Filter("pay_no", orderNo).
+			Filter("tid", uint64(claims.TenantId)).
+			Filter("delete_time__isnull", true).
+			OrderBy("-id").
+			One(&order); err2 != nil {
+			c.jsonErr(404, 404, "订单不存在")
+			return
+		}
 	}

 	// 在线支付：同步渠道状态，支付成功后自动开通
@@ -144,18 +144,68 @@ func GetTenantPackageModules(tid uint64) ([]models.SystemTenantPackageModule, er
 	return rows, err
 }

-// GetTenantModuleCodes 获取租户已开通的功能模块编码集合（如 erp / oa / crm）
+// NormalizeModuleCode 规范化模块编码：去空格 + 统一小写。
+// 模块、套餐模块、产品关联模块来自不同配置入口，编码大小写可能不一致
+// （如模块 code=oa、产品关联功能填 OA），统一规范化后再比对。
+func NormalizeModuleCode(code string) string {
+	return strings.ToLower(strings.TrimSpace(code))
+}
+
+// GetTenantModuleCodes 获取租户已开通的功能模块编码集合（如 erp / oa / crm）。
+//
+// 权益由两部分组成：
+//   - 当前生效套餐包含的模块；
+//   - 已开通单品订单关联产品的模块。
+//
+// 单品购买不会替换租户当前套餐，因此必须与套餐模块取并集；否则产品订单即使
+// 已显示“已开通”，首页仍会把该模块判定为未购买。历史订单同样按产品当前的
+// module_code 识别，保证已存在的已开通订单可立即恢复对应模块权限。
+// 返回的 key 均为 NormalizeModuleCode 规范化后的编码。
 func GetTenantModuleCodes(tid uint64) map[string]bool {
 	codes := map[string]bool{}
+	mark := func(code string) {
+		if c := NormalizeModuleCode(code); c != "" {
+			codes[c] = true
+		}
+	}
 	rows, err := GetTenantPackageModules(tid)
-	if err != nil {
+	if err == nil {
+		for _, r := range rows {
+			mark(r.ModuleCode)
+		}
+	}
+
+	if tid == 0 {
 		return codes
 	}
-	for _, r := range rows {
-		code := strings.TrimSpace(r.ModuleCode)
-		if code != "" {
-			codes[code] = true
+
+	var orders []models.PlatformProductOrder
+	if _, err := models.Orm.QueryTable(new(models.PlatformProductOrder)).
+		Filter("tid", tid).
+		Filter("status", models.ProductOrderStatusActivated).
+		Filter("product_id__gt", 0).
+		Filter("delete_time__isnull", true).
+		All(&orders); err != nil || len(orders) == 0 {
+		return codes
+	}
+
+	productIDs := make([]uint64, 0, len(orders))
+	for _, order := range orders {
+		productIDs = append(productIDs, order.ProductID)
+	}
+	var products []models.PlatformProduct
+	if _, err := models.Orm.QueryTable(new(models.PlatformProduct)).
+		Filter("id__in", productIDs).
+		All(&products); err != nil {
+		return codes
+	}
+	for _, product := range products {
+		code := product.ModuleCode
+		if strings.TrimSpace(code) == "" {
+			// 产品未显式关联功能模块时，按产品编码兜底匹配（产品编码通常即模块编码，如 oa）
+			code = product.Code
 		}
+		mark(code)
 	}
 	return codes
 }
@@ -403,7 +453,7 @@ func FilterMenusByTenantModules(menus []models.SystemMenu, allModules []models.S
 			out = append(out, m)
 			continue
 		}
-		if allowedCodes[code] {
+		if allowedCodes[NormalizeModuleCode(code)] {
 			out = append(out, m)
 		}
 	}
@@ -425,7 +475,7 @@ func MarkMenusLockedByTenantModules(menus []models.SystemMenu, allModules []mode
 			path = *m.Path
 		}
 		code, ok := ownerModuleCode(sorted, path)
-		if ok && !allowedCodes[code] {
+		if ok && !allowedCodes[NormalizeModuleCode(code)] {
 			lockedIDs[m.ID] = true
 		}
 		kept = append(kept, m)