/// <summary>
/// Sends a MessageRejected message to the given receiver address.
/// </summary>
/// <param name="failureMessage">The query failure message.</param>
/// <param name="correlationId">The message correlation identifier.</param>
protected void SendQueryFailure(string failureMessage, Guid correlationId)
{
var response = new QueryFailure(
failureMessage,
correlationId,
Guid.NewGuid(),
this.TimeNow());
this.SendMessage(response);
}
public void SetClosures(SDK.IQBBase qry, QuerySuccess success, QueryFailure failure = null, bool allowEmpty = false)
{
var rqs = _msg.RequestList;
for (int idx = 0; idx < rqs.Count; idx++)
{
var rq = rqs.GetAt(idx);
if (rq.Detail == qry)
{
_clousures.Add(rq.RequestID, new ClosureSpec()
{
success = success,
failure = failure,
allowEmptyReturn = allowEmpty
});
return;
}
}
throw new Exception("Query not found in this batch");
}
internal void CanSerializeAndDeserialize_QueryFailure()
{
// Arrange
var correlationId = Guid.NewGuid();
var message = "data not found";
var response = new QueryFailure(
message,
correlationId,
Guid.NewGuid(),
StubZonedDateTime.UnixEpoch());
// Act
var packed = this.serializer.Serialize(response);
var unpacked = (QueryFailure)this.serializer.Deserialize(packed);
// Assert
Assert.Equal(response, unpacked);
Assert.Equal(correlationId, unpacked.CorrelationId);
Assert.Equal(message, unpacked.Message);
this.Output.WriteLine(Convert.ToBase64String(packed));
this.Output.WriteLine(Encoding.UTF8.GetString(packed));
}
/// <summary>
/// Lists available operations.
/// </summary>
/// <param name='customHeaders'>
/// Headers that will be added to request.
/// </param>
/// <param name='cancellationToken'>
/// The cancellation token.
/// </param>
/// <exception cref="QueryFailureException">
/// Thrown when the operation returned an invalid status code
/// </exception>
/// <exception cref="SerializationException">
/// Thrown when unable to deserialize the response
/// </exception>
/// <return>
/// A response object containing the response body and response headers.
/// </return>
public async Task <AzureOperationResponse <OperationsListResults> > ListWithHttpMessagesAsync(Dictionary <string, List <string> > customHeaders = null, CancellationToken cancellationToken = default(CancellationToken))
{
string apiVersion = "2018-04-04";
// Tracing
bool _shouldTrace = ServiceClientTracing.IsEnabled;
string _invocationId = null;
if (_shouldTrace)
{
_invocationId = ServiceClientTracing.NextInvocationId.ToString();
Dictionary <string, object> tracingParameters = new Dictionary <string, object>();
tracingParameters.Add("apiVersion", apiVersion);
tracingParameters.Add("cancellationToken", cancellationToken);
ServiceClientTracing.Enter(_invocationId, this, "List", tracingParameters);
}
// Construct URL
var _baseUrl = Client.BaseUri.AbsoluteUri;
var _url = new System.Uri(new System.Uri(_baseUrl + (_baseUrl.EndsWith("/") ? "" : "/")), "providers/Microsoft.PolicyInsights/operations").ToString();
List <string> _queryParameters = new List <string>();
if (apiVersion != null)
{
_queryParameters.Add(string.Format("api-version={0}", System.Uri.EscapeDataString(apiVersion)));
}
if (_queryParameters.Count > 0)
{
_url += (_url.Contains("?") ? "&" : "?") + string.Join("&", _queryParameters);
}
// Create HTTP transport objects
var _httpRequest = new HttpRequestMessage();
HttpResponseMessage _httpResponse = null;
_httpRequest.Method = new HttpMethod("GET");
_httpRequest.RequestUri = new System.Uri(_url);
// Set Headers
if (Client.GenerateClientRequestId != null && Client.GenerateClientRequestId.Value)
{
_httpRequest.Headers.TryAddWithoutValidation("x-ms-client-request-id", System.Guid.NewGuid().ToString());
}
if (Client.AcceptLanguage != null)
{
if (_httpRequest.Headers.Contains("accept-language"))
{
_httpRequest.Headers.Remove("accept-language");
}
_httpRequest.Headers.TryAddWithoutValidation("accept-language", Client.AcceptLanguage);
}
if (customHeaders != null)
{
foreach (var _header in customHeaders)
{
if (_httpRequest.Headers.Contains(_header.Key))
{
_httpRequest.Headers.Remove(_header.Key);
}
_httpRequest.Headers.TryAddWithoutValidation(_header.Key, _header.Value);
}
}
// Serialize Request
string _requestContent = null;
// Set Credentials
if (Client.Credentials != null)
{
cancellationToken.ThrowIfCancellationRequested();
await Client.Credentials.ProcessHttpRequestAsync(_httpRequest, cancellationToken).ConfigureAwait(false);
}
// Send Request
if (_shouldTrace)
{
ServiceClientTracing.SendRequest(_invocationId, _httpRequest);
}
cancellationToken.ThrowIfCancellationRequested();
_httpResponse = await Client.HttpClient.SendAsync(_httpRequest, cancellationToken).ConfigureAwait(false);
if (_shouldTrace)
{
ServiceClientTracing.ReceiveResponse(_invocationId, _httpResponse);
}
HttpStatusCode _statusCode = _httpResponse.StatusCode;
cancellationToken.ThrowIfCancellationRequested();
string _responseContent = null;
if ((int)_statusCode != 200)
{
var ex = new QueryFailureException(string.Format("Operation returned an invalid status code '{0}'", _statusCode));
try
{
_responseContent = await _httpResponse.Content.ReadAsStringAsync().ConfigureAwait(false);
QueryFailure _errorBody = Rest.Serialization.SafeJsonConvert.DeserializeObject <QueryFailure>(_responseContent, Client.DeserializationSettings);
if (_errorBody != null)
{
ex.Body = _errorBody;
}
}
catch (JsonException)
{
// Ignore the exception
}
ex.Request = new HttpRequestMessageWrapper(_httpRequest, _requestContent);
ex.Response = new HttpResponseMessageWrapper(_httpResponse, _responseContent);
if (_shouldTrace)
{
ServiceClientTracing.Error(_invocationId, ex);
}
_httpRequest.Dispose();
if (_httpResponse != null)
{
_httpResponse.Dispose();
}
throw ex;
}
// Create Result
var _result = new AzureOperationResponse <OperationsListResults>();
_result.Request = _httpRequest;
_result.Response = _httpResponse;
if (_httpResponse.Headers.Contains("x-ms-request-id"))
{
_result.RequestId = _httpResponse.Headers.GetValues("x-ms-request-id").FirstOrDefault();
}
// Deserialize Response
if ((int)_statusCode == 200)
{
_responseContent = await _httpResponse.Content.ReadAsStringAsync().ConfigureAwait(false);
try
{
_result.Body = Rest.Serialization.SafeJsonConvert.DeserializeObject <OperationsListResults>(_responseContent, Client.DeserializationSettings);
}
catch (JsonException ex)
{
_httpRequest.Dispose();
if (_httpResponse != null)
{
_httpResponse.Dispose();
}
throw new SerializationException("Unable to deserialize the response.", _responseContent, ex);
}
}
if (_shouldTrace)
{
ServiceClientTracing.Exit(_invocationId, _result);
}
return(_result);
}
public void SetClosures(SDK.IQBBase qry, QuerySuccess success, QueryFailure failure = null, bool allowEmpty = false)
{
var rqs = _msg.RequestList;
for (int idx = 0; idx < rqs.Count; idx++)
{
var rq = rqs.GetAt(idx);
if (rq.Detail == qry)
{
_clousures.Add(rq.RequestID, new ClosureSpec()
{
success = success,
failure = failure,
allowEmptyReturn = allowEmpty
});
return;
}
}
throw new Exception("Query not found in this batch");
}
private void OnMessage(QueryFailure response)
{
this.SendMessage(response);
this.Logger.LogInformation(LogId.Network, $"[RES]--> {response}.");
}