close connections properly

WietseWind · Jul 20, 2023 · dcd34d4 · dcd34d4
1 parent 9e94722
commit dcd34d4
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 36 deletions.
diff --git a/index.js b/index.js
@@ -16,6 +16,16 @@ const StartLedger = typeof process.env.LEDGER === 'undefined' ? 32570 : parseInt
 console.log('Fetch XRPL transactions into Google BigQuery')
 
 const Client = new XrplClient(XRPLNodeUrl)
+
+async function safeHalt() {
+  try {
+    await Client.close()
+  } catch(e) {
+    console.error('ERROR closing connection:', e)
+  } finally {
+    process.exit(1)
+  }
+}
 
 Client.ready().then(Connection => {
   let Stopped = false
@@ -176,47 +186,34 @@ Client.ready().then(Connection => {
 
           return _Tx
         })
-
-        // console.dir(Transactions[0], { depth: null })
-        // process.exit(1)
 
         bigquery.dataset(DATASET_NAME).table(TRANSACTION_TABLE_NAME).insert(Transactions)
           .then(r => {
             console.log(`Inserted rows`, r)
             LastLedger = Result.ledger_index
-            // process.exit(0)
           })
           .catch(err => {
             if (err && err.name === 'PartialFailureError') {
               if (err.errors && err.errors.length > 0) {
                 console.log('Insert errors:')
                 err.errors.forEach(err => console.dir(err, { depth: null }))
-                process.exit(1)
+                return safeHalt()
               }
             } else {
               console.error('ERROR:', err)
-              process.exit(1)
+              return safeHalt()
             }
           })
       }
 
-      // retryTimeout = 0
-
       if (Stopped) {
-        return
+        return Connection.close()
       }
 
       return run(ledger_index + 1)
     }).catch(e => {
       console.log(e)
-      process.exit(1)
-
-      // retryTimeout += 500
-      // if (retryTimeout > 5000) retryTimeout = 5000
-      console.log(`Oops... Retry in ${retryTimeout / 1000} sec.`)
-      setTimeout(() => {
-        return run(ledger_index)
-      }, retryTimeout * 1000)
+      return safeHalt()
     })
   }
 
@@ -240,14 +237,13 @@ Client.ready().then(Connection => {
     }
   }).catch(e => {
     console.log('Google BigQuery Error', e)
-    process.exit(1)
+    return safeHalt()
   })
 
   process.on('SIGINT', function() {
     console.log(`\nGracefully shutting down from SIGINT (Ctrl+C)\n -- Wait for remaining BigQuery inserts and XRPL Connection close...`);
 
     Stopped = true  
-    Connection.close()
     if (LastLedger > 0) {
       console.log(`\nLast ledger: [ ${LastLedger} ]\n\nRun your next job with ENV: "LEDGER=${LastLedger+1}"\n\n`)
     }

diff --git a/ledgerInfo.js b/ledgerInfo.js
@@ -14,6 +14,16 @@ const StartLedger = typeof process.env.LEDGER === 'undefined' ? 32570 : parseInt
 console.log('Fetch XRPL Ledger Info into Google BigQuery')
 
 const Client = new XrplClient(XRPLNodeUrl)
+
+async function safeHalt() {
+  try {
+    await Client.close()
+  } catch(e) {
+    console.error('ERROR closing connection:', e)
+  } finally {
+    process.exit(1)
+  }
+}
 
 Client.ready().then(Connection => {
   let Stopped = false
@@ -39,7 +49,6 @@ Client.ready().then(Connection => {
   const run = (ledger_index) => {
     return fetchLedger(ledger_index).then(Result => {
       console.log(`${Result.ledger_index}`)
-      // console.log(Result)
       bigquery.dataset(DATASET_NAME).table(LEDGER_TABLE_NAME).insert([{
         LedgerIndex: parseInt(Result.ledger.ledger_index),
         hash: Result.ledger.hash,
@@ -55,38 +64,28 @@ Client.ready().then(Connection => {
         .then(r => {
           console.log(`Inserted rows`, r)
           LastLedger = Result.ledger_index
-          // process.exit(0)
         })
         .catch(err => {
           if (err && err.name === 'PartialFailureError') {
             if (err.errors && err.errors.length > 0) {
               console.log('Insert errors:')
               err.errors.forEach(err => console.dir(err, { depth: null }))
-              process.exit(1)
+              return safeHalt()
             }
           } else {
             console.error('ERROR:', err)
-            process.exit(1)
+            return safeHalt()
           }
         })
-
-//      retryTimeout = 0
 
       if (Stopped) {
-        return
+        return Connection.close()
       }
 
       return run(ledger_index + 1)
     }).catch(e => {
       console.log(e)
-      process.exit(1)
-
-      // retryTimeout += 500
-//      if (retryTimeout > 5000) retryTimeout = 5000
-      console.log(`Oops... Retry in ${retryTimeout / 1000} sec.`)
-      setTimeout(() => {
-        return run(ledger_index)
-      }, retryTimeout * 1000)
+      return safeHalt()
     })
   }
 
@@ -107,14 +106,13 @@ Client.ready().then(Connection => {
     }
   }).catch(e => {
     console.log('Google BigQuery Error', e)
-    process.exit(1)
+    return safeHalt()
   })
 
   process.on('SIGINT', function() {
     console.log(`\nGracefully shutting down from SIGINT (Ctrl+C)\n -- Wait for remaining BigQuery inserts and XRPL Connection close...`);
 
     Stopped = true  
-    Connection.close()
     if (LastLedger > 0) {
       console.log(`\nLast ledger: [ ${LastLedger} ]\n\nRun your next job with ENV: "LEDGER=${LastLedger+1}"\n\n`)
     }