#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
97
98
99
100
101
102
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.AiCallLog;
import com.vincent.rsf.server.system.service.AiCallLogService;
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.util.Date;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
 
@RestController
public class AiCallLogController extends BaseController {
 
    @Autowired
    private AiCallLogService aiCallLogService;
 
    @PreAuthorize("hasAuthority('system:aiCallLog:list')")
    @PostMapping("/aiCallLog/page")
    public R page(@RequestBody Map<String, Object> map) {
        BaseParam baseParam = buildParam(map, BaseParam.class);
        PageParam<AiCallLog, BaseParam> pageParam = new PageParam<>(baseParam, AiCallLog.class);
        com.baomidou.mybatisplus.core.conditions.query.QueryWrapper<AiCallLog> wrapper = pageParam.buildWrapper(true);
        wrapper.eq("tenant_id", getTenantId());
        return R.ok().add(aiCallLogService.page(pageParam, wrapper));
    }
 
    @PreAuthorize("hasAuthority('system:aiCallLog:list')")
    @GetMapping("/aiCallLog/{id}")
    public R get(@PathVariable("id") Long id) {
        AiCallLog callLog = getTenantRecord(id);
        if (callLog == null) {
            return R.error("record not found");
        }
        return R.ok().add(callLog);
    }
 
    @PreAuthorize("hasAuthority('system:aiCallLog:list')")
    @PostMapping("/aiCallLog/export")
    public void export(@RequestBody Map<String, Object> map, HttpServletResponse response) throws Exception {
        ExcelUtil.build(ExcelUtil.create(aiCallLogService.list(new LambdaQueryWrapper<AiCallLog>()
                .eq(AiCallLog::getTenantId, getTenantId())), AiCallLog.class), response);
    }
 
    @PreAuthorize("hasAuthority('system:aiCallLog:list')")
    @GetMapping("/ai/call-log/list")
    public R customList() {
        return R.ok().add(aiCallLogService.list(new LambdaQueryWrapper<AiCallLog>()
                .eq(AiCallLog::getTenantId, getTenantId())
                .orderByDesc(AiCallLog::getCreateTime, AiCallLog::getId)));
    }
 
    @PreAuthorize("hasAuthority('system:aiCallLog:list')")
    @GetMapping("/ai/call-log/stats")
    public R stats() {
        List<AiCallLog> logs = aiCallLogService.list(new LambdaQueryWrapper<AiCallLog>()
                .eq(AiCallLog::getTenantId, getTenantId())
                .orderByDesc(AiCallLog::getCreateTime, AiCallLog::getId));
        long total = logs.size();
        long successCount = logs.stream().filter(item -> Integer.valueOf(1).equals(item.getResult())).count();
        long failCount = logs.stream().filter(item -> Integer.valueOf(0).equals(item.getResult())).count();
        long modelCount = logs.stream().map(AiCallLog::getModelCode).filter(item -> item != null && !item.trim().isEmpty()).distinct().count();
        long routeCount = logs.stream().map(AiCallLog::getRouteCode).filter(item -> item != null && !item.trim().isEmpty()).distinct().count();
        long totalSpend = logs.stream().map(AiCallLog::getSpendTime).filter(item -> item != null && item > 0L).mapToLong(Long::longValue).sum();
        long spendCount = logs.stream().map(AiCallLog::getSpendTime).filter(item -> item != null && item > 0L).count();
        Date now = new Date();
        long last24hCount = logs.stream()
                .map(AiCallLog::getCreateTime)
                .filter(item -> item != null && now.getTime() - item.getTime() <= 24L * 60L * 60L * 1000L)
                .count();
 
        Map<String, Object> payload = new LinkedHashMap<>();
        payload.put("total", total);
        payload.put("successCount", successCount);
        payload.put("failCount", failCount);
        payload.put("successRate", total <= 0 ? 0D : (successCount * 100D) / total);
        payload.put("avgSpendTime", spendCount <= 0 ? 0L : totalSpend / spendCount);
        payload.put("modelCount", modelCount);
        payload.put("routeCount", routeCount);
        payload.put("last24hCount", last24hCount);
        return R.ok().add(payload);
    }
 
    private AiCallLog getTenantRecord(Long id) {
        if (id == null) {
            return null;
        }
        return aiCallLogService.getOne(new LambdaQueryWrapper<AiCallLog>()
                .eq(AiCallLog::getTenantId, getTenantId())
                .eq(AiCallLog::getId, id)
                .last("limit 1"));
    }
 
}