refactor: Add function to compatible database name

JerryFan626 · JerryFan626 · commit fcf864507c58 · 2023-10-22T16:28:53.000+08:00
diff --git a/chat2db-client/src/utils/IntelliSense/table.ts b/chat2db-client/src/utils/IntelliSense/table.ts
@@ -13,7 +13,7 @@ let intelliSenseTable = monaco.languages.registerCompletionItemProvider('sql', {
 });
 
 /** 根据不同的数据库，插入不同的表名  */
-const handleInsertText = (text, databaseCode: DatabaseTypeCode = DatabaseTypeCode.MYSQL) => {
+const handleInsertText = (text: string, databaseCode: DatabaseTypeCode = DatabaseTypeCode.MYSQL) => {
   if (
     [DatabaseTypeCode.POSTGRESQL, DatabaseTypeCode.ORACLE, DatabaseTypeCode.DB2, DatabaseTypeCode.SQLITE].includes(
       databaseCode,
diff --git a/chat2db-client/src/utils/database.ts b/chat2db-client/src/utils/database.ts
@@ -1,3 +1,4 @@
+import { DatabaseTypeCode } from '@/constants/common';
 import { IWorkspaceModelType } from '@/models/workspace';
 import { Option } from '@/typings/common';
 
@@ -36,3 +37,32 @@ export function handleDatabaseAndSchema(databaseAndSchema: IWorkspaceModelType['
   }
   return newCascaderOptions;
 }
+
+/**
+ * 兼容处理数据库名称
+ * @param databaseName
+ * @param databaseType
+ * @returns
+ */
+export function compatibleDataBaseName(databaseName: string, databaseType: DatabaseTypeCode) {
+  //""  oracele  sqlite postgrsql  h2 dm
+  // ` MYSQL clickhouse MariaDB
+  // [ sqlserver
+  if (
+    [
+      DatabaseTypeCode.ORACLE,
+      DatabaseTypeCode.SQLITE,
+      DatabaseTypeCode.POSTGRESQL,
+      DatabaseTypeCode.H2,
+      DatabaseTypeCode.DB2,
+    ].includes(databaseType)
+  ) {
+    return `"${databaseName}"`;
+  } else if ([DatabaseTypeCode.SQLSERVER].includes(databaseType)) {
+    return `[${databaseName}]`;
+  } else if ([DatabaseTypeCode.MYSQL, DatabaseTypeCode.CLICKHOUSE, DatabaseTypeCode.MARIADB].includes(databaseType)) {
+    return `\`${databaseName}\``;
+  } else {
+    return `${databaseName}`;
+  }
+}
