评论功能已经成为APP和网站开发中的必备功能。本文采用springboot+mybatis-plus框架,通过代码主要介绍评论功能的数据库设计和接口数据返回。我们返回的格式可以分三种方案,第一种方案是先返回评论,再根据评论id返回回复信息,第二种方案是将评论回复直接封装成一个类似于树的数据结构进行返回(如果数据对的话,可以根据评论分页),第三种方案是将所有数据用递归的SQL查出来,再把数据解析成树,返回结果
表结构:
CREATE TABLE `comment` (
`id` bigint(18) NOT NULL AUTO_INCREMENT,
`parent_id` bigint(18) NOT NULL DEFAULT '0',
`content` text NOT NULL COMMENT '内容',
`author` varchar(20) NOT NULL COMMENT '评论人',
`create_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '评论时间',
PRIMARY KEY (`id`)
) ENGINE=InnoDB AUTO_INCREMENT=6 DEFAULT CHARSET=utf8mb4;
数据添加:
INSERT INTO `comment` (`id`, `parent_id`, `content`, `author`, `create_time`) VALUES (1, 0, '这是评论1', '吴名氏', '2023-02-20 17:11:16');
INSERT INTO `comment` (`id`, `parent_id`, `content`, `author`, `create_time`) VALUES (2, 1, '我回复了第一条评论', '吴名氏', '2023-02-20 17:12:00');
INSERT INTO `comment` (`id`, `parent_id`, `content`, `author`, `create_time`) VALUES (3, 2, '我回复了第一条评论的第一条回复', '吴名氏', '2023-02-20 17:12:13');
INSERT INTO `comment` (`id`, `parent_id`, `content`, `author`, `create_time`) VALUES (4, 2, '我回复了第一条评论的第二条回复', '吴名氏', '2023-02-21 09:23:14');
INSERT INTO `comment` (`id`, `parent_id`, `content`, `author`, `create_time`) VALUES (5, 0, '这是评论2', '吴名氏', '2023-02-21 09:41:02');
INSERT INTO `comment` (`id`, `parent_id`, `content`, `author`, `create_time`) VALUES (6, 3, '我回复了第一条评论的第一条回复的第一条回复', '吴名氏', '2023-02-21 09:56:27');
添加后的数据:
方案一先返回评论列表,再根据评论id返回回复列表,以此循环,具体代码下文进行展示
/**
* 方案一
* @author wuKeFan
* @date 2023-02-20 16:58:08
*/
@Slf4j
@RestController
@RequestMapping("/one/comment")
public class CommentOneController {
@Resource
private CommentService commentService;
@GetMapping("/")
public List getList() {
return commentService.getList();
}
@GetMapping("/{id}")
public Comment getCommentById(@PathVariable Long id) {
return commentService.getById(id);
}
@GetMapping("/parent/{parentId}")
public List getCommentByParentId(@PathVariable Long parentId) {
return commentService.getCommentByParentId(parentId);
}
@PostMapping("/")
public void addComment(@RequestBody Comment comment) {
commentService.addComment(comment);
}
}
/**
* service类
* @author wuKeFan
* @date 2023-02-20 16:55:23
*/
public interface CommentService {
List getCommentByParentId(Long parentId);
void addComment(Comment comment);
Comment getById(Long id);
List getList();
}
/**
* @author wuKeFan
* @date 2023-02-20 16:56:00
*/
@Service
public class CommentServiceImpl implements CommentService{
@Resource
private CommentMapper baseMapper;
@Override
public List getCommentByParentId(Long parentId) {
QueryWrapper queryWrapper = new QueryWrapper<>();
queryWrapper.eq("parent_id", parentId);
return baseMapper.selectList(queryWrapper);
}
@Override
public void addComment(Comment comment) {
baseMapper.insert(comment);
}
@Override
public Comment getById(Long id) {
return baseMapper.selectById(id);
}
@Override
public List getList() {
return baseMapper.selectList(new QueryWrapper().lambda().eq(Comment::getParentId, 0));
}
}
/**
* mapper类
* @author wuKeFan
* @date 2023-02-20 16:53:59
*/
@Repository
public interface CommentMapper extends BaseMapper {
}
/**
* 评论表实体类
* @author wuKeFan
* @date 2023-02-20 16:53:24
*/
@Data
@TableName("comment")
public class Comment {
@TableId(type = IdType.AUTO)
private Long id;
private Long parentId;
private String content;
private String author;
@JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss", timezone = "GMT+8")
private Date createTime;
}
方案二采用的是将数据装到一个类似树的数据结构,然后返回,数据如果多的话,可以根据评论列表进行分页
/**
* @author wuKeFan
* @date 2023-02-20 17:30:45
*/
@Slf4j
@RestController
@RequestMapping("/two/comment")
public class CommentTwoController {
@Resource
private CommentService commentService;
@GetMapping("/")
public List getAllComments() {
return commentService.getAllComments();
}
@PostMapping("/")
public void addComment(@RequestBody Comment comment) {
commentService.addComment(comment);
}
}
/**
* service类
* @author wuKeFan
* @date 2023-02-20 16:55:23
*/
public interface CommentService {
void addComment(Comment comment);
void setChildren(CommentDTO commentDTO);
List getAllComments();
}
/**
* @author wuKeFan
* @date 2023-02-20 16:56:00
*/
@Service
public class CommentServiceImpl implements CommentService{
@Resource
private CommentMapper baseMapper;
@Override
public void addComment(Comment comment) {
baseMapper.insert(comment);
}
@Override
public List getAllComments() {
List rootComments = baseMapper.findByParentId(0L);
rootComments.forEach(this::setChildren);
return rootComments;
}
/**
* 递归获取
* @param commentDTO 参数
*/
@Override
public void setChildren(CommentDTO commentDTO){
List children = baseMapper.findByParentId(commentDTO.getId());
if (!children.isEmpty()) {
commentDTO.setChildren(children);
children.forEach(this::setChildren);
}
}
}
/**
* mapper类
* @author wuKeFan
* @date 2023-02-20 16:53:59
*/
@Repository
public interface CommentMapper extends BaseMapper {
@Select("SELECT id, parent_id as parentId, content, author, create_time as createTime FROM comment WHERE parent_id = #{parentId}")
List findByParentId(Long parentId);
}
/**
* 递归方式实体类
* @author wuKeFan
* @date 2023-02-20 17:26:48
*/
@Data
public class CommentDTO {
private Long id;
private Long parentId;
private String content;
private String author;
private List children;
@JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss", timezone = "GMT+8")
private Date createTime;
}
返回的json格式如图:
[
{
"id": 1,
"parentId": 0,
"content": "这是评论1",
"author": "吴名氏",
"children": [
{
"id": 2,
"parentId": 1,
"content": "我回复了第一条评论",
"author": "吴名氏",
"children": [
{
"id": 3,
"parentId": 2,
"content": "我回复了第一条评论的第一条回复",
"author": "吴名氏",
"children": [
{
"id": 6,
"parentId": 3,
"content": "我回复了第一条评论的第一条回复的第一条回复",
"author": "吴名氏",
"children": null,
"createTime": "2023-02-21 09:56:27"
}
],
"createTime": "2023-02-20 17:12:13"
},
{
"id": 4,
"parentId": 2,
"content": "我回复了第一条评论的第二条回复",
"author": "吴名氏",
"children": null,
"createTime": "2023-02-21 09:23:14"
}
],
"createTime": "2023-02-20 17:12:00"
}
],
"createTime": "2023-02-20 17:11:16"
},
{
"id": 5,
"parentId": 0,
"content": "这是评论2",
"author": "吴名氏",
"children": null,
"createTime": "2023-02-21 09:41:02"
}
]
方案三是将所有数据用递归的SQL查出来,再把数据解析成树,返回结果,适合数据较少的情况下,且MySQL版本需要在8.0以上
/**
* @author wuKeFan
* @date 2023-02-20 17:30:45
*/
@Slf4j
@RestController
@RequestMapping("/three/comment")
public class CommentThreeController {
@Resource
private CommentService commentService;
@GetMapping("/")
public List getAllCommentsBySql() {
return commentService.getAllCommentsBySql();
}
@PostMapping("/")
public void addComment(@RequestBody Comment comment) {
commentService.addComment(comment);
}
}
/**
* service类
* @author wuKeFan
* @date 2023-02-20 16:55:23
*/
public interface CommentService {
List getAllCommentsBySql();
}
/**
* @author wuKeFan
* @date 2023-02-20 16:56:00
*/
@Service
public class CommentServiceImpl implements CommentService{
@Resource
private CommentMapper baseMapper;
/**
* 递归获取
* @param commentDTO 参数
*/
@Override
public void setChildren(CommentDTO commentDTO){
List children = baseMapper.findByParentId(commentDTO.getId());
if (!children.isEmpty()) {
commentDTO.setChildren(children);
children.forEach(this::setChildren);
}
}
}
/**
* mapper类
* @author wuKeFan
* @date 2023-02-20 16:53:59
*/
@Repository
public interface CommentMapper extends BaseMapper {
@DS("localhost80")
List getAllCommentsBySql();
}
/**
* 评论表实体类
* @author wuKeFan
* @date 2023-02-20 16:53:24
*/
@Data
@TableName("comment")
public class Comment {
@TableId(type = IdType.AUTO)
private Long id;
private Long parentId;
private String content;
private String author;
@JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss", timezone = "GMT+8")
private Date createTime;
private Integer level;
}
返回的json格式如图:
[
{
"id": 1,
"parentId": 0,
"content": "这是评论1",
"author": "吴名氏",
"children": [
{
"id": 2,
"parentId": 1,
"content": "我回复了第一条评论",
"author": "吴名氏",
"children": [
{
"id": 3,
"parentId": 2,
"content": "我回复了第一条评论的第一条回复",
"author": "吴名氏",
"children": [
{
"id": 6,
"parentId": 3,
"content": "我回复了第一条评论的第一条回复的第一条回复",
"author": "吴名氏",
"children": [],
"createTime": "2023-02-21 09:56:27",
"level": 4
}
],
"createTime": "2023-02-20 17:12:13",
"level": 3
},
{
"id": 4,
"parentId": 2,
"content": "我回复了第一条评论的第二条回复",
"author": "吴名氏",
"children": [],
"createTime": "2023-02-21 09:23:14",
"level": 3
}
],
"createTime": "2023-02-20 17:12:00",
"level": 2
}
],
"createTime": "2023-02-20 17:11:16",
"level": 1
},
{
"id": 5,
"parentId": 0,
"content": "这是评论2",
"author": "吴名氏",
"children": [],
"createTime": "2023-02-21 09:41:02",
"level": 1
}
]
以上三种方案各有优缺点,需要从不同场景中使用不同的方案