diff --git a/src/main/java/com/xydl/cac/controller/IcdConfigController.java b/src/main/java/com/xydl/cac/controller/IcdConfigController.java index 82188a5..854ab34 100644 --- a/src/main/java/com/xydl/cac/controller/IcdConfigController.java +++ b/src/main/java/com/xydl/cac/controller/IcdConfigController.java @@ -86,8 +86,8 @@ public class IcdConfigController extends BasicController { @GetMapping("tableList") @ApiOperation("查询data表名") - public Response> tableList() throws Exception { - List result = dataService.getDataTables(); + public Response> tableList() throws Exception { + List result = dataService.getDataTables(); return Response.success(result); } diff --git a/src/main/java/com/xydl/cac/service/DataService.java b/src/main/java/com/xydl/cac/service/DataService.java index ed289d4..0b60daf 100644 --- a/src/main/java/com/xydl/cac/service/DataService.java +++ b/src/main/java/com/xydl/cac/service/DataService.java @@ -9,7 +9,7 @@ import java.util.Map; public interface DataService { - List getDataTables() throws Exception; + List getDataTables() throws Exception; List getDataTableColumns(String tableName) throws Exception; diff --git a/src/main/java/com/xydl/cac/service/impl/DataServiceImpl.java b/src/main/java/com/xydl/cac/service/impl/DataServiceImpl.java index 1749eef..bb8a0c9 100644 --- a/src/main/java/com/xydl/cac/service/impl/DataServiceImpl.java +++ b/src/main/java/com/xydl/cac/service/impl/DataServiceImpl.java @@ -1,10 +1,12 @@ package com.xydl.cac.service.impl; +import com.xydl.cac.entity.ModevType; import com.xydl.cac.entity.NiecPoint; import com.xydl.cac.entity.NiecSensor; import com.xydl.cac.model.ColumnModel; import com.xydl.cac.model.ConditionModel; import com.xydl.cac.model.OnePage; +import com.xydl.cac.repository.ModevTypeRepository; import com.xydl.cac.service.DataService; import com.xydl.cac.util.DateUtil; import lombok.extern.slf4j.Slf4j; @@ -28,15 +30,26 @@ public class DataServiceImpl implements DataService { @Resource private JdbcTemplate jdbcTemplate; + @Resource + ModevTypeRepository modevTypeRepository; @Override - public List getDataTables() throws Exception { - List tables = new ArrayList<>(); + public List getDataTables() throws Exception { + List typeList = modevTypeRepository.findAll(); + List tables = new ArrayList<>(); String sql = "SHOW TABLES"; List list = jdbcTemplate.queryForList(sql, String.class); - for (String table : list) { - if (table.startsWith("data_")) { - tables.add(table); + for (String name : list) { + if (name.startsWith("data_")) { + ColumnModel item = new ColumnModel(); + item.setName(name); + for (ModevType type : typeList) { + if (name.equals(type.getTablename())) { + item.setComment(type.getMc()); + break; + } + } + tables.add(item); } } return tables;