|
- package friend_circle
-
- import (
- "applet/app/db"
- "applet/app/db/im/model"
- "applet/app/e"
- "applet/app/md/friend_circles"
- "applet/app/svc"
- svc2 "applet/app/svc/friend_circle"
- "applet/app/utils"
- "code.fnuoos.com/EggPlanet/egg_system_rules.git/md"
- svc3 "code.fnuoos.com/EggPlanet/egg_system_rules.git/svc"
- es2 "code.fnuoos.com/EggPlanet/egg_system_rules.git/utils/es"
- "code.fnuoos.com/go_rely_warehouse/zyos_go_es.git/es"
- "context"
- "encoding/json"
- "errors"
- "fmt"
- "github.com/gin-gonic/gin"
- "github.com/olivere/elastic/v7"
- "strconv"
- "strings"
- "time"
- )
-
- // PersonalSendingList
- // @Summary 朋友圈-个人发送朋友圈列表
- // @Tags 朋友圈
- // @Description 个人发送朋友圈列表
- // @Accept json
- // @Produce json
- // @param Authorization header string true "验证参数Bearer和token空格拼接"
- // @Param req body friend_circles.PersonalSendingListReq true "签名上传url"
- // @Success 200 {object} friend_circles.PersonalSendingListResp "返回数据"
- // @Failure 400 {object} md.Response "具体错误"
- // @Router /api/v1/circleFriends/personalSending [POST]
- func PersonalSendingList(c *gin.Context) {
- var req friend_circles.PersonalSendingListReq
- err := c.ShouldBindJSON(&req)
- if err != nil {
- err = svc.HandleValidateErr(err)
- err1 := err.(e.E)
- e.OutErr(c, err1.Code, err1.Error())
- return
- }
-
- resp, err := svc2.GetPersonalSendingList(c, req)
- if err != nil {
- e.OutErr(c, e.ERR, err.Error())
- }
- resp.Page = req.Page
- resp.PageSize = req.PageSize
- e.OutSuc(c, resp, nil)
- }
-
- // MySelfList
- // @Summary 朋友圈-我的朋友圈列表
- // @Tags 朋友圈
- // @Description 我的朋友圈列表
- // @Accept json
- // @Produce json
- // @param Authorization header string true "验证参数Bearer和token空格拼接"
- // @Param req body friend_circles.MySelfListReq true "签名上传url"
- // @Success 200 {object} friend_circles.MySelfListResp "返回数据"
- // @Failure 400 {object} md.Response "具体错误"
- // @Router /api/v1/circleFriends/mySelfList [POST]
- func MySelfList(c *gin.Context) {
- var args friend_circles.MySelfListReq
- err := c.ShouldBindJSON(&args)
- if err != nil {
- err = svc.HandleValidateErr(err)
- err1 := err.(e.E)
- e.OutErr(c, err1.Code, err1.Error())
- return
- }
- resp, err := svc2.MySelfList(c, args)
- if err != nil {
- e.OutErr(c, e.ERR, err.Error())
- return
- }
- imUserIds := make([]int64, len(resp.List))
- for i, friendCircle := range resp.List {
- imUserIds[i] = friendCircle.ImUid
- }
-
- nicknameAndAvatarMap, err := svc2.GetImUserNicknameAndAvatar(imUserIds)
- if err != nil {
- e.OutErr(c, e.ERR, err.Error())
- return
- }
- for _, friendCircle := range resp.List {
- val, ok := nicknameAndAvatarMap[friendCircle.ImUid]
- if ok {
- friendCircle.ImNickname = val.NickName
- friendCircle.ImAvatar = val.Avatar
- }
- }
-
- resp.Page = args.Page
- resp.PageSize = args.PageSize
- e.OutSuc(c, resp, nil)
- }
-
- // IsCanPublish
- // @Summary 朋友圈-是否可以发布朋友圈
- // @Tags 朋友圈
- // @Description 是否可以发布朋友圈
- // @Accept json
- // @Produce json
- // @param Authorization header string true "验证参数Bearer和token空格拼接"
- // @Success 200 {string} "success"
- // @Failure 400 {object} md.Response "具体错误"
- // @Router /api/v1/circleFriends/isCanPublish [GET]
- func IsCanPublish(c *gin.Context) {
- isCan, err := svc2.IsCanPublish(c)
- if err != nil {
- fmt.Println("IsPublish:::::", err.Error())
- }
- resp := friend_circles.IsCanCommentResp{IsCan: isCan}
- e.OutSuc(c, resp, nil)
- }
-
- // FriendCircleLikeList
- // @Summary 朋友圈-朋友圈点赞列表
- // @Tags 朋友圈
- // @Description 朋友圈点赞列表
- // @Accept json
- // @Produce json
- // @param Authorization header string true "验证参数Bearer和token空格拼接"
- // @Param req body friend_circles.FriendCircleLikeListReq true "签名上传url"
- // @Success 200 {object} friend_circles.FriendCircleLikeListResp "返回数据"
- // @Failure 400 {object} md.Response "具体错误"
- // @Router /api/v1/circleFriends/likeList [POST]
- func FriendCircleLikeList(c *gin.Context) {
- var req friend_circles.FriendCircleLikeListReq
- if err1 := c.ShouldBindJSON(&req); err1 != nil {
- e.OutErr(c, e.ERR_INVALID_ARGS, err1.Error())
- return
- }
- //查找朋友圈记录
- doc, err := es.FirstDoc(md.EggFriendCircleEsIndex, req.CircleIndexId)
- if err != nil {
- e.OutErr(c, e.ERR_DB_ORM, err.Error())
- return
- }
- if !doc.Found { // 表示没找到数据
- e.OutErr(c, e.ERR_NOT_FAN, "朋友圈文档记录不存在")
- return
- }
-
- from := (req.Page - 1) * req.PageSize
-
- // 查找点赞列表
- query := elastic.NewBoolQuery()
- query.Must(elastic.NewTermQuery("circle_id", req.CircleIndexId))
-
- searchResult, err := es.EsClient.Search().
- Index(md.EggFriendCircleLikeEsAlias). // 替换为你的索引名称
- Query(query).
- From(from).
- Size(req.PageSize).
- Sort("created_at", false). // 按时间倒排
- Pretty(true).
- Do(context.Background())
- if err != nil {
- return
- }
-
- var docs []*md.EggFriendCircleLikeEs
- var docUserIds []int64
- var total int64
- // 检查是否有结果
- if searchResult.Hits.TotalHits.Value != 0 {
- // 解析结果
- total = searchResult.Hits.TotalHits.Value
- for _, hit := range searchResult.Hits.Hits {
- var like md.EggFriendCircleLikeEs
- err = json.Unmarshal(hit.Source, &like)
- if err != nil {
- return
- }
- docUserIds = append(docUserIds, like.ImUid)
- docs = append(docs, &like)
- }
- }
-
- nicknameAndAvatarMap, err := svc2.GetImUserNicknameAndAvatar(docUserIds)
- if err != nil {
- e.OutErr(c, e.ERR_DB_ORM, err.Error())
- return
- }
- var list []*friend_circles.FriendCircleLikeEsStruct
- for _, like := range docs {
- temp := friend_circles.FriendCircleLikeEsStruct{
- Uid: like.Uid,
- ImUid: like.ImUid,
- CircleId: like.CircleId,
- }
- val, ok := nicknameAndAvatarMap[like.ImUid]
- if ok {
- temp.ImNickname = val.NickName
- temp.ImAvatar = val.Avatar
- }
- list = append(list, &temp)
- }
- resp := friend_circles.FriendCircleLikeListResp{
- List: list,
- Page: req.Page,
- PageSize: req.PageSize,
- Total: total,
- }
- e.OutSuc(c, resp, nil)
- }
-
- // Publish
- // @Summary 朋友圈-发布朋友圈
- // @Tags 朋友圈
- // @Description 发布朋友圈
- // @Accept json
- // @Produce json
- // @param Authorization header string true "验证参数Bearer和token空格拼接"
- // @Param req body friend_circles.PublishReq true "请求参数"
- // @Success 200 {string} "success"
- // @Failure 400 {object} md.Response "具体错误"
- // @Router /api/v1/circleFriends/publish [POST]
- func Publish(c *gin.Context) {
- var req friend_circles.PublishReq
- if err1 := c.ShouldBindJSON(&req); err1 != nil {
- e.OutErr(c, e.ERR_INVALID_ARGS, err1.Error())
- return
- }
- isCan, err := svc2.IsCanPublish(c)
- if err != nil {
- e.OutErr(c, e.ERR, err.Error())
- return
- }
- if !isCan {
- e.OutErr(c, e.ERR, "当前不允许发朋友圈")
- return
- }
-
- user := svc.GetUser(c)
- var imUser model.User
- _, err = db.DbIm.Where("phone_number = ?", user.Phone).Get(&imUser)
- if err != nil {
- e.OutErr(c, e.ERR_DB_ORM, err.Error())
- return
- }
- scheme, domain := svc.ImageBucket(db.Db)
- var imageStr string
- if len(req.ImageList) > 0 {
- imageStr = utils.SerializeStr(req.ImageList)
- imageStr = strings.ReplaceAll(imageStr, fmt.Sprintf("%s://%s/", scheme, domain), "{{tempHost}}")
- }
-
- //插入es记录
- now := time.Now()
- createDocRet, err := es.CreateDoc(md.EggFriendCircleEsIndex, strconv.FormatInt(user.Id, 10)+"_"+utils.Int64ToStr(now.Unix()), md.EggFriendCircleEs{
- Uid: user.Id,
- ImUid: imUser.Id,
- Kind: 1,
- Content: req.Content,
- Image: imageStr,
- Video: req.Video,
- LikesNums: 0,
- ShareNums: 0,
- CommentNums: 0,
- State: 1,
- IsTopUp: 2,
- IsPraise: 2,
- CreatedAt: now.Format("2006-01-02 15:04:05"),
- UpdatedAt: now.Format("2006-01-02 15:04:05"),
- })
-
- fmt.Printf("CreateDoc ==> %+v \n\n", createDocRet)
- // 更新发朋友圈次数
- year, week := time.Now().ISOWeek()
- yearStr := utils.IntToStr(year)
- weekStr := utils.IntToStr(week)
- index := es2.GetAppointIndexFromAlias(yearStr, weekStr)
- id := fmt.Sprintf("%d%d_%d", year, week, user.Id)
- _, err = es.FirstDoc(index, id)
- if err != nil {
- if err.Error() != "elastic: Error 404 (Not Found)" {
- e.OutErr(c, e.ERR, err.Error())
- return
- } else {
- m := md.EggEnergyUserEggScoreEs{
- Uid: user.Id,
- SendCircleOfFriendNums: 1,
- }
- newCreateDoc, err := es.CreateDoc(index, id, &m)
- if err != nil {
- e.OutErr(c, e.ERR_DB_ORM, err.Error())
- return
- }
- fmt.Printf("newCreateDoc ==> %+v \n\n", newCreateDoc)
- }
- } else {
- script := elastic.NewScript("ctx._source.send_circle_of_friend_nums += params.inc").Param("inc", 1)
- _, err = es.EsClient.Update().
- Index(index).
- Id(id).
- Script(script).
- Do(context.Background())
- if err != nil {
- e.OutErr(c, e.ERR, err.Error())
- return
- }
- }
- e.OutSuc(c, "success", nil)
- }
-
- // RecommendList
- // @Summary 朋友圈-推荐列表
- // @Tags 朋友圈
- // @Description 推荐列表
- // @Accept json
- // @Produce json
- // @param Authorization header string true "验证参数Bearer和token空格拼接"
- // @Param req body friend_circles.RecommendListReq true "签名上传url"
- // @Success 200 {object} friend_circles.RecommendListResp "返回数据"
- // @Failure 400 {object} md.Response "具体错误"
- // @Router /api/v1/circleFriends/recommendList [POST]
- func RecommendList(c *gin.Context) {
- var req friend_circles.RecommendListReq
- err := c.ShouldBindJSON(&req)
- if err != nil {
- err = svc.HandleValidateErr(err)
- err1 := err.(e.E)
- e.OutErr(c, err1.Code, err1.Error())
- return
- }
- resp, err := svc2.GetRecommendList(req)
- if err != nil {
- e.OutErr(c, e.ERR, err.Error())
- return
- }
-
- imUserIds := make([]int64, len(resp.List))
- for i, friendCircle := range resp.List {
- imUserIds[i] = friendCircle.ImUid
- }
-
- nicknameAndAvatarMap, err := svc2.GetImUserNicknameAndAvatar(imUserIds)
- if err != nil {
- e.OutErr(c, e.ERR, err.Error())
- return
- }
- for _, friendCircle := range resp.List {
- val, ok := nicknameAndAvatarMap[friendCircle.ImUid]
- if ok {
- friendCircle.ImNickname = val.NickName
- friendCircle.ImAvatar = val.Avatar
- }
- }
- e.OutSuc(c, resp, nil)
- }
-
- // CommentList
- // @Summary 朋友圈-评论列表
- // @Tags 朋友圈
- // @Description 评论列表
- // @Accept json
- // @Produce json
- // @param Authorization header string true "验证参数Bearer和token空格拼接"
- // @Param circle_index_id query string true "朋友圈文档记录"
- // @Param req body friend_circles.CommentListReq true "请求参数"
- // @Success 200 {string} "success"
- // @Failure 400 {object} md.Response "具体错误"
- // @Router /api/v1/circleFriends/commentList [POST]
- func CommentList(c *gin.Context) {
- var req friend_circles.CommentListReq
- if err1 := c.ShouldBindJSON(&req); err1 != nil {
- e.OutErr(c, e.ERR_INVALID_ARGS, err1.Error())
- return
- }
- //查找朋友圈记录
- doc, err := es.FirstDoc(md.EggFriendCircleEsIndex, req.CircleIndexId)
- if err != nil {
- e.OutErr(c, e.ERR_DB_ORM, err.Error())
- return
- }
- if !doc.Found { // 表示没找到数据
- e.OutErr(c, e.ERR_NOT_FAN, "朋友圈文档记录不存在")
- return
- }
-
- resp, err := svc2.CommentList(req)
- if err != nil {
- e.OutErr(c, e.ERR, err.Error())
- return
- }
- e.OutSuc(c, resp, nil)
- }
-
- // CommentDetail
- // @Summary 朋友圈-评论详情
- // @Tags 朋友圈
- // @Description 评论详情
- // @Accept json
- // @Produce json
- // @param Authorization header string true "验证参数Bearer和token空格拼接"
- // @Param comment_index_id query string true "评论文档记录"
- // @Success 200 {string} "success"
- // @Failure 400 {object} md.Response "具体错误"
- // @Router /api/v1/circleFriends/commentDetail [GET]
- func CommentDetail(c *gin.Context) {
- var req friend_circles.CommentDetailReq
- if err1 := c.ShouldBindJSON(&req); err1 != nil {
- e.OutErr(c, e.ERR_INVALID_ARGS, err1.Error())
- return
- }
-
- //查找评论记录
- boolQuery := elastic.NewBoolQuery()
- boolQuery.Must(elastic.NewTermQuery("_id", req.CommentIndexId))
- searchResult, err := es.EsClient.Search().
- Index(md.EggFriendCircleCommentEsAlias).
- Query(boolQuery).
- Pretty(true).
- Do(context.Background())
- if err != nil {
- e.OutErr(c, e.ERR, err.Error())
- return
- }
- var doc md.EggFriendCircleCommentEs
- // 检查是否有结果
- if searchResult.Hits.TotalHits.Value != 0 {
- err = json.Unmarshal(searchResult.Hits.Hits[0].Source, &doc)
- if err != nil {
- e.OutErr(c, e.ERR, err.Error())
- return
- }
- } else {
- e.OutErr(c, e.ERR_NOT_FAN, "评论文档记录不存在")
- return
- }
-
- resp, err := svc2.CommentDetail(req)
- if err != nil {
- e.OutErr(c, e.ERR, err.Error())
- return
- }
- resp.CircleIndexId = doc.CircleId
- e.OutSuc(c, resp, nil)
- }
-
- // Delete
- // @Summary 朋友圈-删除朋友圈
- // @Tags 朋友圈
- // @Description 删除朋友圈
- // @Accept json
- // @Produce json
- // @param Authorization header string true "验证参数Bearer和token空格拼接"
- // @Param req body friend_circles.EggFriendCircleDelReq true "请求参数"
- // @Success 200 {string} "success"
- // @Failure 400 {object} md.Response "具体错误"
- // @Router /api/v1/circleFriends/delete [DELETE]
- func Delete(c *gin.Context) {
- var req friend_circles.EggFriendCircleDelReq
- err := c.ShouldBindJSON(&req)
- if err != nil {
- err = svc.HandleValidateErr(err)
- err1 := err.(e.E)
- e.OutErr(c, err1.Code, err1.Error())
- return
- }
-
- user := svc.GetUser(c)
- aliasName := md.EggFriendCircleEsAlias
- // 1.判断是否是自己文章
- doc, err := es.FirstDoc(md.EggFriendCircleEsIndex, req.CircleIndexId)
- if err != nil {
- e.OutErr(c, e.ERR_DB_ORM, err.Error())
- return
- }
- if !doc.Found { // 表示没找到数据
- e.OutErr(c, e.ERR_NOT_FAN, "朋友圈文档记录不存在")
- return
- }
- var docInfo md.EggFriendCircleEs
- utils.Unserialize(doc.Source, &docInfo)
- if docInfo.Uid != user.Id {
- e.OutErr(c, e.ERR, errors.New("不允许删除别人的朋友圈"))
- return
- }
-
- // 2.删除文章
- deleteDocRet, err := es.DeleteDoc(aliasName, req.CircleIndexId)
- if err != nil {
- e.OutErr(c, e.ERR, err.Error())
- return
- }
- fmt.Printf("CreateDoc ==> %+v \n\n", deleteDocRet)
-
- // 3.将评论id推入mq
- err = svc2.DelFriendCircleComment(req.CircleIndexId)
- if err != nil {
- e.OutErr(c, e.ERR, err.Error())
- return
- }
- e.OutSuc(c, "success", nil)
- }
-
- // Share
- // @Summary 朋友圈-分享后朋友圈分享数(增加)
- // @Tags 朋友圈
- // @Description 分享后朋友圈分享数(增加)
- // @Accept json
- // @Produce json
- // @param Authorization header string true "验证参数Bearer和token空格拼接"
- // @Param req body friend_circles.EggFriendCircleShareReq true "请求参数"
- // @Success 200 {string} "success"
- // @Failure 400 {object} md.Response "具体错误"
- // @Router /api/v1/circleFriends/share [POST]
- func Share(c *gin.Context) {
- var req friend_circles.EggFriendCircleShareReq
- err := c.ShouldBindJSON(&req)
- if err != nil {
- err = svc.HandleValidateErr(err)
- err1 := err.(e.E)
- e.OutErr(c, err1.Code, err1.Error())
- return
- }
-
- script := elastic.NewScript("ctx._source.share_nums += params.inc").Param("inc", 1)
- aliasName := md.EggFriendCircleEsAlias
- _, err = es.EsClient.Update().
- Index(aliasName).
- Id(req.CircleIndexId).
- Script(script).
- Do(context.Background())
- if err != nil {
- e.OutErr(c, e.ERR, err.Error())
- return
- }
- e.OutSuc(c, "success", nil)
- }
-
- // Like
- // @Summary 朋友圈-点赞朋友圈
- // @Tags 朋友圈
- // @Description 点赞朋友圈
- // @Accept json
- // @Produce json
- // @param Authorization header string true "验证参数Bearer和token空格拼接"
- // @Param req body friend_circles.EggFriendCircleLikeReq true "请求参数"
- // @Success 200 {string} "success"
- // @Failure 400 {object} md.Response "具体错误"
- // @Router /api/v1/circleFriends/like [POST]
- func Like(c *gin.Context) {
- var req friend_circles.EggFriendCircleLikeReq
- err := c.ShouldBindJSON(&req)
- if err != nil {
- err = svc.HandleValidateErr(err)
- err1 := err.(e.E)
- e.OutErr(c, err1.Code, err1.Error())
- return
- }
- user := svc.GetUser(c)
-
- likeEsIndex := svc3.GetEggFriendCircleLikeEsIndex(user.Id)
- likeId := svc3.GetEggFriendCircleLikeEsIndexId(user.Id, req.CircleIndexId)
- // 判断是否已经点赞过
- query := elastic.NewBoolQuery().
- Must(
- elastic.NewTermQuery("_id", likeId),
- )
- searchResult, err := es.EsClient.Search().Index(likeEsIndex).
- Query(query).
- Size(0). // 只关心总数,不实际返回文档
- Do(context.Background())
- if err != nil {
- return
- }
- total := searchResult.TotalHits()
- if int(total) > 0 {
- e.OutSuc(c, "success", nil)
- return
- }
-
- var imUser model.User
- _, err = db.DbIm.Where("phone_number = ?", user.Phone).Get(&imUser)
- if err != nil {
- e.OutErr(c, e.ERR_DB_ORM, err.Error())
- return
- }
-
- // 新增点赞信息
- now := time.Now()
- m := md.EggFriendCircleLikeEs{
- Uid: user.Id,
- ImUid: imUser.Id,
- CircleId: req.CircleIndexId,
- CreatedAt: now.Format("2006-01-02 15:04:05"),
- UpdatedAt: now.Format("2006-01-02 15:04:05"),
- }
- createDocRet, err := es.CreateDoc(likeEsIndex, likeId, &m)
- if err != nil {
- e.OutErr(c, e.ERR, err.Error())
- return
- }
- fmt.Printf("CreateDoc ==> %+v \n\n", createDocRet)
-
- // 增加文章点赞数
- script := elastic.NewScript("ctx._source.likes_nums += params.inc").Param("inc", 1)
- aliasName := md.EggFriendCircleEsAlias
- _, err = es.EsClient.Update().
- Index(aliasName).
- Id(req.CircleIndexId).
- Script(script).
- Do(context.Background())
- if err != nil {
- e.OutErr(c, e.ERR, err.Error())
- return
- }
- e.OutSuc(c, "success", nil)
- }
-
- // CancelLike
- // @Summary 朋友圈-取消点赞朋友圈
- // @Tags 朋友圈
- // @Description 取消点赞朋友圈
- // @Accept json
- // @Produce json
- // @param Authorization header string true "验证参数Bearer和token空格拼接"
- // @Param req body friend_circles.EggFriendCircleCancelLikeReq true "请求参数"
- // @Success 200 {string} "success"
- // @Failure 400 {object} md.Response "具体错误"
- // @Router /api/v1/circleFriends/cancelLike [POST]
- func CancelLike(c *gin.Context) {
- // 减少文章点赞数
- var req friend_circles.EggFriendCircleCancelLikeReq
- err := c.ShouldBindJSON(&req)
- if err != nil {
- err = svc.HandleValidateErr(err)
- err1 := err.(e.E)
- e.OutErr(c, err1.Code, err1.Error())
- return
- }
- user := svc.GetUser(c)
-
- // 删除点赞信息
- likeEsIndex := svc3.GetEggFriendCircleLikeEsIndex(user.Id)
- likeId := svc3.GetEggFriendCircleLikeEsIndexId(user.Id, req.CircleIndexId)
- deleteDocRet, err := es.DeleteDoc(likeEsIndex, likeId)
- if err != nil {
- if err.Error() == "elastic: Error 404 (Not Found)" {
- e.OutErr(c, e.ERR_BAD_REQUEST, errors.New("点赞已取消,请勿重复请求").Error())
- return
- }
- e.OutErr(c, e.ERR, err.Error())
- return
- }
- fmt.Printf("DeleteDoc ==> %+v \n\n", deleteDocRet)
-
- // 减少文章点赞数
- script := elastic.NewScript("ctx._source.likes_nums -= params.inc").Param("inc", 1)
- aliasName := md.EggFriendCircleEsAlias
- _, err = es.EsClient.Update().
- Index(aliasName).
- Id(req.CircleIndexId).
- Script(script).
- Do(context.Background())
- if err != nil {
- e.OutErr(c, e.ERR, err.Error())
- return
- }
- e.OutSuc(c, "success", nil)
- }
|