实例增加3个字段：是否验证服务端SSL证书、显示的数据库列表、隐藏的数据库列表

sql/engines/__init__.py

@@ -1,7 +1,9 @@
 """engine base库, 包含一个``EngineBase`` class和一个get_engine函数"""
 
 import importlib
+import re
 from sql.engines.models import ResultSet, ReviewSet
+from sql.models import Instance
 from sql.utils.ssh_tunnel import SSHConnection
 from django.conf import settings
 
@@ -14,11 +16,11 @@ class EngineBase:
     name = "Base"
     info = "base engine"
 
-    def __init__(self, instance=None):
+    def __init__(self, instance: Instance = None):
         self.conn = None
         self.thread_id = None
         if instance:
-            self.instance = instance
+            self.instance = instance  # type: Instance
             self.instance_name = instance.instance_name
             self.host = instance.host
             self.port = int(instance.port)

sql/engines/mongo.py

@@ -848,9 +848,10 @@ def get_all_databases(self):
         result = ResultSet()
         conn = self.get_connection()
         try:
-            result.rows = conn.list_database_names()
+            db_list = conn.list_database_names()
         except OperationFailure:
-            result.rows = [self.db_name]
+            db_list = [self.db_name]
+        result.rows = db_list
         return result
 
     def get_all_tables(self, db_name, **kwargs):

sql/instance.py

@@ -14,6 +14,7 @@
 from common.utils.convert import Convert
 from sql.engines import get_engine
 from sql.plugins.schemasync import SchemaSync
+from sql.utils.sql_utils import filter_db_list
 from .models import Instance, ParamTemplate, ParamHistory
 
 
@@ -332,6 +333,16 @@ def instance_resource(request):
         tb_name = query_engine.escape_string(tb_name)
         if resource_type == "database":
             resource = query_engine.get_all_databases()
+            resource.rows = filter_db_list(
+                db_list=resource.rows,
+                db_name_regex=query_engine.instance.show_db_name_regex,
+                is_match_regex=True,
+            )
+            resource.rows = filter_db_list(
+                db_list=resource.rows,
+                db_name_regex=query_engine.instance.denied_db_name_regex,
+                is_match_regex=False,
+            )
         elif resource_type == "schema" and db_name:
             resource = query_engine.get_all_schemas(db_name=db_name)
         elif resource_type == "table" and db_name:

sql/models.py

@@ -205,7 +205,23 @@ class Instance(models.Model):
         verbose_name="密码", max_length=300, default="", blank=True
     )
     is_ssl = models.BooleanField("是否启用SSL", default=False)
+    verify_ssl = models.BooleanField("是否验证服务端SSL证书", default=True)
     db_name = models.CharField("数据库", max_length=64, default="", blank=True)
+    show_db_name_regex = models.CharField(
+        "显示的数据库列表正则",
+        max_length=1024,
+        default="",
+        blank=True,
+        help_text="正则表达式。示例：^(test_db|dmp_db|za.*)$。Redis示例: ^(0|4|6|11|12|13)$",
+    )
+    denied_db_name_regex = models.CharField(
+        "隐藏的数据库列表正则",
+        max_length=1024,
+        default="",
+        blank=True,
+        help_text="正则表达式。隐藏大于显示，此规则优先。",
+    )
+
     charset = models.CharField("字符集", max_length=20, default="", blank=True)
     service_name = models.CharField(
         "Oracle service name", max_length=50, null=True, blank=True

sql/utils/sql_utils.py

@@ -380,3 +380,37 @@ def get_exec_sqlitem_list(reviewResult, db_name):
             )
         )
     return list
+
+
+def filter_db_list(db_list, db_name_regex: str, is_match_regex: bool, key="value"):
+    """
+    根据配置的数据库列表过滤数据库名称列表。
+
+    :param db_list: 待过滤的数据库名称列表，可能是字符串列表或字典列表。
+    示例数据：
+    1. db_list=[{"value": 0, "text": 0, "value": 1, "text": 1}]
+    2. db_list=["a_db","b_db"]
+    :param db_name_regex: 配置的数据库正则。
+    :param is_match_regex 是匹配还是不匹配的正则
+    :param key: 当 db_list 包含字典时，指定用于匹配的键。默认值为 'value'。
+    :return: 过滤后的数据库名称列表或字典列表。
+    """
+    if not db_name_regex:
+        return db_list  # 如果没有指定 db_name_regex，返回原始 db_list
+
+    try:
+        db_regex = re.compile(db_name_regex)  # 编译正则表达式
+    except re.error:
+        raise ValueError(f"正则表达式解析异常: {db_name_regex}")
+
+    filtered_list = []
+
+    # 根据类型处理 db_list
+    for db in db_list:
+        # 确定要检查的值（字符串或字典中的特定键）
+        db_value = str(db[key]) if isinstance(db, dict) else db
+        is_match = bool(db_regex.match(db_value))
+        # 根据 is_match_regex 参数过滤
+        if (is_match_regex and is_match) or (not is_match_regex and not is_match):
+            filtered_list.append(db)
+    return filtered_list