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.FilterInputStream; 030import java.io.IOException; 031import java.io.InputStream; 032 033class CountingInputStream extends FilterInputStream { 034 035 private final Stats stats; 036 037 CountingInputStream(final InputStream instream, final Stats stats) { 038 super(instream); 039 this.stats = stats; 040 } 041 042 @Override 043 public int read() throws IOException { 044 final int b = this.in.read(); 045 if (b != -1) { 046 this.stats.incTotalBytesRecv(1); 047 } 048 return b; 049 } 050 051 @Override 052 public int read(final byte[] b) throws IOException { 053 final int bytesRead = this.in.read(b); 054 if (bytesRead > 0) { 055 this.stats.incTotalBytesRecv(bytesRead); 056 } 057 return bytesRead; 058 } 059 060 @Override 061 public int read(final byte[] b, final int off, final int len) throws IOException { 062 final int bytesRead = this.in.read(b, off, len); 063 if (bytesRead > 0) { 064 this.stats.incTotalBytesRecv(bytesRead); 065 } 066 return bytesRead; 067 } 068 069 @Override 070 public long skip(final long n) throws IOException { 071 final long bytesRead = this.in.skip(n); 072 if (bytesRead > 0) { 073 this.stats.incTotalBytesRecv(bytesRead); 074 } 075 return bytesRead; 076 } 077 078}