use std::error::Error;
use std::fmt;
use async_trait::async_trait;
use rusoto_core::credential::ProvideAwsCredentials;
use rusoto_core::region;
use rusoto_core::request::{BufferedHttpResponse, DispatchSignedRequest};
use rusoto_core::{Client, RusotoError};
use rusoto_core::proto;
use rusoto_core::signature::SignedRequest;
#[allow(unused_imports)]
use serde::{Deserialize, Serialize};
use serde_json;
#[derive(Default, Debug, Clone, PartialEq, Serialize)]
#[cfg_attr(feature = "deserialize_structs", derive(Deserialize))]
pub struct GetPersonalizedRankingRequest {
#[serde(rename = "campaignArn")]
pub campaign_arn: String,
#[serde(rename = "context")]
#[serde(skip_serializing_if = "Option::is_none")]
pub context: Option<::std::collections::HashMap<String, String>>,
#[serde(rename = "inputList")]
pub input_list: Vec<String>,
#[serde(rename = "userId")]
pub user_id: String,
}
#[derive(Default, Debug, Clone, PartialEq, Deserialize)]
#[cfg_attr(any(test, feature = "serialize_structs"), derive(Serialize))]
pub struct GetPersonalizedRankingResponse {
#[serde(rename = "personalizedRanking")]
#[serde(skip_serializing_if = "Option::is_none")]
pub personalized_ranking: Option<Vec<PredictedItem>>,
}
#[derive(Default, Debug, Clone, PartialEq, Serialize)]
#[cfg_attr(feature = "deserialize_structs", derive(Deserialize))]
pub struct GetRecommendationsRequest {
#[serde(rename = "campaignArn")]
pub campaign_arn: String,
#[serde(rename = "context")]
#[serde(skip_serializing_if = "Option::is_none")]
pub context: Option<::std::collections::HashMap<String, String>>,
#[serde(rename = "itemId")]
#[serde(skip_serializing_if = "Option::is_none")]
pub item_id: Option<String>,
#[serde(rename = "numResults")]
#[serde(skip_serializing_if = "Option::is_none")]
pub num_results: Option<i64>,
#[serde(rename = "userId")]
#[serde(skip_serializing_if = "Option::is_none")]
pub user_id: Option<String>,
}
#[derive(Default, Debug, Clone, PartialEq, Deserialize)]
#[cfg_attr(any(test, feature = "serialize_structs"), derive(Serialize))]
pub struct GetRecommendationsResponse {
#[serde(rename = "itemList")]
#[serde(skip_serializing_if = "Option::is_none")]
pub item_list: Option<Vec<PredictedItem>>,
}
#[derive(Default, Debug, Clone, PartialEq, Deserialize)]
#[cfg_attr(any(test, feature = "serialize_structs"), derive(Serialize))]
pub struct PredictedItem {
#[serde(rename = "itemId")]
#[serde(skip_serializing_if = "Option::is_none")]
pub item_id: Option<String>,
}
#[derive(Debug, PartialEq)]
pub enum GetPersonalizedRankingError {
InvalidInput(String),
ResourceNotFound(String),
}
impl GetPersonalizedRankingError {
pub fn from_response(res: BufferedHttpResponse) -> RusotoError<GetPersonalizedRankingError> {
if let Some(err) = proto::json::Error::parse_rest(&res) {
match err.typ.as_str() {
"InvalidInputException" => {
return RusotoError::Service(GetPersonalizedRankingError::InvalidInput(err.msg))
}
"ResourceNotFoundException" => {
return RusotoError::Service(GetPersonalizedRankingError::ResourceNotFound(
err.msg,
))
}
"ValidationException" => return RusotoError::Validation(err.msg),
_ => {}
}
}
RusotoError::Unknown(res)
}
}
impl fmt::Display for GetPersonalizedRankingError {
#[allow(unused_variables)]
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
match *self {
GetPersonalizedRankingError::InvalidInput(ref cause) => write!(f, "{}", cause),
GetPersonalizedRankingError::ResourceNotFound(ref cause) => write!(f, "{}", cause),
}
}
}
impl Error for GetPersonalizedRankingError {}
#[derive(Debug, PartialEq)]
pub enum GetRecommendationsError {
InvalidInput(String),
ResourceNotFound(String),
}
impl GetRecommendationsError {
pub fn from_response(res: BufferedHttpResponse) -> RusotoError<GetRecommendationsError> {
if let Some(err) = proto::json::Error::parse_rest(&res) {
match err.typ.as_str() {
"InvalidInputException" => {
return RusotoError::Service(GetRecommendationsError::InvalidInput(err.msg))
}
"ResourceNotFoundException" => {
return RusotoError::Service(GetRecommendationsError::ResourceNotFound(err.msg))
}
"ValidationException" => return RusotoError::Validation(err.msg),
_ => {}
}
}
RusotoError::Unknown(res)
}
}
impl fmt::Display for GetRecommendationsError {
#[allow(unused_variables)]
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
match *self {
GetRecommendationsError::InvalidInput(ref cause) => write!(f, "{}", cause),
GetRecommendationsError::ResourceNotFound(ref cause) => write!(f, "{}", cause),
}
}
}
impl Error for GetRecommendationsError {}
#[async_trait]
pub trait PersonalizeRuntime {
async fn get_personalized_ranking(
&self,
input: GetPersonalizedRankingRequest,
) -> Result<GetPersonalizedRankingResponse, RusotoError<GetPersonalizedRankingError>>;
async fn get_recommendations(
&self,
input: GetRecommendationsRequest,
) -> Result<GetRecommendationsResponse, RusotoError<GetRecommendationsError>>;
}
#[derive(Clone)]
pub struct PersonalizeRuntimeClient {
client: Client,
region: region::Region,
}
impl PersonalizeRuntimeClient {
pub fn new(region: region::Region) -> PersonalizeRuntimeClient {
PersonalizeRuntimeClient {
client: Client::shared(),
region,
}
}
pub fn new_with<P, D>(
request_dispatcher: D,
credentials_provider: P,
region: region::Region,
) -> PersonalizeRuntimeClient
where
P: ProvideAwsCredentials + Send + Sync + 'static,
D: DispatchSignedRequest + Send + Sync + 'static,
{
PersonalizeRuntimeClient {
client: Client::new_with(credentials_provider, request_dispatcher),
region,
}
}
pub fn new_with_client(client: Client, region: region::Region) -> PersonalizeRuntimeClient {
PersonalizeRuntimeClient { client, region }
}
}
#[async_trait]
impl PersonalizeRuntime for PersonalizeRuntimeClient {
async fn get_personalized_ranking(
&self,
input: GetPersonalizedRankingRequest,
) -> Result<GetPersonalizedRankingResponse, RusotoError<GetPersonalizedRankingError>> {
let request_uri = "/personalize-ranking";
let mut request = SignedRequest::new("POST", "personalize", &self.region, &request_uri);
request.set_content_type("application/x-amz-json-1.1".to_owned());
request.set_endpoint_prefix("personalize-runtime".to_string());
let encoded = Some(serde_json::to_vec(&input).unwrap());
request.set_payload(encoded);
let mut response = self
.client
.sign_and_dispatch(request)
.await
.map_err(RusotoError::from)?;
if response.status.is_success() {
let response = response.buffer().await.map_err(RusotoError::HttpDispatch)?;
let result = proto::json::ResponsePayload::new(&response)
.deserialize::<GetPersonalizedRankingResponse, _>()?;
Ok(result)
} else {
let response = response.buffer().await.map_err(RusotoError::HttpDispatch)?;
Err(GetPersonalizedRankingError::from_response(response))
}
}
async fn get_recommendations(
&self,
input: GetRecommendationsRequest,
) -> Result<GetRecommendationsResponse, RusotoError<GetRecommendationsError>> {
let request_uri = "/recommendations";
let mut request = SignedRequest::new("POST", "personalize", &self.region, &request_uri);
request.set_content_type("application/x-amz-json-1.1".to_owned());
request.set_endpoint_prefix("personalize-runtime".to_string());
let encoded = Some(serde_json::to_vec(&input).unwrap());
request.set_payload(encoded);
let mut response = self
.client
.sign_and_dispatch(request)
.await
.map_err(RusotoError::from)?;
if response.status.is_success() {
let response = response.buffer().await.map_err(RusotoError::HttpDispatch)?;
let result = proto::json::ResponsePayload::new(&response)
.deserialize::<GetRecommendationsResponse, _>()?;
Ok(result)
} else {
let response = response.buffer().await.map_err(RusotoError::HttpDispatch)?;
Err(GetRecommendationsError::from_response(response))
}
}
}