worker.ts•1.19 kB
import { AppSystemProp, PiecesSource, WorkerSystemProp } from '@activepieces/server-shared'
import { isNil } from '@activepieces/shared'
import { FastifyInstance } from 'fastify'
import { flowWorker, pieceBuilder } from 'server-worker'
import { accessTokenManager } from './authentication/lib/access-token-manager'
import { system } from './helper/system/system'
export const setupWorker = async (app: FastifyInstance): Promise<void> => {
const piecesSource = system.getOrThrow<PiecesSource>(AppSystemProp.PIECES_SOURCE)
const devPieces = system.get(AppSystemProp.DEV_PIECES)?.split(',') ?? []
await pieceBuilder(app, app.io, devPieces, piecesSource)
app.addHook('onClose', async () => {
await flowWorker(app.log).close()
})
}
export async function workerPostBoot(app: FastifyInstance): Promise<void> {
const workerToken = await generateWorkerToken()
await flowWorker(app.log).init({ workerToken })
}
async function generateWorkerToken(): Promise<string> {
const workerToken = system.get(WorkerSystemProp.WORKER_TOKEN)
if (!isNil(workerToken)) {
return workerToken
}
return accessTokenManager.generateWorkerToken()
}