Rename storage-rocksdb to partition-store

Cargo.toml

@@ -49,6 +49,7 @@ restate-network = { path = "crates/network" }
 restate-node = { path = "crates/node" }
 restate-node-protocol = { path = "crates/node-protocol" }
 restate-node-services = { path = "crates/node-services" }
+restate-partition-store = { path = "crates/partition-store" }
 restate-queue = { path = "crates/queue" }
 restate-rocksdb = { path = "crates/rocksdb" }
 restate-schema = { path = "crates/schema" }
@@ -60,7 +61,6 @@ restate-service-protocol = { path = "crates/service-protocol" }
 restate-storage-api = { path = "crates/storage-api" }
 restate-storage-query-datafusion = { path = "crates/storage-query-datafusion" }
 restate-storage-query-postgres = { path = "crates/storage-query-postgres" }
-restate-storage-rocksdb = { path = "crates/storage-rocksdb" }
 restate-test-util = { path = "crates/test-util" }
 restate-timer = { path = "crates/timer" }
 restate-timer-queue = { path = "crates/timer-queue" }

crates/storage-rocksdb/Cargo.toml → crates/partition-store/Cargo.toml

@@ -1,5 +1,5 @@
 [package]
-name = "restate-storage-rocksdb"
+name = "restate-partition-store"
 version.workspace = true
 authors.workspace = true
 edition.workspace = true

crates/storage-rocksdb/benches/basic_benchmark.rs → crates/partition-store/benches/basic_benchmark.rs

@@ -12,12 +12,12 @@ use std::ops::RangeInclusive;
 
 use criterion::{criterion_group, criterion_main, Criterion};
 use restate_core::TaskCenterBuilder;
+use restate_partition_store::{OpenMode, PartitionStore, PartitionStoreManager};
 use restate_rocksdb::RocksDbManager;
 use restate_storage_api::deduplication_table::{
     DedupSequenceNumber, DeduplicationTable, ProducerId,
 };
 use restate_storage_api::Transaction;
-use restate_storage_rocksdb::{OpenMode, PartitionStore, PartitionStoreManager};
 use restate_types::arc_util::Constant;
 use restate_types::config::{CommonOptions, WorkerOptions};
 use restate_types::identifiers::PartitionKey;

crates/storage-rocksdb/src/keys.rs → crates/partition-store/src/keys.rs

@@ -143,7 +143,7 @@ pub trait TableKey: Sized + std::fmt::Debug + Send + 'static {
 /// This macro expands to:
 /// ```ignore
 /// use bytes::{Buf, BufMut, Bytes};
-/// use restate_storage_rocksdb::TableKind;
+/// use restate_partition_store::TableKind;
 /// #[derive(Debug, Eq, PartialEq)]
 /// pub struct FooBarKey {
 ///     pub foo: Option<u32>,

crates/storage-rocksdb/tests/inbox_table_test/mod.rs → crates/partition-store/tests/inbox_table_test/mod.rs

@@ -10,9 +10,9 @@
 
 use crate::{assert_stream_eq, mock_state_mutation};
 use once_cell::sync::Lazy;
+use restate_partition_store::PartitionStore;
 use restate_storage_api::inbox_table::{InboxEntry, InboxTable, SequenceNumberInboxEntry};
 use restate_storage_api::Transaction;
