package userlib import ( "app.yhyue.com/moapp/jyPoints/rpc/integralclient" "app.yhyue.com/moapp/jydocs/services/model" docRpcUtil "app.yhyue.com/moapp/jydocs/services/util" "context" "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) 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 ctx, _ := context.WithTimeout(context.Background(), 5*time.Second) FileSystem :=integralclient.NewIntegral(zrpc.MustNewClient(c.FileSystemConf)) req := &integral.Req{UserId: "111111",PointType:2003,BusinessTypeId:1,BusinessType:"1", Point: 110,EndDate:"2021-03-11",AppId: 10000} res, err := FileSystem.IntegralConsume(ctx, req) log.Println("err ", err) log.Println("req ", res) //查询之前有无收藏记录 有收藏记录类型改为下载,状态改为未删除、有收藏记录插入一条新纪录 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 nil } }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 nil } } return nil }) if err != nil { return false } return true }