shanshanzhong
c582087c0f
Some checks failed
Build docker and publish / build (20.15.1) (push) Failing after 6m27s
feat: 添加版本和构建时间变量 fix: 修正短信队列类型注释错误 style: 清理未使用的代码和测试文件 docs: 更新安装文档中的下载链接 chore: 迁移数据库脚本添加日志和订阅配置
39 lines
938 B
Go
39 lines
938 B
Go
package middleware
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/perfect-panel/server/pkg/constant"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/perfect-panel/server/internal/svc"
|
|
)
|
|
|
|
type PaymentParams struct {
|
|
Platform string `uri:"platform"`
|
|
Token string `uri:"token"`
|
|
}
|
|
|
|
func NotifyMiddleware(svc *svc.ServiceContext) func(c *gin.Context) {
|
|
return func(c *gin.Context) {
|
|
ctx := c.Request.Context()
|
|
var params PaymentParams
|
|
// Get platform and token from uri
|
|
if err := c.ShouldBindUri(¶ms); err != nil {
|
|
c.JSON(400, gin.H{"error": err.Error()})
|
|
c.Abort()
|
|
return
|
|
}
|
|
config, err := svc.PaymentModel.FindOneByPaymentToken(ctx, params.Token)
|
|
if err != nil {
|
|
c.JSON(400, gin.H{"error": err.Error()})
|
|
c.Abort()
|
|
return
|
|
}
|
|
ctx = context.WithValue(ctx, constant.CtxKeyPlatform, config.Platform)
|
|
ctx = context.WithValue(ctx, constant.CtxKeyPayment, config)
|
|
c.Request = c.Request.WithContext(ctx)
|
|
c.Next()
|
|
}
|
|
}
|