add loop tracking datastructure

hydro-project · Jan 14, 2025 · 1763b7a · 1763b7a
1 parent 43aacd2
commit 1763b7a
Show file tree

Hide file tree

Showing 34 changed files with 102 additions and 6 deletions.
diff --git a/...re/src/snapshots/dfir_datalog_core__tests__aggregations_and_comments@datalog_program.snap b/...re/src/snapshots/dfir_datalog_core__tests__aggregations_and_comments@datalog_program.snap
@@ -108,6 +108,7 @@ fn main() {
                 var_expr!(),
                 var_expr!(hoff_1v3_send, hoff_9v3_send, hoff_23v1_send),
                 false,
+                None,
                 move |
                     context,
                     var_args!(),
@@ -371,6 +372,7 @@ fn main() {
                 var_expr!(hoff_9v3_recv, hoff_25v1_recv),
                 var_expr!(),
                 false,
+                None,
                 move |context, var_args!(hoff_9v3_recv, hoff_25v1_recv), var_args!()| {
                     let mut hoff_9v3_recv = hoff_9v3_recv.borrow_mut_swap();
                     let hoff_9v3_recv = hoff_9v3_recv.drain(..);
@@ -688,6 +690,7 @@ fn main() {
                 var_expr!(hoff_23v1_recv),
                 var_expr!(),
                 false,
+                None,
                 move |context, var_args!(hoff_23v1_recv), var_args!()| {
                     let mut hoff_23v1_recv = hoff_23v1_recv.borrow_mut_swap();
                     let hoff_23v1_recv = hoff_23v1_recv.drain(..);
@@ -905,6 +908,7 @@ fn main() {
                 var_expr!(hoff_1v3_recv),
                 var_expr!(hoff_7v3_send),
                 false,
+                None,
                 move |context, var_args!(hoff_1v3_recv), var_args!(hoff_7v3_send)| {
                     let mut hoff_1v3_recv = hoff_1v3_recv.borrow_mut_swap();
                     let hoff_1v3_recv = hoff_1v3_recv.drain(..);
@@ -1008,6 +1012,7 @@ fn main() {
                 var_expr!(hoff_7v3_recv),
                 var_expr!(hoff_6v3_send),
                 false,
+                None,
                 move |context, var_args!(hoff_7v3_recv), var_args!(hoff_6v3_send)| {
                     let mut hoff_7v3_recv = hoff_7v3_recv.borrow_mut_swap();
                     let hoff_7v3_recv = hoff_7v3_recv.drain(..);
@@ -1071,6 +1076,7 @@ fn main() {
                 var_expr!(hoff_6v3_recv),
                 var_expr!(hoff_25v1_send),
                 false,
+                None,
                 move |context, var_args!(hoff_6v3_recv), var_args!(hoff_25v1_send)| {
                     let mut hoff_6v3_recv = hoff_6v3_recv.borrow_mut_swap();
                     let hoff_6v3_recv = hoff_6v3_recv.drain(..);

diff --git a/...src/snapshots/dfir_datalog_core__tests__aggregations_fold_keyed_expr@datalog_program.snap b/...src/snapshots/dfir_datalog_core__tests__aggregations_fold_keyed_expr@datalog_program.snap
@@ -62,6 +62,7 @@ fn main() {
                 var_expr!(),
                 var_expr!(hoff_6v3_send),
                 false,
+                None,
                 move |context, var_args!(), var_args!(hoff_6v3_send)| {
                     let hoff_6v3_send = hydroflow::pusherator::for_each::ForEach::new(|
                         v|
@@ -212,6 +213,7 @@ fn main() {
                 var_expr!(hoff_6v3_recv),
                 var_expr!(),
                 false,
+                None,
                 move |context, var_args!(hoff_6v3_recv), var_args!()| {
                     let mut hoff_6v3_recv = hoff_6v3_recv.borrow_mut_swap();
                     let hoff_6v3_recv = hoff_6v3_recv.drain(..);

diff --git a/dfir_datalog_core/src/snapshots/dfir_datalog_core__tests__anti_join@datalog_program.snap b/dfir_datalog_core/src/snapshots/dfir_datalog_core__tests__anti_join@datalog_program.snap
@@ -152,6 +152,7 @@ fn main() {
                 var_expr!(),
                 var_expr!(hoff_12v3_send),
                 false,
+                None,
                 move |context, var_args!(), var_args!(hoff_12v3_send)| {
                     let hoff_12v3_send = hydroflow::pusherator::for_each::ForEach::new(|
                         v|
@@ -302,6 +303,7 @@ fn main() {
                 var_expr!(),
                 var_expr!(hoff_9v3_send),
                 false,
+                None,
                 move |context, var_args!(), var_args!(hoff_9v3_send)| {
                     let hoff_9v3_send = hydroflow::pusherator::for_each::ForEach::new(|
                         v|
@@ -366,6 +368,7 @@ fn main() {
                 var_expr!(),
                 var_expr!(hoff_6v3_send),
                 false,
+                None,
                 move |context, var_args!(), var_args!(hoff_6v3_send)| {
                     let hoff_6v3_send = hydroflow::pusherator::for_each::ForEach::new(|
                         v|
@@ -430,6 +433,7 @@ fn main() {
                 var_expr!(hoff_6v3_recv, hoff_9v3_recv, hoff_12v3_recv),
                 var_expr!(),
                 false,
+                None,
                 move |
                     context,
                     var_args!(hoff_6v3_recv, hoff_9v3_recv, hoff_12v3_recv),

diff --git a/dfir_datalog_core/src/snapshots/dfir_datalog_core__tests__collect_vec@datalog_program.snap b/dfir_datalog_core/src/snapshots/dfir_datalog_core__tests__collect_vec@datalog_program.snap
@@ -106,6 +106,7 @@ fn main() {
                 var_expr!(),
                 var_expr!(hoff_9v3_send),
                 false,
+                None,
                 move |context, var_args!(), var_args!(hoff_9v3_send)| {
                     let hoff_9v3_send = hydroflow::pusherator::for_each::ForEach::new(|
                         v|
@@ -536,6 +537,7 @@ fn main() {
                 var_expr!(hoff_9v3_recv),
                 var_expr!(),
                 false,
+                None,
                 move |context, var_args!(hoff_9v3_recv), var_args!()| {
                     let mut hoff_9v3_recv = hoff_9v3_recv.borrow_mut_swap();
                     let hoff_9v3_recv = hoff_9v3_recv.drain(..);

diff --git a/dfir_datalog_core/src/snapshots/dfir_datalog_core__tests__detuple@datalog_program.snap b/dfir_datalog_core/src/snapshots/dfir_datalog_core__tests__detuple@datalog_program.snap
@@ -51,6 +51,7 @@ fn main() {
                 var_expr!(),
                 var_expr!(),
                 false,
+                None,
                 move |context, var_args!(), var_args!()| {
                     let op_7v1 = std::iter::from_fn(|| {
                         match hydroflow::futures::stream::Stream::poll_next(

diff --git a/...talog_core/src/snapshots/dfir_datalog_core__tests__detuple_then_flat@datalog_program.snap b/...talog_core/src/snapshots/dfir_datalog_core__tests__detuple_then_flat@datalog_program.snap
@@ -51,6 +51,7 @@ fn main() {
                 var_expr!(),
                 var_expr!(),
                 false,
+                None,
                 move |context, var_args!(), var_args!()| {
                     let op_7v1 = std::iter::from_fn(|| {
                         match hydroflow::futures::stream::Stream::poll_next(

diff --git a/dfir_datalog_core/src/snapshots/dfir_datalog_core__tests__expr_lhs@datalog_program.snap b/dfir_datalog_core/src/snapshots/dfir_datalog_core__tests__expr_lhs@datalog_program.snap
@@ -84,6 +84,7 @@ fn main() {
                     hoff_23v1_send, hoff_24v1_send
                 ),
                 false,
+                None,
                 move |
                     context,
                     var_args!(),
@@ -278,6 +279,7 @@ fn main() {
                 ),
                 var_expr!(),
                 false,
+                None,
                 move |
                     context,
                     var_args!(

diff --git a/..._datalog_core/src/snapshots/dfir_datalog_core__tests__expr_predicate@datalog_program.snap b/..._datalog_core/src/snapshots/dfir_datalog_core__tests__expr_predicate@datalog_program.snap
@@ -71,6 +71,7 @@ fn main() {
                 var_expr!(),
                 var_expr!(hoff_1v3_send, hoff_6v3_send, hoff_21v1_send, hoff_22v1_send),
                 false,
+                None,
                 move |
                     context,
                     var_args!(),
@@ -378,6 +379,7 @@ fn main() {
                 var_expr!(hoff_1v3_recv, hoff_6v3_recv, hoff_21v1_recv, hoff_22v1_recv),
                 var_expr!(),
                 false,
+                None,
                 move |
                     context,
                     var_args!(

diff --git a/...talog_core/src/snapshots/dfir_datalog_core__tests__flat_then_detuple@datalog_program.snap b/...talog_core/src/snapshots/dfir_datalog_core__tests__flat_then_detuple@datalog_program.snap
@@ -51,6 +51,7 @@ fn main() {
                 var_expr!(),
                 var_expr!(),
                 false,
+                None,
                 move |context, var_args!(), var_args!()| {
                     let op_7v1 = std::iter::from_fn(|| {
                         match hydroflow::futures::stream::Stream::poll_next(

diff --git a/dfir_datalog_core/src/snapshots/dfir_datalog_core__tests__flatten@datalog_program.snap b/dfir_datalog_core/src/snapshots/dfir_datalog_core__tests__flatten@datalog_program.snap
@@ -51,6 +51,7 @@ fn main() {
                 var_expr!(),
                 var_expr!(),
                 false,
+                None,
                 move |context, var_args!(), var_args!()| {
                     let op_7v1 = std::iter::from_fn(|| {
                         match hydroflow::futures::stream::Stream::poll_next(

diff --git a/dfir_datalog_core/src/snapshots/dfir_datalog_core__tests__index@datalog_program.snap b/dfir_datalog_core/src/snapshots/dfir_datalog_core__tests__index@datalog_program.snap
@@ -206,6 +206,7 @@ fn main() {
                 var_expr!(),
                 var_expr!(hoff_16v3_send, hoff_21v3_send),
                 false,
+                None,
                 move |context, var_args!(), var_args!(hoff_16v3_send, hoff_21v3_send)| {
                     let hoff_16v3_send = hydroflow::pusherator::for_each::ForEach::new(|
                         v|
@@ -619,6 +620,7 @@ fn main() {
                 var_expr!(hoff_12v3_recv),
                 var_expr!(hoff_15v3_send),
                 false,
+                None,
                 move |context, var_args!(hoff_12v3_recv), var_args!(hoff_15v3_send)| {
                     let mut hoff_12v3_recv = hoff_12v3_recv.borrow_mut_swap();
                     let hoff_12v3_recv = hoff_12v3_recv.drain(..);
@@ -684,6 +686,7 @@ fn main() {
                 var_expr!(hoff_6v3_recv),
                 var_expr!(hoff_9v3_send),
                 false,
+                None,
                 move |context, var_args!(hoff_6v3_recv), var_args!(hoff_9v3_send)| {
                     let mut hoff_6v3_recv = hoff_6v3_recv.borrow_mut_swap();
                     let hoff_6v3_recv = hoff_6v3_recv.drain(..);
@@ -749,6 +752,7 @@ fn main() {
                 var_expr!(hoff_21v3_recv),
                 var_expr!(),
                 false,
+                None,
                 move |context, var_args!(hoff_21v3_recv), var_args!()| {
                     let mut hoff_21v3_recv = hoff_21v3_recv.borrow_mut_swap();
                     let hoff_21v3_recv = hoff_21v3_recv.drain(..);
@@ -1026,6 +1030,7 @@ fn main() {
                 var_expr!(hoff_9v3_recv, hoff_16v3_recv),
                 var_expr!(hoff_6v3_send, hoff_10v3_send, hoff_13v3_send),
                 false,
+                None,
                 move |
                     context,
                     var_args!(hoff_9v3_recv, hoff_16v3_recv),
@@ -1589,6 +1594,7 @@ fn main() {
                 var_expr!(hoff_13v3_recv),
                 var_expr!(),
                 false,
+                None,
                 move |context, var_args!(hoff_13v3_recv), var_args!()| {
                     let mut hoff_13v3_recv = hoff_13v3_recv.borrow_mut_swap();
                     let hoff_13v3_recv = hoff_13v3_recv.drain(..);
@@ -1882,6 +1888,7 @@ fn main() {
                 var_expr!(hoff_10v3_recv, hoff_15v3_recv),
                 var_expr!(hoff_12v3_send),
                 false,
+                None,
                 move |
                     context,
                     var_args!(hoff_10v3_recv, hoff_15v3_recv),

diff --git a/...datalog_core/src/snapshots/dfir_datalog_core__tests__join_with_other@datalog_program.snap b/...datalog_core/src/snapshots/dfir_datalog_core__tests__join_with_other@datalog_program.snap
@@ -95,6 +95,7 @@ fn main() {
                 var_expr!(),
                 var_expr!(),
                 false,
+                None,
                 move |context, var_args!(), var_args!()| {
                     let op_10v1 = std::iter::from_fn(|| {
                         match hydroflow::futures::stream::Stream::poll_next(

diff --git a/..._datalog_core/src/snapshots/dfir_datalog_core__tests__join_with_self@datalog_program.snap b/..._datalog_core/src/snapshots/dfir_datalog_core__tests__join_with_self@datalog_program.snap
@@ -81,6 +81,7 @@ fn main() {
                 var_expr!(),
                 var_expr!(hoff_4v3_send, hoff_6v3_send),
                 false,
+                None,
                 move |context, var_args!(), var_args!(hoff_4v3_send, hoff_6v3_send)| {
                     let hoff_4v3_send = hydroflow::pusherator::for_each::ForEach::new(|
                         v|
@@ -233,6 +234,7 @@ fn main() {
                 var_expr!(hoff_4v3_recv, hoff_6v3_recv),
                 var_expr!(),
                 false,
+                None,
                 move |context, var_args!(hoff_4v3_recv, hoff_6v3_recv), var_args!()| {
                     let mut hoff_4v3_recv = hoff_4v3_recv.borrow_mut_swap();
                     let hoff_4v3_recv = hoff_4v3_recv.drain(..);

diff --git a/...log_core/src/snapshots/dfir_datalog_core__tests__local_constraints@datalog_program-2.snap b/...log_core/src/snapshots/dfir_datalog_core__tests__local_constraints@datalog_program-2.snap
@@ -51,6 +51,7 @@ fn main() {
                 var_expr!(),
                 var_expr!(),
                 false,
+                None,
                 move |context, var_args!(), var_args!()| {
                     let op_7v1 = std::iter::from_fn(|| {
                         match hydroflow::futures::stream::Stream::poll_next(

diff --git a/...talog_core/src/snapshots/dfir_datalog_core__tests__local_constraints@datalog_program.snap b/...talog_core/src/snapshots/dfir_datalog_core__tests__local_constraints@datalog_program.snap
@@ -51,6 +51,7 @@ fn main() {
                 var_expr!(),
                 var_expr!(),
                 false,
+                None,
                 move |context, var_args!(), var_args!()| {
                     let op_7v1 = std::iter::from_fn(|| {
                         match hydroflow::futures::stream::Stream::poll_next(

diff --git a/dfir_datalog_core/src/snapshots/dfir_datalog_core__tests__max@datalog_program.snap b/dfir_datalog_core/src/snapshots/dfir_datalog_core__tests__max@datalog_program.snap
@@ -62,6 +62,7 @@ fn main() {
                 var_expr!(),
                 var_expr!(hoff_6v3_send),
                 false,
+                None,
                 move |context, var_args!(), var_args!(hoff_6v3_send)| {
                     let hoff_6v3_send = hydroflow::pusherator::for_each::ForEach::new(|
                         v|
@@ -212,6 +213,7 @@ fn main() {
                 var_expr!(hoff_6v3_recv),
                 var_expr!(),
                 false,
+                None,
                 move |context, var_args!(hoff_6v3_recv), var_args!()| {
                     let mut hoff_6v3_recv = hoff_6v3_recv.borrow_mut_swap();
                     let hoff_6v3_recv = hoff_6v3_recv.drain(..);

diff --git a/dfir_datalog_core/src/snapshots/dfir_datalog_core__tests__max_all@datalog_program.snap b/dfir_datalog_core/src/snapshots/dfir_datalog_core__tests__max_all@datalog_program.snap
@@ -65,6 +65,7 @@ fn main() {
                 var_expr!(),
                 var_expr!(hoff_6v3_send),
                 false,
+                None,
                 move |context, var_args!(), var_args!(hoff_6v3_send)| {
                     let hoff_6v3_send = hydroflow::pusherator::for_each::ForEach::new(|
                         v|
@@ -215,6 +216,7 @@ fn main() {
                 var_expr!(hoff_6v3_recv),
                 var_expr!(),
                 false,
+                None,
                 move |context, var_args!(hoff_6v3_recv), var_args!()| {
                     let mut hoff_6v3_recv = hoff_6v3_recv.borrow_mut_swap();
                     let hoff_6v3_recv = hoff_6v3_recv.drain(..);

diff --git a/...datalog_core/src/snapshots/dfir_datalog_core__tests__minimal_program@datalog_program.snap b/...datalog_core/src/snapshots/dfir_datalog_core__tests__minimal_program@datalog_program.snap
@@ -51,6 +51,7 @@ fn main() {
                 var_expr!(),
                 var_expr!(),
                 false,
+                None,
                 move |context, var_args!(), var_args!()| {
                     let op_7v1 = std::iter::from_fn(|| {
                         match hydroflow::futures::stream::Stream::poll_next(

diff --git a/dfir_datalog_core/src/snapshots/dfir_datalog_core__tests__multi_detuple@datalog_program.snap b/dfir_datalog_core/src/snapshots/dfir_datalog_core__tests__multi_detuple@datalog_program.snap
@@ -51,6 +51,7 @@ fn main() {
                 var_expr!(),
                 var_expr!(),
                 false,
+                None,
                 move |context, var_args!(), var_args!()| {
                     let op_7v1 = std::iter::from_fn(|| {
                         match hydroflow::futures::stream::Stream::poll_next(

diff --git a/...g_core/src/snapshots/dfir_datalog_core__tests__multiple_contributors@datalog_program.snap b/...g_core/src/snapshots/dfir_datalog_core__tests__multiple_contributors@datalog_program.snap
@@ -75,6 +75,7 @@ fn main() {
                 var_expr!(),
                 var_expr!(),
                 false,
+                None,
                 move |context, var_args!(), var_args!()| {
                     let op_10v1 = std::iter::from_fn(|| {
                         match hydroflow::futures::stream::Stream::poll_next(

diff --git a/...alog_core/src/snapshots/dfir_datalog_core__tests__non_copy_but_clone@datalog_program.snap b/...alog_core/src/snapshots/dfir_datalog_core__tests__non_copy_but_clone@datalog_program.snap
@@ -51,6 +51,7 @@ fn main() {
                 var_expr!(),
                 var_expr!(),
                 false,
+                None,
                 move |context, var_args!(), var_args!()| {
                     let op_7v1 = std::iter::from_fn(|| {
                         match hydroflow::futures::stream::Stream::poll_next(