报表中心

gic-platform-enterprise-service/src/main/java/com/gic/enterprise/service/outer/impl/HiveDataDownloadApiServiceImpl.java

@@ -15,6 +15,7 @@ import com.gic.enterprise.utils.FileUtils;
 import com.gic.enterprise.utils.HiveConnUtils;
 import com.gic.thirdparty.BucketNameEnum;
 import com.gic.thirdparty.FileOperateUtils;
+import org.apache.commons.lang3.StringUtils;
 import org.apache.commons.lang3.concurrent.BasicThreadFactory;
 import org.apache.logging.log4j.LogManager;
 import org.apache.logging.log4j.Logger;
@@ -88,7 +89,7 @@ public class HiveDataDownloadApiServiceImpl implements HiveDataDownloadApiServic
         downloadTaskPOJO.setCanDownload(!DownloadReportDataTypeEnum.COMPLETE_DATA.getCode().equals(hiveDataDownloadDTO.getDataType()));
         downloadTaskPOJO.setDesenField(hiveDataDownloadDTO.getDesenField());
         downloadTaskPOJO.setExcelExtension(hiveDataDownloadDTO.getExcelExtension());
-        downloadTaskPOJO.setTableName(hiveDataDownloadDTO.getSql().substring(hiveDataDownloadDTO.getSql().toLowerCase().indexOf("from")+4, hiveDataDownloadDTO.getSql().toLowerCase().indexOf("where")));
+        downloadTaskPOJO.setTableName(getTableName(hiveDataDownloadDTO.getSql()));
         taskList.add(downloadTaskPOJO);
     }
 
@@ -174,4 +175,15 @@ public class HiveDataDownloadApiServiceImpl implements HiveDataDownloadApiServic
         }, interval*1000, interval*1000, TimeUnit.MILLISECONDS);
     }
 
+    private String getTableName(String sql){
+        if(StringUtils.isBlank(sql)){
+            return null;
+        }
+        if(sql.contains("where")){
+            return sql.substring(sql.toLowerCase().indexOf("from")+4, sql.toLowerCase().indexOf("where"));
+        } else {
+            return sql.substring(sql.toLowerCase().indexOf("from")+4);
+        }
+    }
+
 }