package com.zy.core.task;
|
|
import com.zy.ai.entity.ChatCompletionRequest;
|
import com.zy.ai.service.LlmChatService;
|
import com.zy.common.utils.RedisUtil;
|
import com.zy.core.News;
|
import com.zy.core.enums.RedisKeyType;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.beans.factory.annotation.Value;
|
import org.springframework.scheduling.annotation.Scheduled;
|
import org.springframework.stereotype.Component;
|
|
import java.nio.charset.StandardCharsets;
|
import java.nio.file.Files;
|
import java.nio.file.Paths;
|
import java.util.List;
|
|
@Component
|
public class MakeMainProcessPseudocodeScheduler {
|
|
@Value("${mainProcessPlugin}")
|
private String mainProcessPlugin;
|
@Autowired
|
private LlmChatService llmChatService;
|
@Autowired
|
private RedisUtil redisUtil;
|
|
@Scheduled(cron = "1 * * * * ? ")
|
public void refreshPseudocodeDaily() {
|
try {
|
initMainProcessPseudocode();
|
} catch (Exception e) {
|
e.printStackTrace();
|
}
|
}
|
|
private void initMainProcessPseudocode(){
|
Object object = redisUtil.get(RedisKeyType.MAIN_PROCESS_PSEUDOCODE.key);
|
if (object != null) {
|
return;
|
}
|
|
String plugin = mainProcessPlugin;
|
if (plugin == null) plugin = "NormalProcess";
|
String className = plugin.contains(".") ? plugin : "com.zy.core.plugin." + plugin;
|
String code = null;
|
try {
|
String rel = className.replace('.', '/') + ".java";
|
java.nio.file.Path p = Paths.get(System.getProperty("user.dir"), "src", "main", "java", rel);
|
if (Files.exists(p)) {
|
code = new String(Files.readAllBytes(p), StandardCharsets.UTF_8);
|
}
|
} catch (Exception ignore) {}
|
String result = null;
|
if (code != null && !code.isEmpty()) {
|
List<ChatCompletionRequest.Message> messages = new java.util.ArrayList<>();
|
ChatCompletionRequest.Message system = new ChatCompletionRequest.Message();
|
system.setRole("system");
|
system.setContent("你现在是一名高级 Java 架构师兼伪代码转换专家，专门负责把复杂的 Java 代码转换成结构清晰、适合大模型阅读与推理的伪代码。\n" +
|
"\n" +
|
"请严格遵守以下要求工作：\n" +
|
"\n" +
|
"核心目标\n" +
|
"\n" +
|
"输入是一段或多段 Java 代码。\n" +
|
"\n" +
|
"输出是一段人类可读、逻辑清晰、尽量语言中立的伪代码。\n" +
|
"\n" +
|
"这份伪代码将被用作后续大模型提问的“参考描述”，所以要：\n" +
|
"\n" +
|
"保留关键业务逻辑和判断条件；\n" +
|
"\n" +
|
"弱化语言细节（如具体库、注解、框架细节）；\n" +
|
"\n" +
|
"用自然语言 + 简洁流程结构，帮助大模型快速理解代码意图。\n" +
|
"\n" +
|
"风格要求\n" +
|
"\n" +
|
"使用中文描述逻辑，但可以保留少量关键英文标识（例如类名、方法名、状态枚举）以便跟代码对应。\n" +
|
"\n" +
|
"伪代码要分层分块，尽量按：\n" +
|
"\n" +
|
"类职责说明\n" +
|
"\n" +
|
"重要字段 / 全局变量说明\n" +
|
"\n" +
|
"每个公开方法 / 核心私有方法的伪代码\n" +
|
"\n" +
|
"逻辑上使用类似：\n" +
|
"\n" +
|
"如果 ... 则 ...\n" +
|
"\n" +
|
"否则如果 ...\n" +
|
"\n" +
|
"循环遍历列表 ...\n" +
|
"\n" +
|
"调用服务/方法: ...\n" +
|
"\n" +
|
"返回 ...\n" +
|
"\n" +
|
"不追求严格语法，只追求易懂和准确。\n" +
|
"\n" +
|
"保留信息 & 抽象信息\n" +
|
"\n" +
|
"必须保留：\n" +
|
"\n" +
|
"关键业务含义（例如“生成入库任务”、“检查堆垛机任务是否完成”）\n" +
|
"\n" +
|
"关键条件判断（状态字段、枚举、重要配置开关）\n" +
|
"\n" +
|
"重要数据流向（从哪里读数据、写到哪里、调用了哪些服务）\n" +
|
"\n" +
|
"与外部系统交互（如 HTTP 调用 WMS、写 Redis 锁、写数据库）\n" +
|
"\n" +
|
"可以抽象或省略：\n" +
|
"\n" +
|
"日志打印的具体格式，只保留“记录日志：xxx”即可；\n" +
|
"\n" +
|
"具体框架注解（如 @Component, @Autowired 等）；\n" +
|
"\n" +
|
"泛型、异常栈细节、工具类内部实现；\n" +
|
"\n" +
|
"结构模板（优先遵循）\n" +
|
"\n" +
|
"对于一段较大的 Java 类，请按以下结构输出伪代码：\n" +
|
"\n" +
|
"类整体说明\n" +
|
"\n" +
|
"简要说明这个类的用途和在系统中的角色。\n" +
|
"\n" +
|
"重要字段 / 配置说明\n" +
|
"\n" +
|
"列出关键静态变量 / 配置项 / 状态缓存，并用一行解释它们的含义。\n" +
|
"\n" +
|
"主流程方法（例如 run()）\n" +
|
"\n" +
|
"用有序列表或伪代码，按调用顺序描述主要步骤。\n" +
|
"\n" +
|
"每个核心私有方法\n" +
|
"\n" +
|
"对于每个关键方法：\n" +
|
"\n" +
|
"先用一行中文总结功能；\n" +
|
"\n" +
|
"再给出伪代码流程（条件、循环、关键调用）；\n" +
|
"\n" +
|
"与外部系统交互的说明\n" +
|
"\n" +
|
"单独强调有哪些地方调用了外部服务（HTTP、消息队列、数据库、Redis 等）。\n" +
|
"\n" +
|
"输出格式要求\n" +
|
"\n" +
|
"使用 Markdown 结构，方便复制给其他大模型：\n" +
|
"\n" +
|
"用 ## 标题区分“类说明”、“主流程伪代码”、“方法伪代码”等部分；\n" +
|
"\n" +
|
"伪代码块可以使用缩进和项目符号，或用 pseudo 代码块 包裹；\n" +
|
"\n" +
|
"不要直接逐行翻译代码，而是做抽象和整理；\n" +
|
"\n" +
|
"不要输出无关文本，例如道歉、寒暄或与任务无关的解释。\n" +
|
"\n" +
|
"伪代码示例风格（示意）\n" +
|
"\n" +
|
"例如当输入一个 run() 方法时，期望你的输出风格类似：\n" +
|
"\n" +
|
"函数 run():\n" +
|
" 读取配置 enableFake, fakeRealTaskRequestWms\n" +
|
" 如果 enableFake == \"Y\":\n" +
|
" 调用 checkInStationHasTask() 检测入库站并生成仿真站点数据\n" +
|
" 如果 fakeRealTaskRequestWms == \"N\":\n" +
|
" 调用 generateFakeInTask() 生成本地仿真入库任务\n" +
|
" 调用 generateFakeOutTask() 生成本地仿真出库任务\n" +
|
" 计算所有站点的停留时间 calcAllStationStayTime()\n" +
|
" 检查出库站点是否超时并重置 checkOutStationStayTimeOut()\n" +
|
" 检查入库站点货物是否已被堆垛机取走 checkInStationCrnTake()\n" +
|
" 如果 fakeRealTaskRequestWms == \"Y\":\n" +
|
" 调用 generateStoreWrkFile() 请求 WMS 生成真实任务\n" +
|
" 调用 crnOperateUtils.crnIoExecute() 执行堆垛机任务\n" +
|
" 调用 crnIoExecuteFinish() 处理堆垛机任务完成后的状态更新和仿真站点生成\n" +
|
" 调用 stationOperateProcessUtils.stationInExecute() 执行输送站入库任务\n" +
|
" 调用 stationOperateProcessUtils.stationOutExecute() 执行输送站出库任务\n" +
|
" 调用 stationOperateProcessUtils.stationOutExecuteFinish() 检查输送站出库任务完成\n" +
|
"\n" +
|
"\n" +
|
"对输入的要求\n" +
|
"\n" +
|
"如果用户给出的是多段代码或只给出片段：\n" +
|
"\n" +
|
"先推断这段代码的职责；\n" +
|
"\n" +
|
"再按你能理解到的范围进行伪代码转换；\n" +
|
"\n" +
|
"如果存在明显缺失的类/方法，只需在伪代码中用“调用 XXX（具体逻辑略）”标记即可。\n" +
|
"\n" +
|
"请始终以「让后续大模型能看懂这段代码逻辑并基于伪代码进行推理和提问」为最高优先级来组织你的输出。");
|
messages.add(system);
|
ChatCompletionRequest.Message user = new ChatCompletionRequest.Message();
|
user.setRole("user");
|
user.setContent("主流程插件类源代码：\n\n" + code);
|
messages.add(user);
|
try {
|
result = llmChatService.chat(messages, 0.2, 2048);
|
} catch (Exception ignore) {}
|
}
|
redisUtil.set(RedisKeyType.MAIN_PROCESS_PSEUDOCODE.key, result, 60 * 60 * 24);
|
News.info("主流程伪代码已刷新");
|
}
|
|
}
|
