From 1dbb559b388831b5f15e690d0b33d24b1faaa983 Mon Sep 17 00:00:00 2001
From: 648540858 <648540858@qq.com>
Date: 星期二, 25 一月 2022 15:43:52 +0800
Subject: [PATCH] 优化文件导入判断
---
src/main/java/com/genersoft/iot/vmp/vmanager/streamPush/StreamPushController.java | 83 ++++++++++++++++++++++++-----------------
1 files changed, 48 insertions(+), 35 deletions(-)
diff --git a/src/main/java/com/genersoft/iot/vmp/vmanager/streamPush/StreamPushController.java b/src/main/java/com/genersoft/iot/vmp/vmanager/streamPush/StreamPushController.java
index 9111844..c58bd60 100644
--- a/src/main/java/com/genersoft/iot/vmp/vmanager/streamPush/StreamPushController.java
+++ b/src/main/java/com/genersoft/iot/vmp/vmanager/streamPush/StreamPushController.java
@@ -123,7 +123,7 @@
// 褰曞儚鏌ヨ浠hannelId浣滀负deviceId鏌ヨ
String key = DeferredResultHolder.UPLOAD_FILE_CHANNEL;
String uuid = UUID.randomUUID().toString();
-
+ logger.warn("閫氶亾瀵煎叆鏂囦欢绫诲瀷: {}",file.getContentType() );
if (file.isEmpty()) {
logger.warn("閫氶亾瀵煎叆鏂囦欢涓虹┖");
WVPResult<Object> wvpResult = new WVPResult<>();
@@ -139,16 +139,16 @@
result.setResult(ResponseEntity.status(HttpStatus.BAD_REQUEST).body(wvpResult));
return result;
}
- if (!file.getContentType().endsWith(".xls")
- && !file.getContentType().endsWith(".csv")
- && !file.getContentType().endsWith(".xlsx") ) {
- logger.warn("閫氶亾瀵煎叆鏂囦欢绫诲瀷閿欒");
- WVPResult<Object> wvpResult = new WVPResult<>();
- wvpResult.setCode(-1);
- wvpResult.setMsg("鏂囦欢绫诲瀷閿欒锛岃浣跨敤");
- result.setResult(ResponseEntity.status(HttpStatus.BAD_REQUEST).body(wvpResult));
- return result;
- }
+// if (!file.getContentType().endsWith(".xls")
+// && !file.getContentType().endsWith(".csv")
+// && !file.getContentType().endsWith(".xlsx") ) {
+// logger.warn("閫氶亾瀵煎叆鏂囦欢绫诲瀷閿欒: {}",file.getContentType() );
+// WVPResult<Object> wvpResult = new WVPResult<>();
+// wvpResult.setCode(-1);
+// wvpResult.setMsg("鏂囦欢绫诲瀷閿欒锛岃浣跨敤");
+// result.setResult(ResponseEntity.status(HttpStatus.BAD_REQUEST).body(wvpResult));
+// return result;
+// }
// 鍚屾椂鍙鐞嗕竴涓枃浠�
if (resultHolder.exist(key, null)) {
logger.warn("宸叉湁瀵煎叆浠诲姟姝ｅ湪鎵ц");
@@ -178,30 +178,43 @@
} catch (IOException e) {
e.printStackTrace();
}
- //浼犲叆鍙傛暟
- ExcelReader excelReader = EasyExcel.read(inputStream, StreamPushExcelDto.class,
- new StreamPushUploadFileHandler(streamPushService, mediaServerService.getDefaultMediaServer().getId(), (errorStreams, errorGBs)->{
- logger.info("閫氶亾瀵煎叆鎴愬姛锛屽瓨鍦ㄩ噸澶岮pp+Stream涓簕}涓紝瀛樺湪鍥芥爣ID涓簕}涓�", errorStreams.size(), errorGBs.size());
- RequestMessage msg = new RequestMessage();
- msg.setKey(key);
- WVPResult<Map<String, List<String>>> wvpResult = new WVPResult<>();
- if (errorStreams.size() == 0 && errorGBs.size() == 0) {
- wvpResult.setCode(0);
- wvpResult.setMsg("鎴愬姛");
- }else {
- wvpResult.setCode(1);
- wvpResult.setMsg("瀵煎叆鎴愬姛銆備絾鏄瓨鍦ㄩ噸澶嶆暟鎹�");
- Map<String, List<String>> errorData = new HashMap<>();
- errorData.put("gbId", errorGBs);
- errorData.put("stream", errorStreams);
- wvpResult.setData(errorData);
- }
- msg.setData(wvpResult);
- resultHolder.invokeAllResult(msg);
- })).build();
- ReadSheet readSheet = EasyExcel.readSheet(0).build();
- excelReader.read(readSheet);
- excelReader.finish();
+ try {
+ //浼犲叆鍙傛暟
+ ExcelReader excelReader = EasyExcel.read(inputStream, StreamPushExcelDto.class,
+ new StreamPushUploadFileHandler(streamPushService, mediaServerService.getDefaultMediaServer().getId(), (errorStreams, errorGBs)->{
+ logger.info("閫氶亾瀵煎叆鎴愬姛锛屽瓨鍦ㄩ噸澶岮pp+Stream涓簕}涓紝瀛樺湪鍥芥爣ID涓簕}涓�", errorStreams.size(), errorGBs.size());
+ RequestMessage msg = new RequestMessage();
+ msg.setKey(key);
+ WVPResult<Map<String, List<String>>> wvpResult = new WVPResult<>();
+ if (errorStreams.size() == 0 && errorGBs.size() == 0) {
+ wvpResult.setCode(0);
+ wvpResult.setMsg("鎴愬姛");
+ }else {
+ wvpResult.setCode(1);
+ wvpResult.setMsg("瀵煎叆鎴愬姛銆備絾鏄瓨鍦ㄩ噸澶嶆暟鎹�");
+ Map<String, List<String>> errorData = new HashMap<>();
+ errorData.put("gbId", errorGBs);
+ errorData.put("stream", errorStreams);
+ wvpResult.setData(errorData);
+ }
+ msg.setData(wvpResult);
+ resultHolder.invokeAllResult(msg);
+ })).build();
+ ReadSheet readSheet = EasyExcel.readSheet(0).build();
+ excelReader.read(readSheet);
+ excelReader.finish();
+ }catch (Exception e) {
+ logger.warn("閫氶亾瀵煎叆澶辫触锛�", e);
+ RequestMessage msg = new RequestMessage();
+ msg.setKey(key);
+ WVPResult<Object> wvpResult = new WVPResult<>();
+ wvpResult.setCode(-1);
+ wvpResult.setMsg("閫氶亾瀵煎叆澶辫触: " + e.getMessage() );
+ msg.setData(wvpResult);
+ resultHolder.invokeAllResult(msg);
+ }
+
+
return result;
}
--
Gitblit v1.8.0