Skip to content

Conversation

@stoty
Copy link
Contributor

@stoty stoty commented Jan 31, 2025

No description provided.

@stoty stoty requested a review from julianhyde January 31, 2025 08:06
return requestConfigBuilder.build();
}

@Override public byte[] send(byte[] request) {
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I think I'd prefer an approach like this for adapting to the new ensureOpen(...) syntax since it avoids the need for any pass-by-instance semantics:

@Override public byte[] send(byte[] request) {
  // Create the client with the AuthSchemeRegistry and manager
  HttpPost post = new HttpPost(uri);
  try {
    HttpHost host = RoutingSupport.determineHost(post);

    while (true) {
      ByteArrayEntity entity = new ByteArrayEntity(request, ContentType.APPLICATION_OCTET_STREAM);

      post.setEntity(entity);

      try (ClassicHttpResponse response = executeOpen(host, post, context)) {
        final int statusCode = response.getCode();
        if (HttpURLConnection.HTTP_OK == statusCode
            || HttpURLConnection.HTTP_INTERNAL_ERROR == statusCode) {
          userToken = context.getUserToken();
          return EntityUtils.toByteArray(response.getEntity());
        } else if (HttpURLConnection.HTTP_UNAVAILABLE == statusCode) {
          LOG.debug("Failed to connect to server (HTTP/503), retrying");
          continue;
        }

        throw new RuntimeException(
            "Failed to execute HTTP Request, got HTTP/" + statusCode);
      } catch (NoHttpResponseException e) {
        // This can happen when sitting behind a load balancer and a backend server dies
        LOG.debug("The server failed to issue an HTTP response, retrying");
        continue;
      }
    }
  } catch (RuntimeException e) {
    throw e;
  } catch (Exception e) {
    LOG.debug("Failed to execute HTTP request", e);
    throw new RuntimeException(e);
  }
}

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Can you elaborate @chrisdennis ?

I couldn't find ensureOpen() in either the apache HttpClient or Avatica codebase.
Google found it in some nio implementation classes, but I don't see how that's related.

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

If you mean using the execute() calls that pass a Handler, those don't return error codes, and would make implementing the retry logic unneccarily convoluted.

The response is already in a try-with-resources block, so there is no danger of leakage.

Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Sorry - I've no idea why I wrote ensureOpen... you're right I mean't executeOpen(...). I'm not suggesting migrating to execute(...) calls though. I meant that if you formulate things like above then you don't need to compute the HttpHost in advance using a "dummy" HttpPost/URI, and you don't need to pass the result in via the new instance variable, you can just reorder things slightly, and keep everything contained within the call stack.

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for the clarification.

Yes, that would work, but we're currently caching and pre-computing as much as possible.
Your version is slightly less efficient, as it needs to call RoutingSupport.determineHost for each send() call.

But we can avoid the Dummy post by postponing the host detection, I will do that.

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I have removed the dummy post, please re-check @chrisdennis .

@stoty
Copy link
Contributor Author

stoty commented Feb 11, 2025

merged manually

@stoty stoty closed this Feb 11, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants