all-docs/all-docs-api/src/main/java/com/jiaruiblog/api/controller/CollectController.java at 77875a8efcc72d9fc33965091e3820f1e7e23c32 · Jarrettluo/all-docs · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
package com.jiaruiblog.api.controller;

import com.jiaruiblog.common.ApiResult;
import com.jiaruiblog.domain.entity.po.CollectDocRelationship;
import com.jiaruiblog.domain.entity.dto.CollectDTO;
import com.jiaruiblog.common.exception.BusinessException;
import com.jiaruiblog.common.exception.ErrorCode;
import com.jiaruiblog.application.service.CollectService;
import com.jiaruiblog.application.service.DocumentService;
import com.jiaruiblog.application.service.IUserService;
import io.swagger.v3.oas.annotations.tags.Tag;
import jakarta.annotation.Resource;
import jakarta.servlet.http.HttpServletRequest;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.annotation.*;

import java.util.Date;

/**
 * @author luojiarui
 **/
@Tag(name = "用户收藏模块")
@RestController
@Slf4j
@RequestMapping("/api/v1/collect")
public class CollectController {

    @Resource
    private CollectService collectService;

    @Resource
    private IUserService userService;

    @Resource
    private DocumentService fileService;

    /**
     * @param collect 收藏数据传输对象
     * @param request HTTP请求对象
     * @return ApiResult<Void> 操作结果
     * @author luojiarui
     **/
    @PostMapping(value = "/auth/insert")
    public ApiResult<Void> insert(@RequestBody CollectDTO collect, HttpServletRequest request) {
        log.info("开始执行文档收藏操作，文档ID: {}, 用户ID: {}", collect.getDocId(), request.getAttribute("id"));
        CollectDocRelationship relationship = setRelationshipValue(collect, request);
        // 必须经过userId和docId的校验，否则不予关注
        if (!userService.isExist(relationship.getUserId())) {
            log.error("用户不存在，用户ID: {}", relationship.getUserId());
            throw new BusinessException(ErrorCode.USER_NOT_FOUND);
        }
        if (!fileService.isExist(relationship.getDocId())) {
            log.error("文档不存在，文档ID: {}", relationship.getDocId());
            throw new BusinessException(ErrorCode.DOCUMENT_NOT_FOUND);
        }
        try {
            collectService.insert(relationship);
            log.info("文档收藏成功，文档ID: {}, 用户ID: {}", relationship.getDocId(), relationship.getUserId());
        } catch (Exception e) {
            log.error("文档收藏失败，文档ID: {}, 用户ID: {}, 错误: {}", relationship.getDocId(), relationship.getUserId(), e.getMessage(), e);
            throw e;
        }
        return ApiResult.success();
    }

    /**
     * @param collect 收藏数据传输对象
     * @param request HTTP请求对象
     * @return ApiResult<Void> 操作结果
     * @author luojiarui
     **/
    @DeleteMapping(value = "/auth/remove")
    public ApiResult<Void> remove(@RequestBody CollectDTO collect, HttpServletRequest request) {
        log.info("开始执行取消收藏操作，文档ID: {}, 用户ID: {}", collect.getDocId(), request.getAttribute("id"));
        CollectDocRelationship relationship = setRelationshipValue(collect, request);
        try {
            collectService.remove(relationship);
            log.info("取消收藏成功，文档ID: {}, 用户ID: {}", relationship.getDocId(), relationship.getUserId());
        } catch (Exception e) {
            log.error("取消收藏失败，文档ID: {}, 用户ID: {}, 错误: {}", relationship.getDocId(), relationship.getUserId(), e.getMessage(), e);
            throw e;
        }
        return ApiResult.success();
    }

    /**
     * @param collect 收藏数据传输对象
     * @param request HTTP请求对象
     * @return CollectDocRelationship 收藏文档关系实体
     * @author luojiarui
     **/
    private CollectDocRelationship setRelationshipValue(CollectDTO collect, HttpServletRequest request) {
        log.debug("创建收藏关系实体，文档ID: {}, 用户ID: {}", collect.getDocId(), request.getAttribute("id"));
        CollectDocRelationship relationship = new CollectDocRelationship();
        relationship.setDocId(collect.getDocId());
        relationship.setUserId((String) request.getAttribute("id"));
        relationship.setCreateDate(new Date());
        relationship.setUpdateDate(new Date());
        return relationship;
    }

}