Implement list_root_keys to the AdminKeyValueStore

CLI.md

@@ -55,6 +55,7 @@ This document contains the help content for the `linera` command-line program.
 * [`linera storage check_absence`↴](#linera-storage-check_absence)
 * [`linera storage initialize`↴](#linera-storage-initialize)
 * [`linera storage list_namespaces`↴](#linera-storage-list_namespaces)
+* [`linera storage list_root_keys`↴](#linera-storage-list_root_keys)
 
 ## `linera`
 
@@ -937,6 +938,7 @@ Operation on the storage
 * `check_absence` — Check absence of a namespace in the database
 * `initialize` — Initialize a namespace in the database
 * `list_namespaces` — List the namespaces of the database
+* `list_root_keys` — List the root keys of the database
 
 
 
@@ -1012,6 +1014,18 @@ List the namespaces of the database
 
 
 
+## `linera storage list_root_keys`
+
+List the root keys of the database
+
+**Usage:** `linera storage list_root_keys --storage <STORAGE_CONFIG>`
+
+###### **Options:**
+
+* `--storage <STORAGE_CONFIG>` — Storage configuration for the blockchain history
+
+
+
 <hr/>
 
 <small><i>

Cargo.toml

@@ -47,7 +47,7 @@ async-lock = "3.3.0"
 async-trait = "0.1.77"
 async-tungstenite = { version = "0.22", features = ["tokio-runtime"] }
 aws-config = "1.1.7"
-aws-sdk-dynamodb = "1.16.0"
+aws-sdk-dynamodb = "1.60.0"
 aws-sdk-s3 = "1.17.0"
 aws-smithy-http = "0.60.6"
 aws-types = "1.1.7"

linera-client/src/client_options.rs

@@ -964,6 +964,14 @@ pub enum DatabaseToolCommand {
         #[arg(long = "storage")]
         storage_config: String,
     },
+
+    /// List the root keys of the database
+    #[command(name = "list_root_keys")]
+    ListRootKeys {
+        /// Storage configuration for the blockchain history.
+        #[arg(long = "storage")]
+        storage_config: String,
+    },
 }
 
 impl DatabaseToolCommand {
@@ -975,6 +983,7 @@ impl DatabaseToolCommand {
             DatabaseToolCommand::CheckAbsence { storage_config } => storage_config,
             DatabaseToolCommand::Initialize { storage_config } => storage_config,
             DatabaseToolCommand::ListNamespaces { storage_config } => storage_config,
+            DatabaseToolCommand::ListRootKeys { storage_config } => storage_config,
         };
         Ok(storage_config.parse::<StorageConfigNamespace>()?)
     }

linera-client/src/storage.rs

@@ -554,6 +554,32 @@ impl StoreConfig {
             }
         }
     }
+
+    /// Lists all root keys of the storage
+    pub async fn list_root_keys(self) -> Result<Vec<Vec<u8>>, ViewError> {
+        match self {
+            StoreConfig::Memory(_, _) => Err(ViewError::StoreError {
+                backend: "memory".to_string(),
+                error: "list_root_keys is not supported for the memory storage".to_string(),
+            }),
+            #[cfg(feature = "storage-service")]
+            StoreConfig::Service(config, namespace) => {
+                Ok(ServiceStoreClient::list_root_keys(&config, &namespace).await?)
+            }
+            #[cfg(feature = "rocksdb")]
+            StoreConfig::RocksDb(config, namespace) => {
+                Ok(RocksDbStore::list_root_keys(&config, &namespace).await?)
+            }
+            #[cfg(feature = "dynamodb")]
+            StoreConfig::DynamoDb(config, namespace) => {
+                Ok(DynamoDbStore::list_root_keys(&config, &namespace).await?)
+            }
+            #[cfg(feature = "scylladb")]
+            StoreConfig::ScyllaDb(config, namespace) => {
+                Ok(ScyllaDbStore::list_root_keys(&config, &namespace).await?)
+            }
+        }
+    }
 }
 
 #[async_trait]

linera-indexer/lib/src/indexer.rs

@@ -68,6 +68,7 @@ where
         let root_key = "indexer".as_bytes().to_vec();
         let store = store
             .clone_with_root_key(&root_key)
+            .await
             .map_err(|_e| IndexerError::CloneWithRootKeyError)?;
         let context = ViewContext::create_root_context(store, ())
             .await

linera-indexer/lib/src/plugin.rs

@@ -75,6 +75,7 @@ where
     let root_key = name.as_bytes().to_vec();
     let store = store
         .clone_with_root_key(&root_key)
