#ai 代码优化

zhou zhou

13 小时以前 82624affb0251b75b62b35567d3eb260c06efe78

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
package com.vincent.rsf.server.ai.store;
 
import com.fasterxml.jackson.core.type.TypeReference;
import com.vincent.rsf.server.ai.config.AiDefaults;
import com.vincent.rsf.server.ai.dto.AiMcpConnectivityTestDto;
import com.vincent.rsf.server.ai.dto.AiMcpToolPreviewDto;
import com.vincent.rsf.server.ai.store.support.AiRedisExecutor;
import com.vincent.rsf.server.ai.store.support.AiRedisKeys;
import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Component;
 
import java.util.List;
 
@Component
@RequiredArgsConstructor
public class AiMcpCacheStore {
 
    private final AiRedisExecutor aiRedisExecutor;
    private final AiRedisKeys aiRedisKeys;
 
    public List<AiMcpToolPreviewDto> getToolPreview(Long tenantId, Long mountId) {
        return aiRedisExecutor.readJson(aiRedisKeys.buildMcpPreviewKey(tenantId, mountId), new TypeReference<List<AiMcpToolPreviewDto>>() {
        });
    }
 
    public void cacheToolPreview(Long tenantId, Long mountId, List<AiMcpToolPreviewDto> tools) {
        aiRedisExecutor.writeJson(aiRedisKeys.buildMcpPreviewKey(tenantId, mountId), tools, AiDefaults.MCP_PREVIEW_CACHE_TTL_SECONDS);
    }
 
    public AiMcpConnectivityTestDto getConnectivity(Long tenantId, Long mountId) {
        return aiRedisExecutor.readJson(aiRedisKeys.buildMcpHealthKey(tenantId, mountId), AiMcpConnectivityTestDto.class);
    }
 
    public void cacheConnectivity(Long tenantId, Long mountId, AiMcpConnectivityTestDto connectivity) {
        aiRedisExecutor.writeJson(aiRedisKeys.buildMcpHealthKey(tenantId, mountId), connectivity, AiDefaults.MCP_HEALTH_CACHE_TTL_SECONDS);
    }
 
    public void evictMcpMountCaches(Long tenantId, Long mountId) {
        if (mountId == null) {
            return;
        }
        aiRedisExecutor.delete(aiRedisKeys.buildMcpPreviewKey(tenantId, mountId));
        aiRedisExecutor.delete(aiRedisKeys.buildMcpHealthKey(tenantId, mountId));
    }
}