chore(trace): Support tasks trace

Author: LucienShui

src/tasks/dto/task.dto.ts

@@ -17,26 +17,57 @@ function getTaskStatus(task: Task): string {
   return 'pending';
 }
 
-export class TaskMetaDto {
+class TaskDtoBase {
   id: string;
   namespace_id: string;
   user_id: string;
   priority: number;
   function: string;
-  attrs: Record<string, any> | null;
   status: string;
   created_at: string;
   updated_at: string;
   started_at: string | null;
   ended_at: string | null;
   canceled_at: string | null;
 
-  protected static setValue(obj: TaskMetaDto, task: Task) {
+  protected static setValue(obj: TaskDtoBase, task: Task) {
     obj.id = task.id;
     obj.namespace_id = task.namespaceId;
     obj.user_id = task.userId;
     obj.priority = task.priority;
     obj.function = task.function;
+    obj.status = getTaskStatus(task);
+    obj.created_at = task.createdAt.toISOString();
+    obj.updated_at = task.updatedAt.toISOString();
+    obj.started_at = task.startedAt?.toISOString() || null;
+    obj.ended_at = task.endedAt?.toISOString() || null;
+    obj.canceled_at = task.canceledAt?.toISOString() || null;
+  }
+}
+
+export class InternalTaskDto extends TaskDtoBase {
+  payload: Record<string, any> | null;
+  input: Record<string, any>;
+  output: Record<string, any> | null;
+  exception: Record<string, any> | null;
+
+  static fromEntity(task: Task): InternalTaskDto {
+    const dto = new InternalTaskDto();
+    this.setValue(dto, task);
+
+    dto.payload = task.payload;
+    dto.input = task.input;
+    dto.output = task.output;
+    dto.exception = task.exception;
+    return dto;
+  }
+}
+
+export class TaskMetaDto extends TaskDtoBase {
+  attrs: Record<string, any> | null;
+
+  protected static setValue(obj: TaskMetaDto, task: Task) {
+    super.setValue(obj, task);
 
     if (task.payload) {
       obj.attrs = {};
@@ -46,13 +77,6 @@ export class TaskMetaDto {
         }
       }
     }
-
-    obj.status = getTaskStatus(task);
-    obj.created_at = task.createdAt.toISOString();
-    obj.updated_at = task.updatedAt.toISOString();
-    obj.started_at = task.startedAt?.toISOString() || null;
-    obj.ended_at = task.endedAt?.toISOString() || null;
-    obj.canceled_at = task.canceledAt?.toISOString() || null;
   }
 
   static fromEntity(task: Task): TaskMetaDto {

src/wizard/wizard.service.ts

@@ -19,7 +19,7 @@ import { ResourceType } from 'omniboxd/resources/resources.entity';
 import { AttachmentsService } from 'omniboxd/attachments/attachments.service';
 import { WizardTaskService } from 'omniboxd/tasks/wizard-task.service';
 import { Image, ProcessedImage } from 'omniboxd/wizard/types/wizard.types';
-import { TaskDto } from 'omniboxd/tasks/dto/task.dto';
+import { InternalTaskDto } from 'omniboxd/tasks/dto/task.dto';
 
 @Injectable()
 export class WizardService {
@@ -227,7 +227,7 @@ export class WizardService {
     task.output.images = processedImages;
   }
 
-  async fetchTask(): Promise<TaskDto | null> {
+  async fetchTask(): Promise<InternalTaskDto | null> {
     const rawQuery = `
       WITH running_tasks_sub_query AS (SELECT namespace_id,
                                               COUNT(id) AS running_count
@@ -271,7 +271,7 @@ export class WizardService {
         namespaceId: record.namespace_id,
       });
       const newTask = await this.wizardTaskService.taskRepository.save(task);
-      return TaskDto.fromEntity(newTask);
+      return InternalTaskDto.fromEntity(newTask);
     }
 
     return null;