commit file sinks after rewards db purged (#486)

helium · Apr 25, 2023 · e5eebee · e5eebee
1 parent 148fc17
commit e5eebee
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 31 deletions.
diff --git a/iot_verifier/src/rewarder.rs b/iot_verifier/src/rewarder.rs
@@ -88,28 +88,11 @@ impl Rewarder {
             .await?
             // Await the returned oneshot to ensure we wrote the file
             .await??;
-
         let written_files = self.rewards_sink.commit().await?.await??;
-        // Write the rewards manifest for the completed period
-        self.reward_manifests_sink
-            .write(
-                RewardManifest {
-                    start_timestamp: scheduler.reward_period.start.encode_timestamp(),
-                    end_timestamp: scheduler.reward_period.end.encode_timestamp(),
-                    written_files,
-                },
-                [],
-            )
-            .await?
-            .await??;
-
-        self.reward_manifests_sink.commit().await?;
 
         let mut transaction = self.pool.begin().await?;
-
         // Clear gateway shares table period to end of reward period
         GatewayShares::clear_rewarded_shares(&mut transaction, scheduler.reward_period.end).await?;
-
         save_rewarded_timestamp(
             "last_rewarded_end_time",
             &scheduler.reward_period.end,
@@ -122,9 +105,22 @@ impl Rewarder {
             &mut transaction,
         )
         .await?;
-
         transaction.commit().await?;
 
+        // now that the db has been purged, safe to write out the manifest
+        self.reward_manifests_sink
+            .write(
+                RewardManifest {
+                    start_timestamp: scheduler.reward_period.start.encode_timestamp(),
+                    end_timestamp: scheduler.reward_period.end.encode_timestamp(),
+                    written_files,
+                },
+                [],
+            )
+            .await?
+            .await??;
+        self.reward_manifests_sink.commit().await?;
+
         Ok(())
     }
 }

diff --git a/mobile_verifier/src/verifier.rs b/mobile_verifier/src/verifier.rs
@@ -170,7 +170,17 @@ impl VerifierDaemon {
 
         let written_files = self.mobile_rewards.commit().await?.await??;
 
-        // Write out the manifest file
+        let mut transaction = self.pool.begin().await?;
+        // Clear the heartbeats table:
+        sqlx::query("TRUNCATE TABLE heartbeats;")
+            .execute(&mut transaction)
+            .await?;
+
+        save_last_rewarded_end_time(&mut transaction, &scheduler.reward_period.end).await?;
+        save_next_rewarded_end_time(&mut transaction, &scheduler.next_reward_period().end).await?;
+        transaction.commit().await?;
+
+        // now that the db has been purged, safe to write out the manifest
         self.reward_manifests
             .write(
                 RewardManifest {
@@ -185,18 +195,6 @@ impl VerifierDaemon {
 
         self.reward_manifests.commit().await?;
 
-        let mut transaction = self.pool.begin().await?;
-
-        // Clear the heartbeats table:
-        sqlx::query("TRUNCATE TABLE heartbeats;")
-            .execute(&mut transaction)
-            .await?;
-
-        save_last_rewarded_end_time(&mut transaction, &scheduler.reward_period.end).await?;
-        save_next_rewarded_end_time(&mut transaction, &scheduler.next_reward_period().end).await?;
-
-        transaction.commit().await?;
-
         Ok(())
     }
 }