judge: add retry to postFile

hydro-dev · Jan 2, 2025 · 07165b1 · 07165b1
1 parent 566ebd9
commit 07165b1
Showing 1 changed file with 14 additions and 5 deletions.
diff --git a/packages/hydrojudge/src/hosts/hydro.ts b/packages/hydrojudge/src/hosts/hydro.ts
@@ -116,11 +116,20 @@ export default class Hydro implements Session {
         return target;
     }
 
-    async postFile(target: string, filename: string, file: string) {
-        await this.post('judge/upload')
-            .field('rid', target)
-            .field('name', filename)
-            .attach('file', fs.createReadStream(file));
+    async postFile(target: string, filename: string, file: string, retry = 3) {
+        try {
+            await this.post('judge/upload')
+                .field('rid', target)
+                .field('name', filename)
+                .attach('file', await fs.readFile(file));
+        } catch (e) {
+            if (!retry) {
+                log.error('PostFile Fail: %s %s %o', target, filename, e);
+                throw e;
+            }
+            await new Promise((resolve) => { setTimeout(resolve, 1000); });
+            await this.postFile(target, filename, file, retry - 1);
+        }
     }
 
     getLang(name: string, doThrow = true) {