@@ -2737,8 +2737,7 @@ def transition_processing_memory(
2737
2737
2738
2738
if ws != ts ._processing_on : # someone else has this task
2739
2739
logger .info (
2740
- "Unexpected worker completed task, likely due to "
2741
- "work stealing. Expected: %s, Got: %s, Key: %s" ,
2740
+ "Unexpected worker completed task. Expected: %s, Got: %s, Key: %s" ,
2742
2741
ts ._processing_on ,
2743
2742
ws ,
2744
2743
key ,
@@ -2835,7 +2834,7 @@ def transition_memory_released(self, key, safe: bint = False):
2835
2834
worker_msg = {
2836
2835
"op" : "free-keys" ,
2837
2836
"keys" : [key ],
2838
- "reason " : f"Memory->Released { key } " ,
2837
+ "stimulus_id " : f"memory-released- { time () } " ,
2839
2838
}
2840
2839
for ws in ts ._who_has :
2841
2840
worker_msgs [ws ._address ] = [worker_msg ]
@@ -2935,7 +2934,11 @@ def transition_erred_released(self, key):
2935
2934
if dts ._state == "erred" :
2936
2935
recommendations [dts ._key ] = "waiting"
2937
2936
2938
- w_msg = {"op" : "free-keys" , "keys" : [key ], "reason" : "Erred->Released" }
2937
+ w_msg = {
2938
+ "op" : "free-keys" ,
2939
+ "keys" : [key ],
2940
+ "stimulus_id" : f"erred-released-{ time ()} " ,
2941
+ }
2939
2942
for ws_addr in ts ._erred_on :
2940
2943
worker_msgs [ws_addr ] = [w_msg ]
2941
2944
ts ._erred_on .clear ()
@@ -3013,7 +3016,7 @@ def transition_processing_released(self, key):
3013
3016
{
3014
3017
"op" : "free-keys" ,
3015
3018
"keys" : [key ],
3016
- "reason " : f"processing-released-{ time ()} " ,
3019
+ "stimulus_id " : f"processing-released-{ time ()} " ,
3017
3020
}
3018
3021
]
3019
3022
@@ -4339,9 +4342,9 @@ async def add_worker(
4339
4342
worker_msgs [address ] = []
4340
4343
worker_msgs [address ].append (
4341
4344
{
4342
- "op" : "free-keys " ,
4345
+ "op" : "remove-replicas " ,
4343
4346
"keys" : already_released_keys ,
4344
- "reason " : f"reconnect-already-released-{ time ()} " ,
4347
+ "stimulus_id " : f"reconnect-already-released-{ time ()} " ,
4345
4348
}
4346
4349
)
4347
4350
for ts in list (parent ._unrunnable ):
@@ -4767,7 +4770,7 @@ def stimulus_task_finished(self, key=None, worker=None, **kwargs):
4767
4770
{
4768
4771
"op" : "free-keys" ,
4769
4772
"keys" : [key ],
4770
- "reason " : f"already-released-or-forgotten-{ time ()} " ,
4773
+ "stimulus_id " : f"already-released-or-forgotten-{ time ()} " ,
4771
4774
}
4772
4775
]
4773
4776
elif ts ._state == "memory" :
@@ -5965,7 +5968,7 @@ async def delete_worker_data(
5965
5968
await retry_operation (
5966
5969
self .rpc (addr = worker_address ).free_keys ,
5967
5970
keys = list (keys ),
5968
- reason = "rebalance/replicate " ,
5971
+ stimulus_id = f"delete-data- { time () } " ,
5969
5972
)
5970
5973
except OSError as e :
5971
5974
# This can happen e.g. if the worker is going through controlled shutdown;
@@ -7846,7 +7849,7 @@ def _propagate_forgotten(
7846
7849
{
7847
7850
"op" : "free-keys" ,
7848
7851
"keys" : [key ],
7849
- "reason " : f"propagate-forgotten { ts . key } " ,
7852
+ "stimulus_id " : f"propagate-forgotten- { time () } " ,
7850
7853
}
7851
7854
]
7852
7855
state .remove_all_replicas (ts )
0 commit comments