From 36064fae90cac9920bd753dcc342db83ffd1dff9 Mon Sep 17 00:00:00 2001
From: fengshuonan <sn93@qq.com>
Date: Thu, 27 May 2021 10:43:17 +0800
Subject: [PATCH] =?UTF-8?q?=E3=80=907.0.4=E3=80=91=E3=80=90ds=E3=80=91?=
 =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E4=B8=80=E4=B8=AA=E5=88=A0=E9=99=A4=E6=8E=A5?=
 =?UTF-8?q?=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../roses/kernel/dsctn/api/DataSourceApi.java     |  9 +++++++++
 .../service/impl/DatabaseInfoServiceImpl.java     | 15 +++++++++++++++
 2 files changed, 24 insertions(+)

diff --git a/kernel-d-ds-container/ds-container-api/src/main/java/cn/stylefeng/roses/kernel/dsctn/api/DataSourceApi.java b/kernel-d-ds-container/ds-container-api/src/main/java/cn/stylefeng/roses/kernel/dsctn/api/DataSourceApi.java
index 62b059cff..f20c4752a 100644
--- a/kernel-d-ds-container/ds-container-api/src/main/java/cn/stylefeng/roses/kernel/dsctn/api/DataSourceApi.java
+++ b/kernel-d-ds-container/ds-container-api/src/main/java/cn/stylefeng/roses/kernel/dsctn/api/DataSourceApi.java
@@ -32,4 +32,13 @@ public interface DataSourceApi {
      */
     void add(DatabaseInfoRequest databaseInfoRequest);
 
+    /**
+     * 通过数据源编码删除数据源
+     *
+     * @param datasourceCode 数据源编码
+     * @author fengshuonan
+     * @date 2021/5/27 10:06
+     */
+    void deleteByDatasourceCode(String datasourceCode);
+
 }
diff --git a/kernel-d-ds-container/ds-container-business/src/main/java/cn/stylefeng/roses/kernel/dsctn/modular/service/impl/DatabaseInfoServiceImpl.java b/kernel-d-ds-container/ds-container-business/src/main/java/cn/stylefeng/roses/kernel/dsctn/modular/service/impl/DatabaseInfoServiceImpl.java
index cc6015ec3..4294316e6 100644
--- a/kernel-d-ds-container/ds-container-business/src/main/java/cn/stylefeng/roses/kernel/dsctn/modular/service/impl/DatabaseInfoServiceImpl.java
+++ b/kernel-d-ds-container/ds-container-business/src/main/java/cn/stylefeng/roses/kernel/dsctn/modular/service/impl/DatabaseInfoServiceImpl.java
@@ -84,6 +84,21 @@ public class DatabaseInfoServiceImpl extends ServiceImpl<DatabaseInfoMapper, Dat
         addDataSourceToContext(entity, false);
     }
 
+    @Override
+    @Transactional(rollbackFor = Exception.class)
+    public void deleteByDatasourceCode(String datasourceCode) {
+
+        LambdaQueryWrapper<DatabaseInfo> wrapper = new LambdaQueryWrapper<>();
+        wrapper.eq(DatabaseInfo::getDbName, datasourceCode);
+        DatabaseInfo databaseInfo = this.getOne(wrapper, false);
+
+        // 删除数据源信息
+        this.removeById(databaseInfo.getDbId());
+
+        // 删除容器中的数据源记录
+        DataSourceContext.removeDataSource(datasourceCode);
+    }
+
     @Override
     @Transactional(rollbackFor = Exception.class)
     public void del(DatabaseInfoRequest databaseInfoRequest) {