001/* 002 * ==================================================================== 003 * Licensed to the Apache Software Foundation (ASF) under one 004 * or more contributor license agreements. See the NOTICE file 005 * distributed with this work for additional information 006 * regarding copyright ownership. The ASF licenses this file 007 * to you under the Apache License, Version 2.0 (the 008 * "License"); you may not use this file except in compliance 009 * with the License. You may obtain a copy of the License at 010 * 011 * http://www.apache.org/licenses/LICENSE-2.0 012 * 013 * Unless required by applicable law or agreed to in writing, 014 * software distributed under the License is distributed on an 015 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY 016 * KIND, either express or implied. See the License for the 017 * specific language governing permissions and limitations 018 * under the License. 019 * ==================================================================== 020 * 021 * This software consists of voluntary contributions made by many 022 * individuals on behalf of the Apache Software Foundation. For more 023 * information on the Apache Software Foundation, please see 024 * <http://www.apache.org/>. 025 * 026 */ 027package org.apache.http.benchmark; 028 029import java.io.InputStream; 030import java.io.OutputStream; 031 032import org.apache.http.impl.DefaultBHttpClientConnection; 033import org.apache.http.io.SessionInputBuffer; 034import org.apache.http.io.SessionOutputBuffer; 035 036class BenchmarkConnection extends DefaultBHttpClientConnection { 037 038 private final Stats stats; 039 040 BenchmarkConnection(final int bufsize, final Stats stats) { 041 super(bufsize); 042 this.stats = stats; 043 } 044 045 @Override 046 protected OutputStream createOutputStream(final long len, final SessionOutputBuffer outbuffer) { 047 return new CountingOutputStream(super.createOutputStream(len, outbuffer), this.stats); 048 } 049 050 @Override 051 protected InputStream createInputStream(final long len, final SessionInputBuffer inbuffer) { 052 return new CountingInputStream(super.createInputStream(len, inbuffer), this.stats); 053 } 054 055}