org.apache.inlong.sort.filesystem.shaded.com.amazonaws.http.conn.ClientConnectionManagerFactory Maven / Gradle / Ivy
/*
* Copyright 2010-2022 Amazon.com, Inc. or its affiliates. All Rights Reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License").
* You may not use this file except in compliance with the License.
* A copy of the License is located at
*
* http://aws.amazon.com/apache2.0
*
* or in the "license" file accompanying this file. This file is distributed
* on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
* express or implied. See the License for the specific language governing
* permissions and limitations under the License.
*/
package com.amazonaws.http.conn;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.apache.http.conn.ConnectionRequest;
import org.apache.http.conn.HttpClientConnectionManager;
import org.apache.http.pool.ConnPoolControl;
import java.lang.reflect.InvocationHandler;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.lang.reflect.Proxy;
public class ClientConnectionManagerFactory {
private static final Log log = LogFactory.getLog(ClientConnectionManagerFactory.class);
/**
* Returns a wrapped instance of {@link HttpClientConnectionManager}
* to capture the necessary performance metrics.
*
* @param orig the target instance to be wrapped
*/
public static HttpClientConnectionManager wrap
(HttpClientConnectionManager orig) {
if (orig instanceof Wrapped)
throw new IllegalArgumentException();
final Class[] interfaces;
if (orig instanceof ConnPoolControl) {
interfaces = new Class[]{
HttpClientConnectionManager.class,
ConnPoolControl.class,
Wrapped.class
};
} else {
interfaces = new Class[]{
HttpClientConnectionManager.class,
Wrapped.class
};
}
return (HttpClientConnectionManager) Proxy.newProxyInstance(
// https://github.com/aws/aws-sdk-java/pull/48#issuecomment-29454423
ClientConnectionManagerFactory.class.getClassLoader(),
interfaces,
new Handler(orig));
}
/**
* The handler behind the dynamic proxy for {@link HttpClientConnectionManager}
* so that the any returned instance of {@link ConnectionRequest} can
* further wrapped for capturing performance metrics.
*/
private static class Handler implements InvocationHandler {
private final HttpClientConnectionManager orig;
Handler(HttpClientConnectionManager real) {
this.orig = real;
}
@Override
public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
try {
Object ret = method.invoke(orig, args);
return ret instanceof ConnectionRequest
? ClientConnectionRequestFactory.wrap((ConnectionRequest) ret)
: ret
;
} catch (InvocationTargetException e) {
log.debug("", e);
throw e.getCause();
}
}
}
}