Skip to content

Commit b9de400

Browse files
author
JeongSeok Hong
authored
fix: Patch to resolve issue #13 (#14)
1 parent 22bfb59 commit b9de400

File tree

1 file changed

+22
-4
lines changed

1 file changed

+22
-4
lines changed

app.js

Lines changed: 22 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -46,14 +46,21 @@ app.get('/health', (req, res) => {
4646

4747
app.post('/api/classify', async (req, res, next) => {
4848
let imageFile;
49+
let temp
4950
if (req.body.image_file) {
5051
imageFile = fs.createReadStream(`public/${req.body.image_file}`);
5152
}
5253
else if (req.body.image_data) {
53-
const resource = parseBase64Image(req.body.image_data);
54-
const temp = path.join(os.tmpdir(), `${uuid.v4()}.${resource.type}`);
55-
fs.writeFileSync(temp, resource.data);
56-
imageFile = fs.createReadStream(temp);
54+
try {
55+
const resource = parseBase64Image(req.body.image_data);
56+
temp = path.join(os.tmpdir(), `${uuid.v4()}.${resource.type}`);
57+
fs.writeFileSync(temp, resource.data);
58+
imageFile = fs.createReadStream(temp);
59+
} catch (err) {
60+
console.error('Error creating image file: ', err);
61+
next(err);
62+
return;
63+
}
5764
}
5865
const classifyParams = {
5966
imagesFile: imageFile,
@@ -74,6 +81,17 @@ app.post('/api/classify', async (req, res, next) => {
7481
next(error);
7582
}
7683
next(err);
84+
} finally {
85+
if (temp) {
86+
fs.unlink(temp, (err) => {
87+
if (err) {
88+
console.error(err)
89+
return
90+
}
91+
92+
//file removed
93+
})
94+
}
7795
}
7896
});
7997

0 commit comments

Comments
 (0)