1use crate::partition::Partition;
2use crate::types::{
3 AttemptIndex, BudgetId, EdgeId, ExecutionId, FlowId, LaneId, QuotaPolicyId, SignalId,
4 WaitpointId, WaitpointKey, WorkerInstanceId,
5};
6
7pub struct ExecKeyContext {
19 tag: String,
21 eid: String,
23}
24
25impl ExecKeyContext {
26 pub fn new(partition: &Partition, eid: &ExecutionId) -> Self {
27 Self {
28 tag: partition.hash_tag(),
29 eid: eid.to_string(),
30 }
31 }
32
33 pub fn core(&self) -> String {
37 format!("ff:exec:{}:{}:core", self.tag, self.eid)
38 }
39
40 pub fn payload(&self) -> String {
42 format!("ff:exec:{}:{}:payload", self.tag, self.eid)
43 }
44
45 pub fn result(&self) -> String {
47 format!("ff:exec:{}:{}:result", self.tag, self.eid)
48 }
49
50 pub fn policy(&self) -> String {
52 format!("ff:exec:{}:{}:policy", self.tag, self.eid)
53 }
54
55 pub fn tags(&self) -> String {
57 format!("ff:exec:{}:{}:tags", self.tag, self.eid)
58 }
59
60 pub fn lease_current(&self) -> String {
64 format!("ff:exec:{}:{}:lease:current", self.tag, self.eid)
65 }
66
67 pub fn lease_history(&self) -> String {
69 format!("ff:exec:{}:{}:lease:history", self.tag, self.eid)
70 }
71
72 pub fn claim_grant(&self) -> String {
74 format!("ff:exec:{}:{}:claim_grant", self.tag, self.eid)
75 }
76
77 pub fn attempts(&self) -> String {
81 format!("ff:exec:{}:{}:attempts", self.tag, self.eid)
82 }
83
84 pub fn attempt_hash(&self, index: AttemptIndex) -> String {
86 format!("ff:attempt:{}:{}:{}", self.tag, self.eid, index)
87 }
88
89 pub fn attempt_usage(&self, index: AttemptIndex) -> String {
91 format!("ff:attempt:{}:{}:{}:usage", self.tag, self.eid, index)
92 }
93
94 pub fn attempt_policy(&self, index: AttemptIndex) -> String {
96 format!("ff:attempt:{}:{}:{}:policy", self.tag, self.eid, index)
97 }
98
99 pub fn stream(&self, index: AttemptIndex) -> String {
103 format!("ff:stream:{}:{}:{}", self.tag, self.eid, index)
104 }
105
106 pub fn stream_meta(&self, index: AttemptIndex) -> String {
108 format!("ff:stream:{}:{}:{}:meta", self.tag, self.eid, index)
109 }
110
111 pub fn suspension_current(&self) -> String {
115 format!("ff:exec:{}:{}:suspension:current", self.tag, self.eid)
116 }
117
118 pub fn waitpoints(&self) -> String {
120 format!("ff:exec:{}:{}:waitpoints", self.tag, self.eid)
121 }
122
123 pub fn waitpoint(&self, wp_id: &WaitpointId) -> String {
125 format!("ff:wp:{}:{}", self.tag, wp_id)
126 }
127
128 pub fn waitpoint_signals(&self, wp_id: &WaitpointId) -> String {
130 format!("ff:wp:{}:{}:signals", self.tag, wp_id)
131 }
132
133 pub fn waitpoint_condition(&self, wp_id: &WaitpointId) -> String {
135 format!("ff:wp:{}:{}:condition", self.tag, wp_id)
136 }
137
138 pub fn signal(&self, sig_id: &SignalId) -> String {
142 format!("ff:signal:{}:{}", self.tag, sig_id)
143 }
144
145 pub fn signal_payload(&self, sig_id: &SignalId) -> String {
147 format!("ff:signal:{}:{}:payload", self.tag, sig_id)
148 }
149
150 pub fn exec_signals(&self) -> String {
152 format!("ff:exec:{}:{}:signals", self.tag, self.eid)
153 }
154
155 pub fn signal_dedup(&self, wp_id: &WaitpointId, idempotency_key: &str) -> String {
157 format!("ff:sigdedup:{}:{}:{}", self.tag, wp_id, idempotency_key)
158 }
159
160 pub fn deps_meta(&self) -> String {
164 format!("ff:exec:{}:{}:deps:meta", self.tag, self.eid)
165 }
166
167 pub fn dep_edge(&self, edge_id: &EdgeId) -> String {
169 format!("ff:exec:{}:{}:dep:{}", self.tag, self.eid, edge_id)
170 }
171
172 pub fn deps_unresolved(&self) -> String {
174 format!("ff:exec:{}:{}:deps:unresolved", self.tag, self.eid)
175 }
176
177 pub fn deps_all_edges(&self) -> String {
182 format!("ff:exec:{}:{}:deps:all_edges", self.tag, self.eid)
183 }
184
185 pub fn noop(&self) -> String {
191 format!("ff:noop:{}", self.tag)
192 }
193
194 pub fn hash_tag(&self) -> &str {
195 &self.tag
196 }
197
198 pub fn execution_id_str(&self) -> &str {
199 &self.eid
200 }
201}
202
203pub struct IndexKeys {
207 tag: String,
208}
209
210impl IndexKeys {
211 pub fn new(partition: &Partition) -> Self {
212 Self {
213 tag: partition.hash_tag(),
214 }
215 }
216
217 pub fn lane_eligible(&self, lane_id: &LaneId) -> String {
219 format!("ff:idx:{}:lane:{}:eligible", self.tag, lane_id)
220 }
221
222 pub fn lane_delayed(&self, lane_id: &LaneId) -> String {
224 format!("ff:idx:{}:lane:{}:delayed", self.tag, lane_id)
225 }
226
227 pub fn lane_active(&self, lane_id: &LaneId) -> String {
229 format!("ff:idx:{}:lane:{}:active", self.tag, lane_id)
230 }
231
232 pub fn lane_terminal(&self, lane_id: &LaneId) -> String {
234 format!("ff:idx:{}:lane:{}:terminal", self.tag, lane_id)
235 }
236
237 pub fn lane_blocked_dependencies(&self, lane_id: &LaneId) -> String {
239 format!("ff:idx:{}:lane:{}:blocked:dependencies", self.tag, lane_id)
240 }
241
242 pub fn lane_blocked_budget(&self, lane_id: &LaneId) -> String {
244 format!("ff:idx:{}:lane:{}:blocked:budget", self.tag, lane_id)
245 }
246
247 pub fn lane_blocked_quota(&self, lane_id: &LaneId) -> String {
249 format!("ff:idx:{}:lane:{}:blocked:quota", self.tag, lane_id)
250 }
251
252 pub fn lane_blocked_route(&self, lane_id: &LaneId) -> String {
254 format!("ff:idx:{}:lane:{}:blocked:route", self.tag, lane_id)
255 }
256
257 pub fn lane_blocked_operator(&self, lane_id: &LaneId) -> String {
259 format!("ff:idx:{}:lane:{}:blocked:operator", self.tag, lane_id)
260 }
261
262 pub fn lane_suspended(&self, lane_id: &LaneId) -> String {
264 format!("ff:idx:{}:lane:{}:suspended", self.tag, lane_id)
265 }
266
267 pub fn waitpoint_hmac_secrets(&self) -> String {
276 format!("ff:sec:{}:waitpoint_hmac", self.tag)
277 }
278
279 pub fn lease_expiry(&self) -> String {
281 format!("ff:idx:{}:lease_expiry", self.tag)
282 }
283
284 pub fn worker_leases(&self, wid: &WorkerInstanceId) -> String {
286 format!("ff:idx:{}:worker:{}:leases", self.tag, wid)
287 }
288
289 pub fn suspension_timeout(&self) -> String {
291 format!("ff:idx:{}:suspension_timeout", self.tag)
292 }
293
294 pub fn pending_waitpoint_expiry(&self) -> String {
296 format!("ff:idx:{}:pending_waitpoint_expiry", self.tag)
297 }
298
299 pub fn attempt_timeout(&self) -> String {
301 format!("ff:idx:{}:attempt_timeout", self.tag)
302 }
303
304 pub fn execution_deadline(&self) -> String {
306 format!("ff:idx:{}:execution_deadline", self.tag)
307 }
308
309 pub fn all_executions(&self) -> String {
311 format!("ff:idx:{}:all_executions", self.tag)
312 }
313}
314
315pub struct FlowKeyContext {
319 tag: String,
320 fid: String,
321}
322
323impl FlowKeyContext {
324 pub fn new(partition: &Partition, fid: &FlowId) -> Self {
325 Self {
326 tag: partition.hash_tag(),
327 fid: fid.to_string(),
328 }
329 }
330
331 pub fn core(&self) -> String {
333 format!("ff:flow:{}:{}:core", self.tag, self.fid)
334 }
335
336 pub fn members(&self) -> String {
338 format!("ff:flow:{}:{}:members", self.tag, self.fid)
339 }
340
341 pub fn member(&self, eid: &ExecutionId) -> String {
343 format!("ff:flow:{}:{}:member:{}", self.tag, self.fid, eid)
344 }
345
346 pub fn edge(&self, edge_id: &EdgeId) -> String {
348 format!("ff:flow:{}:{}:edge:{}", self.tag, self.fid, edge_id)
349 }
350
351 pub fn outgoing(&self, upstream_eid: &ExecutionId) -> String {
353 format!("ff:flow:{}:{}:out:{}", self.tag, self.fid, upstream_eid)
354 }
355
356 pub fn incoming(&self, downstream_eid: &ExecutionId) -> String {
358 format!("ff:flow:{}:{}:in:{}", self.tag, self.fid, downstream_eid)
359 }
360
361 pub fn events(&self) -> String {
363 format!("ff:flow:{}:{}:events", self.tag, self.fid)
364 }
365
366 pub fn summary(&self) -> String {
368 format!("ff:flow:{}:{}:summary", self.tag, self.fid)
369 }
370
371 pub fn grant(&self, mutation_id: &str) -> String {
373 format!("ff:flow:{}:{}:grant:{}", self.tag, self.fid, mutation_id)
374 }
375
376 pub fn hash_tag(&self) -> &str {
377 &self.tag
378 }
379}
380
381pub struct FlowIndexKeys {
383 tag: String,
384}
385
386impl FlowIndexKeys {
387 pub fn new(partition: &Partition) -> Self {
388 Self {
389 tag: partition.hash_tag(),
390 }
391 }
392
393 pub fn flow_index(&self) -> String {
396 format!("ff:idx:{}:flow_index", self.tag)
397 }
398}
399
400pub struct BudgetKeyContext {
404 tag: String,
405 bid: String,
406}
407
408impl BudgetKeyContext {
409 pub fn new(partition: &Partition, bid: &BudgetId) -> Self {
410 Self {
411 tag: partition.hash_tag(),
412 bid: bid.to_string(),
413 }
414 }
415
416 pub fn definition(&self) -> String {
418 format!("ff:budget:{}:{}", self.tag, self.bid)
419 }
420
421 pub fn limits(&self) -> String {
423 format!("ff:budget:{}:{}:limits", self.tag, self.bid)
424 }
425
426 pub fn usage(&self) -> String {
428 format!("ff:budget:{}:{}:usage", self.tag, self.bid)
429 }
430
431 pub fn executions(&self) -> String {
433 format!("ff:budget:{}:{}:executions", self.tag, self.bid)
434 }
435
436 pub fn hash_tag(&self) -> &str {
437 &self.tag
438 }
439}
440
441pub fn budget_attach_key(tag: &str, scope_type: &str, scope_id: &str) -> String {
443 format!("ff:budget_attach:{}:{}:{}", tag, scope_type, scope_id)
444}
445
446pub fn budget_resets_key(tag: &str) -> String {
448 format!("ff:idx:{}:budget_resets", tag)
449}
450
451pub fn budget_policies_index(tag: &str) -> String {
454 format!("ff:idx:{}:budget_policies", tag)
455}
456
457pub struct QuotaKeyContext {
461 tag: String,
462 qid: String,
463}
464
465impl QuotaKeyContext {
466 pub fn new(partition: &Partition, qid: &QuotaPolicyId) -> Self {
467 Self {
468 tag: partition.hash_tag(),
469 qid: qid.to_string(),
470 }
471 }
472
473 pub fn definition(&self) -> String {
475 format!("ff:quota:{}:{}", self.tag, self.qid)
476 }
477
478 pub fn window(&self, dimension: &str) -> String {
480 format!("ff:quota:{}:{}:window:{}", self.tag, self.qid, dimension)
481 }
482
483 pub fn concurrency(&self) -> String {
485 format!("ff:quota:{}:{}:concurrency", self.tag, self.qid)
486 }
487
488 pub fn admitted(&self, eid: &ExecutionId) -> String {
490 format!("ff:quota:{}:{}:admitted:{}", self.tag, self.qid, eid)
491 }
492
493 pub fn admitted_set(&self) -> String {
496 format!("ff:quota:{}:{}:admitted_set", self.tag, self.qid)
497 }
498
499 pub fn hash_tag(&self) -> &str {
500 &self.tag
501 }
502}
503
504pub fn quota_policies_index(tag: &str) -> String {
507 format!("ff:idx:{}:quota_policies", tag)
508}
509
510pub fn quota_attach_key(tag: &str, scope_type: &str, scope_id: &str) -> String {
512 format!("ff:quota_attach:{}:{}:{}", tag, scope_type, scope_id)
513}
514
515pub fn lane_config_key(lane_id: &LaneId) -> String {
519 format!("ff:lane:{}:config", lane_id)
520}
521
522pub fn lane_counts_key(lane_id: &LaneId) -> String {
524 format!("ff:lane:{}:counts", lane_id)
525}
526
527pub fn worker_key(wid: &WorkerInstanceId) -> String {
529 format!("ff:worker:{}", wid)
530}
531
532pub fn worker_caps_key(wid: &WorkerInstanceId) -> String {
540 format!("ff:worker:{}:caps", wid)
541}
542
543pub fn workers_index_key() -> String {
550 "ff:idx:workers".to_owned()
551}
552
553pub fn workers_capability_key(key: &str, value: &str) -> String {
555 format!("ff:idx:workers:cap:{}:{}", key, value)
556}
557
558pub fn lanes_index_key() -> String {
560 "ff:idx:lanes".to_owned()
561}
562
563pub fn global_config_partitions() -> String {
566 "ff:config:partitions".to_owned()
567}
568
569pub fn namespace_executions_key(namespace: &str) -> String {
573 format!("ff:ns:{}:executions", namespace)
574}
575
576pub fn idempotency_key(tag: &str, namespace: &str, idem_key: &str) -> String {
581 format!("ff:idem:{}:{}:{}", tag, namespace, idem_key)
582}
583
584pub fn noop_key(tag: &str) -> String {
590 format!("ff:noop:{}", tag)
591}
592
593pub fn tag_index_key(namespace: &str, key: &str, value: &str) -> String {
595 format!("ff:tag:{}:{}:{}", namespace, key, value)
596}
597
598pub fn waitpoint_key_resolution(tag: &str, wp_key: &WaitpointKey) -> String {
600 format!("ff:wpkey:{}:{}", tag, wp_key)
601}
602
603pub const USAGE_DEDUP_KEY_PREFIX: &str = "ff:usagededup:";
609
610pub fn usage_dedup_key(hash_tag: &str, dedup_id: &str) -> String {
616 format!("{USAGE_DEDUP_KEY_PREFIX}{hash_tag}:{dedup_id}")
617}
618
619#[cfg(test)]
620mod tests {
621 use super::*;
622 use crate::partition::{execution_partition, flow_partition, PartitionConfig};
623
624 #[test]
625 fn exec_key_context_core_format() {
626 let config = PartitionConfig::default();
627 let eid = ExecutionId::parse("{fp:0}:550e8400-e29b-41d4-a716-446655440000").unwrap();
628 let partition = execution_partition(&eid, &config);
629 let ctx = ExecKeyContext::new(&partition, &eid);
630
631 let core_key = ctx.core();
632 assert!(core_key.starts_with("ff:exec:{fp:"));
634 assert!(core_key.ends_with(":core"));
635 assert!(core_key.contains("550e8400-e29b-41d4-a716-446655440000"));
636 }
637
638 #[test]
639 fn exec_key_all_keys_share_hash_tag() {
640 let config = PartitionConfig::default();
641 let eid = ExecutionId::for_flow(&FlowId::new(), &config);
642 let partition = execution_partition(&eid, &config);
643 let ctx = ExecKeyContext::new(&partition, &eid);
644 let tag = ctx.hash_tag();
645
646 assert!(ctx.core().contains(tag));
648 assert!(ctx.payload().contains(tag));
649 assert!(ctx.result().contains(tag));
650 assert!(ctx.policy().contains(tag));
651 assert!(ctx.lease_current().contains(tag));
652 assert!(ctx.lease_history().contains(tag));
653 assert!(ctx.attempts().contains(tag));
654 assert!(ctx.suspension_current().contains(tag));
655 assert!(ctx.exec_signals().contains(tag));
656 }
657
658 #[test]
659 fn attempt_key_includes_index() {
660 let config = PartitionConfig::default();
661 let eid = ExecutionId::for_flow(&FlowId::new(), &config);
662 let partition = execution_partition(&eid, &config);
663 let ctx = ExecKeyContext::new(&partition, &eid);
664
665 let key = ctx.attempt_hash(AttemptIndex::new(3));
666 assert!(key.contains(":3"), "attempt key should contain index");
667 }
668
669 #[test]
670 fn stream_key_format() {
671 let config = PartitionConfig::default();
672 let eid = ExecutionId::for_flow(&FlowId::new(), &config);
673 let partition = execution_partition(&eid, &config);
674 let ctx = ExecKeyContext::new(&partition, &eid);
675
676 let key = ctx.stream(AttemptIndex::new(0));
677 assert!(key.starts_with("ff:stream:{fp:"));
678 assert!(key.ends_with(":0"));
679 }
680
681 #[test]
682 fn index_keys_format() {
683 let config = PartitionConfig::default();
684 let eid = ExecutionId::for_flow(&FlowId::new(), &config);
685 let partition = execution_partition(&eid, &config);
686 let idx = IndexKeys::new(&partition);
687 let lane = LaneId::new("default");
688
689 assert!(idx.lane_eligible(&lane).contains(":lane:default:eligible"));
690 assert!(idx.lane_delayed(&lane).contains(":lane:default:delayed"));
691 assert!(idx.lease_expiry().contains(":lease_expiry"));
692 assert!(idx.all_executions().contains(":all_executions"));
693 }
694
695 #[test]
696 fn flow_key_context_format() {
697 let config = PartitionConfig::default();
698 let fid = FlowId::new();
699 let partition = flow_partition(&fid, &config);
700 let ctx = FlowKeyContext::new(&partition, &fid);
701
702 assert!(ctx.core().starts_with("ff:flow:{fp:"));
703 assert!(ctx.core().ends_with(":core"));
704 assert!(ctx.members().ends_with(":members"));
705 }
706
707 #[test]
708 fn global_keys_no_hash_tag() {
709 let lane = LaneId::new("default");
710 let key = lane_config_key(&lane);
711 assert_eq!(key, "ff:lane:default:config");
712 assert!(!key.contains('{'));
713 }
714
715 #[test]
716 fn usage_dedup_key_format() {
717 let key = usage_dedup_key("{bp:7}", "dedup-123");
719 assert_eq!(key, "ff:usagededup:{bp:7}:dedup-123");
720 assert!(key.starts_with(USAGE_DEDUP_KEY_PREFIX));
721 assert_eq!(key.matches('{').count(), 1);
723 assert_eq!(key.matches('}').count(), 1);
724 }
725}