Java tutorial
/* * Copyright 2010-2016 ksyun.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://ksyun.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.ksc.http.protocol; import java.io.IOException; import java.net.Socket; import org.apache.http.HttpClientConnection; import org.apache.http.HttpException; import org.apache.http.HttpRequest; import org.apache.http.HttpResponse; import org.apache.http.conn.ManagedHttpClientConnection; import org.apache.http.protocol.HttpContext; import org.apache.http.protocol.HttpRequestExecutor; import com.ksc.internal.SdkMetricsSocket; import com.ksc.internal.SdkSSLMetricsSocket; import com.ksc.util.KscRequestMetrics; import com.ksc.util.KscRequestMetrics.Field; /** * Used to capture the http send-request and receive-response latency metrics * of the http client library, with no retries involved. */ public class SdkHttpRequestExecutor extends HttpRequestExecutor { @Override protected HttpResponse doSendRequest(final HttpRequest request, final HttpClientConnection conn, final HttpContext context) throws IOException, HttpException { KscRequestMetrics awsRequestMetrics = (KscRequestMetrics) context .getAttribute(KscRequestMetrics.class.getSimpleName()); if (awsRequestMetrics == null) { return super.doSendRequest(request, conn, context); } if (conn instanceof ManagedHttpClientConnection) { ManagedHttpClientConnection managedConn = (ManagedHttpClientConnection) conn; Socket sock = managedConn.getSocket(); if (sock instanceof SdkMetricsSocket) { SdkMetricsSocket sdkMetricsSocket = (SdkMetricsSocket) sock; sdkMetricsSocket.setMetrics(awsRequestMetrics); } else if (sock instanceof SdkSSLMetricsSocket) { SdkSSLMetricsSocket sdkSSLMetricsSocket = (SdkSSLMetricsSocket) sock; sdkSSLMetricsSocket.setMetrics(awsRequestMetrics); } } awsRequestMetrics.startEvent(Field.HttpClientSendRequestTime); try { return super.doSendRequest(request, conn, context); } finally { awsRequestMetrics.endEvent(Field.HttpClientSendRequestTime); } } @Override protected HttpResponse doReceiveResponse(final HttpRequest request, final HttpClientConnection conn, final HttpContext context) throws HttpException, IOException { KscRequestMetrics awsRequestMetrics = (KscRequestMetrics) context .getAttribute(KscRequestMetrics.class.getSimpleName()); if (awsRequestMetrics == null) { return super.doReceiveResponse(request, conn, context); } awsRequestMetrics.startEvent(Field.HttpClientReceiveResponseTime); try { return super.doReceiveResponse(request, conn, context); } finally { awsRequestMetrics.endEvent(Field.HttpClientReceiveResponseTime); } } }