+        .await
         .map_err(|_e| IndexerError::CloneWithRootKeyError)?;
     let context = ViewContext::create_root_context(store, ())
         .await

linera-service/src/linera/main.rs

@@ -1770,6 +1770,14 @@ async fn run(options: &ClientOptions) -> Result<i32, anyhow::Error> {
                     );
                     println!("The list of namespaces is {:?}", namespaces);
                 }
+                DatabaseToolCommand::ListRootKeys { .. } => {
+                    let root_keys = Box::pin(full_storage_config.list_root_keys()).await?;
+                    info!(
+                        "Root keys listed in {} ms",
+                        start_time.elapsed().as_millis()
+                    );
+                    println!("The list of root keys is {:?}", root_keys);
+                }
             }
             Ok(0)
         }

linera-storage-service/proto/key_value_store.proto

@@ -1,6 +1,8 @@
 syntax = "proto3";
 package key_value_store.v1;
 
+import "google/protobuf/empty.proto";
+
 message KeyValue {
   bytes key = 1;
   bytes value = 2;
@@ -92,9 +94,6 @@ message RequestWriteBatchExtended {
   repeated Statement statements = 1;
 }
 
-message ReplyWriteBatchExtended {
-}
-
 
 message RequestSpecificChunk {
   int64 message_index = 1;
@@ -110,9 +109,6 @@ message RequestCreateNamespace {
   bytes namespace = 1;
 }
 
-message ReplyCreateNamespace {
-}
-
 
 message RequestExistsNamespace {
   bytes namespace = 1;
@@ -127,23 +123,26 @@ message RequestDeleteNamespace {
   bytes namespace = 1;
 }
 
-message ReplyDeleteNamespace {
+
+message ReplyListAll {
+  repeated bytes namespaces = 1;
 }
 
 
-message RequestListAll {
+message RequestListRootKeys {
+  bytes namespace = 1;
 }
 
-message ReplyListAll {
-  repeated bytes namespaces = 1;
+message ReplyListRootKeys {
+  repeated bytes root_keys = 1;
 }
 
 
-message RequestDeleteAll {
+message RequestInsertRootKey {
+  bytes namespace = 1;
+  bytes root_key = 2;
 }
 
-message ReplyDeleteAll {
-}
 
 service StoreProcessor {
   rpc ProcessReadValue (RequestReadValue) returns (ReplyReadValue) {}
@@ -152,11 +151,13 @@ service StoreProcessor {
   rpc ProcessReadMultiValues (RequestReadMultiValues) returns (ReplyReadMultiValues) {}
   rpc ProcessFindKeysByPrefix (RequestFindKeysByPrefix) returns (ReplyFindKeysByPrefix) {}
   rpc ProcessFindKeyValuesByPrefix (RequestFindKeyValuesByPrefix) returns (ReplyFindKeyValuesByPrefix) {}
-  rpc ProcessWriteBatchExtended (RequestWriteBatchExtended) returns (ReplyWriteBatchExtended) {}
+  rpc ProcessWriteBatchExtended (RequestWriteBatchExtended) returns (google.protobuf.Empty) {}
   rpc ProcessSpecificChunk (RequestSpecificChunk) returns (ReplySpecificChunk) {}
-  rpc ProcessCreateNamespace (RequestCreateNamespace) returns (ReplyCreateNamespace) {}
+  rpc ProcessCreateNamespace (RequestCreateNamespace) returns (google.protobuf.Empty) {}
   rpc ProcessExistsNamespace (RequestExistsNamespace) returns (ReplyExistsNamespace) {}
-  rpc ProcessDeleteNamespace (RequestDeleteNamespace) returns (ReplyDeleteNamespace) {}
-  rpc ProcessListAll (RequestListAll) returns (ReplyListAll) {}
-  rpc ProcessDeleteAll (RequestDeleteAll) returns (ReplyDeleteAll) {}
+  rpc ProcessDeleteNamespace (RequestDeleteNamespace) returns (google.protobuf.Empty) {}
+  rpc ProcessListAll (google.protobuf.Empty) returns (ReplyListAll) {}
+  rpc ProcessListRootKeys (RequestListRootKeys) returns (ReplyListRootKeys) {}
+  rpc ProcessInsertRootKey (RequestInsertRootKey) returns (google.protobuf.Empty) {}
+  rpc ProcessDeleteAll (google.protobuf.Empty) returns (google.protobuf.Empty) {}
 }