异步函数防止重入

y3-editor · Jul 4, 2024 · 4eec7c6 · 4eec7c6
1 parent de157a6
commit 4eec7c6
Show file tree

Hide file tree

Showing 3 changed files with 46 additions and 5 deletions.
diff --git a/src/editorTable/language.ts b/src/editorTable/language.ts
@@ -73,7 +73,7 @@ class Language extends vscode.Disposable {
         return hash(value);
     }
 
-    @throttle(1000)
+    @throttle(500)
     private updateFile() {
         let content = JSON.stringify(this._language, null, 4);
         y3.fs.writeFile(this.uri!, content);

diff --git a/src/env.ts b/src/env.ts
@@ -5,7 +5,7 @@ import path from 'path';
 import * as tools from './tools';
 import { Template } from './constants';
 import { isPathValid } from './utility';
-import { throttle } from './utility/decorators';
+import { throttle, queue } from './utility/decorators';
 
 type EditorVersion = '1.0' | '2.0' | 'unknown';
 
@@ -247,7 +247,7 @@ class Env {
         }, 100);
     }
 
-    @throttle(1000)
+    @queue()
     public async updateEditor(askUser = false) {
         let editorUri = await this.searchEditorUri(askUser);
         if (!editorUri) {
@@ -262,14 +262,15 @@ class Env {
         this.fireOnDidReload();
     }
 
+    @queue()
     public async editorReady(askUser = false) {
         if (this.editorUri) {
             return;
         }
         await this.updateEditor(askUser);
     }
 
-    @throttle(1000)
+    @queue()
     public async updateMap(search: boolean, askUser: boolean) {
         let mapUri = await this.searchProjectPath(search, askUser);
         if (!mapUri) {
@@ -294,6 +295,7 @@ class Env {
         this.fireOnDidReload();
     }
 
+    @queue()
     public async mapReady(askUser = false) {
         if (this.mapUri) {
             return;

diff --git a/src/utility/decorators.ts b/src/utility/decorators.ts
@@ -1,4 +1,3 @@
-
 // 每隔一段时间只能执行一次
 export function throttle(wait: number) {
     return function(target: any, propertyKey: string, descriptor: PropertyDescriptor) {
@@ -17,3 +16,43 @@ export function throttle(wait: number) {
         };
     };
 }
+
+// 请求需要排队，禁止重入（只能用于异步函数）
+export function queue() {
+    return function(target: any, propertyKey: string, descriptor: PropertyDescriptor) {
+        const originalMethod: (...args: any[]) => Promise<any> = descriptor.value;
+        let runningMap = new Map<any, (() => void)[]>();
+
+        descriptor.value = async function(...args: any[]) {
+            let queue = runningMap.get(this);
+            if (!queue) {
+                queue = [];
+                runningMap.set(this, queue);
+            }
+
+            let myTurn = new Promise<void>((resolve) => {
+                queue.push(resolve);
+            });
+
+            let promise = new Promise(async (resolve) => {
+                // 等待轮到自己
+                await myTurn;
+                // 执行真正的操作
+                let result = await originalMethod.apply(this, args);
+                // 返回当前的请求
+                resolve(result);
+                // 把自己移出队列
+                queue.shift();
+                // 处理下一个请求
+                queue[0]?.();
+            });
+
+            if (queue.length === 1) {
+                // 处理第一个请求
+                queue[0]?.();
+            }
+
+            return promise;
+        };
+    };
+}