#ai 代码优化

zhou zhou

7 小时以前 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
package com.vincent.rsf.server.ai.store;
 
import com.vincent.rsf.server.ai.config.AiDefaults;
import com.vincent.rsf.server.ai.dto.AiResolvedConfig;
import com.vincent.rsf.server.ai.store.support.AiRedisExecutor;
import com.vincent.rsf.server.ai.store.support.AiRedisIndexSupport;
import com.vincent.rsf.server.ai.store.support.AiRedisKeys;
import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Component;
 
@Component
@RequiredArgsConstructor
public class AiConfigCacheStore {
 
    private final AiRedisExecutor aiRedisExecutor;
    private final AiRedisIndexSupport aiRedisIndexSupport;
    private final AiRedisKeys aiRedisKeys;
 
    public AiResolvedConfig getResolvedConfig(Long tenantId, String promptCode, Long aiParamId) {
        return aiRedisExecutor.readJson(aiRedisKeys.buildConfigKey(tenantId, promptCode, aiParamId), AiResolvedConfig.class);
    }
 
    public void cacheResolvedConfig(Long tenantId, String promptCode, Long aiParamId, AiResolvedConfig config) {
        String key = aiRedisKeys.buildConfigKey(tenantId, promptCode, aiParamId);
        aiRedisExecutor.writeJson(key, config, AiDefaults.CONFIG_CACHE_TTL_SECONDS);
        aiRedisIndexSupport.remember(aiRedisKeys.buildConfigIndexKey(tenantId), key);
    }
 
    public void evictTenantConfigCaches(Long tenantId) {
        aiRedisIndexSupport.deleteTrackedKeys(aiRedisKeys.buildConfigIndexKey(tenantId));
    }
}