jy-docs/jydocs-back/rpc/stdDocRpc.go

package rpc

import (
	"app.yhyue.com/moapp/jy_docs/rpc/stdlib/stdlib"
	"app.yhyue.com/moapp/jy_docs/rpc/stdlib/stdlibclient"
	"context"
	"fmt"
	"github.com/tal-tech/go-zero/core/discov"
	"github.com/tal-tech/go-zero/zrpc"
	"jy-docs/config"
	"log"
)

// 标准库RPC接口
var jyStdDocStdlib stdlibclient.Stdlib

func init() {
	jyStdDocStdlib = stdlibclient.NewStdlib(zrpc.MustNewClient(zrpc.RpcClientConf{
		Etcd: discov.EtcdConf{
			Key:   config.JyDocsAppConfig.RpcServers.StdDoc.Key,
			Hosts: config.JyDocsAppConfig.RpcServers.StdDoc.Address,
		},
	}))
}

/*
检索文库
param

	userId  用户id
	keyWord 关键词
	tag	分类
	pageNum 页码
	pageSize 每页数量
	tSort 时间排序
	dSort 下载排序
	vSort 浏览量排序
*/
func GetDocQuery(userId, keyWord, tag string, pageNum, pageSize int64, sort string) ([]*stdlib.Doc, int64, error) {
	param := &stdlib.DocQueryRequest{
		AppId:    config.JyDocsAppConfig.AppId,
		KeyWord:  keyWord,
		PageSize: pageSize,
		PageNum:  pageNum,
	}
	if tag != "" {
		param.DocTag = []string{tag}
	}
	sortArr := []string{}
	switch sort { //倒序字段前加-，uploadDate：上架时间 viewTimes：浏览量 downTimes：下载量
	case "dSort": //下载量倒叙
		sortArr = append(sortArr, "-downTimes")
	case "vSort": //浏览量倒叙
		sortArr = append(sortArr, "-viewTimes")
	default: // "tSort"上传时间倒叙
		sortArr = append(sortArr, "-uploadDate")
	}
	param.Sort = sortArr

	resp, err := jyStdDocStdlib.DocQuery(context.Background(), param)
	if err != nil {
		log.Printf("%s SetUserCollect call error %v\n", userId, err)
		return nil, -1, err
	}
	if resp.Code != 1 {
		log.Printf("%s SetUserCollect fail Message %v\n", userId, resp.Msg)
		return nil, -1, fmt.Errorf("查询失败")
	}
	for _, v := range resp.Docs {
		if v.PreviewImgId != "" {
			v.PreviewImgId = fmt.Sprintf("https://%s.%s/%s", config.JyDocsAppConfig.OssBucket.Priv, config.JyDocsAppConfig.OssAdmin, v.PreviewImgId)
		}
	}
	return resp.Docs, resp.Total, nil
}

/*
获取活动列表
param

	userId  用户id
	code	活动编号
	pageNum 页码
	pageSize 每页数量

return

	文库列表、异常
*/
func GeActivityList(userId string, code, pageNum, pageSize int64) ([]*stdlib.DocActivity, error) {
	resp, err := jyStdDocStdlib.DocActivity(context.Background(), &stdlib.DocActivityReq{
		AppId:      config.JyDocsAppConfig.AppId,
		ActivityId: code,
		PageNum:    pageNum,
		PageSize:   pageSize,
		UserId:     userId,
	})
	if err != nil {
		log.Printf("%s GeActivityList call error %v\n", userId, err)
		return nil, err
	}
	if resp.Code != 1 {
		log.Printf("%s GeActivityList fail Message %v\n", userId, resp.Msg)
		return nil, fmt.Errorf("获取列表失败")
	}
	return resp.Docs, nil
}

/*
获取文库详情
param

	userId  用户id
	docId	文库id

return

	DocInfo 文库详情
	error 异常
*/
func GetDocDetail(userId, docId string) (*stdlib.DocInfo, bool, bool, error) {
	resp, err := jyStdDocStdlib.DocGetCheck(context.Background(), &stdlib.DocGetCheckReq{
		AppId:  config.JyDocsAppConfig.AppId,
		UserId: userId,
		DocId:  docId,
	})
	if err != nil {
		log.Printf("%s GetDocDetail call error %v\n", userId, err)
		return nil, false, false, err
	}
	if resp.Code != 1 {
		log.Printf("%s GetDocDetail fail Message %v\n", userId, resp.Msg)
		return nil, false, false, fmt.Errorf("获取内容失败")
	}
	if resp.DocDeail == nil {
		return nil, false, false, fmt.Errorf("查询文档异常")
	}
	return resp.DocDeail, resp.IsBuy, resp.IsCollect, nil
}

/*
文库浏览次数、下载次数统计
param

	userId  用户id
	docId	文库id
	sign	1增加下载次数 2增加浏览次数 3评分
*/
const Down int32 = 1
const View int32 = 2

func DocStatistics(userId, docId string, sign int32) {
	resp, err := jyStdDocStdlib.DocStatistics(context.Background(), &stdlib.DocStatisticsReq{
		AppId:             config.JyDocsAppConfig.AppId,
		DocId:             docId,
		DocStatisticsType: sign,
	})
	if err != nil {
		log.Printf("%s DocStatistics call error %v\n", userId, err)
	}
	if !resp.State {
		log.Printf("%s DocStatistics fail Message %v\n", userId, resp)
	}
}