Export non existence query logic (#403)

Author: nikkolasg

mp2-v1/Cargo.toml

@@ -22,10 +22,15 @@ serde.workspace = true
 mp2_common = { path = "../mp2-common" }
 recursion_framework = { path = "../recursion-framework" }
 ryhope = { path = "../ryhope" }
+parsil = { path = "../parsil" }
 verifiable-db = { path = "../verifiable-db" }
 derive_more = "0.99.18"
 hex.workspace = true
 serde_json.workspace = true
+bb8 = "0.8.5"
+bb8-postgres = "0.8.1"
+tokio-postgres = "0.7.12"
+futures = "0.3.30"
 
 [dev-dependencies]
 alloy.workspace = true

mp2-v1/src/lib.rs

@@ -1,10 +1,12 @@
 //! Circuits for v1 of Lagrange Proof Network (LPN)
-
+#![allow(incomplete_features)]
 // Add this to allow generic const expressions, e.g. `PAD_LEN(NODE_LEN)`.
 #![feature(generic_const_exprs)]
 // Add this so we don't need to always specify const generic in generic
 // parameters (i.e. use "_")
 #![feature(generic_arg_infer)]
+// stylistic feature
+#![feature(async_closure)]
 use mp2_common::mpt_sequential::PAD_LEN;
 
 pub const MAX_BRANCH_NODE_LEN: usize = 532;
@@ -21,4 +23,5 @@ pub mod contract_extraction;
 pub mod final_extraction;
 pub mod indexing;
 pub mod length_extraction;
+pub mod query;
 pub mod values_extraction;

mp2-v1/src/query/mod.rs

@@ -0,0 +1 @@
+pub mod planner;

mp2-v1/src/query/planner.rs

@@ -1,22 +1,15 @@
 use alloy::primitives::U256;
 use anyhow::Result;
 use mp2_common::{
-    eth::{left_pad_generic, BlockUtil},
+    eth::BlockUtil,
     proof::deserialize_proof,
-    u256,
     utils::{Endianness, Packer, ToFields},
     C, D, F,
 };
 use mp2_v1::{api, block_extraction, indexing::block::BlockPrimaryIndex};
 
 use super::TestContext;
 
-pub(crate) fn block_number_to_u256_limbs(number: u64) -> Vec<F> {
-    const NUM_LIMBS: usize = u256::NUM_LIMBS;
-    let block_number_buff = number.to_be_bytes();
-    left_pad_generic::<u32, NUM_LIMBS>(&block_number_buff.pack(Endianness::Big)).to_fields()
-}
-
 impl TestContext {
     pub(crate) async fn prove_block_extraction(&self, bn: BlockPrimaryIndex) -> Result<Vec<u8>> {
         let block = self

mp2-v1/tests/common/block_extraction.rs

@@ -69,9 +69,6 @@ pub(crate) const MAPPING_VALUE_COLUMN: &str = "map_value";
 pub(crate) const MAPPING_KEY_COLUMN: &str = "map_key";
 
 impl TableIndexing {
-    pub fn table(&self) -> &Table {
-        &self.table
-    }
     pub(crate) async fn merge_table_test_case(
         ctx: &mut TestContext,
     ) -> Result<(Self, Vec<TableRowUpdate<BlockPrimaryIndex>>)> {
@@ -429,7 +426,7 @@ impl TableIndexing {
         updates: Vec<TableRowUpdate<BlockPrimaryIndex>>,
         expected_metadata_hash: &HashOutput,
     ) -> Result<()> {
-        let current_block = ctx.block_number().await;
+        let current_block = ctx.block_number().await as BlockPrimaryIndex;
         // apply the new cells to the trees
         // NOTE ONLY the rest of the cells, not including the secondary one !
         let mut rows_update = Vec::new();
@@ -462,7 +459,7 @@ impl TableIndexing {
                     let row_payload = ctx
                         .prove_cells_tree(
                             &self.table,
-                            current_block as usize,
+                            current_block,
                             previous_row,
                             new_cell_collection,
                             tree_update,
@@ -495,7 +492,7 @@ impl TableIndexing {
                     let row_payload = ctx
                         .prove_cells_tree(
                             &self.table,
-                            current_block as usize,
+                            current_block,
                             Row {
                                 k: new_cells.previous_row_key.clone(),
                                 payload: old_row,

mp2-v1/tests/common/cases/indexing.rs

@@ -121,7 +121,7 @@ impl TreeInfo<RowTreeKey, RowPayload<BlockPrimaryIndex>>
         planner: &mut QueryPlanner<'a>,
         primary: BlockPrimaryIndex,
         k: &RowTreeKey,
-        v: &RowPayload<BlockPrimaryIndex>,
+        _v: &RowPayload<BlockPrimaryIndex>,
     ) -> Result<Option<Vec<u8>>> {
         // TODO export that in single function
         Ok(if self.is_satisfying_query(k) {
@@ -274,7 +274,7 @@ impl TreeInfo<BlockPrimaryIndex, IndexNode<BlockPrimaryIndex>>
         &self,
         ctx: &mut TestContext,
         query_id: &QueryID,
-        primary: BlockPrimaryIndex,
+        _primary: BlockPrimaryIndex,
         key: &BlockPrimaryIndex,
         placeholder_values: PlaceholderValues,
         proof: Vec<u8>,
@@ -345,7 +345,7 @@ impl<'b> TreeInfo<BlockPrimaryIndex, IndexNode<BlockPrimaryIndex>> for IndexInfo
         &self,
         ctx: &mut TestContext,
         query_id: &QueryID,
-        primary: BlockPrimaryIndex,
+        _primary: BlockPrimaryIndex,
         key: &BlockPrimaryIndex,
         placeholder_values: PlaceholderValues,
         proof: Vec<u8>,