Browse Source

保存算法文件添加try-catch

Huhu 3 years ago
parent
commit
9e5560a83f

+ 24 - 8
src/main/java/io/renren/modules/sys/controller/algs/algTrainController.java

@@ -300,9 +300,17 @@ public class algTrainController {
         String tensorboardFile=entry.getFilename();
         Long fileSize=entry.getAttrs().getSize();
         System.out.println(tensorboardFile+","+fileSize);
-        InputStream inputStream=FTPUtils.downloadFile("/opt/uploadFile/algTrain"+algorithmTrainingId+"/testTensorboard/"+tensorboardFile);
-        MinIoUtils.uploadFileByInputStream(inputStream,fileSize,"tensorboardfile","tensorboardFile"+algorithmTrainingId+"/"+tensorboardFile);
-        inputStream.close();//记得关闭inputStream 不然后续会报错
+        InputStream inputStream=null;
+        try {
+            inputStream=FTPUtils.downloadFile("/opt/uploadFile/algTrain"+algorithmTrainingId+"/testTensorboard/"+tensorboardFile);
+            MinIoUtils.uploadFileByInputStream(inputStream,fileSize,"tensorboardfile","tensorboardFile"+algorithmTrainingId+"/"+tensorboardFile);
+            inputStream.close();//记得关闭inputStream 不然后续会报错
+        }catch (Exception e){
+            e.printStackTrace();
+        }finally {
+            inputStream.close();
+        }
+
         return R.ok();
     }
 
@@ -468,11 +476,19 @@ public class algTrainController {
             String fileName=vector.get(i).getFilename();
             Long fileSize=vector.get(i).getAttrs().getSize();
             if(models.contains(fileName)||pythonFiles.contains(fileName)){
-                InputStream inputStream=FTPUtils.downloadFile("/opt/uploadFile/algTrain"+algorithmTrainingId+"/"+fileName);
-                //同时上传到publishedalgorithm桶和algmodel桶
-                MinIoUtils.uploadFileByInputStream(inputStream,fileSize,"publishedalgorithm","algorithm"+algorithmTrainingId+"/"+fileName);
-                MinIoUtils.copyFile("algmodel","model"+modelId+"/"+fileName,"publishedalgorithm","algorithm"+algorithmTrainingId+"/"+fileName);
-                inputStream.close();
+                InputStream inputStream=null;
+                try {
+                    inputStream=FTPUtils.downloadFile("/opt/uploadFile/algTrain"+algorithmTrainingId+"/"+fileName);
+                    //同时上传到publishedalgorithm桶和algmodel桶
+                    MinIoUtils.uploadFileByInputStream(inputStream,fileSize,"publishedalgorithm","algorithm"+algorithmTrainingId+"/"+fileName);
+                    MinIoUtils.copyFile("algmodel","model"+modelId+"/"+fileName,"publishedalgorithm","algorithm"+algorithmTrainingId+"/"+fileName);
+                    inputStream.close();
+                }catch (Exception e){
+                    e.printStackTrace();
+                }finally {
+                    inputStream.close();
+                }
+
             }
         }