resource_fetch.h

Go to the documentation of this file.

/*
 * Copyright 2011 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_REWRITER_PUBLIC_RESOURCE_FETCH_H_
#define NET_INSTAWEB_REWRITER_PUBLIC_RESOURCE_FETCH_H_

#include "net/instaweb/http/public/async_fetch.h"
#include "net/instaweb/http/public/request_context.h"
#include "pagespeed/kernel/base/basictypes.h"
#include "pagespeed/kernel/http/google_url.h"

namespace net_instaweb {

class MessageHandler;
class ServerContext;
class RewriteDriver;
class RewriteOptions;
class SyncFetcherAdapterCallback;
class Timer;

class ResourceFetch : public SharedAsyncFetch {
 public:
  enum CleanupMode {
    kAutoCleanupDriver,
    kDontAutoCleanupDriver
  };

  static void Start(const GoogleUrl& url,
                    RewriteOptions* custom_options,
                    ServerContext* server_context,
                    AsyncFetch* async_fetch);

  static void StartWithDriver(const GoogleUrl& url,
                              CleanupMode cleanup_mode,
                              ServerContext* server_context,
                              RewriteDriver* driver,
                              AsyncFetch* async_fetch);

  static bool BlockingFetch(const GoogleUrl& url,
                            ServerContext* server_context,
                            RewriteDriver* driver,
                            SyncFetcherAdapterCallback* async_fetch);

  static RewriteDriver* GetDriver(const GoogleUrl& url,
                                  RewriteOptions* custom_options,
                                  ServerContext* server_context,
                                  const RequestContextPtr& request_ctx);

 protected:
  virtual void HandleHeadersComplete();
  virtual void HandleDone(bool success);

 private:
  ResourceFetch(const GoogleUrl& url, CleanupMode cleanup_mode,
                RewriteDriver* driver, Timer* timer,
                MessageHandler* handler, AsyncFetch* async_fetch);
  virtual ~ResourceFetch();

  static void ApplyExperimentOptions(const GoogleUrl& url,
                                     const RequestContextPtr& request_ctx,
                                     ServerContext* server_context,
                                     RewriteOptions** custom_options);

  GoogleUrl resource_url_;
  RewriteDriver* driver_;
  Timer* timer_;
  MessageHandler* message_handler_;

  int64 start_time_ms_;
  int redirect_count_;
  CleanupMode cleanup_mode_;

  
};

}  

#endif