HttpConnectionMetricsImpl.java revision 417f3b92ba4549b2f22340e3107d869d2b9c5bb8 
1/*
2 * $HeadURL: http://svn.apache.org/repos/asf/httpcomponents/httpcore/trunk/module-main/src/main/java/org/apache/http/impl/HttpConnectionMetricsImpl.java $
3 * $Revision: 548031 $
4 * $Date: 2007-06-17 04:28:38 -0700 (Sun, 17 Jun 2007) $
5 *
6 * ====================================================================
7 * Licensed to the Apache Software Foundation (ASF) under one
8 * or more contributor license agreements.  See the NOTICE file
9 * distributed with this work for additional information
10 * regarding copyright ownership.  The ASF licenses this file
11 * to you under the Apache License, Version 2.0 (the
12 * "License"); you may not use this file except in compliance
13 * with the License.  You may obtain a copy of the License at
14 *
15 *   http://www.apache.org/licenses/LICENSE-2.0
16 *
17 * Unless required by applicable law or agreed to in writing,
18 * software distributed under the License is distributed on an
19 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
20 * KIND, either express or implied.  See the License for the
21 * specific language governing permissions and limitations
22 * under the License.
23 * ====================================================================
24 *
25 * This software consists of voluntary contributions made by many
26 * individuals on behalf of the Apache Software Foundation.  For more
27 * information on the Apache Software Foundation, please see
28 * <http://www.apache.org/>.
29 *
30 */
31
32package org.apache.http.impl;
33
34import java.util.HashMap;
35import org.apache.http.HttpConnectionMetrics;
36import org.apache.http.io.HttpTransportMetrics;
37
38/**
39 * Implementation of the metrics interface.
40 */
41public class HttpConnectionMetricsImpl implements HttpConnectionMetrics {
42
43    public static final String REQUEST_COUNT = "http.request-count";
44    public static final String RESPONSE_COUNT = "http.response-count";
45    public static final String SENT_BYTES_COUNT = "http.sent-bytes-count";
46    public static final String RECEIVED_BYTES_COUNT = "http.received-bytes-count";
47
48    private final HttpTransportMetrics inTransportMetric;
49    private final HttpTransportMetrics outTransportMetric;
50    private long requestCount = 0;
51    private long responseCount = 0;
52
53    /**
54     * The cache map for all metrics values.
55     */
56    private HashMap metricsCache;
57
58    public HttpConnectionMetricsImpl(
59            final HttpTransportMetrics inTransportMetric,
60            final HttpTransportMetrics outTransportMetric) {
61        super();
62        this.inTransportMetric = inTransportMetric;
63        this.outTransportMetric = outTransportMetric;
64    }
65
66    /* ------------------  Public interface method -------------------------- */
67
68    public long getReceivedBytesCount() {
69        if (this.inTransportMetric != null) {
70            return this.inTransportMetric.getBytesTransferred();
71        } else {
72            return -1;
73        }
74    }
75
76    public long getSentBytesCount() {
77        if (this.outTransportMetric != null) {
78            return this.outTransportMetric.getBytesTransferred();
79        } else {
80            return -1;
81        }
82    }
83
84    public long getRequestCount() {
85        return this.requestCount;
86    }
87
88    public void incrementRequestCount() {
89        this.requestCount++;
90    }
91
92    public long getResponseCount() {
93        return this.responseCount;
94    }
95
96    public void incrementResponseCount() {
97        this.responseCount++;
98    }
99
100    public Object getMetric(final String metricName) {
101        Object value = null;
102        if (this.metricsCache != null) {
103            value = this.metricsCache.get(metricName);
104        }
105        if (value == null) {
106            if (REQUEST_COUNT.equals(metricName)) {
107                value = new Long(requestCount);
108            } else if (RESPONSE_COUNT.equals(metricName)) {
109                value = new Long(responseCount);
110            } else if (RECEIVED_BYTES_COUNT.equals(metricName)) {
111                if (this.inTransportMetric != null) {
112                    return new Long(this.inTransportMetric.getBytesTransferred());
113                } else {
114                    return null;
115                }
116            } else if (SENT_BYTES_COUNT.equals(metricName)) {
117                if (this.outTransportMetric != null) {
118                    return new Long(this.outTransportMetric.getBytesTransferred());
119                } else {
120                    return null;
121                }
122            }
123        }
124        return value;
125    }
126
127    public void setMetric(final String metricName, Object obj) {
128        if (this.metricsCache == null) {
129            this.metricsCache = new HashMap();
130        }
131        this.metricsCache.put(metricName, obj);
132    }
133
134    public void reset() {
135        if (this.outTransportMetric != null) {
136            this.outTransportMetric.reset();
137        }
138        if (this.inTransportMetric != null) {
139            this.inTransportMetric.reset();
140        }
141        this.requestCount = 0;
142        this.responseCount = 0;
143        this.metricsCache = null;
144    }
145
146}
147