Update endpoint + listeners

twentyhq · Aug 1, 2024 · 8cdc8fe · 8cdc8fe
1 parent 47db158
commit 8cdc8fe
Show file tree

Hide file tree

Showing 4 changed files with 21 additions and 9 deletions.
diff --git a/...dules/workflow/dtos/workflow-input.dto.ts → ...ow/dtos/run-workflow-version-input.dto.ts b/...dules/workflow/dtos/workflow-input.dto.ts → ...ow/dtos/run-workflow-version-input.dto.ts
@@ -3,7 +3,13 @@ import { Field, InputType } from '@nestjs/graphql';
 import graphqlTypeJson from 'graphql-type-json';
 
 @InputType()
-export class WorkflowInput {
+export class RunWorkflowVersionInput {
+  @Field(() => String, {
+    description: 'Workflow version ID',
+    nullable: false,
+  })
+  workflowVersionId: string;
+
   @Field(() => graphqlTypeJson, {
     description: 'Execution result in JSON format',
     nullable: true,

diff --git a/packages/twenty-server/src/engine/core-modules/workflow/workflow-trigger.resolver.ts b/packages/twenty-server/src/engine/core-modules/workflow/workflow-trigger.resolver.ts
@@ -1,7 +1,7 @@
 import { UseGuards } from '@nestjs/common';
 import { Args, Mutation, Resolver } from '@nestjs/graphql';
 
-import { WorkflowInput } from 'src/engine/core-modules/workflow/dtos/workflow-input.dto';
+import { RunWorkflowVersionInput } from 'src/engine/core-modules/workflow/dtos/run-workflow-version-input.dto';
 import { WorkflowTriggerResultDTO } from 'src/engine/core-modules/workflow/dtos/workflow-trigger-result.dto';
 import { workflowTriggerGraphqlApiExceptionHandler } from 'src/engine/core-modules/workflow/utils/workflow-trigger-graphql-api-exception-handler.util';
 import { Workspace } from 'src/engine/core-modules/workspace/workspace.entity';
@@ -32,14 +32,13 @@ export class WorkflowTriggerResolver {
   }
 
   @Mutation(() => WorkflowTriggerResultDTO)
-  async triggerWorkflow(
+  async runWorkflowVersion(
     @AuthWorkspace() { id: workspaceId }: Workspace,
-    @Args('workflowVersionId') workflowVersionId: string,
-    @Args('input') { payload }: WorkflowInput,
+    @Args('input') { workflowVersionId, payload }: RunWorkflowVersionInput,
   ) {
     try {
       return {
-        result: await this.workflowTriggerService.runWorkflow(
+        result: await this.workflowTriggerService.runWorkflowVersion(
           workspaceId,
           workflowVersionId,
           payload ?? {},

diff --git a/...server/src/modules/workflow/workflow-trigger/listeners/database-event-trigger.listener.ts b/...server/src/modules/workflow/workflow-trigger/listeners/database-event-trigger.listener.ts
@@ -4,6 +4,8 @@ import { OnEvent } from '@nestjs/event-emitter';
 import { FeatureFlagKey } from 'src/engine/core-modules/feature-flag/enums/feature-flag-key.enum';
 import { IsFeatureEnabledService } from 'src/engine/core-modules/feature-flag/services/is-feature-enabled.service';
 import { ObjectRecordCreateEvent } from 'src/engine/integrations/event-emitter/types/object-record-create.event';
+import { ObjectRecordDeleteEvent } from 'src/engine/integrations/event-emitter/types/object-record-delete.event';
+import { ObjectRecordUpdateEvent } from 'src/engine/integrations/event-emitter/types/object-record-update.event';
 import { InjectMessageQueue } from 'src/engine/integrations/message-queue/decorators/message-queue.decorator';
 import { MessageQueue } from 'src/engine/integrations/message-queue/message-queue.constants';
 import { MessageQueueService } from 'src/engine/integrations/message-queue/services/message-queue.service';
@@ -25,8 +27,13 @@ export class DatabaseEventTriggerListener {
     private readonly isFeatureFlagEnabledService: IsFeatureEnabledService,
   ) {}
 
-  @OnEvent('**')
-  async handleEvent(payload: ObjectRecordCreateEvent<any>) {
+  @OnEvent(['*.created', '*.updated', '*.deleted'])
+  async handleEvent(
+    payload:
+      | ObjectRecordCreateEvent<any>
+      | ObjectRecordUpdateEvent<any>
+      | ObjectRecordDeleteEvent<any>,
+  ) {
     const workspaceId = payload.workspaceId;
     const eventName = payload.name;
 

diff --git a/packages/twenty-server/src/modules/workflow/workflow-trigger/workflow-trigger.service.ts b/packages/twenty-server/src/modules/workflow/workflow-trigger/workflow-trigger.service.ts
@@ -22,7 +22,7 @@ export class WorkflowTriggerService {
     private readonly workflowRunnerService: WorkflowRunnerService,
   ) {}
 
-  async runWorkflow(
+  async runWorkflowVersion(
     workspaceId: string,
     workflowVersionId: string,
     payload: object,