feat: update types

lfglabs-dev · Mar 6, 2024 · e68b5cb · e68b5cb
1 parent c5d72e6
commit e68b5cb
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/src/models.rs b/src/models.rs
@@ -1,5 +1,4 @@
 use mongodb::{bson, Database};
-use mongodb::bson::DateTime;
 use serde::{Deserialize, Serialize};
 use serde_json::Value;
 use starknet::{
@@ -54,7 +53,7 @@ pub_struct!(Deserialize; CompletedTasks {
 pub struct CompletedTaskDocument {
     address: String,
     task_id: u32,
-    timestamp: DateTime,
+    timestamp: i64,
 }
 
 pub_struct!(Serialize; Reward {
@@ -95,7 +94,7 @@ pub_struct!(Deserialize; VerifyAchievementQuery {
 pub_struct!(Debug, Serialize, Deserialize; AchievedDocument {
     addr: String,
     achievement_id: u32,
-    timestamp:DateTime,
+    timestamp:i64,
 });
 
 pub_struct!(Debug, Serialize, Deserialize; AchievementDocument {

diff --git a/src/utils.rs b/src/utils.rs
@@ -125,9 +125,9 @@ impl CompletedTasksTrait for AppState {
     ) -> Result<UpdateResult, mongodb::error::Error> {
         let completed_tasks_collection: Collection<CompletedTasks> =
             self.db.collection("completed_tasks");
-        let timestamp = Utc::now().timestamp_millis() as f64;
+        let created_at = Utc::now().timestamp_millis();
         let filter = doc! { "address": addr.to_string(), "task_id": task_id };
-        let update = doc! { "$setOnInsert": { "address": addr.to_string(), "task_id": task_id , "timestamp":timestamp} };
+        let update = doc! { "$setOnInsert": { "address": addr.to_string(), "task_id": task_id , "timestamp":created_at} };
         let options = UpdateOptions::builder().upsert(true).build();
 
         let result = completed_tasks_collection
@@ -292,10 +292,11 @@ impl AchievementsTrait for AppState {
         achievement_id: u32,
     ) -> Result<UpdateResult, mongodb::error::Error> {
         let achieved_collection: Collection<CompletedTasks> = self.db.collection("achieved");
+        let created_at = Utc::now().timestamp_millis();
         let filter = doc! { "addr": addr.to_string(), "achievement_id": achievement_id };
         let timestamp = Utc::now().timestamp_millis() as f64;
         let update =
-            doc! { "$setOnInsert": { "addr": addr.to_string(), "achievement_id": achievement_id , "timestamp":timestamp } };
+            doc! { "$setOnInsert": { "addr": addr.to_string(), "achievement_id": achievement_id , "timestamp":created_at } };
         let options = UpdateOptions::builder().upsert(true).build();
 
         let result = achieved_collection