third_codes/PowerJob
1
0
Fork 1
mirror of https://github.com/PowerJob/PowerJob.git synced 2025-07-17 00:00:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

minor code refactor

Browse Source
This commit is contained in:
hxuanyu 2020-06-04 13:25:17 +10:00
parent d999f18f93
commit 8de01e2860
1 changed files with 15 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
oh-my-scheduler-server/src/main/java/com/github/kfcfans/oms/server/persistence/mongodb/GridFsManager.java
View File

@ -60,12 +60,11 @@ public class GridFsManager {
* @throws IOException 异常 * @throws IOException 异常
*/ */
public void store(File localFile, String bucketName, String fileName) throws IOException { public void store(File localFile, String bucketName, String fileName) throws IOException {
if (db == null) { if (available()) {
return; GridFSBucket bucket = getBucket(bucketName);
} try (BufferedInputStream bis = new BufferedInputStream(new FileInputStream(localFile))) {
GridFSBucket bucket = getBucket(bucketName); bucket.uploadFromStream(fileName, bis);
try (BufferedInputStream bis = new BufferedInputStream(new FileInputStream(localFile))) { }
bucket.uploadFromStream(fileName, bis);
} }
} }
@ -77,18 +76,17 @@ public class GridFsManager {
* @throws IOException 异常 * @throws IOException 异常
*/ */
public void download(File targetFile, String bucketName, String fileName) throws IOException { public void download(File targetFile, String bucketName, String fileName) throws IOException {
if (db == null) { if (available()) {
return; GridFSBucket bucket = getBucket(bucketName);
} byte[] buffer = new byte[1024];
GridFSBucket bucket = getBucket(bucketName); try (GridFSDownloadStream gis = bucket.openDownloadStream(fileName);
byte[] buffer = new byte[1024]; BufferedOutputStream bos = new BufferedOutputStream(new FileOutputStream(targetFile))
try (GridFSDownloadStream gis = bucket.openDownloadStream(fileName); ) {
BufferedOutputStream bos = new BufferedOutputStream(new FileOutputStream(targetFile)) while (gis.read(buffer) != -1) {
) { bos.write(buffer);
while (gis.read(buffer) != -1) { }
bos.write(buffer); bos.flush();
} }
bos.flush();
} }
} }