1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
/*use tonic::{Request, Response, Status};
use actix_web::web::Data;
use ant_core::data::Wallet;
use crate::service::key_value_service::KeyValueService;
use crate::controller::StoreType;
use bytes::Bytes;
pub mod key_value_proto {
tonic::include_proto!("key_value");
}
use key_value_proto::key_value_service_server::KeyValueService as KeyValueServiceTrait;
pub use key_value_proto::key_value_service_server::KeyValueServiceServer;
use key_value_proto::{CreateKeyValueRequest, GetKeyValueRequest, KeyValueResponse, GetKeyValueBinaryResponse};
use crate::error::public_data_error::PublicDataError;
pub struct KeyValueHandler {
key_value_service: Data<KeyValueService>,
evm_wallet: Data<Wallet>,
}
impl KeyValueHandler {
pub fn new(key_value_service: Data<KeyValueService>, evm_wallet: Data<Wallet>) -> Self {
Self { key_value_service, evm_wallet }
}
}
fn to_status(error: PublicDataError) -> Status {
Status::internal(error.to_string())
}
#[tonic::async_trait]
impl KeyValueServiceTrait for KeyValueHandler {
async fn create_key_value(
&self,
request: Request<CreateKeyValueRequest>,
) -> Result<Response<KeyValueResponse>, Status> {
let req = request.into_inner();
self.key_value_service.create_key_value_binary(
req.bucket.clone(),
req.object.clone(),
Bytes::from(req.content),
self.evm_wallet.get_ref().clone(),
StoreType::Network,
).await.map_err(to_status)?;
Ok(Response::new(KeyValueResponse {
bucket: req.bucket,
object: req.object,
}))
}
async fn get_key_value(
&self,
request: Request<GetKeyValueRequest>,
) -> Result<Response<GetKeyValueBinaryResponse>, Status> {
let req = request.into_inner();
let content = self.key_value_service.get_key_value_binary(
req.bucket.clone(),
req.object.clone(),
).await.map_err(to_status)?;
Ok(Response::new(GetKeyValueBinaryResponse {
bucket: req.bucket,
object: req.object,
content: content.to_vec(),
}))
}
}
*/