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 stream_summary(&self, index: AttemptIndex) -> String {
118 format!("ff:attempt:{}:{}:{}:summary", self.tag, self.eid, index)
119 }
120
121 pub fn suspension_current(&self) -> String {
125 format!("ff:exec:{}:{}:suspension:current", self.tag, self.eid)
126 }
127
128 pub fn suspension_satisfied_set(&self) -> String {
134 format!(
135 "ff:exec:{}:{}:suspension:current:satisfied_set",
136 self.tag, self.eid
137 )
138 }
139
140 pub fn suspension_member_map(&self) -> String {
145 format!(
146 "ff:exec:{}:{}:suspension:current:member_map",
147 self.tag, self.eid
148 )
149 }
150
151 pub fn waitpoints(&self) -> String {
153 format!("ff:exec:{}:{}:waitpoints", self.tag, self.eid)
154 }
155
156 pub fn waitpoint(&self, wp_id: &WaitpointId) -> String {
158 format!("ff:wp:{}:{}", self.tag, wp_id)
159 }
160
161 pub fn waitpoint_signals(&self, wp_id: &WaitpointId) -> String {
163 format!("ff:wp:{}:{}:signals", self.tag, wp_id)
164 }
165
166 pub fn waitpoint_condition(&self, wp_id: &WaitpointId) -> String {
168 format!("ff:wp:{}:{}:condition", self.tag, wp_id)
169 }
170
171 pub fn signal(&self, sig_id: &SignalId) -> String {
175 format!("ff:signal:{}:{}", self.tag, sig_id)
176 }
177
178 pub fn signal_payload(&self, sig_id: &SignalId) -> String {
180 format!("ff:signal:{}:{}:payload", self.tag, sig_id)
181 }
182
183 pub fn exec_signals(&self) -> String {
185 format!("ff:exec:{}:{}:signals", self.tag, self.eid)
186 }
187
188 pub fn signal_dedup(&self, wp_id: &WaitpointId, idempotency_key: &str) -> String {
190 format!("ff:sigdedup:{}:{}:{}", self.tag, wp_id, idempotency_key)
191 }
192
193 pub fn suspend_dedup(&self, idempotency_key: &str) -> String {
200 format!(
201 "ff:dedup:suspend:{}:{}:{}",
202 self.tag, self.eid, idempotency_key
203 )
204 }
205
206 pub fn deps_meta(&self) -> String {
210 format!("ff:exec:{}:{}:deps:meta", self.tag, self.eid)
211 }
212
213 pub fn dep_edge(&self, edge_id: &EdgeId) -> String {
215 format!("ff:exec:{}:{}:dep:{}", self.tag, self.eid, edge_id)
216 }
217
218 pub fn deps_unresolved(&self) -> String {
220 format!("ff:exec:{}:{}:deps:unresolved", self.tag, self.eid)
221 }
222
223 pub fn deps_all_edges(&self) -> String {
228 format!("ff:exec:{}:{}:deps:all_edges", self.tag, self.eid)
229 }
230
231 pub fn noop(&self) -> String {
237 format!("ff:noop:{}", self.tag)
238 }
239
240 pub fn hash_tag(&self) -> &str {
241 &self.tag
242 }
243
244 pub fn execution_id_str(&self) -> &str {
245 &self.eid
246 }
247}
248
249pub struct IndexKeys {
253 tag: String,
254}
255
256impl IndexKeys {
257 pub fn new(partition: &Partition) -> Self {
258 Self {
259 tag: partition.hash_tag(),
260 }
261 }
262
263 pub fn lane_eligible(&self, lane_id: &LaneId) -> String {
265 format!("ff:idx:{}:lane:{}:eligible", self.tag, lane_id)
266 }
267
268 pub fn lane_delayed(&self, lane_id: &LaneId) -> String {
270 format!("ff:idx:{}:lane:{}:delayed", self.tag, lane_id)
271 }
272
273 pub fn lane_active(&self, lane_id: &LaneId) -> String {
275 format!("ff:idx:{}:lane:{}:active", self.tag, lane_id)
276 }
277
278 pub fn lane_terminal(&self, lane_id: &LaneId) -> String {
280 format!("ff:idx:{}:lane:{}:terminal", self.tag, lane_id)
281 }
282
283 pub fn lane_blocked_dependencies(&self, lane_id: &LaneId) -> String {
285 format!("ff:idx:{}:lane:{}:blocked:dependencies", self.tag, lane_id)
286 }
287
288 pub fn lane_blocked_budget(&self, lane_id: &LaneId) -> String {
290 format!("ff:idx:{}:lane:{}:blocked:budget", self.tag, lane_id)
291 }
292
293 pub fn lane_blocked_quota(&self, lane_id: &LaneId) -> String {
295 format!("ff:idx:{}:lane:{}:blocked:quota", self.tag, lane_id)
296 }
297
298 pub fn lane_blocked_route(&self, lane_id: &LaneId) -> String {
300 format!("ff:idx:{}:lane:{}:blocked:route", self.tag, lane_id)
301 }
302
303 pub fn lane_blocked_operator(&self, lane_id: &LaneId) -> String {
305 format!("ff:idx:{}:lane:{}:blocked:operator", self.tag, lane_id)
306 }
307
308 pub fn lane_suspended(&self, lane_id: &LaneId) -> String {
310 format!("ff:idx:{}:lane:{}:suspended", self.tag, lane_id)
311 }
312
313 pub fn waitpoint_hmac_secrets(&self) -> String {
322 format!("ff:sec:{}:waitpoint_hmac", self.tag)
323 }
324
325 pub fn lease_expiry(&self) -> String {
327 format!("ff:idx:{}:lease_expiry", self.tag)
328 }
329
330 pub fn worker_leases(&self, wid: &WorkerInstanceId) -> String {
332 format!("ff:idx:{}:worker:{}:leases", self.tag, wid)
333 }
334
335 pub fn suspension_timeout(&self) -> String {
337 format!("ff:idx:{}:suspension_timeout", self.tag)
338 }
339
340 pub fn pending_waitpoint_expiry(&self) -> String {
342 format!("ff:idx:{}:pending_waitpoint_expiry", self.tag)
343 }
344
345 pub fn attempt_timeout(&self) -> String {
347 format!("ff:idx:{}:attempt_timeout", self.tag)
348 }
349
350 pub fn execution_deadline(&self) -> String {
352 format!("ff:idx:{}:execution_deadline", self.tag)
353 }
354
355 pub fn all_executions(&self) -> String {
357 format!("ff:idx:{}:all_executions", self.tag)
358 }
359}
360
361pub struct FlowKeyContext {
365 tag: String,
366 fid: String,
367}
368
369impl FlowKeyContext {
370 pub fn new(partition: &Partition, fid: &FlowId) -> Self {
371 Self {
372 tag: partition.hash_tag(),
373 fid: fid.to_string(),
374 }
375 }
376
377 pub fn core(&self) -> String {
379 format!("ff:flow:{}:{}:core", self.tag, self.fid)
380 }
381
382 pub fn members(&self) -> String {
384 format!("ff:flow:{}:{}:members", self.tag, self.fid)
385 }
386
387 pub fn tags(&self) -> String {
392 format!("ff:flow:{}:{}:tags", self.tag, self.fid)
393 }
394
395 pub fn member(&self, eid: &ExecutionId) -> String {
397 format!("ff:flow:{}:{}:member:{}", self.tag, self.fid, eid)
398 }
399
400 pub fn edge(&self, edge_id: &EdgeId) -> String {
402 format!("ff:flow:{}:{}:edge:{}", self.tag, self.fid, edge_id)
403 }
404
405 pub fn outgoing(&self, upstream_eid: &ExecutionId) -> String {
407 format!("ff:flow:{}:{}:out:{}", self.tag, self.fid, upstream_eid)
408 }
409
410 pub fn incoming(&self, downstream_eid: &ExecutionId) -> String {
412 format!("ff:flow:{}:{}:in:{}", self.tag, self.fid, downstream_eid)
413 }
414
415 pub fn edgegroup(&self, downstream_eid: &ExecutionId) -> String {
421 format!(
422 "ff:flow:{}:{}:edgegroup:{}",
423 self.tag, self.fid, downstream_eid
424 )
425 }
426
427 pub fn events(&self) -> String {
429 format!("ff:flow:{}:{}:events", self.tag, self.fid)
430 }
431
432 pub fn summary(&self) -> String {
434 format!("ff:flow:{}:{}:summary", self.tag, self.fid)
435 }
436
437 pub fn grant(&self, mutation_id: &str) -> String {
439 format!("ff:flow:{}:{}:grant:{}", self.tag, self.fid, mutation_id)
440 }
441
442 pub fn pending_cancels(&self) -> String {
449 format!("ff:flow:{}:{}:pending_cancels", self.tag, self.fid)
450 }
451
452 pub fn hash_tag(&self) -> &str {
453 &self.tag
454 }
455
456 pub fn flow_id(&self) -> &str {
458 &self.fid
459 }
460}
461
462pub struct FlowIndexKeys {
464 tag: String,
465}
466
467impl FlowIndexKeys {
468 pub fn new(partition: &Partition) -> Self {
469 Self {
470 tag: partition.hash_tag(),
471 }
472 }
473
474 pub fn flow_index(&self) -> String {
477 format!("ff:idx:{}:flow_index", self.tag)
478 }
479
480 pub fn cancel_backlog(&self) -> String {
488 format!("ff:idx:{}:cancel_backlog", self.tag)
489 }
490
491 pub fn pending_cancel_groups(&self) -> String {
501 format!("ff:idx:{}:pending_cancel_groups", self.tag)
502 }
503}
504
505pub struct BudgetKeyContext {
509 tag: String,
510 bid: String,
511}
512
513impl BudgetKeyContext {
514 pub fn new(partition: &Partition, bid: &BudgetId) -> Self {
515 Self {
516 tag: partition.hash_tag(),
517 bid: bid.to_string(),
518 }
519 }
520
521 pub fn definition(&self) -> String {
523 format!("ff:budget:{}:{}", self.tag, self.bid)
524 }
525
526 pub fn limits(&self) -> String {
528 format!("ff:budget:{}:{}:limits", self.tag, self.bid)
529 }
530
531 pub fn usage(&self) -> String {
533 format!("ff:budget:{}:{}:usage", self.tag, self.bid)
534 }
535
536 pub fn executions(&self) -> String {
538 format!("ff:budget:{}:{}:executions", self.tag, self.bid)
539 }
540
541 pub fn hash_tag(&self) -> &str {
542 &self.tag
543 }
544}
545
546pub fn budget_attach_key(tag: &str, scope_type: &str, scope_id: &str) -> String {
548 format!("ff:budget_attach:{}:{}:{}", tag, scope_type, scope_id)
549}
550
551pub fn budget_resets_key(tag: &str) -> String {
553 format!("ff:idx:{}:budget_resets", tag)
554}
555
556pub fn budget_policies_index(tag: &str) -> String {
559 format!("ff:idx:{}:budget_policies", tag)
560}
561
562pub struct QuotaKeyContext {
566 tag: String,
567 qid: String,
568}
569
570impl QuotaKeyContext {
571 pub fn new(partition: &Partition, qid: &QuotaPolicyId) -> Self {
572 Self {
573 tag: partition.hash_tag(),
574 qid: qid.to_string(),
575 }
576 }
577
578 pub fn definition(&self) -> String {
580 format!("ff:quota:{}:{}", self.tag, self.qid)
581 }
582
583 pub fn window(&self, dimension: &str) -> String {
585 format!("ff:quota:{}:{}:window:{}", self.tag, self.qid, dimension)
586 }
587
588 pub fn concurrency(&self) -> String {
590 format!("ff:quota:{}:{}:concurrency", self.tag, self.qid)
591 }
592
593 pub fn admitted(&self, eid: &ExecutionId) -> String {
595 format!("ff:quota:{}:{}:admitted:{}", self.tag, self.qid, eid)
596 }
597
598 pub fn admitted_set(&self) -> String {
601 format!("ff:quota:{}:{}:admitted_set", self.tag, self.qid)
602 }
603
604 pub fn hash_tag(&self) -> &str {
605 &self.tag
606 }
607}
608
609pub fn quota_policies_index(tag: &str) -> String {
612 format!("ff:idx:{}:quota_policies", tag)
613}
614
615pub fn quota_attach_key(tag: &str, scope_type: &str, scope_id: &str) -> String {
617 format!("ff:quota_attach:{}:{}:{}", tag, scope_type, scope_id)
618}
619
620pub fn lane_config_key(lane_id: &LaneId) -> String {
624 format!("ff:lane:{}:config", lane_id)
625}
626
627pub fn lane_counts_key(lane_id: &LaneId) -> String {
629 format!("ff:lane:{}:counts", lane_id)
630}
631
632pub fn worker_key(wid: &WorkerInstanceId) -> String {
634 format!("ff:worker:{}", wid)
635}
636
637pub fn worker_caps_key(wid: &WorkerInstanceId) -> String {
645 format!("ff:worker:{}:caps", wid)
646}
647
648pub fn workers_index_key() -> String {
655 "ff:idx:workers".to_owned()
656}
657
658pub fn workers_capability_key(key: &str, value: &str) -> String {
660 format!("ff:idx:workers:cap:{}:{}", key, value)
661}
662
663pub fn lanes_index_key() -> String {
665 "ff:idx:lanes".to_owned()
666}
667
668pub fn global_config_partitions() -> String {
671 "ff:config:partitions".to_owned()
672}
673
674pub fn namespace_executions_key(namespace: &str) -> String {
678 format!("ff:ns:{}:executions", namespace)
679}
680
681pub fn idempotency_key(tag: &str, namespace: &str, idem_key: &str) -> String {
686 format!("ff:idem:{}:{}:{}", tag, namespace, idem_key)
687}
688
689pub fn noop_key(tag: &str) -> String {
695 format!("ff:noop:{}", tag)
696}
697
698pub fn tag_index_key(namespace: &str, key: &str, value: &str) -> String {
700 format!("ff:tag:{}:{}:{}", namespace, key, value)
701}
702
703pub fn waitpoint_key_resolution(tag: &str, wp_key: &WaitpointKey) -> String {
705 format!("ff:wpkey:{}:{}", tag, wp_key)
706}
707
708pub const USAGE_DEDUP_KEY_PREFIX: &str = "ff:usagededup:";
714
715pub fn usage_dedup_key(hash_tag: &str, dedup_id: &str) -> String {
721 format!("{USAGE_DEDUP_KEY_PREFIX}{hash_tag}:{dedup_id}")
722}
723
724#[cfg(test)]
725mod tests {
726 use super::*;
727 use crate::partition::{execution_partition, flow_partition, PartitionConfig};
728
729 #[test]
730 fn exec_key_context_core_format() {
731 let config = PartitionConfig::default();
732 let eid = ExecutionId::parse("{fp:0}:550e8400-e29b-41d4-a716-446655440000").unwrap();
733 let partition = execution_partition(&eid, &config);
734 let ctx = ExecKeyContext::new(&partition, &eid);
735
736 let core_key = ctx.core();
737 assert!(core_key.starts_with("ff:exec:{fp:"));
739 assert!(core_key.ends_with(":core"));
740 assert!(core_key.contains("550e8400-e29b-41d4-a716-446655440000"));
741 }
742
743 #[test]
744 fn exec_key_all_keys_share_hash_tag() {
745 let config = PartitionConfig::default();
746 let eid = ExecutionId::for_flow(&FlowId::new(), &config);
747 let partition = execution_partition(&eid, &config);
748 let ctx = ExecKeyContext::new(&partition, &eid);
749 let tag = ctx.hash_tag();
750
751 assert!(ctx.core().contains(tag));
753 assert!(ctx.payload().contains(tag));
754 assert!(ctx.result().contains(tag));
755 assert!(ctx.policy().contains(tag));
756 assert!(ctx.lease_current().contains(tag));
757 assert!(ctx.lease_history().contains(tag));
758 assert!(ctx.attempts().contains(tag));
759 assert!(ctx.suspension_current().contains(tag));
760 assert!(ctx.exec_signals().contains(tag));
761 }
762
763 #[test]
764 fn attempt_key_includes_index() {
765 let config = PartitionConfig::default();
766 let eid = ExecutionId::for_flow(&FlowId::new(), &config);
767 let partition = execution_partition(&eid, &config);
768 let ctx = ExecKeyContext::new(&partition, &eid);
769
770 let key = ctx.attempt_hash(AttemptIndex::new(3));
771 assert!(key.contains(":3"), "attempt key should contain index");
772 }
773
774 #[test]
775 fn stream_key_format() {
776 let config = PartitionConfig::default();
777 let eid = ExecutionId::for_flow(&FlowId::new(), &config);
778 let partition = execution_partition(&eid, &config);
779 let ctx = ExecKeyContext::new(&partition, &eid);
780
781 let key = ctx.stream(AttemptIndex::new(0));
782 assert!(key.starts_with("ff:stream:{fp:"));
783 assert!(key.ends_with(":0"));
784 }
785
786 #[test]
787 fn index_keys_format() {
788 let config = PartitionConfig::default();
789 let eid = ExecutionId::for_flow(&FlowId::new(), &config);
790 let partition = execution_partition(&eid, &config);
791 let idx = IndexKeys::new(&partition);
792 let lane = LaneId::new("default");
793
794 assert!(idx.lane_eligible(&lane).contains(":lane:default:eligible"));
795 assert!(idx.lane_delayed(&lane).contains(":lane:default:delayed"));
796 assert!(idx.lease_expiry().contains(":lease_expiry"));
797 assert!(idx.all_executions().contains(":all_executions"));
798 }
799
800 #[test]
801 fn flow_key_context_format() {
802 let config = PartitionConfig::default();
803 let fid = FlowId::new();
804 let partition = flow_partition(&fid, &config);
805 let ctx = FlowKeyContext::new(&partition, &fid);
806
807 assert!(ctx.core().starts_with("ff:flow:{fp:"));
808 assert!(ctx.core().ends_with(":core"));
809 assert!(ctx.members().ends_with(":members"));
810 }
811
812 #[test]
813 fn global_keys_no_hash_tag() {
814 let lane = LaneId::new("default");
815 let key = lane_config_key(&lane);
816 assert_eq!(key, "ff:lane:default:config");
817 assert!(!key.contains('{'));
818 }
819
820 #[test]
821 fn usage_dedup_key_format() {
822 let key = usage_dedup_key("{bp:7}", "dedup-123");
824 assert_eq!(key, "ff:usagededup:{bp:7}:dedup-123");
825 assert!(key.starts_with(USAGE_DEDUP_KEY_PREFIX));
826 assert_eq!(key.matches('{').count(), 1);
828 assert_eq!(key.matches('}').count(), 1);
829 }
830}