mirror of
https://github.com/Litlyx/litlyx
synced 2025-12-09 23:48:36 +01:00
.
This commit is contained in:
2
.gitignore
vendored
2
.gitignore
vendored
@@ -1,2 +1,4 @@
|
||||
steps
|
||||
PROCESS_EVENT
|
||||
docker
|
||||
dev
|
||||
7
producer/Dockerfile
Normal file
7
producer/Dockerfile
Normal file
@@ -0,0 +1,7 @@
|
||||
FROM node:21-alpine
|
||||
WORKDIR /home/app
|
||||
COPY package.json pnpm-lock.yaml ./
|
||||
RUN npm install -g pnpm && pnpm install --prod --frozen-lockfile
|
||||
COPY ./dist /home/app/dist
|
||||
EXPOSE ${PORT}
|
||||
CMD ["node", "dist/producer/src/index.js"]
|
||||
@@ -20,7 +20,8 @@
|
||||
"dev": "node scripts/start_dev.js",
|
||||
"compile": "tsc",
|
||||
"build": "ts-node scripts/build.ts",
|
||||
"build_all": "npm run compile && npm run build"
|
||||
"build_all": "npm run compile && npm run build",
|
||||
"docker-build": "node scripts/prepare_docker.js && docker build -t litlyx-producer ."
|
||||
},
|
||||
"keywords": [],
|
||||
"author": "Emily",
|
||||
|
||||
7
producer/scripts/prepare_docker.js
Normal file
7
producer/scripts/prepare_docker.js
Normal file
@@ -0,0 +1,7 @@
|
||||
|
||||
const child = require('child_process');
|
||||
|
||||
const p = child.exec('pnpm run compile && pnpm run build');
|
||||
|
||||
p.stdout.on('data', (e) => { console.log(e.toString()); });
|
||||
p.stderr.on('data', (e) => { console.log(e.toString()); });
|
||||
@@ -7,7 +7,8 @@ export type ReadingLoopOptions = {
|
||||
empty?: number
|
||||
},
|
||||
readBlock?: number,
|
||||
streamName: string
|
||||
streamName: string,
|
||||
consumer?: string
|
||||
}
|
||||
|
||||
export class RedisStreamService {
|
||||
@@ -23,7 +24,7 @@ export class RedisStreamService {
|
||||
}
|
||||
|
||||
private static async readingLoop(options: ReadingLoopOptions, processFunction: (content: Record<string, string>) => Promise<any>) {
|
||||
const result = await this.readFromStream(options.streamName, options.readBlock || 2500);
|
||||
const result = await this.readFromStream(options.streamName, options.readBlock || 2500, options.consumer || 'base_consumer');
|
||||
if (!result) {
|
||||
await new Promise(r => setTimeout(r, options.delay?.empty || 5000));
|
||||
setTimeout(() => this.readingLoop(options, processFunction), 1);
|
||||
@@ -35,17 +36,29 @@ export class RedisStreamService {
|
||||
return;
|
||||
}
|
||||
|
||||
static startReadingLoop(options: ReadingLoopOptions, processFunction: (content: Record<string, string>) => Promise<any>) {
|
||||
static async startReadingLoop(options: ReadingLoopOptions, processFunction: (content: Record<string, string>) => Promise<any>) {
|
||||
|
||||
try {
|
||||
await this.client.xGroupCreate(options.streamName, 'broker', '0', { MKSTREAM: true, });
|
||||
} catch (ex) {
|
||||
|
||||
}
|
||||
|
||||
this.readingLoop(options, processFunction)
|
||||
}
|
||||
|
||||
private static async readFromStream(streamName: string, readBlock: number) {
|
||||
const result = await this.client.xRead({ id: '0', key: streamName }, { COUNT: 1, BLOCK: readBlock });
|
||||
private static async readFromStream(streamName: string, readBlock: number, consumer: string) {
|
||||
const result = await this.client.xReadGroup(
|
||||
'broker', consumer,
|
||||
[{ key: streamName, id: '>' }],
|
||||
{ COUNT: 1, BLOCK: readBlock }
|
||||
);
|
||||
if (!result) return;
|
||||
if (result.length == 0) return;
|
||||
if (!result[0].messages) return;
|
||||
if (result[0].messages.length == 0) return;
|
||||
const message = result[0].messages[0];
|
||||
await this.client.xAck(streamName, 'broker', message.id);
|
||||
await this.client.xDel(streamName, message.id);
|
||||
const content = message.message;
|
||||
return content;
|
||||
|
||||
Reference in New Issue
Block a user