jy_docs/services/userlib/userDocService.go

package userlib

import (
	"app.yhyue.com/moapp/jyPoints/rpc/integral"
	"app.yhyue.com/moapp/jyPoints/rpc/integralclient"
	"app.yhyue.com/moapp/jy_docs/services/model"
	docRpcUtil "app.yhyue.com/moapp/jy_docs/services/util"
	"context"
	"github.com/olivere/elastic/v7/config"
	"github.com/tal-tech/go-zero/core/discov"
	"github.com/tal-tech/go-zero/zrpc"
	"gorm.io/gorm"
	"log"
	"time"
)

func UserDocCollect(userDoc *model.UserDoc, cost int) bool {
	log.Println("UserDocCollect exec ......")
	err := docRpcUtil.GetJyDocsDB().Transaction(func(tx *gorm.DB) error {
		//用户文库表添加记录(需要检查是否重复)
		err := docRpcUtil.GetJyDocsDB().Create(userDoc).Error
		if err != nil {
			log.Println("userDocCollect error:", err)
			tx.Rollback()
			return err
		}
		//用户收藏、转存记录表添加记录
		err = docRpcUtil.GetJyDocsDB().Exec("insert into download_collection_record (docId,userId,sourceUserId,category,cost) values (?,?,?,?,?)", userDoc.DocId, userDoc.UserId, userDoc.DocSourceUserId, userDoc.DocCategory, cost).Error
		if err != nil {
			log.Println("userDocCollect record insert error:", err)
			tx.Rollback()
			return err
		}
		return nil
	})
	if err != nil {
		return false
	}
	return true
}

//转存操作
func UserDocDownload(userDoc *model.UserDoc, cost int,hosts []string,key string) bool {
	log.Println("UserDocCollect exec ......")
	err := docRpcUtil.GetJyDocsDB().Transaction(func(tx *gorm.DB) error {
		//用户文库表添加记录
		//获取文档有关信息
		//查询之前有无下载记录  0和1只要有一条不能下载，如果有未删除可以下载
		userDocDownloadList := []model.UserDoc{}
		err := docRpcUtil.GetJyDocsDB().Table("user_doc").Where(" userId=? and docId=? and docCategory=? and (isDelete=? or isDelete=?)", userDoc.UserId, userDoc.DocId, model.UserDocCategory_Download, model.UserDocStatus_Normal, model.UserDocStatus_LogicDelete).Find(&userDocDownloadList)
		if err.Error != nil {
			return err.Error
		}
		if len(userDocDownloadList) > 0 {
			return nil
		}
		//先扣除积分
		//rpc
		client := zrpc.MustNewClient(zrpc.RpcClientConf{
			Etcd: discov.EtcdConf{
				Hosts: hosts,
				Key:  key,
			},
		})
		integralLib := integralclient.NewIntegral(client)
		req := &integral.Req{UserId: userDoc.UserId,
			PointType:      2003,
			BusinessTypeId: 1,
			BusinessType:   "1",
			Point:          int64(cost),
			AppId:          10000}
		res, pointsErr := integralLib.IntegralConsume(context.Background(), req)
		log.Println("err ", pointsErr)
		log.Println("req ", res)
		if (pointsErr != nil || res.Code == 0) {
			return pointsErr
		}

		//查询之前有无收藏记录 有收藏记录类型改为下载，状态改为未删除、有收藏记录插入一条新纪录
		userDocCollectionList := []model.UserDoc{}
		err = docRpcUtil.GetJyDocsDB().Table("user_doc").Where(" userId=? and docId=? and docCategory=? and (isDelete=? or isDelete=?)", userDoc.UserId, userDoc.DocId, model.UserDocCategory_Collect, model.UserDocStatus_Normal, model.UserDocStatus_LogicDelete).Find(&userDocCollectionList)
		if err.Error != nil {
			return err.Error
		}
		if len(userDocCollectionList) > 0 {
			//修改收藏为下载
			if err := tx.Exec("update user_doc set docCategory=? , isDelete=?  ,update_at=?  where  id =?", model.UserDocCategory_Download, model.UserDocStatus_Normal, time.Now(), userDocCollectionList[0].ID).Error; err != nil {
				return err
			}
			if err.Error != nil {
				log.Println("userDocCollect error:", err)
				tx.Rollback()
				return err.Error
			}
			//用户收藏、转存记录表添加记录
			err := docRpcUtil.GetJyDocsDB().Exec("insert into download_collection_record (docId,userId,sourceUserId,category,cost,date) values (?,?,?,?,?,?)", userDoc.DocId, userDoc.UserId, userDoc.DocSourceUserId, model.UserDocCategory_Download, cost, time.Now())
			if err.Error != nil {
				log.Println("userDocCollect record insert error:", err)
				tx.Rollback()
				return err.Error
			}
		} else {
			//用户文库表添加记录
			userDoc.CreateAt = time.Now()
			userDoc.UpdateAt = time.Now()
			userDoc.DeletedAt = time.Now()
			err = docRpcUtil.GetJyDocsDB().Create(userDoc)
			if err.Error != nil {
				log.Println("userDocCollect error:", err)
				tx.Rollback()
				return err.Error
			}
			//用户收藏、转存记录表添加记录
			err := docRpcUtil.GetJyDocsDB().Exec("insert into download_collection_record (docId,userId,sourceUserId,category,cost,date) values (?,?,?,?,?,?)", userDoc.DocId, userDoc.UserId, userDoc.DocSourceUserId, model.UserDocCategory_Download, cost, time.Now())
			if err.Error != nil {
				log.Println("userDocCollect record insert error:", err)
				tx.Rollback()
				return err.Error
			}
		}
		return nil
	})
	if err != nil {
		return false
	}
	return true
}
func UserDocDelete(docId string) bool {
	err := docRpcUtil.GetJyDocsDB().Transaction(func(tx *gorm.DB) error {
		orm := docRpcUtil.GetJyDocsDB()
		docData := model.UserDoc{}
		//查询用户文档详情
		orm.Select("userId,docId").First(&docData, docId)
		log.Println(docData.UserId)
		log.Println(docData.DocId)

		//逻辑删除
		err := docRpcUtil.GetJyDocsDB().Exec("UPDATE user_doc SET isDelete = 1 WHERE id = ?", docId).Error
		if err != nil {
			tx.Rollback()
			return err
		}
		//删除记录
		err = docRpcUtil.GetJyDocsDB().Exec("insert into del_record (docId, userId, date, operate, viewType, source, shareUserId) values (?,?,?,?,?,?,?)", docId, docData.UserId, time.Now(), 1, 1, 1, 1).Error
		if err != nil {
			tx.Rollback()
			return err
		}
		return nil
	})
	if err != nil {
		return false
	}
	return true

}