improved listen to create for tasks

cvat-ai · May 23, 2024 · 8b02cec · 8b02cec
1 parent 358e046
commit 8b02cec
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 8 deletions.
diff --git a/cvat-core/src/requests-manager.ts b/cvat-core/src/requests-manager.ts
@@ -278,12 +278,16 @@ class RequestsManager {
                 });
             };
 
-            this.listening[storedID] = {
-                onUpdate: callback ? [callback] : [],
-                timeout: window.setTimeout(timeoutCallback),
-                request: initialRequest,
-                requestDelayIdx: 0,
-            };
+            if (initialRequest?.status === RQStatus.FAILED) {
+                reject(new RequestError(initialRequest?.message));
+            } else {
+                this.listening[storedID] = {
+                    onUpdate: callback ? [callback] : [],
+                    timeout: window.setTimeout(timeoutCallback),
+                    request: initialRequest,
+                    requestDelayIdx: 0,
+                };
+            }
         });
 
         this.listening[storedID] = {

diff --git a/cvat-core/src/session-implementation.ts b/cvat-core/src/session-implementation.ts
@@ -516,10 +516,11 @@ export function implementTask(Task) {
     };
 
     Task.prototype.listenToCreate.implementation = async function (
+        rqID,
         options,
     ): Promise<TaskClass> {
         if (Number.isInteger(this.id) && this.size === 0) {
-            const request = await requestsManager.listen(null, options);
+            const request = await requestsManager.listen(rqID, options);
             const [serializedTask] = await serverProxy.tasks.get({ id: request.operation.taskID });
             return new Task(omit(serializedTask, ['labels', 'jobs']));
         }

diff --git a/cvat-core/src/session.ts b/cvat-core/src/session.ts
@@ -1072,9 +1072,10 @@ export class Task extends Session {
     }
 
     async listenToCreate(
+        rqID,
         options,
     ): Promise<Task> {
-        const result = await PluginRegistry.apiWrapper.call(this, Task.prototype.listenToCreate, options);
+        const result = await PluginRegistry.apiWrapper.call(this, Task.prototype.listenToCreate, rqID, options);
         return result;
     }
 

diff --git a/cvat-ui/src/components/tasks-page/task-item.tsx b/cvat-ui/src/components/tasks-page/task-item.tsx
@@ -81,6 +81,7 @@ class TaskItemComponent extends React.PureComponent<TaskItemProps & RouteCompone
                         });
                     }
                 },
+                initialRequest: activeRequest,
             },
             ).then((createdTask: Task) => {
                 if (!this.#isUnmounted) {
-Original file line number
+Diff line change
@@ Expand Up @@
                             });
                         }
                     },
+                    initialRequest: activeRequest,
                 },
                 ).then((createdTask: Task) => {
                     if (!this.#isUnmounted) {
@@ Expand Down @@