74 lines
2.1 KiB
Go
74 lines
2.1 KiB
Go
package callback
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
|
|
"github.com/gofiber/fiber/v2"
|
|
"github.com/valyala/fasthttp/fasthttpadaptor"
|
|
|
|
"github.com/break/junhong_cmp_fiber/internal/model"
|
|
orderService "github.com/break/junhong_cmp_fiber/internal/service/order"
|
|
"github.com/break/junhong_cmp_fiber/pkg/errors"
|
|
"github.com/break/junhong_cmp_fiber/pkg/response"
|
|
"github.com/break/junhong_cmp_fiber/pkg/wechat"
|
|
)
|
|
|
|
type PaymentHandler struct {
|
|
orderService *orderService.Service
|
|
wechatPayment wechat.PaymentServiceInterface
|
|
}
|
|
|
|
func NewPaymentHandler(orderService *orderService.Service, wechatPayment wechat.PaymentServiceInterface) *PaymentHandler {
|
|
return &PaymentHandler{
|
|
orderService: orderService,
|
|
wechatPayment: wechatPayment,
|
|
}
|
|
}
|
|
|
|
// WechatPayCallback 微信支付回调（带签名验证）
|
|
// POST /api/callback/wechat-pay
|
|
func (h *PaymentHandler) WechatPayCallback(c *fiber.Ctx) error {
|
|
if h.wechatPayment == nil {
|
|
return errors.New(errors.CodeWechatCallbackInvalid, "微信支付服务未配置")
|
|
}
|
|
|
|
var httpReq http.Request
|
|
fasthttpadaptor.ConvertRequest(c.Context(), &httpReq, true)
|
|
|
|
ctx := context.Background()
|
|
_, err := h.wechatPayment.HandlePaymentNotify(&httpReq, func(result *wechat.PaymentNotifyResult) error {
|
|
if result.TradeState != "SUCCESS" {
|
|
return nil
|
|
}
|
|
return h.orderService.HandlePaymentCallback(ctx, result.OutTradeNo, model.PaymentMethodWechat)
|
|
})
|
|
|
|
if err != nil {
|
|
return errors.Wrap(errors.CodeWechatCallbackInvalid, err, "处理微信支付回调失败")
|
|
}
|
|
|
|
return response.Success(c, map[string]string{"return_code": "SUCCESS"})
|
|
}
|
|
|
|
type AlipayCallbackRequest struct {
|
|
OrderNo string `json:"out_trade_no" form:"out_trade_no"`
|
|
}
|
|
|
|
func (h *PaymentHandler) AlipayCallback(c *fiber.Ctx) error {
|
|
var req AlipayCallbackRequest
|
|
if err := c.BodyParser(&req); err != nil {
|
|
return errors.New(errors.CodeInvalidParam, "请求参数解析失败")
|
|
}
|
|
|
|
if req.OrderNo == "" {
|
|
return errors.New(errors.CodeInvalidParam, "订单号不能为空")
|
|
}
|
|
|
|
if err := h.orderService.HandlePaymentCallback(c.UserContext(), req.OrderNo, model.PaymentMethodAlipay); err != nil {
|
|
return err
|
|
}
|
|
|
|
return c.SendString("success")
|
|
}
|