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
	var collectCount int64
	isbuy, isCollect := false, false
	jyDocsRpcUtil.GetJyDocsDB().Table("download_collection_record").Where("docId = ? and userId = ? and appId = ? and category = 1", docId, userId, appId).Count(&buyCount)
	if buyCount > 0 {
		isbuy = true
	}
	jyDocsRpcUtil.GetJyDocsDB().Table("download_collection_record").Where("docId = ? and userId = ? and appId = ? and category = 2", docId, userId, appId).Count(&collectCount)
	if collectCount > 0 {
		isCollect = true
	}
	//已购买
	doc := &model.Doc{}
	jyDocsRpcUtil.GetJyDocsDB().Where("id = ? and appId = ?", docId, appId).Find(doc)
	if doc.DocName != "" {
		// docPrice = docData["price"].(int)
		docStatistics := model.DocStatistics{}
		jyDocsRpcUtil.GetJyDocsDB().First(&docStatistics, &model.DocStatistics{AppId: appId, DocId: docId})
		ossPdfId, ossDocId := "", ""
		if isbuy {
			ossPdfId = doc.OssPdfId
			ossDocId = doc.OssDocId
		}
		docCheck := stdlib.DocGetCheckResp{
			Code:      1,
			Msg:       "ok",
			IsBuy:     isbuy,
			IsCollect: isCollect,
			DocDeail: &stdlib.DocInfo{
				DocId:        doc.Id,
				DocName:      doc.DocName,
				Price:        int64(doc.Price),
				DocFileSize:  int64(doc.DocFileSize),
				DocPageSize:  int64(doc.DocPageSize),
				Tags:         doc.DocTags,
				DownTimes:    int64(docStatistics.DownTimes),
				ViewTimes:    int64(docStatistics.ViewTimes),
				UploadDate:   doc.UploadDate.Format("2006-01-02 15:04:05"),
				DocSummary:   doc.DocSummary,
				DocFileType:  int64(doc.DocFileType),
				PreviewImgId: doc.PreviewImgId,
				OssPdfId:     ossPdfId,
				OssDocId:     ossDocId,
			},
		}
		return docCheck
	}
	return stdlib.DocGetCheckResp{Code: 0, Msg: "查询失败"}
}

func DocGetContent(docId, userId, bucketName, appId string) stdlib.DocGetContentResp {
	doc := &model.Doc{}
	docContent := stdlib.DocGetContentResp{Code: 0, 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 = 1
			docContent.Msg = "ok"
			docContent.DocContent = res.RawFileContent
		}
	}
	return docContent
}