feat: Implement process worker with bullmq

This commit is contained in:
Faruk AYDIN
2022-03-05 00:15:46 +03:00
committed by Ömer Faruk Aydın
parent b288dc8c35
commit 20e725b590
9 changed files with 194 additions and 3 deletions

View File

@@ -16,6 +16,8 @@ type AppConfig = {
baseUrl?: string;
encryptionKey: string;
serveWebAppSeparately: boolean;
redisHost: string;
redisPort: number;
};
const appConfig: AppConfig = {
@@ -33,6 +35,8 @@ const appConfig: AppConfig = {
encryptionKey: process.env.ENCRYPTION_KEY,
serveWebAppSeparately:
process.env.SERVE_WEB_APP_SEPARATELY === 'true' ? true : false,
redisHost: process.env.REDIS_HOST || '127.0.0.1',
redisPort: parseInt(process.env.REDIS_PORT) || 6379,
};
if (appConfig.serveWebAppSeparately) {

View File

@@ -0,0 +1,8 @@
import appConfig from './app';
const redisConfig = {
host: appConfig.redisHost,
port: appConfig.redisPort,
};
export default redisConfig;

View File

@@ -2,6 +2,7 @@ import { GraphQLString, GraphQLNonNull } from 'graphql';
import RequestWithCurrentUser from '../../types/express/request-with-current-user';
import executeFlowType from '../types/execute-flow';
import Processor from '../../services/processor';
import processorQueue from '../../queues/processor';
type Params = {
stepId: string;
@@ -21,6 +22,12 @@ const executeFlowResolver = async (
const flow = await step.$relatedQuery('flow');
const data = await new Processor(flow, step).run();
// TODO: Use this snippet to execute flows with the background job.
// const data = processorQueue.add('processorJob', {
// flowId: flow.id,
// stepId: step.id,
// });
await step.$query().patch({
status: 'completed',
});

View File

@@ -0,0 +1,8 @@
import { Queue } from 'bullmq';
import redisConfig from '../config/redis';
const processorQueue = new Queue('processor', {
connection: redisConfig,
});
export default processorQueue;

View File

@@ -0,0 +1,2 @@
import './config/database';
import './workers/processor';

View File

@@ -0,0 +1,31 @@
import { Worker } from 'bullmq';
import Processor from '../services/processor';
import redisConfig from '../config/redis';
import Step from '../models/step';
import logger from '../helpers/logger';
const worker = new Worker(
'processor',
async (job) => {
const step = await Step.query()
.withGraphFetched('connection')
.findOne({
'steps.id': job.data.stepId,
})
.throwIfNotFound();
const flow = await step.$relatedQuery('flow');
const data = await new Processor(flow, step).run();
return data;
},
{ connection: redisConfig }
);
worker.on('completed', (job) => {
logger.info(`${job.id} has completed!`);
});
worker.on('failed', (job, err) => {
logger.info(`${job.id} has failed with ${err.message}`);
});