Weishen/Hydro
2
0 Fork 0
mirror of https://gitcode.com/gh_mirrors/hy/Hydro.git synced 2026-10-20 01:38:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

为model.file添加错误处理

Browse Source
This commit is contained in:
undefined committed 2020-07-15 22:10:31 +08:00
1 parent 8f6968bbbd
commit 9428efb156
2 files changed
+16 -6
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

hydro/model/file.ts
+14 -4
View File
@@ -69,7 +69,7 @@ export async function dec(_id: ObjectID): Promise<number> {
return file.value.count;
}
export async function getWithSecret(_id: ObjectID, secret: string) {
export async function getWithSecret(_id: ObjectID, secret: string, reject?: Function) {
const file = await coll.findOne({ _id });
if (!file) throw new NotFoundError(_id);
const timestamp = _timestamp();
@@ -78,13 +78,23 @@ export async function getWithSecret(_id: ObjectID, secret: string) {
throw new ForbiddenError();
}
}
return gridfs.openDownloadStream(_id);
const stream = gridfs.openDownloadStream(_id);
stream.on('error', (err) => {
console.error(err);
if (reject) reject();
});
return stream;
}
export async function get(_id: ObjectID) {
export async function get(_id: ObjectID, reject?: Function) {
const file = await coll.findOne({ _id });
if (!file) throw new NotFoundError(_id);
return gridfs.openDownloadStream(_id);
const stream = gridfs.openDownloadStream(_id);
stream.on('error', (err) => {
console.error(err);
if (reject) reject();
});
return stream;
}
export function getMeta(_id: ObjectID): Promise<Ufdoc> {