diff --git a/chat2db-client/src/blocks/Setting/BaseSetting/index.tsx b/chat2db-client/src/blocks/Setting/BaseSetting/index.tsx index aa4ff38e..bb2c1547 100644 --- a/chat2db-client/src/blocks/Setting/BaseSetting/index.tsx +++ b/chat2db-client/src/blocks/Setting/BaseSetting/index.tsx @@ -105,6 +105,9 @@ export default function BaseSetting() { function changeLang(e: any) { setLangLocalStorage(e.target.value); + //切换语言时,需要设置cookie,用来改变后台服务的Locale + const date = new Date('2030-12-30 12:30:00').toUTCString(); + document.cookie = `CHAT2DB.LOCALE=${e.target.value};Expires=${date}`; location.reload(); } diff --git a/chat2db-server/chat2db-server-start/src/main/java/ai/chat2db/server/start/config/util/CopyTemplate.java b/chat2db-server/chat2db-server-start/src/main/java/ai/chat2db/server/start/config/util/CopyTemplate.java index b0f6c164..d723e590 100644 --- a/chat2db-server/chat2db-server-start/src/main/java/ai/chat2db/server/start/config/util/CopyTemplate.java +++ b/chat2db-server/chat2db-server-start/src/main/java/ai/chat2db/server/start/config/util/CopyTemplate.java @@ -21,14 +21,9 @@ public class CopyTemplate { /** * 模板文件 **/ - public static final List TEMPLATE_FILE = Arrays.asList("template.html", "template_diy.docx", "sub_template_diy.docx"); + private static final List TEMPLATE_FILE = Arrays.asList("template.html", "template_diy.docx", "sub_template_diy.docx"); - static { - //复制模板 - copyTemplateFile(); - } - - public static void copyTemplateFile() { + public void copyTemplateFile() { String templateDir = ConfigUtils.CONFIG_BASE_PATH + File.separator + "template"; File file = new File(templateDir); if (!file.exists()) { @@ -39,7 +34,7 @@ public class CopyTemplate { } } - public static void saveFile(String dir, String path, boolean isOverride) { + public void saveFile(String dir, String path, boolean isOverride) { if (!isOverride) { File file = new File(dir + File.separator + path); if (file.exists()) { @@ -49,7 +44,7 @@ public class CopyTemplate { try (// 模板文件输入输出地址 读取resources下文件 FileOutputStream outputStream = new FileOutputStream(dir + File.separator + path); //返回读取指定资源的输入流 - InputStream inputStream = ConfigUtils.class.getClassLoader().getResourceAsStream("template" + File.separator + path)) { + InputStream inputStream = CopyTemplate.class.getClassLoader().getResourceAsStream("template/" + path)) { byte[] buffer = new byte[4096]; int n = 0; while (-1 != (n = inputStream.read(buffer))) { diff --git a/chat2db-server/chat2db-server-start/src/main/java/ai/chat2db/server/start/listener/TemplateListener.java b/chat2db-server/chat2db-server-start/src/main/java/ai/chat2db/server/start/listener/TemplateListener.java new file mode 100644 index 00000000..2f0783c7 --- /dev/null +++ b/chat2db-server/chat2db-server-start/src/main/java/ai/chat2db/server/start/listener/TemplateListener.java @@ -0,0 +1,26 @@ +package ai.chat2db.server.start.listener; + +import ai.chat2db.server.start.config.util.CopyTemplate; +import ai.chat2db.server.web.api.controller.rdb.doc.event.TemplateEvent; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.context.event.EventListener; +import org.springframework.stereotype.Component; + +/** + * TemplateListener + * + * @author lzy + **/ +@Component +public class TemplateListener { + + @Autowired + private CopyTemplate copyTemplate; + + @EventListener(classes = TemplateEvent.class) + public void copyTemplate() { + //复制模板 + copyTemplate.copyTemplateFile(); + } + +} diff --git a/chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/rdb/RdbDocController.java b/chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/rdb/RdbDocController.java index a765b153..4dd39690 100644 --- a/chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/rdb/RdbDocController.java +++ b/chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/rdb/RdbDocController.java @@ -11,6 +11,7 @@ import ai.chat2db.server.web.api.aspect.ConnectionInfoAspect; import ai.chat2db.server.web.api.controller.rdb.converter.RdbWebConverter; import ai.chat2db.server.web.api.controller.rdb.doc.DatabaseExportService; import ai.chat2db.server.web.api.controller.rdb.doc.conf.ExportOptions; +import ai.chat2db.server.web.api.controller.rdb.doc.event.TemplateEvent; import ai.chat2db.server.web.api.controller.rdb.factory.ExportServiceFactory; import ai.chat2db.server.web.api.controller.rdb.request.DataExportRequest; import ai.chat2db.server.web.api.controller.rdb.vo.TableVO; @@ -20,6 +21,7 @@ import jakarta.servlet.http.HttpServletResponse; import jakarta.validation.Valid; import lombok.extern.slf4j.Slf4j; import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.context.ApplicationContext; import org.springframework.stereotype.Controller; import org.springframework.web.bind.annotation.PostMapping; import org.springframework.web.bind.annotation.RequestBody; @@ -48,6 +50,9 @@ public class RdbDocController { @Autowired private RdbWebConverter rdbWebConverter; + @Autowired + private ApplicationContext applicationContext; + /** * export data * @@ -55,6 +60,8 @@ public class RdbDocController { */ @PostMapping("/export") public void export(@Valid @RequestBody DataExportRequest request, HttpServletResponse response) throws Exception { + //复制模板 + applicationContext.publishEvent(new TemplateEvent("copy")); ExportTypeEnum exportType = EasyEnumUtils.getEnum(ExportTypeEnum.class, request.getExportType()); response.setCharacterEncoding("utf-8"); String fileName = URLEncoder.encode( diff --git a/chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/rdb/doc/event/TemplateEvent.java b/chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/rdb/doc/event/TemplateEvent.java new file mode 100644 index 00000000..e480ac1b --- /dev/null +++ b/chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/rdb/doc/event/TemplateEvent.java @@ -0,0 +1,14 @@ +package ai.chat2db.server.web.api.controller.rdb.doc.event; + +import org.springframework.context.ApplicationEvent; + +/** + * TemplateEvent + * + * @author lzy + **/ +public class TemplateEvent extends ApplicationEvent { + public TemplateEvent(String key) { + super(key); + } +}