Add VSS Http thin client implementation for get/put/listKeyVersions api's

Author: G8XSU

vss-accessor/Cargo.toml

@@ -5,6 +5,9 @@ edition = "2021"
 build = "build.rs"
 
 [dependencies]
+prost = "0.11.9"
+#bytes = "1.4.0"
+reqwest = { version = "0.11.13", features = ["rustls-tls", "blocking"] }
 
 [dev-dependencies]
 

vss-accessor/src/lib.rs

@@ -1,2 +1,94 @@
 #![deny(rustdoc::broken_intra_doc_links)]
 #![deny(rustdoc::private_intra_doc_links)]
+
+use std::error::Error;
+
+use ::prost::Message;
+use reqwest;
+use reqwest::Client;
+
+use crate::vss::{
+	GetObjectRequest, GetObjectResponse, ListKeyVersionsRequest, ListKeyVersionsResponse, PutObjectRequest,
+	PutObjectResponse,
+};
+use crate::vss_error::VssError;
+
+mod vss_error;
+
+/// Import auto-generated code
+pub mod vss {
+	include!("generated-src/org.vss.rs");
+}
+
+/// Thin-client to access hosted instance of `Versioned Storage Service (VSS)`.
+/// Api for VssClient is minimalistic and directly mimics vss-server-side api's as it is.
+pub struct VssClient {
+	base_url: String,
+	client: Client,
+}
+
+impl VssClient {
+	/// Constructs new instance of VssClient.
+	/// `base_url` is the vss-server endpoint to be used.
+	pub fn new(base_url: &str) -> Result<Self, Box<dyn Error>> {
+		let client = Client::new();
+		Ok(Self { base_url: String::from(base_url), client })
+	}
+
+	/// Fetches a value against a given `key` in `request`.
+	/// Makes a service call to GetObject endpoint of vss-server.
+	/// For api-contract/usage, refer docs for: [`GetObjectRequest`] and [`GetObjectResponse`]
+	pub async fn get_object(&self, request: GetObjectRequest) -> Result<GetObjectResponse, VssError> {
+		let url = format!("{}/getObject", self.base_url);
+
+		let response_raw = self.client.post(url).body(request.encode_to_vec()).send().await?;
+		let status = response_raw.status();
+		let payload = response_raw.bytes().await?;
+
+		if status.is_success() {
+			let response = GetObjectResponse::decode(&payload[..])?;
+			Ok(response)
+		} else {
+			Err(VssError::new(status, payload))
+		}
+	}
+
+	/// Writes multiple `transaction_items` of `request` in single transaction.
+	/// Makes a service call to PutObject endpoint of vss-server, with multiple items.
+	/// Items in the `request` are written in a single all-or-nothing transaction.
+	/// For api-contract/usage, refer docs for: [`PutObjectRequest`] and [`PutObjectResponse`]
+	pub async fn put_object(&self, request: PutObjectRequest) -> Result<PutObjectResponse, VssError> {
+		let url = format!("{}/putObjects", self.base_url);
+
+		let response_raw = self.client.post(url).body(request.encode_to_vec()).send().await?;
+		let status = response_raw.status();
+		let payload = response_raw.bytes().await?;
+
+		if status.is_success() {
+			let response = PutObjectResponse::decode(&payload[..])?;
+			Ok(response)
+		} else {
+			Err(VssError::new(status, payload))
+		}
+	}
+
+	/// Lists keys and their corresponding version for a given `store_id` in `request`.
+	/// Makes a service call to ListKeyVersions endpoint of vss-server.
+	/// For api-contract/usage, refer docs for: [`ListKeyVersionsRequest`] and [`ListKeyVersionsResponse`]
+	pub async fn list_key_versions(
+		&self, request: ListKeyVersionsRequest,
+	) -> Result<ListKeyVersionsResponse, VssError> {
+		let url = format!("{}/listKeyVersions", self.base_url);
+
+		let response_raw = self.client.post(url).body(request.encode_to_vec()).send().await?;
+		let status = response_raw.status();
+		let payload = response_raw.bytes().await?;
+
+		if status.is_success() {
+			let response = ListKeyVersionsResponse::decode(&payload[..])?;
+			Ok(response)
+		} else {
+			Err(VssError::new(status, payload))
+		}
+	}
+}

vss-accessor/src/vss_error.rs

@@ -0,0 +1,78 @@
+use crate::vss::{ErrorCode, ErrorResponse};
+use prost::bytes::Bytes;
+use prost::{DecodeError, Message};
+use reqwest::StatusCode;
+use std::error::Error;
+use std::fmt::{Display, Formatter};
+
+/// When there is an error while writing to vss-storage, the response contains a relevant error code.
+/// This enum is direct mapping of vss-server error codes. Refer docs for [`ErrorResponse`] for more
+/// information regarding each error code and corresponding use-cases.
+#[derive(Debug)]
+pub enum VssError {
+	InvalidRequestError(String),
+	ConflictError(String),
+	InternalServerError(String),
+	InternalError(String),
+}
+
+impl VssError {
+	/// Create new instance of VssError
+	pub fn new(status: StatusCode, payload: Bytes) -> VssError {
+		match ErrorResponse::decode(&payload[..]) {
+			Ok(error_response) => VssError::from(error_response),
+			Err(e) => {
+				let message =
+					format!("Unable to decode ErrorResponse from server, HttpStatusCode: {}, DecodeErr: {}", status, e);
+				VssError::InternalError(message)
+			}
+		}
+	}
+}
+
+impl Display for VssError {
+	fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
+		match self {
+			VssError::InvalidRequestError(message) => {
+				write!(f, "Request sent to VSS Storage was invalid : {}", message)
+			}
+			VssError::ConflictError(message) => {
+				write!(f, "Potential version conflict in write operation : {}", message)
+			}
+			VssError::InternalServerError(message) => {
+				write!(f, "InternalServerError : {}", message)
+			}
+			VssError::InternalError(message) => {
+				write!(f, "InternalError : {}", message)
+			}
+		}
+	}
+}
+
+impl Error for VssError {}
+
+impl From<ErrorResponse> for VssError {
+	fn from(error_response: ErrorResponse) -> Self {
+		return match error_response.error_code() {
+			ErrorCode::InvalidRequestException => VssError::InvalidRequestError(error_response.message),
+			ErrorCode::ConflictException => VssError::ConflictError(error_response.message),
+			ErrorCode::InternalServerException => VssError::InternalServerError(error_response.message),
+			_ => VssError::InternalError(format!(
+				"VSS responded with an unknown error code: {}, message: {}",
+				error_response.error_code, error_response.message
+			)),
+		};
+	}
+}
+
+impl From<DecodeError> for VssError {
+	fn from(err: DecodeError) -> Self {
+		VssError::InternalError(err.to_string())
+	}
+}
+
+impl From<reqwest::Error> for VssError {
+	fn from(err: reqwest::Error) -> Self {
+		VssError::InternalError(err.to_string())
+	}
+}