From 75ee368c96a3db085c51a6ee32efb3f341a8333e Mon Sep 17 00:00:00 2001
From: 648540858 <648540858@qq.com>
Date: 星期二, 25 一月 2022 20:21:36 +0800
Subject: [PATCH] 优化默认节点的展示
---
src/main/java/com/genersoft/iot/vmp/vmanager/streamPush/StreamPushController.java | 81 +++++++++++++++++++++++++++-------------
1 files changed, 55 insertions(+), 26 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 c0fafd3..3ea207a 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
@@ -118,13 +118,12 @@
@ResponseBody
public DeferredResult<ResponseEntity<WVPResult<Object>>> uploadChannelFile(@RequestParam(value = "file") MultipartFile file){
-
// 鏈�澶氬鐞嗘枃浠朵竴涓皬鏃�
DeferredResult<ResponseEntity<WVPResult<Object>>> result = new DeferredResult<>(60*60*1000L);
// 褰曞儚鏌ヨ浠hannelId浣滀负deviceId鏌ヨ
String key = DeferredResultHolder.UPLOAD_FILE_CHANNEL;
String uuid = UUID.randomUUID().toString();
-
+ logger.info("閫氶亾瀵煎叆鏂囦欢绫诲瀷: {}",file.getContentType() );
if (file.isEmpty()) {
logger.warn("閫氶亾瀵煎叆鏂囦欢涓虹┖");
WVPResult<Object> wvpResult = new WVPResult<>();
@@ -133,6 +132,23 @@
result.setResult(ResponseEntity.status(HttpStatus.BAD_REQUEST).body(wvpResult));
return result;
}
+ if (file.getContentType() == null) {
+ 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("宸叉湁瀵煎叆浠诲姟姝e湪鎵ц");
@@ -162,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