refactor: subscribe module

server/service/subscribe.go

@@ -1,23 +1,12 @@
 package service
 
 import (
+	"crm/common"
 	"crm/dao"
-	"crm/global"
 	"crm/models"
 	"crm/response"
 	"encoding/json"
-	"fmt"
-	"log"
-	"net/url"
 	"time"
-
-	"github.com/smartwalle/alipay/v3"
-	"github.com/smartwalle/xid"
-)
-
-const (
-	Paying = 1 // 支付中
-	Payed  = 2 // 已支付
 )
 
 type SubscribeService struct {
@@ -37,134 +26,69 @@ func NewSubscribeService() *SubscribeService {
 func (s *SubscribeService) Pay(param models.SubscribePayParam) (*models.SubscribePayUrl, int) {
 
 	// 构建订单支付信息
-	tradeNo := fmt.Sprintf("%d", xid.Next())
-	var p = alipay.TradePagePay{}
-	p.NotifyURL = global.Config.Alipay.NotifyURL
-	p.ReturnURL = global.Config.Alipay.ReturnURL
-	p.Subject = "支付测试:" + tradeNo
-	p.OutTradeNo = tradeNo
-	p.ProductCode = "FAST_INSTANT_TRADE_PAY"
-	switch param.Version {
-	case 2:
-		p.TotalAmount = "18.00"
-	case 3:
-		p.TotalAmount = "198.00"
-	}
-
-	// 缓存订单信息
-	order := &models.SubscribePayOrder{
-		Uid:     param.Uid,
-		Version: param.Version,
-	}
-	if err := s.subscribeDao.SetOrder(tradeNo, order); err != nil {
-		return nil, response.ErrCodeFailed
-	}
-
-	// 返回支付链接
-	payUrl, err := global.Alipay.TradePagePay(p)
+	tradeNo := common.GetAlipay().GenTradeNo()
+	totalAmount := float64(param.Duration) * 0.6
+	payUrl, err := common.GetAlipay().PagePay(tradeNo, totalAmount)
 	if err != nil {
 		return nil, response.ErrCodeFailed
 	}
+
+	order := models.SubscribePayOrder{
+		Uid:      param.Uid,
+		TradeNo:  tradeNo,
+		Duration: param.Duration,
+	}
+
+	if err := s.subscribeDao.SetOrder(tradeNo, &order); err != nil {
+		return nil, response.ErrCodeFailed
+	}
+
 	subscribePayUrl := models.SubscribePayUrl{
-		PayUrl: payUrl.String(),
+		PayUrl: payUrl,
 	}
 	return &subscribePayUrl, response.ErrCodeSuccess
 }
 
 // 支付成功回调
-func (s *SubscribeService) Callback(outTradeNo string) (string, int) {
-
-	// 验证订单信息
-	var p = alipay.TradeQuery{}
-	p.OutTradeNo = outTradeNo
-	rsp, err := global.Alipay.TradeQuery(p)
-	if err != nil {
-		log.Printf("验证订单 %s 信息发生错误: %s", outTradeNo, err.Error())
-		return StringNull, response.ErrCodeFailed
-	}
-	if !rsp.IsSuccess() {
-		log.Printf("验证订单 %s 信息发生错误: %s-%s", outTradeNo, rsp.Content.Msg, rsp.Content.SubMsg)
-		return StringNull, response.ErrCodeFailed
-	}
+func (s *SubscribeService) PayBack(outTradeNo string) int {
 
 	// 获取订单信息
 	var order models.SubscribePayOrder
 	orderJson, err := s.subscribeDao.GetOrder(outTradeNo)
 	if err != nil {
-		return StringNull, response.ErrCodeFailed
+		return response.ErrCodeFailed
 	}
 	if err := json.Unmarshal([]byte(orderJson), &order); err != nil {
-		return StringNull, response.ErrCodeFailed
+		return response.ErrCodeFailed
 	}
 
 	// 创建订阅信息
-	var expired int64
-	var content string
-	switch order.Version {
-	case 2:
-		expired = time.Now().Unix() + int64(2592000)
-		content = SUBSCRIBE_NOTICE_TEMPLATE1
-	case 3:
-		expired = time.Now().Unix() + int64(31104000)
-		content = SUBSCRIBE_NOTICE_TEMPLATE2
-	}
+	expired := order.Duration * 24 * 60 * 60
 	if !s.subscribeDao.IsExists(order.Uid) {
 		subscribe := models.SubscribeCreateParam{
 			Uid:     order.Uid,
-			Version: order.Version,
+			Version: 2,
 			Expired: expired,
 		}
 		if err := s.subscribeDao.Create(&subscribe); err != nil {
-			return StringNull, response.ErrCodeFailed
+			return response.ErrCodeFailed
 		}
 	} else {
 		subscribe := models.SubscribeUpdateParam{
 			Uid:     order.Uid,
-			Version: order.Version,
 			Expired: expired,
 		}
 		if err := s.subscribeDao.Update(&subscribe); err != nil {
-			return StringNull, response.ErrCodeFailed
+			return response.ErrCodeFailed
 		}
 	}
 
 	// 订阅通知
 	s.noticeDao.Create(&models.NoticeCreateParam{
-		Content: content,
+		Content: SUBSCRIBE_NOTICE_TEMPLATE1,
 		Creator: order.Uid,
 	})
 
-	return global.Config.Alipay.PaySuccessURL, response.ErrCodeSuccess
-}
-
-// 异步验签
-func (s *SubscribeService) Notify(data url.Values, outTradeNo string) int {
-	ok, err := global.Alipay.VerifySign(data)
-	if err != nil {
-		log.Println("异步通知验证签名发生错误", err)
-		return response.ErrCodeFailed
-	}
-
-	if !ok {
-		log.Println("异步通知验证签名未通过")
-		return response.ErrCodeFailed
-	}
-
-	log.Println("异步通知验证签名通过")
-
-	var p = alipay.TradeQuery{}
-	p.OutTradeNo = outTradeNo
-	rsp, err := global.Alipay.TradeQuery(p)
-	if err != nil {
-		log.Printf("异步通知验证订单 %s 信息发生错误: %s \n", outTradeNo, err.Error())
-		return response.ErrCodeFailed
-	}
-	if !rsp.IsSuccess() {
-		log.Printf("异步通知验证订单 %s 信息发生错误: %s-%s \n", outTradeNo, rsp.Content.Msg, rsp.Content.SubMsg)
-		return response.ErrCodeFailed
-	}
-
-	log.Printf("订单 %s 支付成功 \n", outTradeNo)
 	return response.ErrCodeSuccess
 }