add ssl conn support to redis (#2215)

hhyo · Jul 17, 2023 · cedc1f3 · cedc1f3
1 parent cdbd12e
commit cedc1f3
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 0 deletions.
diff --git a/sql/engines/__init__.py b/sql/engines/__init__.py
@@ -20,6 +20,7 @@ def __init__(self, instance=None):
  self.password = instance.password
  self.db_name = instance.db_name
  self.mode = instance.mode
+ self.is_ssl = instance.is_ssl
 
  # 判断如果配置了隧道则连接隧道，只测试了MySQL
  if self.instance.tunnel:

diff --git a/sql/engines/redis.py b/sql/engines/redis.py
@@ -33,6 +33,7 @@ def get_connection(self, db_name=None):
  encoding_errors="ignore",
  decode_responses=True,
  socket_connect_timeout=10,
+ ssl=self.is_ssl,
  )
  else:
  return redis.Redis(
@@ -43,6 +44,7 @@ def get_connection(self, db_name=None):
  encoding_errors="ignore",
  decode_responses=True,
  socket_connect_timeout=10,
+ ssl=self.is_ssl,
  )
 
  @property

diff --git a/sql/models.py b/sql/models.py
@@ -195,6 +195,7 @@ class Instance(models.Model):
  password = fields.EncryptedCharField(
  verbose_name="密码", max_length=300, default="", blank=True
  )
+ is_ssl = models.BooleanField("是否启用SSL", default=False)
  db_name = models.CharField("数据库", max_length=64, default="", blank=True)
  charset = models.CharField("字符集", max_length=20, default="", blank=True)
  service_name = models.CharField(

diff --git a/src/init_sql/v1.9.3.sql b/src/init_sql/v1.9.3.sql
@@ -0,0 +1 @@
+ALTER TABLE sql_instance ADD is_ssl tinyint(1) DEFAULT 0 COMMENT '是否启用SSL';