feat: init task type.

cnpm · Jul 26, 2023 · 4458d05 · 4458d05
1 parent 55ea2d9
commit 4458d05
Show file tree

Hide file tree

Showing 3 changed files with 93 additions and 1 deletion.
diff --git a/app/common/enum/Task.ts b/app/common/enum/Task.ts
@@ -2,7 +2,7 @@ export enum TaskType {
   SyncPackage = 'sync_package',
   ChangesStream = 'changes_stream',
   SyncBinary = 'sync_binary',
-  UpdateProxy = 'update_proxy',
+  UpdateProxyCache = 'update_proxy_cache',
   CreateHook = 'create_hook',
   TriggerHook = 'trigger_hook',
 }

diff --git a/app/core/entity/Task.ts b/app/core/entity/Task.ts
@@ -242,6 +242,26 @@ export class Task<T extends TaskBaseData = TaskBaseData> extends Entity {
     return task;
   }
 
+  public static createUpdateProxyCache(targetName: string, options: CreateUpdateProxyCacheTaskData):CreateUpdateProxyCacheTask {
+    const data = {
+      type: TaskType.UpdateProxyCache,
+      state: TaskState.Waiting,
+      targetName,
+      authorId: `pid_${PID}`,
+      authorIp: HOST_NAME,
+      data: {
+        taskWorker: '',
+        targetName,
+        version: options?.version,
+        fileType: options.fileType,
+        filePath: options.filePath,
+      },
+    };
+    const task = this.create(data);
+    task.logPath = `/packages/${targetName}/update-manifests/${dayjs().format('YYYY/MM/DDHHmm')}-${task.taskId}.log`;
+    return task;
+  }
+
   start(): TaskUpdateCondition {
     const condition = {
       taskId: this.taskId,

diff --git a/app/core/service/ProxyModeService.ts b/app/core/service/ProxyModeService.ts
@@ -5,12 +5,16 @@ import { downloadToTempfile } from '../../common/FileUtil';
 import { NPMRegistry, RegistryResponse } from '../../common/adapter/NPMRegistry';
 import { ProxyModeCachedFiles } from '../entity/ProxyModeCachedFiles';
 import { ProxyModeCachedFilesRepository } from '../../repository/ProxyModeCachedFilesRepository';
+import { TaskRepository } from '../../repository/TaskRepository';
 import { AbstractService } from '../../common/AbstractService';
+import { TaskService } from './TaskService';
 import { readFile, rm } from 'node:fs/promises';
 import { NFSAdapter } from '../../common/adapter/NFSAdapter';
 import { PROXY_MODE_CACHED_PACKAGE_DIR_NAME } from '../../common/constants';
 import { DIST_NAMES } from '../entity/Package';
 import type { PackageJSONType } from '../../repository/PackageRepository';
+import { TaskType, TaskState } from '../../common/enum/Task';
+import { Task } from '../entity/Task';
 
 @SingletonProto({
   accessLevel: AccessLevel.PUBLIC,
@@ -24,6 +28,10 @@ export class ProxyModeService extends AbstractService {
   private readonly nfsAdapter: NFSAdapter;
   @Inject()
   private readonly proxyModeCachedFiles: ProxyModeCachedFilesRepository;
+  @Inject()
+  private readonly taskRepository: TaskRepository;
+  @Inject()
+  private readonly taskService: TaskService;
 
   async getPackageVersionTarAndTempFilePath(fullname: string, url: string): Promise<{ tgzBuffer:Buffer| null }> {
     if (this.config.cnpmcore.syncPackageBlockList.includes(fullname)) {
@@ -165,4 +173,68 @@ export class ProxyModeService extends AbstractService {
     return { storeKey, proxyBytes, pkgManifest };
   }
 
+  public async createTask(targetName, options) {
+    const existsTask = await this.taskRepository.findTaskByTargetName(targetName, TaskType.UpdateProxyCache);
+    if (existsTask) {
+      return existsTask;
+    }
+    try {
+      return await this.taskService.createTask(Task.createSyncBinary(targetName, options), false);
+    } catch (e) {
+      this.logger.error('[ProxyModeService.createTask] targetName: %s, error: %s', targetName, e);
+    }
+  }
+
+  public async findTask(taskId: string) {
+    return await this.taskService.findTask(taskId);
+  }
+
+  public async findTaskLog(task: Task) {
+    return await this.taskService.findTaskLog(task);
+  }
+
+  public async findExecuteTask() {
+    return await this.taskService.findExecuteTask(TaskType.UpdateProxyCache);
+  }
+
+  public async executeTask(task: Task) {
+    const logs: string[] = [];
+    await this.taskService.finishTask(task, TaskState.Fail, logs.join('\n'));
+    //   const binaryName = task.targetName as BinaryName;
+    //   const binaryAdapter = await this.getBinaryAdapter(binaryName);
+    //   const logUrl = `${this.config.cnpmcore.registry}/-/binary/${binaryName}/syncs/${task.taskId}/log`;
+    //   let logs: string[] = [];
+    //   logs.push(`[${isoNow()}] 🚧🚧🚧🚧🚧 Start sync binary "${binaryName}" 🚧🚧🚧🚧🚧`);
+    //   if (!binaryAdapter) {
+    //     task.error = 'unknow binaryName';
+    //     logs.push(`[${isoNow()}] ❌ Synced "${binaryName}" fail, ${task.error}, log: ${logUrl}`);
+    //     logs.push(`[${isoNow()}] ❌❌❌❌❌ "${binaryName}" ❌❌❌❌❌`);
+    //     this.logger.error('[BinarySyncerService.executeTask:fail] taskId: %s, targetName: %s, %s',
+    //       task.taskId, task.targetName, task.error);
+    //     await this.taskService.finishTask(task, TaskState.Fail, logs.join('\n'));
+    //     return;
+    //   }
+
+  //   await this.taskService.appendTaskLog(task, logs.join('\n'));
+  //   logs = [];
+  //   this.logger.info('[BinarySyncerService.executeTask:start] taskId: %s, targetName: %s, log: %s',
+  //     task.taskId, task.targetName, logUrl);
+  //   try {
+  //     await this.syncDir(binaryAdapter, task, '/');
+  //     logs.push(`[${isoNow()}] 🟢 log: ${logUrl}`);
+  //     logs.push(`[${isoNow()}] 🟢🟢🟢🟢🟢 "${binaryName}" 🟢🟢🟢🟢🟢`);
+  //     await this.taskService.finishTask(task, TaskState.Success, logs.join('\n'));
+  //     this.logger.info('[BinarySyncerService.executeTask:success] taskId: %s, targetName: %s, log: %s',
+  //       task.taskId, task.targetName, logUrl);
+  //   } catch (err: any) {
+  //     task.error = err.message;
+  //     logs.push(`[${isoNow()}] ❌ Synced "${binaryName}" fail, ${task.error}, log: ${logUrl}`);
+  //     logs.push(`[${isoNow()}] ❌❌❌❌❌ "${binaryName}" ❌❌❌❌❌`);
+  //     this.logger.error('[BinarySyncerService.executeTask:fail] taskId: %s, targetName: %s, %s',
+  //       task.taskId, task.targetName, task.error);
+  //     this.logger.error(err);
+  //     await this.taskService.finishTask(task, TaskState.Fail, logs.join('\n'));
+  //   }
+  }
+
 }