优化代码

src/main/java/com/hisense/dataservice/bo/ColumnDataRangeFilter.java

@@ -117,6 +117,9 @@ public class ColumnDataRangeFilter {
 
     public static List<List<String>> generateExcelHead(List<ColumnDataRangeFilter> columnDataRangeFilters) {
         List<List<String>> result = new ArrayList<>();
+        if(CollectionUtils.isEmpty(columnDataRangeFilters)){
+            return result;
+        }
         for (ColumnDataRangeFilter columnDataRangeFilter : columnDataRangeFilters) {
             result.addAll(columnDataRangeFilter.generateExcelHead());
         }

src/main/java/com/hisense/dataservice/entity/DataApiDataSource.java

@@ -39,6 +39,9 @@ public class DataApiDataSource extends BaseEntity {
     // 数据驱动
     @Column(name = "driver", columnDefinition = "varchar2(255)", nullable = false)
     private String driver;
+    // 环境 0: 测试，1： 生产
+    @Column(name = "environment", columnDefinition = "number(1) default 0 ", nullable = false)
+    private Integer environment;
 
     public DataSourceVo toItem(){
         DataSourceVo dataSourceVo = new DataSourceVo();

src/main/java/com/hisense/dataservice/service/impl/DataApiServiceManagementServiceImpl.java

@@ -446,6 +446,7 @@ public class DataApiServiceManagementServiceImpl implements DataApiServiceManage
         queryDataTask.setDataApiDataSource(dataSource);
         queryDataTask.setQuerySql(dataSql);
         queryDataTask.setPageNumber(1L);
+        queryDataTask.setColumnHeads(ColumnDataRangeFilter.generateExcelHead(dataApiModelConfig.getColumnDataRangeConfig()));
         queryDataTask.setJdbcParamValues(jdbcParamValues);
 
         Future<?> future = debugQueryDataThreadPool.addExecuteTask(queryDataTask);