/*
* Copyright 2014 NAVER Corp.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License 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.navercorp.pinpoint.plugin.httpclient4.interceptor;
import java.net.URI;
import com.navercorp.pinpoint.bootstrap.interceptor.scope.InterceptorScope;
import org.apache.http.HttpRequest;
import org.apache.http.client.methods.HttpUriRequest;
import com.navercorp.pinpoint.bootstrap.context.MethodDescriptor;
import com.navercorp.pinpoint.bootstrap.context.TraceContext;
import com.navercorp.pinpoint.bootstrap.interceptor.annotation.Scope;
import com.navercorp.pinpoint.bootstrap.pair.NameIntValuePair;
import com.navercorp.pinpoint.plugin.httpclient4.HttpClient4Constants;
/**
* InstrumentMethod interceptor
* <p/>
* <p/>
*
* <pre>
* org.apache.http.impl.client.AbstractHttpClient.
* public final HttpResponse execute(HttpUriRequest request) throws IOException, ClientProtocolException
* </pre>
* @author emeroad
* @author minwoo.jung
* @author jaehong.kim
*/
@Scope(HttpClient4Constants.HTTP_CLIENT4_SCOPE)
public class HttpClientExecuteMethodWithHttpUriRequestInterceptor extends AbstractHttpClientExecuteMethodInterceptor {
private static final int HTTP_URI_REQUEST_INDEX = 0;
public HttpClientExecuteMethodWithHttpUriRequestInterceptor(boolean isHasCallbackParam, TraceContext context, MethodDescriptor methodDescriptor, InterceptorScope interceptorScope) {
super(HttpClientExecuteMethodWithHttpUriRequestInterceptor.class, isHasCallbackParam, context, methodDescriptor, interceptorScope);
}
@Override
protected NameIntValuePair<String> getHost(Object[] args) {
final HttpUriRequest httpUriRequest = getHttpUriRequest(args);
if (httpUriRequest == null) {
return null;
}
final URI uri = httpUriRequest.getURI();
return extractHost(uri);
}
@Override
protected HttpRequest getHttpRequest(Object[] args) {
return getHttpUriRequest(args);
}
private HttpUriRequest getHttpUriRequest(Object[] args) {
final Object arg = args[HTTP_URI_REQUEST_INDEX];
if (arg != null && arg instanceof HttpUriRequest) {
return (HttpUriRequest) arg;
}
return null;
}
/**
* copy
* org.apache.http.client.utils.URIUtils#extractHost(java.net.URI)
* @param uri
* @return
*/
private NameIntValuePair<String> extractHost(final URI uri) {
if (uri == null) {
return null;
}
NameIntValuePair<String> target = null;
if (uri.isAbsolute()) {
int port = uri.getPort(); // may be overridden later
String host = uri.getHost();
if (host == null) { // normal parse failed; let's do it ourselves
// authority does not seem to care about the valid character-set
// for host names
host = uri.getAuthority();
if (host != null) {
// Strip off any leading user credentials
int at = host.indexOf('@');
if (at >= 0) {
if (host.length() > at + 1) {
host = host.substring(at + 1);
} else {
host = null; // @ on its own
}
}
// Extract the port suffix, if present
if (host != null) {
int colon = host.indexOf(':');
if (colon >= 0) {
int pos = colon + 1;
int len = 0;
for (int i = pos; i < host.length(); i++) {
if (Character.isDigit(host.charAt(i))) {
len++;
} else {
break;
}
}
if (len > 0) {
try {
port = Integer.parseInt(host.substring(pos, pos + len));
} catch (NumberFormatException ignore) {
// skip
}
}
host = host.substring(0, colon);
}
}
}
}
if (host != null) {
target = new NameIntValuePair<String>(host, port);
}
}
return target;
}
}