Fix NSX rest client to not reset execution counter after a login

This commit is contained in:
Miguel Ferreira 2015-12-02 14:53:56 +01:00
parent d248e61a31
commit cfe14462a3

View File

@ -90,7 +90,7 @@ public class NiciraRestClient extends BasicRestClient {
if (HttpStatusCodeHelper.isUnauthorized(statusCode)) { if (HttpStatusCodeHelper.isUnauthorized(statusCode)) {
return handleUnauthorizedResponse(request, previousStatusCode, response, statusCode); return handleUnauthorizedResponse(request, previousStatusCode, response, statusCode);
} else if (HttpStatusCodeHelper.isSuccess(statusCode)) { } else if (HttpStatusCodeHelper.isSuccess(statusCode)) {
return handleSuccessResponse(response); return handleSuccessResponse(request, response);
} else { } else {
throw new CloudstackRESTException("Unexpecetd status code: " + statusCode); throw new CloudstackRESTException("Unexpecetd status code: " + statusCode);
} }
@ -110,8 +110,10 @@ public class NiciraRestClient extends BasicRestClient {
return execute(request, loginStatusCode); return execute(request, loginStatusCode);
} }
private CloseableHttpResponse handleSuccessResponse(final CloseableHttpResponse response) { private CloseableHttpResponse handleSuccessResponse(final HttpUriRequest request, final CloseableHttpResponse response) {
counter.resetExecutionCounter(); if (!request.getURI().getPath().contains(loginUrl)) {
counter.resetExecutionCounter();
}
return response; return response;
} }