From 35c668c73a11db08a9891be58ae20d421f34c993 Mon Sep 17 00:00:00 2001
From: Junjie <fallin.jie@qq.com>
Date: 星期四, 12 三月 2026 14:06:48 +0800
Subject: [PATCH] #
---
src/main/java/com/zy/ai/config/AiPromptTemplateInitializer.java | 50 ++++++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 50 insertions(+), 0 deletions(-)
diff --git a/src/main/java/com/zy/ai/config/AiPromptTemplateInitializer.java b/src/main/java/com/zy/ai/config/AiPromptTemplateInitializer.java
new file mode 100644
index 0000000..7e84560
--- /dev/null
+++ b/src/main/java/com/zy/ai/config/AiPromptTemplateInitializer.java
@@ -0,0 +1,50 @@
+package com.zy.ai.config;
+
+import com.zy.ai.service.AiPromptTemplateService;
+import jakarta.annotation.PostConstruct;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.stereotype.Component;
+
+import javax.sql.DataSource;
+import java.sql.Connection;
+import java.sql.DatabaseMetaData;
+import java.sql.ResultSet;
+
+@Slf4j
+@Component
+public class AiPromptTemplateInitializer {
+
+ private final DataSource dataSource;
+ private final AiPromptTemplateService aiPromptTemplateService;
+
+ public AiPromptTemplateInitializer(DataSource dataSource, AiPromptTemplateService aiPromptTemplateService) {
+ this.dataSource = dataSource;
+ this.aiPromptTemplateService = aiPromptTemplateService;
+ }
+
+ @PostConstruct
+ public void init() {
+ try (Connection connection = dataSource.getConnection()) {
+ if (!hasTable(connection, "sys_ai_prompt_template")) {
+ log.warn("Skip AI prompt initialization because table sys_ai_prompt_template does not exist");
+ return;
+ }
+ int changed = aiPromptTemplateService.initDefaultsIfMissing();
+ log.info("AI prompt templates initialized, insertedOrRecovered={}", changed);
+ } catch (Exception e) {
+ log.warn("Failed to initialize AI prompt templates", e);
+ }
+ }
+
+ private boolean hasTable(Connection connection, String tableName) throws Exception {
+ DatabaseMetaData metaData = connection.getMetaData();
+ try (ResultSet resultSet = metaData.getTables(connection.getCatalog(), null, tableName, new String[]{"TABLE"})) {
+ while (resultSet.next()) {
+ if (tableName.equalsIgnoreCase(resultSet.getString("TABLE_NAME"))) {
+ return true;
+ }
+ }
+ }
+ return false;
+ }
+}
--
Gitblit v1.9.1