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 string) stdlib.DocGetCheckResp {
	// docPrice := 0
	var buyCount int64
	jyDocsRpcUtil.GetJyDocsDB().Table("doc").Where("docId = ? and userId = ?", docId, userId).Count(&buyCount)
	if buyCount > 0 {
		//已购买
		doc := &model.Doc{}
		jyDocsRpcUtil.GetJyDocsDB().Where("id = ?", docId).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,
			}
			return docCheck
		}
	}
	return stdlib.DocGetCheckResp{Code: -1, Msg: "查询失败"}
}

func DocGetContent(docId, userId, bucketName string) stdlib.DocGetContentResp {
	doc := &model.Doc{}
	docContent := stdlib.DocGetContentResp{Code: -1, Msg: "查询失败"}
	jyDocsRpcUtil.GetJyDocsDB().Where("id = ?", docId).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
}