parent
17e1eb95e3
commit
48c63ead2f
@ -0,0 +1,103 @@ |
||||
/* |
||||
* Copyright (c) 2022-2024 Niko Bonnieure, Par le Peuple, NextGraph.org developers |
||||
* All rights reserved. |
||||
* Licensed under the Apache License, Version 2.0 |
||||
* <LICENSE-APACHE2 or http://www.apache.org/licenses/LICENSE-2.0>
|
||||
* or the MIT license <LICENSE-MIT or http://opensource.org/licenses/MIT>,
|
||||
* at your option. All files in the project carrying such |
||||
* notice may not be copied, modified, or distributed except |
||||
* according to those terms. |
||||
*/ |
||||
|
||||
use std::sync::Arc; |
||||
|
||||
use async_std::sync::Mutex; |
||||
|
||||
use ng_repo::errors::*; |
||||
use ng_repo::log::*; |
||||
use ng_repo::object::Object; |
||||
use ng_repo::store::Store; |
||||
use ng_repo::types::Block; |
||||
|
||||
use super::super::StartProtocol; |
||||
|
||||
use crate::broker::BROKER; |
||||
use crate::connection::NoiseFSM; |
||||
use crate::types::*; |
||||
use crate::{actor::*, types::ProtocolMessage}; |
||||
|
||||
impl ExtObjectGetV0 { |
||||
pub fn get_actor(&self) -> Box<dyn EActor> { |
||||
Actor::<ExtObjectGetV0, Vec<Block>>::new_responder(0) |
||||
} |
||||
} |
||||
|
||||
impl TryFrom<ProtocolMessage> for ExtObjectGetV0 { |
||||
type Error = ProtocolError; |
||||
fn try_from(msg: ProtocolMessage) -> Result<Self, Self::Error> { |
||||
if let ProtocolMessage::Start(StartProtocol::Ext(ExtRequest::V0(ExtRequestV0 { |
||||
content: ExtRequestContentV0::ExtObjectGet(a), |
||||
.. |
||||
}))) = msg |
||||
{ |
||||
Ok(a) |
||||
} else { |
||||
log_debug!("INVALID {:?}", msg); |
||||
Err(ProtocolError::InvalidValue) |
||||
} |
||||
} |
||||
} |
||||
|
||||
impl From<ExtObjectGetV0> for ProtocolMessage { |
||||
fn from(_msg: ExtObjectGetV0) -> ProtocolMessage { |
||||
unimplemented!(); |
||||
} |
||||
} |
||||
|
||||
impl From<ExtObjectGetV0> for ExtRequestContentV0 { |
||||
fn from(msg: ExtObjectGetV0) -> ExtRequestContentV0 { |
||||
ExtRequestContentV0::ExtObjectGet(msg) |
||||
} |
||||
} |
||||
|
||||
impl TryFrom<ProtocolMessage> for Vec<Block> { |
||||
type Error = ProtocolError; |
||||
fn try_from(msg: ProtocolMessage) -> Result<Vec<Block>, Self::Error> { |
||||
let content: ExtResponseContentV0 = msg.try_into()?; |
||||
if let ExtResponseContentV0::Blocks(res) = content { |
||||
Ok(res) |
||||
} else { |
||||
Err(ProtocolError::InvalidValue) |
||||
} |
||||
} |
||||
} |
||||
|
||||
impl Actor<'_, ExtObjectGetV0, Vec<Block>> {} |
||||
|
||||
#[async_trait::async_trait] |
||||
impl EActor for Actor<'_, ExtObjectGetV0, Vec<Block>> { |
||||
async fn respond( |
||||
&mut self, |
||||
msg: ProtocolMessage, |
||||
fsm: Arc<Mutex<NoiseFSM>>, |
||||
) -> Result<(), ProtocolError> { |
||||
let req = ExtObjectGetV0::try_from(msg)?; |
||||
let sb = { |
||||
let broker = BROKER.read().await; |
||||
broker.get_server_broker()? |
||||
}; |
||||
let lock = sb.read().await; |
||||
let store = Store::new_from_overlay_id(&req.overlay, lock.get_block_storage()); |
||||
let mut blocks = Vec::new(); |
||||
for obj_id in req.ids { |
||||
// TODO: deal with RandomAccessFiles (or is it just working?)
|
||||
if let Ok(obj) = Object::load_without_header(obj_id, None, &store) { |
||||
blocks.append(&mut obj.into_blocks()); |
||||
//TODO: load the obj.files too (if req.include_files)
|
||||
} |
||||
} |
||||
let response: ExtResponseV0 = Ok(ExtResponseContentV0::Blocks(blocks)).into(); |
||||
fsm.lock().await.send(response.into()).await?; |
||||
Ok(()) |
||||
} |
||||
} |
||||
@ -1 +1,3 @@ |
||||
pub mod wallet_get_export; |
||||
|
||||
pub mod get; |
||||
|
||||
@ -0,0 +1,8 @@ |
||||
// Copyright (c) 2022-2024 Niko Bonnieure, Par le Peuple, NextGraph.org developers
|
||||
// All rights reserved.
|
||||
// Licensed under the Apache License, Version 2.0
|
||||
// <LICENSE-APACHE2 or http://www.apache.org/licenses/LICENSE-2.0>
|
||||
// or the MIT license <LICENSE-MIT or http://opensource.org/licenses/MIT>,
|
||||
// at your option. All files in the project carrying such
|
||||
// notice may not be copied, modified, or distributed except
|
||||
// according to those terms.
|
||||
Loading…
Reference in new issue