| /* |
| * Copyright 2012 Google Inc. |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License 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. |
| */ |
| |
| // Author: morlovich@google.com (Maksim Orlovich) |
| // |
| // ModSpdyFetchController coordinates a threadpool and a rate controller between |
| // multiple ModSpdyFetcher objects. The basic usage pattern is that |
| // ModSpdyFetcher::Fetch calls ModSpdyFetchController::ScheduleBlockingFetch, |
| // which will then cause ModSpdyFetcher::BlockingFetch to be called on a |
| // thread in a hopefully intelligent manner. |
| |
| #ifndef PAGESPEED_APACHE_MOD_SPDY_FETCH_CONTROLLER_H_ |
| #define PAGESPEED_APACHE_MOD_SPDY_FETCH_CONTROLLER_H_ |
| |
| #include "net/instaweb/http/public/rate_controller.h" |
| #include "pagespeed/kernel/base/atomic_bool.h" |
| #include "pagespeed/kernel/base/basictypes.h" |
| #include "pagespeed/kernel/base/string.h" |
| #include "pagespeed/kernel/thread/queued_worker_pool.h" |
| |
| namespace net_instaweb { |
| |
| class AsyncFetch; |
| class MessageHandler; |
| class ModSpdyFetcher; |
| class Statistics; |
| class Timer; |
| class ThreadSystem; |
| |
| class ModSpdyFetchController { |
| public: |
| // Note: RateController::InitStats must have been called before using this. |
| ModSpdyFetchController(int num_threads, |
| ThreadSystem* thread_system, |
| Timer* timer, |
| Statistics* statistics); |
| ~ModSpdyFetchController(); |
| |
| // This must be called for every statistics object in use before using this. |
| static void InitStats(Statistics* statistics); |
| |
| // Arranges for fetcher->BlockingFetch to be called on our thread pool. |
| void ScheduleBlockingFetch( |
| ModSpdyFetcher* fetcher, const GoogleString& url, Statistics* stats, |
| MessageHandler* message_handler, AsyncFetch* fetch); |
| |
| // Makes any further fetches quick-fail, and makes us more careful about |
| // using external dependencies on things like stats. |
| void ShutDown() { shutdown_.set_value(true); } |
| bool is_shut_down() const { return shutdown_.value(); } |
| |
| Timer* timer() const { return timer_; } |
| |
| private: |
| class FetchDispatcher; |
| |
| Timer* timer_; |
| RateController rate_controller_; |
| QueuedWorkerPool thread_pool_; |
| AtomicBool shutdown_; |
| DISALLOW_COPY_AND_ASSIGN(ModSpdyFetchController); |
| }; |
| |
| } // namespace net_instaweb |
| |
| #endif // PAGESPEED_APACHE_MOD_SPDY_FETCH_CONTROLLER_H_ |