Spring Boot 微信小程序 MyBatis 代码

package com.plant.api.controller; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.plant.api.bean.dto.CommentDto; import com.plant.api.bean.dto.PostDto; import com.plant.api.bean.general.result.Result; import com.plant.api.bean.general.result.ResultStatus; import com.plant.api.bean.threadLocal.UserLocal; import com.plant.api.bean.vto.PostVto; import com.plant.api.entity.Comment; import com.plant.api.entity.Post; import com.plant.api.service.CommentService; import com.plant.api.service.PostService; import com.plant.api.service.UserService; import com.plant.config.filter.login.LoginFilter; import com.plant.util.Rules; import javafx.geometry.Pos; import org.apache.commons.lang3.ArrayUtils; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.web.bind.annotation.PostMapping; import org.springframework.web.bind.annotation.RequestBody; import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.bind.annotation.RestController; import javax.annotation.Resource; import java.util.ArrayList; import java.util.Arrays; import java.util.Date; import java.util.List; import static org.apache.commons.lang3.StringUtils.join; /** * <p> * 发帖表 前端控制器 * </p> * * @author ${author} * @since 2021-02-21 */ @RestController @RequestMapping("/post") public class PostController { @Autowired private PostService postService; @Autowired private UserService userService; @Autowired private CommentService commentService; @Resource private Rules rules; /** * 发帖请求 */ @LoginFilter @PostMapping("/add") public Result add(@RequestBody PostVto postVto) { //发送前执行规则判断 if (!rules.setRulePost(postVto.getPostName(), postVto.getPostContent())) { return Result.failure(ResultStatus.RULES); } else { Post post = new Post(); if (postVto.getImgList() != null) { String photoString = ArrayUtils.toString(postVto.getImgList(), ","); post.setPostFile(photoString.replace("[", "").trim().replace("]", "")); } else { post.setPostFile(""); } post.setPostContent(postVto.getPostContent()); post.setPostName(postVto.getPostName()); post.setPostCreaterId(UserLocal.getUser().getUserId()); post.setCreateTime(new Date()); postService.save(post); return Result.success(); } } /** * 获取所有帖子 * 小程序接口 */ @PostMapping("/list") public List<PostDto> postDtoList(@RequestBody Post post) { List<PostDto> postDtoList = new ArrayList<>(); if (post.getPostContent() != null) { postService.list(new QueryWrapper<Post>().like("Post_content", post.getPostContent())).forEach(x -> { PostDto postDto = new PostDto(); postDto.setPostId(x.getPostId()); if (x.getPostFile() != null) { String[] photoArr = x.getPostFile().split(","); postDto.setPostPhotoList(Arrays.asList(photoArr)); } postDto.setPostContent(x.getPostContent()); postDto.setCreateTime(x.getCreateTime()); postDto.setPostName(x.getPostName()); postDto.setPostUserName(userService.getById(x.getPostCreaterId()).getUserName()); postDto.setCommentTotal(commentService.count(new QueryWrapper<Comment>().eq("Comment_post_id", x.getPostId()))); postDtoList.add(postDto); }); } else { postService.list().forEach(x -> { PostDto postDto = new PostDto(); postDto.setPostId(x.getPostId()); if (x.getPostFile() != null) { String[] photoArr = x.getPostFile().split(","); postDto.setPostPhotoList(Arrays.asList(photoArr)); } postDto.setPostContent(x.getPostContent()); postDto.setCreateTime(x.getCreateTime()); postDto.setPostName(x.getPostName()); postDto.setPostUserName(userService.getById(x.getPostCreaterId()).getUserName()); postDto.setCommentTotal(commentService.count(new QueryWrapper<Comment>().eq("Comment_post_id", x.getPostId()))); postDtoList.add(postDto); }); } return postDtoList; } /** * 查看帖子详情 */ @PostMapping("/detail") public PostDto postDtoDetail(@RequestBody Post postVto) { PostDto postDto = new PostDto(); Post post = postService.getById(postVto.getPostId()); postDto.setPostId(post.getPostId()); if (post.getPostFile() != null) { String[] photoArr = post.getPostFile().split(","); postDto.setPostPhotoList(Arrays.asList(photoArr)); } postDto.setPostContent(post.getPostContent()); postDto.setCreateTime(post.getCreateTime()); postDto.setPostName(post.getPostName()); postDto.setPostUserName(userService.getById(post.getPostCreaterId()).getUserName()); postDto.setCommentTotal(commentService.count(new QueryWrapper<Comment>().eq("Comment_post_id", post.getPostId()))); List<CommentDto> commentDtoList = new ArrayList<>(); commentService.list(new QueryWrapper<Comment>().eq("Comment_post_id", post.getPostId())).forEach(x -> { CommentDto commentDto = new CommentDto(); commentDto.setCommentId(x.getCommentId()); commentDto.setCommentContent(x.getCommentContent()); commentDto.setCommentUserName(userService.getById(x.getCommentUserId()).getUserName()); commentDtoList.add(commentDto); }); postDto.setComments(commentDtoList); return postDto; } }