feat: add ability to search workflows by activity id

Signed-off-by: ItsRoy69 <jyotirmoyroy649@gmail.com>

Author: ItsRoy69

src/lib/stores/search-attributes.ts

@@ -24,6 +24,7 @@ export const searchAttributes: Readable<SearchAttributes> = derived(
   ([$allSearchAttributes]) => ({
     ...$allSearchAttributes.customAttributes,
     ...$allSearchAttributes.systemAttributes,
+    ActivityId: SEARCH_ATTRIBUTE_TYPE.KEYWORD,
   }),
 );
 
@@ -172,6 +173,7 @@ export const sortedSearchAttributeOptions: Readable<SearchAttributeOption[]> =
       'WorkflowId',
       'WorkflowType',
       'RunId',
+      'ActivityId',
       'StartTime',
       'CloseTime',
     ];

src/lib/types/workflows.ts

@@ -118,6 +118,7 @@ export type FilterParameters = {
   executionStatus?: WorkflowStatus;
   timeRange?: Duration | string;
   query?: string;
+  activityId?: string;
 };
 
 export type ArchiveFilterParameters = Omit<FilterParameters, 'timeRange'> & {

src/lib/utilities/query/filter-workflow-query.ts

@@ -18,7 +18,8 @@ export type QueryKey =
   | 'CloseTime'
   | 'ExecutionTime'
   | 'ExecutionStatus'
-  | 'RunId';
+  | 'RunId'
+  | 'ActivityId';
 
 type FilterValue = string | Duration;
 
@@ -29,6 +30,7 @@ const filterKeys: Readonly<Record<string, QueryKey>> = {
   executionStatus: 'ExecutionStatus',
   closeTime: 'CloseTime',
   runId: 'RunId',
+  activityId: 'ActivityId',
 } as const;
 
 const isValid = (value: unknown, conditional: string): boolean => {

src/lib/utilities/query/list-workflow-query.ts

@@ -14,14 +14,16 @@ export type QueryKey =
   | 'StartTime'
   | 'CloseTime'
   | 'ExecutionTime'
-  | 'ExecutionStatus';
+  | 'ExecutionStatus'
+  | 'ActivityId';
 
 export type FilterKey =
   | 'workflowId'
   | 'workflowType'
   | 'timeRange'
   | 'executionStatus'
-  | 'closeTime';
+  | 'closeTime'
+  | 'activityId';
 
 type FilterValue = string | Duration;
 
@@ -31,6 +33,7 @@ const queryKeys: Readonly<Record<string, QueryKey>> = {
   timeRange: 'StartTime',
   executionStatus: 'ExecutionStatus',
   closeTime: 'CloseTime',
+  activityId: 'ActivityId',
 } as const;
 
 const filterKeys: readonly FilterKey[] = [
@@ -39,6 +42,7 @@ const filterKeys: readonly FilterKey[] = [
   'timeRange',
   'executionStatus',
   'closeTime',
+  'activityId',
 ] as const;
 
 const isValid = (value: unknown): boolean => {

src/lib/utilities/query/to-list-workflow-filters.ts

@@ -79,6 +79,7 @@ const DefaultAttributes: SearchAttributes = {
   WorkflowId: SEARCH_ATTRIBUTE_TYPE.KEYWORD,
   WorkflowType: SEARCH_ATTRIBUTE_TYPE.KEYWORD,
   RunId: SEARCH_ATTRIBUTE_TYPE.KEYWORD,
+  ActivityId: SEARCH_ATTRIBUTE_TYPE.KEYWORD,
 };
 
 export const toListWorkflowFilters = (

src/lib/utilities/query/to-list-workflow-parameters.test.ts

@@ -17,7 +17,8 @@ const defaultParameters = {
   workflowId: '',
   workflowType: '',
   executionStatus: null,
-  timeRange: null,
+  timeRange: undefined,
+  activityId: '',
 };
 
 describe('toListWorkflowParameters', () => {

src/lib/utilities/query/to-list-workflow-parameters.ts

@@ -32,14 +32,16 @@ const isWorkflowTypeStatement = is('WorkflowType');
 const isWorkflowIdStatement = is('WorkflowId');
 const isStartTimeStatement = is('StartTime');
 const isExecutionStatusStatement = is('ExecutionStatus');
+const isActivityIdStatement = is('ActivityId');
 
 export const toListWorkflowParameters = (query: string): ParsedParameters => {
   const tokens = tokenize(query);
   const parameters: ParsedParameters = {
     workflowId: '',
     workflowType: '',
     executionStatus: null,
-    timeRange: null,
+    timeRange: undefined,
+    activityId: '',
   };
 
   tokens.forEach((token, index) => {
@@ -66,6 +68,9 @@ export const toListWorkflowParameters = (query: string): ParsedParameters => {
         console.error('Error parsing StartTime from query', error);
       }
     }
+
+    if (isActivityIdStatement(token))
+      parameters.activityId = getTwoAhead(tokens, index);
   });
 
   return parameters;