浏览代码

mysql传参修改

liweiquan 2 年之前
父节点
当前提交
91ff823e1d
共有 1 个文件被更改,包括 8 次插入2 次删除
  1. 8 2
      app/crud/job_jdbc_datasource.py

+ 8 - 2
app/crud/job_jdbc_datasource.py

@@ -22,8 +22,9 @@ def _format_datasource(db: Session, item: schemas.JobJdbcDatasourceBase, ds_id:
     host, port = item.jdbc_url.split(':')
     if not host or not port:
         raise Exception('jdbc_url无效')
-
-    ds = DataSrouceFactory.create(item.datasource, {'port': port, 'host': host, 'username': item.jdbc_username,
+    ds = None
+    if item.datasource == 'hive':
+        ds = DataSrouceFactory.create(item.datasource, {'port': port, 'host': host, 'username': item.jdbc_username,
                                                     'password': item.jdbc_password,
                                                     'database_name': item.database_name,
                                                     'kerberos': item.kerberos,
@@ -31,6 +32,11 @@ def _format_datasource(db: Session, item: schemas.JobJdbcDatasourceBase, ds_id:
                                                     'krb5config': item.krb5config,
                                                     'kerberos_service_name': item.kerberos_service_name,
                                                     'principal':item.principal})
+    else:
+        ds = DataSrouceFactory.create(item.datasource, {'port': port, 'host': host, 'username': item.jdbc_username,
+                                                    'password': item.jdbc_password,
+                                                    'database_name': item.database_name,
+                                                    })
     item.jdbc_url = ds.jdbc_url
     item.jdbc_username = ds.jdbc_username if item.kerberos == 0 else None
     item.jdbc_password = ds.jdbc_password if item.kerberos == 0 else None