simulated_delay_fetcher.h

Go to the documentation of this file.

/*
 * Copyright 2014 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.
 */


#ifndef NET_INSTAWEB_HTTP_PUBLIC_SIMULATED_DELAY_FETCHER_H_
#define NET_INSTAWEB_HTTP_PUBLIC_SIMULATED_DELAY_FETCHER_H_

#include <map>

#include "net/instaweb/http/public/url_async_fetcher.h"
#include "pagespeed/kernel/base/basictypes.h"
#include "pagespeed/kernel/base/file_system.h"
#include "pagespeed/kernel/base/scoped_ptr.h"
#include "pagespeed/kernel/base/string.h"
#include "pagespeed/kernel/base/string_util.h"
#include "pagespeed/kernel/base/thread_annotations.h"

namespace net_instaweb {

class AbstractMutex;
class AsyncFetch;
class MessageHandler;
class Scheduler;
class ThreadSystem;
class Timer;

class SimulatedDelayFetcher : public UrlAsyncFetcher {
 public:
  static const char kPayload[];

  SimulatedDelayFetcher(ThreadSystem* thread_system,
                        Timer* timer,
                        Scheduler* scheduler,
                        MessageHandler* handler,
                        FileSystem* file_system,
                        StringPiece delay_map_path,
                        StringPiece request_log_path,
                        int request_log_flush_frequency);

  virtual ~SimulatedDelayFetcher();

  virtual void Fetch(const GoogleString& url,
                     MessageHandler* message_handler,
                     AsyncFetch* fetch);

 private:
  typedef std::map<GoogleString, int> DelayMap;
  void ProduceReply(AsyncFetch* fetch);
  void ParseDelayMap(StringPiece delay_map_path);

  Timer* timer_;
  Scheduler* scheduler_;
  MessageHandler* message_handler_;
  FileSystem* file_system_;
  DelayMap delays_ms_;
  int request_log_flush_frequency_;

  scoped_ptr<AbstractMutex> mutex_;
  int request_log_outstanding_ GUARDED_BY(mutex_.get());
  FileSystem::OutputFile* request_log_ PT_GUARDED_BY(mutex_.get());

  
};

}  

#endif