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 }