Page Speed Optimization Libraries
1.7.30.4
|
00001 /* 00002 * Copyright 2011 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 00018 00019 #ifndef NET_INSTAWEB_REWRITER_PUBLIC_IN_PLACE_REWRITE_CONTEXT_H_ 00020 #define NET_INSTAWEB_REWRITER_PUBLIC_IN_PLACE_REWRITE_CONTEXT_H_ 00021 00022 #include "net/instaweb/http/public/async_fetch.h" 00023 #include "net/instaweb/http/public/content_type.h" 00024 #include "net/instaweb/http/public/http_value.h" 00025 #include "net/instaweb/http/public/http_value_writer.h" 00026 #include "net/instaweb/http/public/response_headers.h" 00027 #include "net/instaweb/rewriter/public/output_resource_kind.h" 00028 #include "net/instaweb/rewriter/public/resource.h" 00029 #include "net/instaweb/rewriter/public/server_context.h" 00030 #include "net/instaweb/rewriter/public/resource_slot.h" 00031 #include "net/instaweb/rewriter/public/rewrite_context.h" 00032 #include "net/instaweb/rewriter/public/rewrite_options.h" 00033 #include "net/instaweb/rewriter/public/single_rewrite_context.h" 00034 #include "net/instaweb/util/public/basictypes.h" 00035 #include "net/instaweb/util/public/proto_util.h" 00036 #include "net/instaweb/util/public/scoped_ptr.h" 00037 #include "net/instaweb/util/public/string.h" 00038 #include "net/instaweb/util/public/string_util.h" 00039 00040 namespace net_instaweb { 00041 00042 class CachedResult; 00043 class CacheUrlAsyncFetcher; 00044 class InputInfo; 00045 class MessageHandler; 00046 class ResourceContext; 00047 class RewriteDriver; 00048 class RewriteFilter; 00049 class Statistics; 00050 class Variable; 00051 00055 class InPlaceRewriteResourceSlot : public ResourceSlot { 00056 public: 00057 static const char kIproSlotLocation[]; 00058 explicit InPlaceRewriteResourceSlot(const ResourcePtr& resource); 00059 00061 virtual void Render(); 00062 00064 virtual GoogleString LocationString(); 00065 00066 protected: 00067 virtual ~InPlaceRewriteResourceSlot(); 00068 00069 private: 00070 DISALLOW_COPY_AND_ASSIGN(InPlaceRewriteResourceSlot); 00071 }; 00072 00074 class InPlaceRewriteContext : public SingleRewriteContext { 00075 public: 00079 static const char kInPlaceOversizedOptStream[]; 00080 static const char kInPlaceUncacheableRewrites[]; 00081 00082 InPlaceRewriteContext(RewriteDriver* driver, const StringPiece& url); 00083 virtual ~InPlaceRewriteContext(); 00084 00086 virtual void RewriteSingle(const ResourcePtr& input, 00087 const OutputResourcePtr& output); 00089 virtual const char* id() const { return RewriteOptions::kInPlaceRewriteId; } 00091 virtual OutputResourceKind kind() const { return kRewrittenResource; } 00093 virtual bool DecodeFetchUrls(const OutputResourcePtr& output_resource, 00094 MessageHandler* message_handler, 00095 GoogleUrlStarVector* url_vector); 00097 virtual void StartFetchReconstruction(); 00098 00099 static void InitStats(Statistics* statistics); 00100 00101 bool proxy_mode() const { return proxy_mode_; } 00102 void set_proxy_mode(bool x) { proxy_mode_ = x; } 00103 00104 virtual int64 GetRewriteDeadlineAlarmMs() const; 00105 00106 virtual GoogleString UserAgentCacheKey( 00107 const ResourceContext* resource_context) const; 00108 virtual void EncodeUserAgentIntoResourceContext(ResourceContext* context); 00109 00112 virtual bool CreationLockBeforeStartFetch() { return false; } 00113 00114 private: 00115 friend class RecordingFetch; 00117 virtual void Harvest(); 00118 void StartFetchReconstructionParent(); 00120 virtual void FixFetchFallbackHeaders(const CachedResult& cached_result, 00121 ResponseHeaders* headers); 00123 virtual void FetchTryFallback(const GoogleString& url, 00124 const StringPiece& hash); 00126 virtual void FetchCallbackDone(bool success); 00127 00128 RewriteFilter* GetRewriteFilter(const ContentType& type); 00129 00131 void UpdateDateAndExpiry(const protobuf::RepeatedPtrField<InputInfo>& inputs, 00132 int64* date_ms, int64* expiry_ms); 00135 bool InPlaceOptimizeForBrowserEnabled() const; 00138 void AddVaryIfRequired(const CachedResult& cached_result, 00139 ResponseHeaders* headers) const; 00140 00141 RewriteDriver* driver_; 00142 GoogleString url_; 00144 bool is_rewritten_; 00147 GoogleString rewritten_hash_; 00148 00150 ResourcePtr input_resource_; 00151 OutputResourcePtr output_resource_; 00152 00153 scoped_ptr<CacheUrlAsyncFetcher> cache_fetcher_; 00154 00164 bool proxy_mode_; 00165 00166 DISALLOW_COPY_AND_ASSIGN(InPlaceRewriteContext); 00167 }; 00168 00171 class RecordingFetch : public SharedAsyncFetch { 00172 public: 00173 RecordingFetch(bool proxy_mode, 00174 AsyncFetch* async_fetch, 00175 const ResourcePtr& resource, 00176 InPlaceRewriteContext* context, 00177 MessageHandler* handler); 00178 00179 virtual ~RecordingFetch(); 00180 00182 virtual void HandleHeadersComplete(); 00184 virtual bool HandleWrite(const StringPiece& content, MessageHandler* handler); 00186 virtual bool HandleFlush(MessageHandler* handler); 00188 virtual void HandleDone(bool success); 00189 00190 private: 00191 void FreeDriver(); 00192 00193 bool CanInPlaceRewrite(); 00194 00198 bool ShouldStream() const; 00199 00200 bool proxy_mode_; 00201 MessageHandler* handler_; 00202 ResourcePtr resource_; 00203 InPlaceRewriteContext* context_; 00204 00207 bool can_in_place_rewrite_; 00208 00210 bool streaming_; 00211 HTTPValue cache_value_; 00212 HTTPValueWriter cache_value_writer_; 00213 ResponseHeaders saved_headers_; 00214 Variable* in_place_oversized_opt_stream_; 00215 Variable* in_place_uncacheable_rewrites_; 00216 DISALLOW_COPY_AND_ASSIGN(RecordingFetch); 00217 }; 00218 00219 } 00220 00221 #endif ///< NET_INSTAWEB_REWRITER_PUBLIC_IN_PLACE_REWRITE_CONTEXT_H_