Fix bug

qiaoruntao · Jun 21, 2023 · 4b54b27 · 4b54b27
1 parent 7c88f62
commit 4b54b27
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/tasker/consumer.rs b/src/tasker/consumer.rs
@@ -401,6 +401,7 @@ impl<T: DeserializeOwned + Send + Unpin + Sync + Clone + 'static, K: Serialize +
         doc! {
             "$match":{
                 "$and":[
+                     {"task_state.worker_states":{"$exists":true}},
                      {
                         "$or":[
                             { "task_option.min_worker_version": { "$exists": false } },
@@ -424,7 +425,7 @@ impl<T: DeserializeOwned + Send + Unpin + Sync + Clone + 'static, K: Serialize +
                                             "$filter": {
                                                 "input": "$task_state.worker_states",
                                                 "as": "item",
-                                                "cond": { "$gt": vec!["$$item.ping_expire_time", "$$NOW"] }
+                                                "cond": { "$gt": ["$$item.ping_expire_time", "$$NOW"] }
                                             }
                                         }
                                     }, "$task_option.concurrent_worker_cnt"]

diff --git a/src/tasker/producer.rs b/src/tasker/producer.rs
@@ -71,7 +71,7 @@ impl<T: Serialize, K: Serialize> TaskProducer<T, K> {
                 "key":key.as_ref(),
                 "task_state.create_time":now,
                 "task_state.start_time":start_time,
-                "task_state.worker_states":Bson::Array(vec![]),
+                "task_state.worker_states":[],
                 "task_option":to_document(&task_option).unwrap(),
             },
         };