refactor: use ExecutionStep model to get for-each steps

Author: kevinkim-ogp

packages/backend/src/graphql/queries/get-execution-steps.ts

@@ -5,6 +5,7 @@ import {
   TOOLBOX_APP_KEY,
 } from '@/apps/toolbox/common/constants'
 import paginate from '@/helpers/pagination'
+import ExecutionStep from '@/models/execution-step'
 
 import type { QueryResolvers } from '../__generated__/types.generated'
 
@@ -55,60 +56,16 @@ const getExecutionSteps: QueryResolvers['getExecutionSteps'] = async (
   // NOTE: use a separate query for for-each
   // as there are multiple execution steps with the same step_id in the same execution
   if (hasForEach) {
-    const forEachExecutionSteps = execution
-      .$relatedQuery('executionSteps')
-      .with('latest_steps', (builder) => {
-        /**
-         * NOTE: there is a known issue with knex where 'groupBy' are placed at the end of the 'unionAll' query.
-         * the workaround is to unionAll both queries with 'true' to wrap the subequery.
-         */
-        builder
-          .unionAll((qb) => {
-            qb.select(
-              'step_id',
-              raw('max(created_at) as max_created_at'),
-              raw('min(created_at) as min_created_at'),
-            )
-              .from('execution_steps')
-              .groupBy('step_id')
-              .where('execution_id', '=', execution.id)
-              .where(raw("metadata = '{}'::jsonb"))
-              .withSoftDeleted()
-          }, true)
-          .unionAll((qb) => {
-            qb.select(
-              'step_id',
-              raw('max(created_at) as max_created_at'),
-              raw('min(created_at) as min_created_at'),
-            )
-              .from('execution_steps')
-              .groupBy('step_id', raw("metadata->>'iteration'"))
-              .where('execution_id', '=', execution.id)
-              .where(raw("metadata != '{}'::jsonb"))
-              .withSoftDeleted()
-          }, true)
-          .withSoftDeleted()
-      })
-      .join('latest_steps', (builder) => {
-        builder
-          .on('execution_steps.step_id', '=', 'latest_steps.step_id')
-          .andOn(
-            'execution_steps.created_at',
-            '=',
-            'latest_steps.max_created_at',
-          )
-      })
-      .select('execution_steps.*', 'min_created_at')
-      .withSoftDeleted()
-      .orderBy('min_created_at', 'asc')
+    const forEachExecutionSteps = await ExecutionStep.getForEachExecutionSteps(
+      execution.id,
+    )
 
-    const allRecords = await forEachExecutionSteps
     return {
       pageInfo: {
         currentPage: 1,
         totalCount: 100,
       },
-      edges: allRecords.map((record) => ({
+      edges: forEachExecutionSteps.map((record) => ({
         node: record,
       })),
     }