smeb_y/0g-storage-node
1
0
Fork 0
mirror of https://github.com/0glabs/0g-storage-node.git synced 2024-11-10 10:05:17 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: add all_shards in admin_getFileLocation (#138)

Browse Source
This commit is contained in:
MiniFrenchBread 2024-07-22 11:28:05 +08:00 committed by GitHub
parent 085c34beb0
commit a782cf2d54
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
node/rpc/src/admin/api.rs
View File

@ -40,5 +40,9 @@ pub trait Rpc {
async fn get_peers(&self) -> RpcResult<HashMap<String, PeerInfo>>;
#[method(name = "getFileLocation")]
async fn get_file_location(&self, tx_seq: u64) -> RpcResult<Option<Vec<LocationInfo>>>;
async fn get_file_location(
&self,
tx_seq: u64,
all_shards: bool,
) -> RpcResult<Option<Vec<LocationInfo>>>;
}

8
node/rpc/src/admin/impl.rs
View File

@ -185,7 +185,11 @@ impl RpcServer for RpcServerImpl {
.collect())
}
async fn get_file_location(&self, tx_seq: u64) -> RpcResult<Option<Vec<LocationInfo>>> {
async fn get_file_location(
&self,
tx_seq: u64,
all_shards: bool,
) -> RpcResult<Option<Vec<LocationInfo>>> {
let tx = match self.ctx.log_store.get_tx_by_seq_number(tx_seq).await? {
Some(tx) => tx,
None => {
@ -221,7 +225,7 @@ impl RpcServer for RpcServerImpl {
shard_config: shard_config.unwrap(),
})
.collect();
if all_shards_available(info.iter().map(|info| info.shard_config).collect()) {
if all_shards && all_shards_available(info.iter().map(|info| info.shard_config).collect()) {
Ok(Some(info))
} else {
Ok(None)