chore: src/modules/demo/controller/open/queue.ts

src/modules/demo/controller/open/queue.ts

@@ -0,0 +1,49 @@
+import { Get, Inject, Post, Provide } from '@midwayjs/core';
+import { CoolController, BaseController } from '@cool-midway/core';
+import { DemoCommQueue } from '../../queue/comm';
+import { DemoGetterQueue } from '../../queue/getter';
+
+/**
+ * 队列
+ */
+@CoolController()
+export class OpenDemoQueueController extends BaseController {
+  // 普通队列
+  @Inject()
+  demoCommQueue: DemoCommQueue;
+
+  // 主动消费队列
+  @Inject()
+  demoGetterQueue: DemoGetterQueue;
+
+  /**
+   * 发送数据到队列
+   */
+  @Post('/add', { summary: '发送队列数据' })
+  async queue() {
+    this.demoCommQueue.add({ a: 2 });
+    return this.ok();
+  }
+
+  @Post('/addGetter')
+  async addGetter() {
+    await this.demoGetterQueue.add({ a: new Date() });
+    return this.ok();
+  }
+
+  /**
+   * 获得队列中的数据，只有当队列类型为getter时有效
+   */
+  @Get('/getter')
+  async getter() {
+    const job = await this.demoGetterQueue.getters.getJobs(
+      ['wait'],
+      0,
+      0,
+      true
+    );
+    // 获得完将数据从队列移除
+    await job[0]?.remove();
+    return this.ok(job[0]?.data);
+  }
+}