apache_fetch.h

Go to the documentation of this file.

// Copyright 2015 Google Inc.

#ifndef PAGESPEED_APACHE_FETCH_H_
#define PAGESPEED_APACHE_FETCH_H_

#include "net/instaweb/http/public/async_fetch.h"
#include "net/instaweb/http/public/request_context.h"
#include "net/instaweb/rewriter/public/rewrite_driver.h"
#include "net/instaweb/rewriter/public/rewrite_options.h"
#include "pagespeed/apache/apache_writer.h"
#include "pagespeed/kernel/base/basictypes.h"
#include "pagespeed/kernel/base/message_handler.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"
#include "pagespeed/kernel/http/request_headers.h"
#include "pagespeed/kernel/http/response_headers.h"
#include "pagespeed/kernel/thread/scheduler.h"

namespace net_instaweb {

class ApacheFetch : public AsyncFetch {
 public:
  ApacheFetch(const GoogleString& mapped_url, StringPiece debug_info,
              RewriteDriver* driver, ApacheWriter* apache_writer,
              RequestHeaders* request_headers,
              const RequestContextPtr& request_context,
              const RewriteOptions* options, MessageHandler* handler);
  virtual ~ApacheFetch();

  void set_handle_error(bool x) { handle_error_ = x; }

  void set_buffered(bool x) { buffered_ = x; }

  void Wait() LOCKS_EXCLUDED(scheduler_->mutex());

  bool status_ok() const { return status_ok_; }

  virtual bool IsCachedResultValid(const ResponseHeaders& headers)
      LOCKS_EXCLUDED(scheduler_->mutex());

  void set_is_proxy(bool x) { is_proxy_ = x; }

 protected:
  virtual void HandleHeadersComplete() LOCKS_EXCLUDED(scheduler_->mutex());
  virtual void HandleDone(bool success) LOCKS_EXCLUDED(scheduler_->mutex());
  virtual bool HandleFlush(MessageHandler* handler)
      LOCKS_EXCLUDED(scheduler_->mutex());
  virtual bool HandleWrite(const StringPiece& sp, MessageHandler* handler)
      LOCKS_EXCLUDED(scheduler_->mutex());

 private:
  void SendOutHeaders();

  GoogleString mapped_url_;
  scoped_ptr<ApacheWriter> apache_writer_;
  const RewriteOptions* options_ GUARDED_BY(scheduler_->mutex());

  MessageHandler* message_handler_;

  bool done_ GUARDED_BY(scheduler_->mutex());
  bool wait_called_;
  bool handle_error_;
  bool squelch_output_;
  bool status_ok_;
  bool is_proxy_;
  bool buffered_;
  GoogleString debug_info_;
  GoogleString output_bytes_;
  RewriteDriver* driver_;
  Scheduler* scheduler_;

  
};

}  

#endif