jy_docs/services/stdlib/docGet.go

package stdlib

import (
	"context"

	"app.yhyue.com/moapp/jy_docs/rpc/stdlib/stdlib"
	"app.yhyue.com/moapp/jy_docs/services/model"
	jyDocsRpcUtil "app.yhyue.com/moapp/jy_docs/services/util"
	"app.yhyue.com/moapp/jyfs/rpc/filesystem"
)

func DocGetCheck(docId, userId, appId string) stdlib.DocGetCheckResp {
	// docPrice := 0
	var buyCount int64
	isbuy := false
	jyDocsRpcUtil.GetJyDocsDB().Table("download_collection_record").Where("docId = ? and userId = ? and appId = ?", docId, userId, appId).Count(&buyCount)
	if buyCount > 0 {
		isbuy = true
	}
	//已购买
	doc := &model.Doc{}
	jyDocsRpcUtil.GetJyDocsDB().Where("id = ? and appId = ?", docId, appId).Find(doc)
	if doc.DocName != "" {
		// docPrice = docData["price"].(int)
		docCheck := stdlib.DocGetCheckResp{
			Code:         0,
			Msg:          "ok",
			DocName:      doc.DocName,
			DocType:      int32(doc.DocFileType),
			DocFileSize:  int32(doc.DocFileSize),
			DocPageSize:  int32(doc.DocPageSize),
			Tags:         doc.DocTags,
			UserId:       doc.UserId,
			DocSummary:   doc.DocSummary,
			PreviewImgId: doc.PreviewImgId,
			IsBuy:        isbuy,
		}
		return docCheck
	}
	return stdlib.DocGetCheckResp{Code: -1, Msg: "查询失败"}
}

func DocGetContent(docId, userId, bucketName, appId string) stdlib.DocGetContentResp {
	doc := &model.Doc{}
	docContent := stdlib.DocGetContentResp{Code: -1, Msg: "查询失败"}
	jyDocsRpcUtil.GetJyDocsDB().Where("id = ? and appId = ?", docId, appId).Find(doc)
	if doc.DocName != "" {
		file := filesystem.LoadFileReq{
			Domain: bucketName,
			FileId: doc.OssDocId,
		}
		res, err := jyDocsRpcUtil.FileSystem.GetFile(context.Background(), &file)
		if err == nil {
			docContent.Code = 0
			docContent.Msg = "ok"
			docContent.DocContent = res.RawFileContent
		}
	}
	return docContent
}