压缩历史至 95081aff

2026-10-09 00:56:37 +08:00 · 2026-06-08 20:34:27 +08:00
commit 8a782525de
435 changed files with 71146 additions and 0 deletions
@@ -0,0 +1,142 @@
+/*
+Copyright 2025 linux.do
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package payment
+
+import (
+	"cmp"
+	"context"
+	"encoding/json"
+	"errors"
+	"fmt"
+	"io"
+	"net/http"
+	"net/url"
+	"strconv"
+	"strings"
+
+	"github.com/hibiken/asynq"
+	"github.com/linux-do/credit/internal/common"
+	"github.com/linux-do/credit/internal/config"
+	"github.com/linux-do/credit/internal/db"
+	"github.com/linux-do/credit/internal/logger"
+	"github.com/linux-do/credit/internal/model"
+	"github.com/linux-do/credit/internal/util"
+	"gorm.io/gorm"
+)
+
+// HandleMerchantPaymentNotify 处理商户支付回调任务
+func HandleMerchantPaymentNotify(ctx context.Context, t *asynq.Task) error {
+	// 解析任务参数
+	var payload struct {
+		OrderID  uint64 `json:"order_id"`
+		ClientID string `json:"client_id"`
+	}
+	if err := json.Unmarshal(t.Payload(), &payload); err != nil {
+		logger.ErrorF(ctx, "解析商户回调任务参数失败: %v", err)
+		return fmt.Errorf("解析任务参数失败: %w", err)
+	}
+
+	// 查询订单信息
+	var order model.Order
+	if err := db.DB(ctx).Where("id = ? AND status = ?", payload.OrderID, model.OrderStatusSuccess).First(&order).Error; err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			logger.ErrorF(ctx, "订单[ID:%d]不存在，跳过回调", payload.OrderID)
+			return nil
+		}
+		return fmt.Errorf("查询订单失败: %w", err)
+	}
+
+	// 查询商户API Key信息
+	var apiKey model.MerchantAPIKey
+	if err := apiKey.GetByClientID(db.DB(ctx), payload.ClientID); err != nil {
+		logger.ErrorF(ctx, "查询商户[ClientID:%s]失败: %v", payload.ClientID, err)
+		return fmt.Errorf("查询商户信息失败: %w", err)
+	}
+
+	// 回调 URL
+	callbackURL := cmp.Or(order.NotifyURL, apiKey.NotifyURL)
+
+	// 判断是否需要回调
+	if config.Config.App.IsProduction() && util.IsLocalhost(callbackURL) {
+		return nil
+	}
+
+	// 构建回调参数
+	callbackParams := map[string]string{
+		"pid":          payload.ClientID,
+		"trade_no":     strconv.FormatUint(order.ID, 10),
+		"out_trade_no": util.DerefString(order.MerchantOrderNo),
+		"type":         common.PayTypeEPay,
+		"name":         order.OrderName,
+		"money":        order.Amount.Truncate(2).StringFixed(2),
+		"trade_status": "TRADE_SUCCESS",
+	}
+	callbackParams["sign"] = GenerateSignature(callbackParams, apiKey.ClientSecret, true)
+
+	// 回调
+	if err := sendCallbackRequest(ctx, callbackURL, callbackParams); err != nil {
+		retried, _ := asynq.GetRetryCount(ctx)
+		logger.ErrorF(ctx, "商户回调失败: 订单[ID:%d] 重试次数[%d] 错误: %v",
+			payload.OrderID, retried+1, err)
+		return err
+	}
+
+	logger.InfoF(ctx, "商户回调成功: 订单[ID:%d] ClientID[%s]", payload.OrderID, payload.ClientID)
+	return nil
+}
+
+// sendCallbackRequest 发送HTTP回调请求
+func sendCallbackRequest(ctx context.Context, callbackURL string, params map[string]string) error {
+	vals := url.Values{}
+	for k, v := range params {
+		vals.Add(k, v)
+	}
+
+	// 拼接URL
+	separator := "?"
+	if strings.Contains(callbackURL, "?") {
+		separator = "&"
+	}
+	targetURL := callbackURL + separator + vals.Encode()
+
+	headers := map[string]string{
+		"User-Agent": "LinuxDo-Credit/1.0",
+	}
+
+	resp, err := util.Request(ctx, http.MethodGet, targetURL, nil, headers, nil)
+	if err != nil {
+		return err
+	}
+	defer resp.Body.Close()
+
+	if resp.StatusCode != http.StatusOK {
+		return fmt.Errorf("回调返回异常状态码: %d", resp.StatusCode)
+	}
+
+	respBody, err := io.ReadAll(resp.Body)
+	if err != nil {
+		return fmt.Errorf("读取响应失败: %w", err)
+	}
+
+	responseText := strings.TrimSpace(strings.ToLower(string(respBody)))
+	if responseText != "success" {
+		return fmt.Errorf("回调返回非成功响应: %s", string(respBody))
+	}
+
+	logger.InfoF(ctx, "商户回调请求成功: URL[%s] 响应[%s]", callbackURL, string(respBody))
+	return nil
+}