feat(mcp): 支持 PostgreSQL 和 ClickHouse 数据库连接 (higress-group#579)

Co-authored-by: xujingfeng <xujingfeng95@gmail.com>

Author: JayLi52

frontend/src/pages/mcp/components/DatabaseConfig.tsx

@@ -58,8 +58,23 @@ export const DB_FIXED_FIELDS = [
 
 export const computeDSN = (values: any) => {
   const { db_type, db_user_name, db_password, db_server_host, db_database, db_server_port } = values;
-  if (!db_type || !db_user_name || !db_password || !db_server_host || !db_database) return '';
-  // eslint-disable-next-line max-len
+  if (!db_type || !db_user_name || !db_password || !db_server_host || !db_server_port) return '';
+  // PostgreSQL: postgres://username:password@host:port/dbname
+  if (db_type === 'POSTGRESQL') {
+    if (!db_database) return '';
+    return `postgres://${db_user_name}:${db_password}@${db_server_host}:${db_server_port}/${db_database}`;
+  }
+  // ClickHouse: tcp://host:port?database=xxx&username=xxx&password=xxx
+  if (db_type === 'CLICKHOUSE') {
+    const query = new URLSearchParams({
+      database: db_database || '',
+      username: db_user_name,
+      password: db_password,
+    }).toString();
+    return `tcp://${db_server_host}:${db_server_port}?${query}`;
+  }
+  // MySQL (默认): username:password@tcp(host:port)/dbname?...
+  if (!db_database) return '';
   return `${db_user_name}:${db_password}@tcp(${db_server_host}:${db_server_port})/${db_database}?charset=utf8mb4&parseTime=True&loc=Local`;
 };
 

frontend/src/pages/mcp/components/McpFormDrawer.tsx

@@ -65,13 +65,35 @@ const McpFormDrawer: React.FC<McpFormDrawerProps> = ({ visible, mode, name, onCl
         form.resetFields();
       } else {
         if (record && record.dsn) {
-          const match = record.dsn.match(REG_DSN_STRING.DEFAULT);
-          if (match) {
-            form.setFieldsValue({
-              db_user_name: match[1],
-              db_password: match[2],
-              db_database: match[5],
-            });
+          const type = record?.dbType;
+          let m: RegExpMatchArray | null = null;
+          if (type === 'MYSQL') {
+            m = record.dsn.match(REG_DSN_STRING.MYSQL);
+            if (m) {
+              form.setFieldsValue({
+                db_user_name: m[1],
+                db_password: m[2],
+                db_database: m[5],
+              });
+            }
+          } else if (type === 'POSTGRESQL') {
+            m = record.dsn.match(REG_DSN_STRING.POSTGRESQL);
+            if (m) {
+              form.setFieldsValue({
+                db_user_name: m[1],
+                db_password: m[2],
+                db_database: m[5],
+              });
+            }
+          } else if (type === 'CLICKHOUSE') {
+            m = record.dsn.match(REG_DSN_STRING.CLICKHOUSE);
+            if (m) {
+              form.setFieldsValue({
+                db_user_name: m[4],
+                db_password: m[5],
+                db_database: m[3],
+              });
+            }
           }
         }
         form.setFieldsValue({
@@ -331,7 +353,20 @@ const McpFormDrawer: React.FC<McpFormDrawerProps> = ({ visible, mode, name, onCl
               {
                 required: true,
                 validator: (_, value) => {
-                  if (!value || !value.match(REG_DSN_STRING.DEFAULT)) {
+                  const values = form.getFieldsValue();
+                  const type = values?.db_type;
+                  if (!value) {
+                    return Promise.reject(new Error(t('mcp.form.databaseConfigInvalid')!));
+                  }
+                  let valid = false;
+                  if (type === 'MYSQL') {
+                    valid = REG_DSN_STRING.MYSQL.test(value);
+                  } else if (type === 'POSTGRESQL') {
+                    valid = REG_DSN_STRING.POSTGRESQL.test(value);
+                  } else if (type === 'CLICKHOUSE') {
+                    valid = REG_DSN_STRING.CLICKHOUSE.test(value);
+                  }
+                  if (!valid) {
                     return Promise.reject(new Error(t('mcp.form.databaseConfigInvalid')!));
                   }
                   return Promise.resolve();

frontend/src/pages/mcp/constant.ts

@@ -13,12 +13,11 @@ export function getServiceTypeMap(directRouteText: string) {
     [SERVICE_TYPE.DIRECT_ROUTE]: directRouteText,
   };
 }
-// MYSQL/PostgreSQL/Sqlite/Clickhouse
+// MYSQL/PostgreSQL/Clickhouse
 export const DB_TYPE_OPTIONS = [
   { label: 'MySQL', value: 'MYSQL' },
-  { label: 'PostgreSQL', value: 'PostgreSQL' },
-  { label: 'SQLite', value: 'Sqlite' },
-  { label: 'ClickHouse', value: 'Clickhouse' },
+  { label: 'PostgreSQL', value: 'POSTGRESQL' },
+  { label: 'ClickHouse', value: 'CLICKHOUSE' },
 ];
 
 export const DOMAIN_PROTOCOL_MAP = {
@@ -28,11 +27,12 @@ export const DOMAIN_PROTOCOL_MAP = {
 
 // 数据库连接字符串正则 eg: mysql:user:pass@tcp(host:port)/database?charset=utf8mb4&parseTime=True&loc=Local
 export const REG_DSN_STRING = {
+  // username:password@tcp(host:port)/dbname?param1=value1&param2=value2
   MYSQL: /^(\w+):([^@]+)@tcp\(([^:]+):(\d+)\)\/([^?]+)\?(.+)$/,
-  POSTGRESQL: /^(\w+):([^@]+)@tcp\(([^:]+):(\d+)\)\/([^?]+)\?(.+)$/,
-  SQLITE: /^(\w+):([^@]+)@tcp\(([^:]+):(\d+)\)\/([^?]+)\?(.+)$/,
-  CLICKHOUSE: /^(\w+):([^@]+)@tcp\(([^:]+):(\d+)\)\/([^?]+)\?(.+)$/,
-  DEFAULT: /^(\w+):([^@]+)@tcp\(([^:]+):(\d+)\)\/([^?]+)\?(.+)$/,
+  // postgres://username:password@host:port/dbname
+  POSTGRESQL: /^postgres:\/\/([^:]+):([^@]+)@([^:]+):(\d+)\/([^?]+)$/,
+  // tcp://localhost:9000?database=default&username=default&password=
+  CLICKHOUSE: /^tcp:\/\/([^:]+):(\d+)\?database=([^&]+)&username=([^&]*)&password=([^&]*)/,
 };