#AI

zhou zhou

4 小时以前 51877df13075ad10ef51107f15bcd21f1661febe

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
package com.vincent.rsf.server.system.controller;
 
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.vincent.rsf.framework.common.R;
import com.vincent.rsf.server.common.domain.BaseParam;
import com.vincent.rsf.server.common.domain.PageParam;
import com.vincent.rsf.server.common.utils.ExcelUtil;
import com.vincent.rsf.server.system.entity.AiDiagnosisRecord;
import com.vincent.rsf.server.system.service.AiDiagnosisRecordService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.web.bind.annotation.*;
 
import javax.servlet.http.HttpServletResponse;
import java.net.URLEncoder;
import java.nio.charset.StandardCharsets;
import java.util.Map;
 
@RestController
public class AiDiagnosisController extends BaseController {
 
    @Autowired
    private AiDiagnosisRecordService aiDiagnosisRecordService;
 
    @PreAuthorize("hasAuthority('system:aiDiagnosis:list')")
    @PostMapping("/aiDiagnosis/page")
    public R page(@RequestBody Map<String, Object> map) {
        BaseParam baseParam = buildParam(map, BaseParam.class);
        PageParam<AiDiagnosisRecord, BaseParam> pageParam = new PageParam<>(baseParam, AiDiagnosisRecord.class);
        com.baomidou.mybatisplus.core.conditions.query.QueryWrapper<AiDiagnosisRecord> wrapper = pageParam.buildWrapper(true);
        wrapper.eq("tenant_id", getTenantId());
        return R.ok().add(aiDiagnosisRecordService.page(pageParam, wrapper));
    }
 
    @PreAuthorize("hasAuthority('system:aiDiagnosis:list')")
    @GetMapping("/aiDiagnosis/{id}")
    public R get(@PathVariable("id") Long id) {
        AiDiagnosisRecord record = getTenantRecord(id);
        if (record == null) {
            return R.error("record not found");
        }
        return R.ok().add(record);
    }
 
    @PreAuthorize("hasAuthority('system:aiDiagnosis:list')")
    @PostMapping("/aiDiagnosis/export")
    public void export(@RequestBody Map<String, Object> map, HttpServletResponse response) throws Exception {
        ExcelUtil.build(ExcelUtil.create(aiDiagnosisRecordService.list(new LambdaQueryWrapper<AiDiagnosisRecord>()
                .eq(AiDiagnosisRecord::getTenantId, getTenantId())), AiDiagnosisRecord.class), response);
    }
 
    @PreAuthorize("hasAuthority('system:aiDiagnosis:list')")
    @GetMapping("/ai/diagnosis/list")
    public R diagnosisList() {
        return R.ok().add(aiDiagnosisRecordService.list(new LambdaQueryWrapper<AiDiagnosisRecord>()
                .eq(AiDiagnosisRecord::getTenantId, getTenantId())
                .orderByDesc(AiDiagnosisRecord::getCreateTime, AiDiagnosisRecord::getId)));
    }
 
    @PreAuthorize("hasAuthority('system:aiDiagnosis:list')")
    @GetMapping("/ai/diagnosis/detail/{id}")
    public R detail(@PathVariable("id") Long id) {
        AiDiagnosisRecord record = getTenantRecord(id);
        if (record == null) {
            return R.error("record not found");
        }
        return R.ok().add(record);
    }
 
    @PreAuthorize("hasAuthority('system:aiDiagnosis:list')")
    @GetMapping("/ai/diagnosis/report/export/{id}")
    public void exportReport(@PathVariable("id") Long id, HttpServletResponse response) throws Exception {
        AiDiagnosisRecord record = getTenantRecord(id);
        if (record == null) {
            response.setStatus(404);
            return;
        }
        String fileName = URLEncoder.encode((record.getDiagnosisNo() == null ? "ai-diagnosis-report" : record.getDiagnosisNo()) + ".md", StandardCharsets.UTF_8.name());
        response.setCharacterEncoding(StandardCharsets.UTF_8.name());
        response.setContentType("text/markdown; charset=utf-8");
        response.setHeader("Content-Disposition", "attachment; filename=" + fileName);
        response.getWriter().write(record.getReportMarkdown() == null ? "" : record.getReportMarkdown());
        response.getWriter().flush();
    }
 
    private AiDiagnosisRecord getTenantRecord(Long id) {
        if (id == null) {
            return null;
        }
        return aiDiagnosisRecordService.getOne(new LambdaQueryWrapper<AiDiagnosisRecord>()
                .eq(AiDiagnosisRecord::getTenantId, getTenantId())
                .eq(AiDiagnosisRecord::getId, id)
                .last("limit 1"));
    }
 
}