mirror of
https://github.com/bufanyun/hotgo.git
synced 2025-09-17 16:56:39 +08:00
145 lines
3.5 KiB
Go
145 lines
3.5 KiB
Go
// Package middleware
|
|
// @Link https://github.com/bufanyun/hotgo
|
|
// @Copyright Copyright (c) 2023 HotGo CLI
|
|
// @Author Ms <133814250@qq.com>
|
|
// @License https://github.com/bufanyun/hotgo/blob/master/LICENSE
|
|
package middleware
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/gogf/gf/v2/errors/gcode"
|
|
"github.com/gogf/gf/v2/errors/gerror"
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
"github.com/gogf/gf/v2/net/ghttp"
|
|
"github.com/gogf/gf/v2/util/gconv"
|
|
"hotgo/internal/consts"
|
|
"hotgo/internal/library/contexts"
|
|
"hotgo/internal/library/response"
|
|
"hotgo/internal/model/input/payin"
|
|
"hotgo/utility/charset"
|
|
"hotgo/utility/simple"
|
|
"net/http"
|
|
)
|
|
|
|
// ResponseHandler HTTP响应预处理
|
|
func (s *sMiddleware) ResponseHandler(r *ghttp.Request) {
|
|
r.Middleware.Next()
|
|
|
|
// 已存在响应
|
|
if r.Response.BufferLength() > 0 && contexts.Get(r.Context()).Response != nil {
|
|
return
|
|
}
|
|
|
|
// html模板响应
|
|
if r.Response.Header().Get("Content-Type") == "text/html" {
|
|
s.responseHtml(r)
|
|
return
|
|
}
|
|
|
|
// 支付通知响应
|
|
if _, ok := s.PayNotifyRoutes[r.Router.Uri]; ok {
|
|
s.responsePayNotify(r)
|
|
return
|
|
}
|
|
|
|
// 默认json响应
|
|
responseJson(r)
|
|
}
|
|
|
|
// responseHtml html模板响应
|
|
func (s *sMiddleware) responseHtml(r *ghttp.Request) {
|
|
code, message, resp := parseResponse(r)
|
|
if code == gcode.CodeOK.Code() {
|
|
return
|
|
}
|
|
|
|
r.Response.ClearBuffer()
|
|
_ = r.Response.WriteTplContent(simple.DefaultErrorTplContent(r.Context()), g.Map{
|
|
"code": code,
|
|
"message": message,
|
|
"stack": resp,
|
|
})
|
|
return
|
|
}
|
|
|
|
// responsePayNotify 支付通知响应
|
|
func (s *sMiddleware) responsePayNotify(r *ghttp.Request) {
|
|
var (
|
|
ctx = r.Context()
|
|
err error
|
|
data *payin.PayNotifyModel
|
|
)
|
|
|
|
code, message, resp := parseResponse(r)
|
|
if code != gcode.CodeOK.Code() {
|
|
response.RJson(r, code, message, data)
|
|
return
|
|
}
|
|
|
|
if err = gconv.Scan(resp, &data); err != nil || data == nil {
|
|
g.Log("exception").Errorf(ctx, "middleware.responsePayNotify Scan err:%+v, data:%+v", err, data)
|
|
r.Response.ClearBuffer()
|
|
r.Response.WriteStatus(http.StatusInternalServerError, err.Error())
|
|
return
|
|
}
|
|
|
|
switch data.PayType {
|
|
case consts.PayTypeAliPay:
|
|
response.RText(r, data.Message)
|
|
|
|
case consts.PayTypeWxPay:
|
|
r.Response.ClearBuffer()
|
|
r.Response.WriteJson(fmt.Sprintf(`{"code": "%v","message": "%v"}`, data.Code, data.Message))
|
|
|
|
case consts.PayTypeQQPay:
|
|
r.Response.ClearBuffer()
|
|
r.Response.Write(`<?xml version="1.0" encoding="UTF-8"?>`)
|
|
r.Response.WriteXml(g.Map{
|
|
"return_code": data.Message,
|
|
})
|
|
|
|
default:
|
|
err = gerror.Newf("无效的支付方式，这可能是没有配置通知回调响应方式导致的：%+v", data)
|
|
g.Log("exception").Error(ctx, err)
|
|
r.Response.ClearBuffer()
|
|
r.Response.WriteStatus(http.StatusInternalServerError, err.Error())
|
|
}
|
|
}
|
|
|
|
// responseJson json响应
|
|
func responseJson(r *ghttp.Request) {
|
|
code, message, data := parseResponse(r)
|
|
response.RJson(r, code, message, data)
|
|
}
|
|
|
|
// parseResponse 解析响应数据
|
|
func parseResponse(r *ghttp.Request) (code int, message string, resp interface{}) {
|
|
var (
|
|
ctx = r.Context()
|
|
err = r.GetError()
|
|
)
|
|
|
|
if err == nil {
|
|
return gcode.CodeOK.Code(), "操作成功", r.GetHandlerResponse()
|
|
}
|
|
|
|
// 是否输出错误堆栈到页面
|
|
if g.Cfg().MustGet(ctx, "hotgo.debug", true).Bool() {
|
|
message = gerror.Current(err).Error()
|
|
resp = charset.ParseErrStack(err)
|
|
} else {
|
|
message = consts.ErrorMessage(gerror.Current(err))
|
|
}
|
|
|
|
// 解析错误状态码
|
|
code = gerror.Code(err).Code()
|
|
|
|
// 记录异常日志
|
|
if code == gcode.CodeNil.Code() {
|
|
g.Log().Stdout(false).Printf(ctx, "exception:%v", err)
|
|
} else {
|
|
g.Log().Errorf(ctx, "exception:%v", err)
|
|
}
|
|
return
|
|
}
|