LingXi-CRM/server/service/subscribe.go

package service

import (
	"crm/common"
	"crm/dao"
	"crm/models"
	"crm/response"
	"encoding/json"
	"time"
)

type SubscribeService struct {
	subscribeDao *dao.SubscribeDao
	noticeDao    *dao.NoticeDao
}

func NewSubscribeService() *SubscribeService {
	subscribeService := SubscribeService{
		subscribeDao: dao.NewSubscribeDao(),
		noticeDao:    dao.NewNoticeDao(),
	}
	return &subscribeService
}

// 订阅专业版，发起支付
func (s *SubscribeService) Pay(param models.SubscribePayParam) (*models.SubscribePayUrl, int) {

	// 构建订单支付信息
	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,
	}
	return &subscribePayUrl, response.ErrCodeSuccess
}

// 支付成功回调
func (s *SubscribeService) PayBack(outTradeNo string) int {

	// 获取订单信息
	var order models.SubscribePayOrder
	orderJson, err := s.subscribeDao.GetOrder(outTradeNo)
	if err != nil {
		return response.ErrCodeFailed
	}
	if err := json.Unmarshal([]byte(orderJson), &order); err != nil {
		return response.ErrCodeFailed
	}

	// 创建订阅信息
	duration := order.Duration * 24 * 60 * 60
	if !s.subscribeDao.IsExists(order.Uid) {
		subscribe := models.SubscribeCreateParam{
			Uid:     order.Uid,
			Version: 2,
			Expired: time.Now().Unix() + duration,
		}
		if err := s.subscribeDao.Create(&subscribe); err != nil {
			return response.ErrCodeFailed
		}
	} else {
		si, err := s.subscribeDao.GetInfo(order.Uid)
		if err != nil {
			return response.ErrCodeFailed
		}
		subscribe := models.SubscribeUpdateParam{
			Uid:     order.Uid,
			Version: 2,
			Expired: si.Expired + duration,
		}
		if err := s.subscribeDao.Update(&subscribe); err != nil {
			return response.ErrCodeFailed
		}
	}

	// 订阅通知
	s.noticeDao.Create(&models.NoticeCreateParam{
		Content: SUBSCRIBE_NOTICE_TEMPLATE1,
		Creator: order.Uid,
	})

	return response.ErrCodeSuccess
}

// 获取订阅信息
func (s *SubscribeService) GetInfo(uid int64) (*models.SubscribeInfo, int) {
	// Always return Professional version with far expiration
	si := &models.SubscribeInfo{
		Version: 2,
		Expired: 4102444800, // 2100-01-01
	}
	return si, response.ErrCodeSuccess
}