00001 /* 00002 * Copyright 2010 Google Inc. 00003 * 00004 * Licensed under the Apache License, Version 2.0 (the "License"); 00005 * you may not use this file except in compliance with the License. 00006 * You may obtain a copy of the License at 00007 * 00008 * http:///www.apache.org/licenses/LICENSE-2.0 00009 * 00010 * Unless required by applicable law or agreed to in writing, software 00011 * distributed under the License is distributed on an "AS IS" BASIS, 00012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 00013 * See the License for the specific language governing permissions and 00014 * limitations under the License. 00015 */ 00016 00020 00021 #ifndef NET_INSTAWEB_HTTP_PUBLIC_COUNTING_URL_ASYNC_FETCHER_H_ 00022 #define NET_INSTAWEB_HTTP_PUBLIC_COUNTING_URL_ASYNC_FETCHER_H_ 00023 00024 #include "net/instaweb/util/public/basictypes.h" 00025 #include "net/instaweb/http/public/url_async_fetcher.h" 00026 #include "net/instaweb/util/public/string.h" 00027 00028 namespace net_instaweb { 00029 00030 class MessageHandler; 00031 class RequestHeaders; 00032 class ResponseHeaders; 00033 class Writer; 00034 00035 class CountingUrlAsyncFetcher : public UrlAsyncFetcher { 00036 public: 00037 explicit CountingUrlAsyncFetcher(UrlAsyncFetcher* fetcher) 00038 : fetcher_(fetcher) { 00039 Clear(); 00040 } 00041 virtual ~CountingUrlAsyncFetcher(); 00042 00043 void set_fetcher(UrlAsyncFetcher* fetcher) { fetcher_ = fetcher; } 00044 00045 virtual bool SupportsHttps() const { return fetcher_->SupportsHttps(); } 00046 00047 virtual bool StreamingFetch(const GoogleString& url, 00048 const RequestHeaders& request_headers, 00049 ResponseHeaders* response_headers, 00050 Writer* fetched_content_writer, 00051 MessageHandler* message_handler, 00052 Callback* callback); 00053 int fetch_count() const { return fetch_count_; } 00054 int byte_count() const { return byte_count_; } 00055 int failure_count() const { return failure_count_; } 00056 00057 void Clear(); 00058 00059 class Fetch; 00060 friend class Fetch; 00061 00062 private: 00063 UrlAsyncFetcher* fetcher_; 00064 int fetch_count_; 00065 int byte_count_; 00066 int failure_count_; 00067 00068 DISALLOW_COPY_AND_ASSIGN(CountingUrlAsyncFetcher); 00069 }; 00070 00071 } 00072 00073 #endif ///< NET_INSTAWEB_HTTP_PUBLIC_COUNTING_URL_ASYNC_FETCHER_H_