wangjia
239 lines
8.9 KiB
Go
239 lines
8.9 KiB
Go
package main
|
|
|
|
import (
|
|
"errors"
|
|
"log"
|
|
"time"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/glebarez/sqlite"
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/logger"
|
|
|
|
"github.com/wangjia/pay/config"
|
|
"github.com/wangjia/pay/internal/accounts"
|
|
"github.com/wangjia/pay/internal/channel"
|
|
"github.com/wangjia/pay/internal/gateway"
|
|
"github.com/wangjia/pay/internal/model"
|
|
"github.com/wangjia/pay/internal/providerbuild"
|
|
"github.com/wangjia/pay/internal/router"
|
|
"github.com/wangjia/pay/internal/store"
|
|
"github.com/wangjia/pay/internal/webhook"
|
|
)
|
|
|
|
func main() {
|
|
config.Load()
|
|
|
|
db := initDB()
|
|
autoMigrate(db)
|
|
seed(db)
|
|
seedWechat(db)
|
|
|
|
reg := channel.NewRegistry(db)
|
|
|
|
gin.SetMode(config.C.Server.Mode)
|
|
r := gin.New()
|
|
r.Use(gin.Logger(), gin.Recovery())
|
|
|
|
orderSvc := router.Setup(r, db, reg)
|
|
|
|
// v2 统一网关装配(P2 骨架,P3 配置驱动注册接线):provider 注册表 + gateway + webhook notifier。
|
|
orderStore := store.NewOrderStore(db)
|
|
webhookStore := store.NewWebhookStore(db)
|
|
notifier := webhook.NewNotifier(webhookStore, config.C.BizByName, func(no string) (bool, error) {
|
|
o, err := orderStore.GetOrder(no) // 投递门禁:订单已付才发(enqueue-before-flip 不变量)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
return o.Status == model.OrderPaidV2, nil
|
|
})
|
|
notifier.Start(60 * time.Second)
|
|
productResolver := gateway.NewDBProductResolver(db) // 币种由下单渠道结算能力驱动(设计 §4.1)
|
|
acctReg := accounts.New(config.C.Accounts)
|
|
pReg := providerbuild.BuildRegistry(acctReg) // 配置驱动:有 enabled 账户才注册对应渠道(P3)
|
|
// P5 多账户路由:按 config.routing.<channel> 选策略(缺省 round_robin)。
|
|
// limit_aware 用量数据源 P6 对账就绪前用空源(NopUsage,退化为 round_robin)。
|
|
acctPicker := accounts.NewRouter(acctReg, config.C.Routing, accounts.NopUsage{})
|
|
gw := gateway.New(orderStore, pReg, acctPicker, productResolver, notifier, "cn")
|
|
router.SetupV2(r, gw)
|
|
|
|
if config.C.QuerySync.Enabled {
|
|
orderSvc.StartQuerySync(
|
|
time.Duration(config.C.QuerySync.IntervalSec)*time.Second,
|
|
time.Duration(config.C.QuerySync.MaxAgeMin)*time.Minute,
|
|
)
|
|
log.Printf("查单兜底已启动：每 %ds 一次", config.C.QuerySync.IntervalSec)
|
|
}
|
|
|
|
// 业务回调重试兜底（webhook 丢失/业务方短暂不可用时补推）。
|
|
orderSvc.StartBizNotifyRetry(60 * time.Second)
|
|
|
|
addr := ":" + config.C.Server.Port
|
|
log.Printf("支付服务启动 %s (mode=%s, base_url=%s)", addr, config.C.Server.Mode, config.C.Server.BaseURL)
|
|
if err := r.Run(addr); err != nil {
|
|
log.Fatalf("启动失败: %v", err)
|
|
}
|
|
}
|
|
|
|
func initDB() *gorm.DB {
|
|
logLevel := logger.Silent
|
|
if config.C.Server.Mode == "debug" {
|
|
logLevel = logger.Info
|
|
}
|
|
gormCfg := &gorm.Config{Logger: logger.Default.LogMode(logLevel), TranslateError: true}
|
|
|
|
switch config.C.Database.Driver {
|
|
case "sqlite", "":
|
|
db, err := gorm.Open(sqlite.Open(config.C.Database.DSN), gormCfg)
|
|
if err != nil {
|
|
log.Fatalf("连接 sqlite 失败: %v", err)
|
|
}
|
|
return db
|
|
case "mysql":
|
|
log.Fatal("mysql 驱动尚未启用：go get gorm.io/driver/mysql 后在 initDB 接入")
|
|
return nil
|
|
default:
|
|
log.Fatalf("未知 database.driver: %s", config.C.Database.Driver)
|
|
return nil
|
|
}
|
|
}
|
|
|
|
func autoMigrate(db *gorm.DB) {
|
|
if err := db.AutoMigrate(
|
|
&model.Merchant{},
|
|
&model.Product{},
|
|
&model.ProductPrice{},
|
|
&model.Order{},
|
|
&model.NotifyLog{},
|
|
&model.BizNotifyLog{},
|
|
&model.OrderV2{}, &model.Attempt{}, &model.Account{}, &model.Refund{}, &model.WebhookDelivery{}, // v2
|
|
); err != nil {
|
|
log.Fatalf("自动迁移失败: %v", err)
|
|
}
|
|
log.Println("AutoMigrate 完成")
|
|
}
|
|
|
|
// seed 据 config 的 alipay_sandbox upsert 一个支付宝商户，并为其补两个测试套餐。
|
|
func seed(db *gorm.DB) {
|
|
sb := config.C.AlipaySandbox
|
|
if !sb.Enabled {
|
|
log.Println("[seed] alipay_sandbox.enabled=false，跳过沙箱商户初始化")
|
|
return
|
|
}
|
|
var m model.Merchant
|
|
err := db.Where("code = ?", sb.MerchantCode).First(&m).Error
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
m = model.Merchant{
|
|
Code: sb.MerchantCode,
|
|
Name: sb.MerchantName,
|
|
Channel: "alipay",
|
|
}
|
|
} else if err != nil {
|
|
log.Fatalf("[seed] 查询商户失败: %v", err)
|
|
}
|
|
m.Production = sb.Production
|
|
m.Enabled = true
|
|
m.AppID = sb.AppID
|
|
m.AppPrivateKey = sb.AppPrivateKey
|
|
m.AlipayPublicKey = sb.AlipayPublicKey
|
|
if err := db.Save(&m).Error; err != nil {
|
|
log.Fatalf("[seed] 保存商户失败: %v", err)
|
|
}
|
|
log.Printf("[seed] 支付宝沙箱商户就绪: code=%s app_id=%s", m.Code, m.AppID)
|
|
|
|
var cnt int64
|
|
db.Model(&model.Product{}).Where("merchant_id = ?", m.ID).Count(&cnt)
|
|
if cnt == 0 {
|
|
samples := []model.Product{
|
|
{MerchantID: m.ID, Name: "测试套餐 A", Description: "沙箱联调用", Price: "0.01", Active: true, Sort: 1},
|
|
{MerchantID: m.ID, Name: "测试套餐 B", Description: "沙箱联调用", Price: "0.02", Active: true, Sort: 2},
|
|
}
|
|
if err := db.Create(&samples).Error; err != nil {
|
|
log.Fatalf("[seed] 创建测试套餐失败: %v", err)
|
|
}
|
|
log.Printf("[seed] 已为商户 %s 创建 %d 个测试套餐", m.Code, len(samples))
|
|
}
|
|
|
|
seedPlans(db, m.ID)
|
|
}
|
|
|
|
// seedPlans 幂等 upsert 业务套餐（带 biz_code，供 jiu 等业务系统按 biz_code 查 product）。
|
|
// 价格权威在此表；权益（时长/档位/设备/功能）由业务方按 biz_code 映射。
|
|
func seedPlans(db *gorm.DB, merchantID uint64) {
|
|
// Name 会成为支付宝收银台的 subject（一行约 12-14 汉字可见）：
|
|
// 带品牌前缀、短句式，避免截断（2026-07-04 用户反馈收银台信息太少/被截断）。
|
|
plans := []model.Product{
|
|
{Name: "岩美酒库·联调测试", Description: "1 分钱端到端联调用（标准权益）", Price: "0.01", BizCode: "test_liandiao", Sort: 0},
|
|
{Name: "岩美酒库·首月特惠30天", Description: "标准权益 · 30 天 · 每门店限一次（限次由 jiu 侧校验）", Price: "1.00", BizCode: "promo_first_month", Sort: 1},
|
|
{Name: "岩美酒库·标准版月付", Description: "单店/单仓库 · 2 客户端 · 千张图片分享", Price: "299.00", BizCode: "monthly_standard", Sort: 2},
|
|
{Name: "岩美酒库·高级版月付", Description: "单店/多仓库 · 5 客户端 · 万张照片 · 免费 AI 周/月度商业分析", Price: "599.00", BizCode: "monthly_pro", Sort: 3},
|
|
{Name: "岩美酒库·标准版年付", Description: "单店/单仓库 · 2 客户端 · 千张图片分享", Price: "2999.00", BizCode: "annual_standard", Sort: 4},
|
|
{Name: "岩美酒库·高级版年付", Description: "单店/多仓库 · 5 客户端 · 万张照片 · 免费 AI 周/月度商业分析", Price: "5999.00", BizCode: "annual_pro", Sort: 5},
|
|
}
|
|
for _, p := range plans {
|
|
var existing model.Product
|
|
err := db.Where("biz_code = ?", p.BizCode).First(&existing).Error
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
p.MerchantID = merchantID
|
|
p.Active = true
|
|
if err := db.Create(&p).Error; err != nil {
|
|
log.Fatalf("[seed] 创建套餐 %s 失败: %v", p.BizCode, err)
|
|
}
|
|
continue
|
|
} else if err != nil {
|
|
log.Fatalf("[seed] 查询套餐 %s 失败: %v", p.BizCode, err)
|
|
}
|
|
// 已存在：更新价格/文案/归属（价格权威可从此改）。
|
|
db.Model(&existing).Updates(map[string]any{
|
|
"merchant_id": merchantID, "name": p.Name, "description": p.Description,
|
|
"price": p.Price, "active": true, "sort": p.Sort,
|
|
})
|
|
}
|
|
log.Printf("[seed] 业务套餐就绪：%d 个（含联调 0.01）", len(plans))
|
|
}
|
|
|
|
// seedWechat 据 config 的 wechat upsert 一个微信商户（V3 Native），并为其补两个测试套餐。
|
|
// 微信无沙箱：wechat.enabled=true 即视为正式（Production=true），只能用真实商户号做 1 分钱实测。
|
|
func seedWechat(db *gorm.DB) {
|
|
wc := config.C.Wechat
|
|
if !wc.Enabled {
|
|
log.Println("[seed] wechat.enabled=false，跳过微信商户初始化")
|
|
return
|
|
}
|
|
var m model.Merchant
|
|
err := db.Where("code = ?", wc.MerchantCode).First(&m).Error
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
m = model.Merchant{
|
|
Code: wc.MerchantCode,
|
|
Name: wc.MerchantName,
|
|
Channel: "wechat",
|
|
}
|
|
} else if err != nil {
|
|
log.Fatalf("[seed] 查询微信商户失败: %v", err)
|
|
}
|
|
m.Production = true // 微信无沙箱，一律正式网关
|
|
m.Enabled = true
|
|
m.MchID = wc.MchID
|
|
m.WxAppID = wc.AppID
|
|
m.CertSerial = wc.CertSerial
|
|
m.WxPrivateKey = wc.PrivateKey
|
|
m.APIv3Key = wc.APIv3Key
|
|
if err := db.Save(&m).Error; err != nil {
|
|
log.Fatalf("[seed] 保存微信商户失败: %v", err)
|
|
}
|
|
log.Printf("[seed] 微信商户就绪: code=%s mch_id=%s", m.Code, m.MchID)
|
|
|
|
var cnt int64
|
|
db.Model(&model.Product{}).Where("merchant_id = ?", m.ID).Count(&cnt)
|
|
if cnt == 0 {
|
|
samples := []model.Product{
|
|
{MerchantID: m.ID, Name: "微信测试套餐 A", Description: "微信联调用", Price: "0.01", Active: true, Sort: 1},
|
|
{MerchantID: m.ID, Name: "微信测试套餐 B", Description: "微信联调用", Price: "0.02", Active: true, Sort: 2},
|
|
}
|
|
if err := db.Create(&samples).Error; err != nil {
|
|
log.Fatalf("[seed] 创建微信测试套餐失败: %v", err)
|
|
}
|
|
log.Printf("[seed] 已为微信商户 %s 创建 %d 个测试套餐", m.Code, len(samples))
|
|
}
|
|
}
|