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_UTIL_PUBLIC_CHUNKING_WRITER_H_ 00020 #define NET_INSTAWEB_UTIL_PUBLIC_CHUNKING_WRITER_H_ 00021 00022 #include "net/instaweb/util/public/basictypes.h" 00023 #include "net/instaweb/util/public/string_util.h" 00024 #include "net/instaweb/util/public/writer.h" 00025 00026 namespace net_instaweb { 00027 00028 class MessageHandler; 00029 00032 class ChunkingWriter : public Writer { 00033 public: 00037 ChunkingWriter(Writer* writer, int flush_limit); 00038 virtual ~ChunkingWriter(); 00039 00040 virtual bool Write(const StringPiece& str, MessageHandler* handler); 00041 virtual bool Flush(MessageHandler* handler); 00042 00043 private: 00045 bool FlushIfNeeded(MessageHandler* handler); 00046 00047 Writer* const writer_; 00048 const int flush_limit_; 00049 int unflushed_bytes_; 00050 00051 DISALLOW_COPY_AND_ASSIGN(ChunkingWriter); 00052 }; 00053 00054 } 00055 00056 #endif ///< NET_INSTAWEB_UTIL_PUBLIC_CHUNKING_WRITER_H_