From 9aee7740f6414329c04bdff880ad286249615421 Mon Sep 17 00:00:00 2001
From: xiaohan815 <403673749@qq.com>
Date: Mon, 27 Apr 2020 12:09:12 +0800
Subject: [PATCH] =?UTF-8?q?checkDataSource=E4=B9=8B=E5=90=8E=E5=85=B3?=
 =?UTF-8?q?=E9=97=ADconnection?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../service/impl/GenDatasourceConfServiceImpl.java     | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/pigx-visual/pigx-codegen/src/main/java/com/pig4cloud/pigx/codegen/service/impl/GenDatasourceConfServiceImpl.java b/pigx-visual/pigx-codegen/src/main/java/com/pig4cloud/pigx/codegen/service/impl/GenDatasourceConfServiceImpl.java
index 3eae2701c..a1ac26bff 100755
--- a/pigx-visual/pigx-codegen/src/main/java/com/pig4cloud/pigx/codegen/service/impl/GenDatasourceConfServiceImpl.java
+++ b/pigx-visual/pigx-codegen/src/main/java/com/pig4cloud/pigx/codegen/service/impl/GenDatasourceConfServiceImpl.java
@@ -30,6 +30,7 @@ import org.jasypt.encryption.StringEncryptor;
 import org.springframework.stereotype.Service;
 
 import javax.sql.DataSource;
+import java.sql.Connection;
 import java.sql.DriverManager;
 import java.sql.SQLException;
 
@@ -134,11 +135,18 @@ public class GenDatasourceConfServiceImpl extends ServiceImpl<GenDatasourceConfM
 	 */
 	@Override
 	public Boolean checkDataSource(GenDatasourceConf conf) {
+		Connection connection =null;
 		try {
-			DriverManager.getConnection(conf.getUrl(),conf.getUsername(),conf.getPassword());
+			connection = DriverManager.getConnection(conf.getUrl(), conf.getUsername(), conf.getPassword());
 		} catch (SQLException e) {
 			log.error("数据源配置 {} , 获取链接失败", conf.getName(), e);
 			return Boolean.FALSE;
+		}finally {
+			try {
+				connection.close();
+			} catch (Exception e) {
+				e.printStackTrace();
+			}
 		}
 		return Boolean.TRUE;
 	}