From 8de01e2860954e2a74147637f4db106a675e712f Mon Sep 17 00:00:00 2001 From: hxuanyu Date: Thu, 4 Jun 2020 13:25:17 +1000 Subject: [PATCH] minor code refactor --- .../persistence/mongodb/GridFsManager.java | 32 +++++++++---------- 1 file changed, 15 insertions(+), 17 deletions(-) diff --git a/oh-my-scheduler-server/src/main/java/com/github/kfcfans/oms/server/persistence/mongodb/GridFsManager.java b/oh-my-scheduler-server/src/main/java/com/github/kfcfans/oms/server/persistence/mongodb/GridFsManager.java index faaf6902..55ff19c4 100644 --- a/oh-my-scheduler-server/src/main/java/com/github/kfcfans/oms/server/persistence/mongodb/GridFsManager.java +++ b/oh-my-scheduler-server/src/main/java/com/github/kfcfans/oms/server/persistence/mongodb/GridFsManager.java @@ -60,12 +60,11 @@ public class GridFsManager { * @throws IOException 异常 */ public void store(File localFile, String bucketName, String fileName) throws IOException { - if (db == null) { - return; - } - GridFSBucket bucket = getBucket(bucketName); - try (BufferedInputStream bis = new BufferedInputStream(new FileInputStream(localFile))) { - bucket.uploadFromStream(fileName, bis); + if (available()) { + GridFSBucket bucket = getBucket(bucketName); + try (BufferedInputStream bis = new BufferedInputStream(new FileInputStream(localFile))) { + bucket.uploadFromStream(fileName, bis); + } } } @@ -77,18 +76,17 @@ public class GridFsManager { * @throws IOException 异常 */ public void download(File targetFile, String bucketName, String fileName) throws IOException { - if (db == null) { - return; - } - GridFSBucket bucket = getBucket(bucketName); - byte[] buffer = new byte[1024]; - try (GridFSDownloadStream gis = bucket.openDownloadStream(fileName); - BufferedOutputStream bos = new BufferedOutputStream(new FileOutputStream(targetFile)) - ) { - while (gis.read(buffer) != -1) { - bos.write(buffer); + if (available()) { + GridFSBucket bucket = getBucket(bucketName); + byte[] buffer = new byte[1024]; + try (GridFSDownloadStream gis = bucket.openDownloadStream(fileName); + BufferedOutputStream bos = new BufferedOutputStream(new FileOutputStream(targetFile)) + ) { + while (gis.read(buffer) != -1) { + bos.write(buffer); + } + bos.flush(); } - bos.flush(); } }