package postservicelogic import ( "app-cloudep-tweeting-service/internal/domain" model "app-cloudep-tweeting-service/internal/model/mongo" "context" ers "code.30cm.net/digimon/library-go/errs" "go.mongodb.org/mongo-driver/bson/primitive" "app-cloudep-tweeting-service/gen_result/pb/tweeting" "app-cloudep-tweeting-service/internal/svc" "github.com/zeromicro/go-zero/core/logx" ) type UpdatePostLogic struct { ctx context.Context svcCtx *svc.ServiceContext logx.Logger } func NewUpdatePostLogic(ctx context.Context, svcCtx *svc.ServiceContext) *UpdatePostLogic { return &UpdatePostLogic{ ctx: ctx, svcCtx: svcCtx, Logger: logx.WithContext(ctx), } } type checkPostID struct { PostID string `validate:"required"` Content string `json:"content,omitempty" validate:"lte=500"` } // UpdatePost 更新貼文 func (l *UpdatePostLogic) UpdatePost(in *tweeting.UpdatePostReq) (*tweeting.OKResp, error) { // 驗證資料 if err := l.svcCtx.Validate.ValidateAll(&checkPostID{ PostID: in.GetPostId(), Content: in.GetContent(), }); err != nil { // 錯誤代碼 05-011-00 return nil, ers.InvalidFormat(err.Error()) } // 沒有就沒有,有就走全覆蓋 update := model.Post{} oid, err := primitive.ObjectIDFromHex(in.GetPostId()) if err != nil { // 錯誤代碼 05-011-00 return nil, ers.InvalidFormat("failed to get correct post id") } update.ID = oid update.Tags = in.GetTags() // 將 Media 存入 media := make([]model.Media, 0, len(in.GetMedia())) for _, item := range in.GetMedia() { media = append(media, model.Media{ Links: item.Url, Type: item.Type, }) } update.MediaURL = media update.Content = in.GetContent() // 因為 0 也有意義,所以如果是真的沒帶進來,用 -1 帶進去表示不作動 if in.LikeCount == nil { update.Like = -1 } else { update.Like = in.GetLikeCount() } if in.DislikeCount == nil { update.DisLike = -1 } else { update.DisLike = in.GetDislikeCount() } _, err = l.svcCtx.PostModel.UpdateOptional(l.ctx, &update) if err != nil { // 錯誤代碼 05-021-04 e := domain.CommentErrorL( domain.UpdatePostError, logx.WithContext(l.ctx), []logx.LogField{ {Key: "req", Value: in}, {Key: "func", Value: "PostModel.UpdateOptional"}, {Key: "err", Value: err}, }, "failed to update post", in.PostId).Wrap(err) return nil, e } return &tweeting.OKResp{}, nil }