'commit'

src/main/java/com/dsideal/base/YunXiao/Controller/YunXiaoController.java

@@ -22,39 +22,28 @@ public class YunXiaoController extends Controller {
     YunXiaoModel ym = new YunXiaoModel();
 
     /**
-     * 可以维护的数据集名称
+     * 获取数据集树
      */
-    // http://10.10.21.20:9000/dsBase/yx/getDataSet
-    @Before({GET.class})
+    @Before(GET.class)
     @IsLoginInterface({})
-    public void getDataSet() {
-        List<Record> list = ym.getDataSet();
+    public void datasetTree() {
+        List<Record> list = ym.datasetTree();
         renderJson(CommonUtil.renderJsonForLayUI(list));
     }
 
     /**
      * 获取数据集下的数据表
-     *
-     * @param id         数据集id
-     * @param pageNumber 第几页
-     * @param keyword    关键字
-     * @param pageSize   每页多少条数据
+     * @param datasetId 数据集id
+     * @param page 第几页
+     * @param limit 每页显示多少条数据
      */
     @Before(GET.class)
     @IsLoginInterface({})
-    @IsNumericInterface({"id"})
-    public void getDataSetContent(int id, String keyword, int pageNumber, int pageSize) {
-        if (StrKit.isBlank(keyword)) keyword = "";
-        if (pageNumber == 0) pageNumber = 1;
-        if (pageSize == 0) pageSize = 20;
-        if (SqlInjectionUtils.hasSqlInjectionRisk(keyword)) {
-            renderJson("输入的查询关键字存在SQL注入攻击，无法执行！");
-            return;
+    public void getDataSetContent(long datasetId, int page, int limit) {
+        Page<Record> p = ym.getDataSetContent(datasetId, page, limit);
+        renderJson(CommonUtil.renderJsonForLayUI(p));
     }
 
-        Page<Record> pageList = ym.getDataSetContent(id, keyword, pageNumber, pageSize);
-        renderJson(CommonUtil.renderJsonForLayUI(pageList));
-    }
 
     /**
      * 保存数据集下的数据表

src/main/java/com/dsideal/base/YunXiao/Model/YunXiaoModel.java

@@ -226,14 +226,14 @@ public class YunXiaoModel {
     /**
      * 获取数据集对应的表
      *
-     * @param id 数据集id
      * @return
      */
-    public Page<Record> getDataSetContent(int id, String keyword, int pageNumber, int pageSize) {
-        Record record = Db.findById("t_dp_yx_dataset", "id", id);
+    public Page<Record> getDataSetContent(long datasetId, int pageNumber, int pageSize) {
+        String sql = "select * from t_dp_yx_dataset where dataset_group_id=?";
+        Record record = Db.findFirst(sql, datasetId);
         String tableName = record.getStr("table_name");
         Page<Record> p = Db.paginate(pageNumber, pageSize,
-                "SELECT *", "from " + DB_NAME + ".`" + tableName + "` where `行政区划` like '%" + keyword + "%'");
+                "SELECT *", "from " + DB_NAME + ".`" + tableName);
         return p;
     }
 
@@ -256,4 +256,14 @@ public class YunXiaoModel {
         if (parent_id == null) return null;
         return getAreaById(parent_id).getStr("area_name");
     }
+
+    /**
+     * 获取数据集树结构
+     *
+     * @return
+     */
+    public List<Record> datasetTree() {
+        SqlPara sqlPara = Db.getSqlPara("YunXiao.datasetTree");
+        return Db.find(sqlPara);
+    }
 }

src/main/resources/Sql/YunXiao.sql

@@ -0,0 +1,16 @@
+#namespace("YunXiao")
+    #sql("datasetTree")
+          WITH RECURSIVE tree_cte AS (
+          SELECT *
+          FROM dataease.core_dataset_group
+          WHERE id = 1072161072829566976 -- 上线
+          UNION ALL
+          SELECT c.*
+          FROM dataease.core_dataset_group c
+          INNER JOIN tree_cte t ON c.pid = t.id
+      )
+    SELECT t1.id  ,t1.name ,t1.pid,t1.node_type,t2.name as parent_name FROM tree_cte as t1
+        inner join dataease.core_dataset_group as t2
+         on t1.pid=t2.id
+    #end
+#end