make sure we bill all cases (#1277)

Author: alexcos20

src/components/c2d/compute_engine_docker.ts

@@ -361,13 +361,14 @@ export class C2DEngineDocker extends C2DEngine {
       }
     }
 
-    // get all jobs that are in the settle status
-    const jobs = await this.db.getJobs(
-      envs,
-      undefined,
-      undefined,
+    // get all jobs that needs to be paid
+    const jobs = await this.db.getJobsByStatus(envs, [
+      C2DStatusNumber.AlgorithmFailed,
+      C2DStatusNumber.DiskQuotaExceeded,
+      C2DStatusNumber.ResultsFetchFailed,
+      C2DStatusNumber.ResultsUploadFailed,
       C2DStatusNumber.JobSettle
-    )
+    ])
     CORE_LOGGER.info(`ClaimPayments:  Got ${jobs.length} jobs to check`)
     if (jobs.length > 0) {
       const providerAddress = this.getKeyManager().getEthAddress()

src/components/database/C2DDatabase.ts

@@ -89,6 +89,13 @@ export class C2DDatabase extends AbstractDatabase {
     return await this.provider.getJobs(environments, fromTimestamp, consumerAddrs, status)
   }
 
+  async getJobsByStatus(
+    environments: string[],
+    status: C2DStatusNumber[]
+  ): Promise<DBComputeJob[]> {
+    return await this.provider.getJobsByStatus(environments, status)
+  }
+
   async updateImage(image: string): Promise<void> {
     return await this.provider.updateImage(image)
   }

src/components/database/sqliteCompute.ts

@@ -439,7 +439,7 @@ export class SQLiteCompute implements ComputeDatabaseProvider {
     })
   }
 
-  getJobs(
+  async getJobs(
     environments?: string[],
     fromTimestamp?: string,
     consumerAddrs?: string[],
@@ -476,7 +476,44 @@ export class SQLiteCompute implements ComputeDatabaseProvider {
       selectSQL += ` WHERE ${conditions.join(' AND ')}`
     }
     selectSQL += ` ORDER BY dateCreated DESC`
+    return await this.doQuery(selectSQL, params, environments)
+  }
+
+  async getJobsByStatus(
+    environments: string[],
+    status: C2DStatusNumber[]
+  ): Promise<DBComputeJob[]> {
+    let selectSQL = `SELECT * FROM ${this.schema.name}`
+
+    // sqlite3 bindings accept both strings and numbers; `status` is a numeric enum.
+    const params: Array<string | number> = []
+    const conditions: string[] = []
+
+    if (environments && environments.length > 0) {
+      const placeholders = environments.map(() => '?').join(',')
+      conditions.push(`environment IN (${placeholders})`)
+      params.push(...environments)
+    }
+
+    if (status && status.length > 0) {
+      const placeholders = status.map(() => '?').join(',')
+      conditions.push(`status IN (${placeholders})`)
+      params.push(...status)
+    }
+
+    if (conditions.length > 0) {
+      selectSQL += ` WHERE ${conditions.join(' AND ')}`
+    }
+    selectSQL += ` ORDER BY dateCreated DESC`
+
+    return await this.doQuery(selectSQL, params, environments)
+  }
 
+  private doQuery(
+    selectSQL: string,
+    params: Array<string | number>,
+    environments: string[]
+  ) {
     return new Promise<DBComputeJob[]>((resolve, reject) => {
       this.db.all(selectSQL, params, (err, rows: any[] | undefined) => {
         if (err) {