-use restate_storage_rocksdb::PartitionStore;
 use restate_types::identifiers::{InvocationId, ServiceId};
 
 static INBOX_ENTRIES: Lazy<Vec<SequenceNumberInboxEntry>> = Lazy::new(|| {

crates/storage-rocksdb/tests/integration_test.rs → crates/partition-store/tests/integration_test.rs

@@ -11,9 +11,9 @@
 use bytes::Bytes;
 use futures::Stream;
 use restate_core::TaskCenterBuilder;
+use restate_partition_store::{OpenMode, PartitionStore, PartitionStoreManager};
 use restate_rocksdb::RocksDbManager;
 use restate_storage_api::StorageError;
-use restate_storage_rocksdb::{OpenMode, PartitionStore, PartitionStoreManager};
 use restate_types::arc_util::Constant;
 use restate_types::config::{CommonOptions, WorkerOptions};
 use restate_types::identifiers::{InvocationId, PartitionKey, ServiceId};

crates/storage-rocksdb/tests/invocation_status_table_test/mod.rs → crates/partition-store/tests/invocation_status_table_test/mod.rs

@@ -16,11 +16,11 @@ use super::assert_stream_eq;
 
 use bytestring::ByteString;
 use once_cell::sync::Lazy;
+use restate_partition_store::PartitionStore;
 use restate_storage_api::invocation_status_table::{
     InFlightInvocationMetadata, InvocationStatus, InvocationStatusTable, JournalMetadata,
     StatusTimestamps,
 };
-use restate_storage_rocksdb::PartitionStore;
 use restate_types::identifiers::InvocationId;
 use restate_types::invocation::{
     HandlerType, InvocationTarget, ServiceInvocationSpanContext, Source,

crates/storage-rocksdb/tests/journal_table_test/mod.rs → crates/partition-store/tests/journal_table_test/mod.rs

@@ -12,9 +12,9 @@ use bytes::Bytes;
 use bytestring::ByteString;
 use futures_util::StreamExt;
 use once_cell::sync::Lazy;
+use restate_partition_store::PartitionStore;
 use restate_storage_api::journal_table::{JournalEntry, JournalTable};
 use restate_storage_api::Transaction;
-use restate_storage_rocksdb::PartitionStore;
 use restate_types::identifiers::{InvocationId, InvocationUuid};
 use restate_types::invocation::{InvocationTarget, ServiceInvocationSpanContext};
 use restate_types::journal::enriched::{

crates/storage-rocksdb/tests/outbox_table_test/mod.rs → crates/partition-store/tests/outbox_table_test/mod.rs

@@ -9,9 +9,9 @@
 // by the Apache License, Version 2.0.
 
 use crate::mock_random_service_invocation;
+use restate_partition_store::PartitionStore;
 use restate_storage_api::outbox_table::{OutboxMessage, OutboxTable};
 use restate_storage_api::Transaction;
-use restate_storage_rocksdb::PartitionStore;
 
 fn mock_outbox_message() -> OutboxMessage {
     OutboxMessage::ServiceInvocation(mock_random_service_invocation())

crates/storage-rocksdb/tests/state_table_test/mod.rs → crates/partition-store/tests/state_table_test/mod.rs

@@ -10,9 +10,9 @@
 
 use crate::{assert_stream_eq, storage_test_environment};
 use bytes::Bytes;
+use restate_partition_store::PartitionStore;
 use restate_storage_api::state_table::{ReadOnlyStateTable, StateTable};
 use restate_storage_api::Transaction;
-use restate_storage_rocksdb::PartitionStore;
 use restate_types::identifiers::ServiceId;
 
 async fn populate_data<T: StateTable>(table: &mut T) {

crates/storage-rocksdb/tests/timer_table_test/mod.rs → crates/partition-store/tests/timer_table_test/mod.rs

@@ -10,9 +10,9 @@
 
 use crate::mock_service_invocation;
 use futures_util::StreamExt;
+use restate_partition_store::PartitionStore;
 use restate_storage_api::timer_table::{Timer, TimerKey, TimerTable};
 use restate_storage_api::Transaction;
-use restate_storage_rocksdb::PartitionStore;
 use restate_types::identifiers::{InvocationUuid, ServiceId};
 use restate_types::invocation::ServiceInvocation;
 use std::pin::pin;

crates/storage-rocksdb/tests/virtual_object_status_table_test/mod.rs → crates/partition-store/tests/virtual_object_status_table_test/mod.rs

@@ -8,8 +8,8 @@
 // the Business Source License, use of this software will be governed
 // by the Apache License, Version 2.0.
 
+use restate_partition_store::PartitionStore;
 use restate_storage_api::service_status_table::{VirtualObjectStatus, VirtualObjectStatusTable};
-use restate_storage_rocksdb::PartitionStore;
 use restate_types::identifiers::{InvocationId, InvocationUuid, ServiceId};
 
 const FIXTURE_INVOCATION: InvocationUuid =

crates/storage-query-datafusion/Cargo.toml

@@ -12,12 +12,12 @@ default = []
 options_schema = ["dep:schemars"]
 
 [dependencies]
-restate-storage-rocksdb = { workspace = true }
-restate-types = { workspace = true }
+restate-invoker-api = { workspace = true }
+restate-partition-store = { workspace = true }
 restate-schema-api = { workspace = true, features = ["deployment"] }
 restate-service-protocol = { workspace = true, features = ["codec"] }
 restate-storage-api = { workspace = true }
-restate-invoker-api = { workspace = true }
+restate-types = { workspace = true }
 
 ahash = { workspace = true }                                                    # Required to due a yanked version used by datafusion
 async-trait = { workspace = true }

crates/storage-query-datafusion/src/context.rs

@@ -19,9 +19,9 @@ use datafusion::physical_plan::SendableRecordBatchStream;
 use datafusion::prelude::{SessionConfig, SessionContext};
 
 use restate_invoker_api::StatusHandle;
+use restate_partition_store::PartitionStore;
 use restate_schema_api::deployment::DeploymentResolver;
 use restate_schema_api::service::ServiceMetadataResolver;
-use restate_storage_rocksdb::PartitionStore;
 use restate_types::config::QueryEngineOptions;
 
 use crate::{analyzer, physical_optimizer};

crates/storage-query-datafusion/src/inbox/table.rs

@@ -23,9 +23,9 @@ use datafusion::physical_plan::stream::RecordBatchReceiverStream;
 use datafusion::physical_plan::SendableRecordBatchStream;
 pub use datafusion_expr::UserDefinedLogicalNode;
 use futures::{Stream, StreamExt};
+use restate_partition_store::PartitionStore;
 use restate_storage_api::inbox_table::{InboxTable, SequenceNumberInboxEntry};
 use restate_storage_api::StorageError;
-use restate_storage_rocksdb::PartitionStore;
 use restate_types::identifiers::PartitionKey;
 use tokio::sync::mpsc::Sender;
 

crates/storage-query-datafusion/src/invocation_status/row.rs

@@ -10,10 +10,10 @@
 
 use crate::invocation_status::schema::{InvocationStatusBuilder, InvocationStatusRowBuilder};
 use crate::table_util::format_using;
+use restate_partition_store::invocation_status_table::OwnedInvocationStatusRow;
 use restate_storage_api::invocation_status_table::{
     InFlightInvocationMetadata, InvocationStatus, JournalMetadata, StatusTimestamps,
 };
-use restate_storage_rocksdb::invocation_status_table::OwnedInvocationStatusRow;
 use restate_types::identifiers::InvocationId;
 use restate_types::invocation::{ServiceType, Source, TraceId};
 

crates/storage-query-datafusion/src/invocation_status/table.rs

@@ -22,8 +22,8 @@ use crate::invocation_status::schema::InvocationStatusBuilder;
 use datafusion::physical_plan::stream::RecordBatchReceiverStream;
 use datafusion::physical_plan::SendableRecordBatchStream;
 pub use datafusion_expr::UserDefinedLogicalNode;
-use restate_storage_rocksdb::invocation_status_table::OwnedInvocationStatusRow;
-use restate_storage_rocksdb::PartitionStore;
+use restate_partition_store::invocation_status_table::OwnedInvocationStatusRow;
+use restate_partition_store::PartitionStore;
 use restate_types::identifiers::PartitionKey;
 use tokio::sync::mpsc::Sender;
 

crates/storage-query-datafusion/src/journal/row.rs

@@ -12,8 +12,8 @@ use crate::journal::schema::JournalBuilder;
 
 use restate_service_protocol::codec::ProtobufRawEntryCodec;
 
+use restate_partition_store::journal_table::OwnedJournalRow;
 use restate_storage_api::journal_table::JournalEntry;
-use restate_storage_rocksdb::journal_table::OwnedJournalRow;
 use restate_types::identifiers::WithPartitionKey;
 use restate_types::journal::enriched::{EnrichedEntryHeader, EnrichedRawEntry};
 

crates/storage-query-datafusion/src/journal/table.rs

@@ -22,8 +22,8 @@ use crate::journal::schema::JournalBuilder;
 use datafusion::physical_plan::stream::RecordBatchReceiverStream;
 use datafusion::physical_plan::SendableRecordBatchStream;
 pub use datafusion_expr::UserDefinedLogicalNode;
-use restate_storage_rocksdb::journal_table::OwnedJournalRow;
-use restate_storage_rocksdb::PartitionStore;
+use restate_partition_store::journal_table::OwnedJournalRow;
+use restate_partition_store::PartitionStore;
 use restate_types::identifiers::PartitionKey;
 use tokio::sync::mpsc::Sender;
 

crates/storage-query-datafusion/src/keyed_service_status/row.rs

@@ -10,8 +10,8 @@
 
 use crate::keyed_service_status::schema::KeyedServiceStatusBuilder;
 use crate::table_util::format_using;
+use restate_partition_store::service_status_table::OwnedVirtualObjectStatusRow;
 use restate_storage_api::service_status_table::VirtualObjectStatus;
-use restate_storage_rocksdb::service_status_table::OwnedVirtualObjectStatusRow;
 
 #[inline]
 pub(crate) fn append_virtual_object_status_row(

crates/storage-query-datafusion/src/keyed_service_status/table.rs

@@ -22,8 +22,8 @@ use crate::keyed_service_status::schema::KeyedServiceStatusBuilder;
 use datafusion::physical_plan::stream::RecordBatchReceiverStream;
 use datafusion::physical_plan::SendableRecordBatchStream;
 pub use datafusion_expr::UserDefinedLogicalNode;
-use restate_storage_rocksdb::service_status_table::OwnedVirtualObjectStatusRow;
-use restate_storage_rocksdb::PartitionStore;
+use restate_partition_store::service_status_table::OwnedVirtualObjectStatusRow;
+use restate_partition_store::PartitionStore;
 use restate_types::identifiers::PartitionKey;
 use tokio::sync::mpsc::Sender;
 

crates/storage-query-datafusion/src/mocks.rs

@@ -17,12 +17,12 @@ use googletest::matcher::{Matcher, MatcherResult};
 use restate_core::task_center;
 use restate_invoker_api::status_handle::mocks::MockStatusHandle;
 use restate_invoker_api::StatusHandle;
+use restate_partition_store::{OpenMode, PartitionStore, PartitionStoreManager};
 use restate_rocksdb::RocksDbManager;
 use restate_schema_api::deployment::mocks::MockDeploymentMetadataRegistry;
 use restate_schema_api::deployment::{Deployment, DeploymentResolver};
 use restate_schema_api::service::mocks::MockServiceMetadataResolver;
 use restate_schema_api::service::{ServiceMetadata, ServiceMetadataResolver};
-use restate_storage_rocksdb::{OpenMode, PartitionStore, PartitionStoreManager};
 use restate_types::arc_util::Constant;
 use restate_types::config::{CommonOptions, QueryEngineOptions, WorkerOptions};
 use restate_types::identifiers::{DeploymentId, PartitionKey, ServiceRevision};