jy_docs/rpc/userlib/internal/logic/userdocslogic.go

package logic

import (
	"app.yhyue.com/moapp/jy_docs/rpc/userlib/internal/svc"
	"app.yhyue.com/moapp/jy_docs/rpc/userlib/userlib"
	userLibService "app.yhyue.com/moapp/jy_docs/services/userlib"
	"context"
	"strings"

	"github.com/zeromicro/go-zero/core/logx"
)

type UserDocsLogic struct {
	ctx    context.Context
	svcCtx *svc.ServiceContext
	logx.Logger
}

func NewUserDocsLogic(ctx context.Context, svcCtx *svc.ServiceContext) *UserDocsLogic {
	return &UserDocsLogic{
		ctx:    ctx,
		svcCtx: svcCtx,
		Logger: logx.WithContext(ctx),
	}
}

const (
	ProductTypeMemberFree = 1 // 商品类型 会员免费
	ProductTypePremium    = 2 // 商品类型 精品文档
)

// 我的文档 收藏和兑换
func (l *UserDocsLogic) UserDocs(in *userlib.UserDocsRequest) (*userlib.UserDocsResponse, error) {
	// todo: add your logic here and delete this line
	result := &userlib.UserDocsResponse{}
	data, count, code, msg := userLibService.UserDocsList(in)
	docList := []*userlib.UserDocs{}
	for _, value := range data {
		if value.IsDownload == 1 {
			value.Cost = "已购买"
		}
		// 因为tidb的历史数据没有处理 收藏列表需要展示商品类型 所以需要处理一下
		if value.ProductType == 0 && in.UserDocCategory == int64(1) {
			var flag bool
			tags := strings.Split(value.DocTags, ",")
			for i := 0; i < len(tags); i++ {
				if l.svcCtx.Config.JyPremiumTag[tags[i]] {
					// 说明是剑鱼精品文档
					flag = true
					value.ProductType = ProductTypePremium
					break
				}
			}
			if !flag {
				// 否则是剑鱼会员免费文档
				value.ProductType = ProductTypeMemberFree
			}
		}
		docList = append(docList, &userlib.UserDocs{
			CreateAt:      value.CreateAt.Unix(),
			UpdateAt:      value.UpdateAt.Unix(),
			DeletedAt:     value.DeletedAt.Unix(),
			DocId:         value.DocId,
			IsDelete:      int64(value.IsDelete),
			DocName:       value.DocName,
			DocFileType:   int64(value.DocFileType),
			DocFileSuffix: value.DocFileSuffix,
			DocFileSize:   int64(value.DocFileSize),
			DocPageSize:   int64(value.DocPageSize),
			DocSummary:    value.DocSummary,
			IsDownload:    int64(value.IsDownload),
			IsCollection:  int64(value.IsCollection),
			Cost:          value.Cost,
			ProductType:   value.ProductType,
			Source:        value.Source,
		})
	}
	if code {
		result.Code = 1
		result.Message = msg
		result.Data = docList
		result.Count = count
		return result, nil
	}
	result.Code = 0
	result.Message = msg
	result.Data = docList
	result.Count = count
	return result, nil
}