simple progress tracking

Author: jordanverasamy

packages/store/src/cli/commands/store/execute.ts

@@ -93,7 +93,10 @@ export default class Execute extends Command {
 
       if (operationType === 'query') {
         this.log('starting bulk query...')
-        const results = await runBulkQuery(query, adminSession)
+        const results = await runBulkQuery(query, adminSession, (status, objectCount) => {
+          this.log(`status: ${status} | objects: ${objectCount}`)
+        })
+        this.log('\nresults:')
         this.log(results)
       } else {
         this.error('bulk mutations not yet implemented')

packages/store/src/cli/services/bulk-operations.ts

@@ -43,6 +43,7 @@ interface BulkOperationStatus {
 export async function runBulkQuery(
   query: string,
   session: AdminSession,
+  onProgress?: (status: string, objectCount: string) => void,
 ): Promise<string> {
   const startResult = await adminRequest<any>(BULK_QUERY_MUTATION, session, {query})
 
@@ -51,16 +52,34 @@ export async function runBulkQuery(
     throw new Error(`bulk operation failed: ${errors}`)
   }
 
+  console.log('bulk operation started', startResult.bulkOperationRunQuery.bulkOperation.id)
+
+  let lastStatus = ''
+
   while (true) {
-    await sleep(1000)
+    await sleep(1)
+
+    console.log('polling bulk operation status')
 
     const statusResult = await adminRequest<BulkOperationStatus>(BULK_OPERATION_STATUS_QUERY, session)
     const operation = statusResult.currentBulkOperation
 
+    if (operation.status !== lastStatus && onProgress) {
+      onProgress(operation.status, operation.objectCount)
+      lastStatus = operation.status
+    }
+
+    if (operation.status === 'RUNNING' && onProgress) {
+      onProgress(operation.status, operation.objectCount)
+    }
+
     if (operation.status === 'COMPLETED') {
       if (!operation.url) {
         throw new Error('bulk operation completed but no results url')
       }
+      if (onProgress) {
+        onProgress('DOWNLOADING', operation.objectCount)
+      }
       const response = await fetch(operation.url)
       return await response.text()
     }