add getsubset request helper

This commit is contained in:
mycognosist 2021-12-27 19:20:01 +02:00 committed by adria0.eth
parent cdaf2fb259
commit a02db1606a
1 changed files with 12 additions and 0 deletions

View File

@ -11,6 +11,7 @@ const MAX_RPC_BODY_LEN: usize = 65536;
#[derive(Debug)]
pub enum ApiMethod {
GetSubset,
Publish,
WhoAmI,
Get,
@ -25,6 +26,7 @@ impl ApiMethod {
pub fn selector(&self) -> &'static [&'static str] {
use ApiMethod::*;
match self {
GetSubset => &["partialReplication", "getSubset"],
Publish => &["publish"],
WhoAmI => &["whoami"],
Get => &["get"],
@ -38,6 +40,7 @@ impl ApiMethod {
pub fn from_selector(s: &[&str]) -> Option<Self> {
use ApiMethod::*;
match s {
["partialReplication", "getSubset"] => Some(GetSubset),
["publish"] => Some(Publish),
["whoami"] => Some(WhoAmI),
["get"] => Some(Get),
@ -68,6 +71,15 @@ impl<W: Write + Unpin> ApiCaller<W> {
&mut self.rpc
}
/// Send ["partialReplication", "getSubset"] request.
pub async fn getsubset_req_send(&mut self, query: SubsetQuery) -> Result<RequestNo> {
let req_no = self
.rpc
.send_request(ApiMethod::GetSubset.selector(), RpcType::Source, &query)
.await?;
Ok(req_no)
}
/// Send ["publish"] request.
pub async fn publish_req_send(&mut self, msg: TypedMessage) -> Result<RequestNo> {
let req_no = self