feat: add ephemeral store (#1723)

Author: ogenev

crates/storage/src/versioned/ephemeral_v1/config.rs

@@ -0,0 +1,32 @@
+use std::path::PathBuf;
+
+use ethportal_api::types::network::Subnetwork;
+use r2d2::Pool;
+use r2d2_sqlite::SqliteConnectionManager;
+
+use crate::{versioned::ContentType, PortalStorageConfig};
+
+/// The config for the EphemeralV1Store
+#[derive(Clone, Debug)]
+pub struct EphemeralV1StoreConfig {
+    pub content_type: ContentType,
+    pub subnetwork: Subnetwork,
+    pub node_data_dir: PathBuf,
+    pub sql_connection_pool: Pool<SqliteConnectionManager>,
+}
+
+#[allow(unused)]
+impl EphemeralV1StoreConfig {
+    pub fn new(
+        content_type: ContentType,
+        subnetwork: Subnetwork,
+        config: PortalStorageConfig,
+    ) -> Self {
+        Self {
+            content_type,
+            subnetwork,
+            node_data_dir: config.node_data_dir,
+            sql_connection_pool: config.sql_connection_pool,
+        }
+    }
+}

crates/storage/src/versioned/ephemeral_v1/mod.rs

@@ -0,0 +1,7 @@
+mod config;
+pub(super) mod sql;
+mod store;
+
+pub use config::EphemeralV1StoreConfig;
+#[allow(unused)]
+pub use store::EphemeralV1Store;

crates/storage/src/versioned/ephemeral_v1/sql.rs

@@ -0,0 +1,78 @@
+use crate::versioned::ContentType;
+
+/// The name of the sql table. The `eph1` stands for `ephemeral_v1`.
+pub fn table_name(content_type: &ContentType) -> String {
+    format!("eph1_{content_type}")
+}
+
+pub fn create_table(content_type: &ContentType) -> String {
+    format!(
+        "
+        CREATE TABLE IF NOT EXISTS {0} (
+            content_id BLOB PRIMARY KEY,
+            content_key BLOB NOT NULL,
+            content_value BLOB NOT NULL,
+            type INTEGER NOT NULL,
+            slot INTEGER NOT NULL,
+            content_size INTEGER NOT NULL
+        );
+        CREATE INDEX IF NOT EXISTS {0}_type_idx ON {0} (type);
+        CREATE INDEX IF NOT EXISTS {0}_slot_idx ON {0} (slot);
+        CREATE INDEX IF NOT EXISTS {0}_content_size_idx ON {0} (content_size);
+        ",
+        table_name(content_type)
+    )
+}
+
+pub fn insert(content_type: &ContentType) -> String {
+    format!(
+        "
+        INSERT OR IGNORE INTO {} (
+            content_id,
+            content_key,
+            content_value,
+            type,
+            slot,
+            content_size
+        )
+        VALUES (
+            :content_id,
+            :content_key,
+            :content_value,
+            :type,
+            :slot,
+            :content_size
+        )",
+        table_name(content_type)
+    )
+}
+
+pub fn delete(content_type: &ContentType) -> String {
+    format!(
+        "DELETE FROM {}
+        WHERE content_id = :content_id
+        RETURNING content_size",
+        table_name(content_type)
+    )
+}
+
+pub fn lookup_key(content_type: &ContentType) -> String {
+    format!(
+        "SELECT content_key FROM {} WHERE content_id = :content_id LIMIT 1",
+        table_name(content_type)
+    )
+}
+
+pub fn lookup_value(content_type: &ContentType) -> String {
+    format!(
+        "SELECT content_value FROM {} WHERE content_id = :content_id LIMIT 1",
+        table_name(content_type)
+    )
+}
+
+pub fn entry_count_and_size(content_type: &ContentType) -> String {
+    format!(
+        "SELECT COUNT(*) as count, TOTAL(content_size) as used_capacity FROM {}",
+        table_name(content_type)
+    )
+}