Skip to content

anchan828/nest-cloud-run-queue

Repository files navigation

@anchan828/nest-cloud-run-queue

Create a Queue/Worker for NestJS application in Cloud Run.

nest-cloud-run-queue

Overview

I am wondering how to implement Queue when running an application with Cloud Run. While @nestjs/bull is a very good library, Cloud Run, which is serverless, cannot be used because the server is not always running. You can use "CPU always allocated" for resolve this issue, but it doesn't make sense to use Cloud Run.

Therefore, I used Cloud Pub/Sub or Cloud Tasks so that I could implement Queue via HTTP requests. This package supports both, so you can choose whichever you prefer.

Choose Cloud Tasks or Pub/Sub

Of course, these packages can work without using Cloud Run on workers since they handle tasks via HTTP requests.

Demo

See https://github.com/anchan828/nest-cloud-run-queue/tree/master/packages/demo

This demo uses an emulator, which runs PubSub and Tasks locally. Please see docker-compose.yml if you are interested.

Packages

There are two types of packages.

Publisher

Package Description
@anchan828/nest-cloud-run-queue-pubsub-publisher Library for sending messages using Cloud Pub/Sub.
@anchan828/nest-cloud-run-queue-tasks-publisher Library for sending messages using Cloud Tasks.

Worker

Package Description
@anchan828/nest-cloud-run-queue-worker Library for creating applications that receive and process messages.

Getting started

1. Create publisher application

Using Cloud Pub/Sub

See: @anchan828/nest-cloud-run-queue-pubsub-publisher - README.md

Using Cloud Tasks

See: @anchan828/nest-cloud-run-queue-tasks-publisher - README.md

2. Create worker application

Import worker module

@Module({
  imports: [QueueWorkerModule.register()],
})
export class WorkerAppModule {}

Create worker provider

@QueueWorker("Worker name")
class Worker {
  @QueueWorkerProcess()
  public async process(message: string | object, raw: QueueWorkerRawMessage): Promise<void> {
    console.log("Message:", message);
    console.log("Raw message:", raw);
  }
}

Add as provider

@Module({
  imports: [QueueWorkerModule.register()],
  providers: [Worker],
})
export class WorkerAppModule {}

Customize worker controller

The Controller who receives the message is automatically defined. You can customize it.

@Module({
  imports: [
    QueueWorkerModule.register({
      workerController: {
        method: RequestMethod.GET,
        path: "/worker",
      },

      // Default
      // workerController: {
      //   method: RequestMethod.POST,
      //   path: "/",
      // },
    }),
  ],
  providers: [Worker],
})
export class WorkerAppModule {}

You can also define your own Controller. In that case, set workerController to null.

@Controller("/worker")
class WorkerController {
  constructor(private readonly service: QueueWorkerService) {}

  @Post()
  public async execute(@Body() body: QueueWorkerReceivedMessage): Promise<void> {
    await this.service.execute(body.message);

    // or

    // const decodedMessage = this.service.decodeMessage(body.message);
    // await this.service.execute(decodedMessage);
  }
}

@Module({
  controllers: [WorkerController],
  imports: [
    QueueWorkerModule.register({
      workerController: null,
    }),
  ],
  providers: [Worker],
})
export class WorkerAppModule {}

Create bootstrap function

async function bootstrap(): Promise<void> {
  const app = await NestFactory.create(WorkerAppModule);
  await app.listen(process.env.PORT || 8080);
}

bootstrap();

Using Cloud Scheduler

You can use Cloud Scheduler as trigger.

Payload is JSON string {"name": "worker name", "data": "str"}

Global Events

This package is defined special event handlers.

Note: throwModuleError: true is not working if you set global events.

UNHANDLED_QUEUE_WORKER

You can listen to undefined worker name

@QueueWorker(UNHANDLED_QUEUE_WORKER)
class Worker {
  @QueueWorkerProcess()
  public async process(message: Message<any>, raw: QueueWorkerRawMessage): Promise<void> {
    console.log("Message:", message);
    console.log("Raw message:", raw);
  }
}

ALL_QUEUE_WORKERS

You can listen to all workers

@QueueWorker(ALL_QUEUE_WORKERS)
class Worker {
  @QueueWorkerProcess()
  public async process(message: Message<any>, raw: QueueWorkerRawMessage): Promise<void> {
    console.log("Message:", message);
    console.log("Raw message:", raw);
  }
}