LineIterator allow different delimiter

ronggh · Jul 19, 2016 · fac3d94 · fac3d94
1 parent b638cec
commit fac3d94
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/util/file_piece.cc b/util/file_piece.cc
@@ -32,7 +32,7 @@ ParseNumberException::ParseNumberException(StringPiece value) throw() {
 }
 
 LineIterator &LineIterator::operator++() {
-  if (!backing_->ReadLineOrEOF(line_))
+  if (!backing_->ReadLineOrEOF(line_, delim_))
     backing_ = NULL;
   return *this;
 }

diff --git a/util/file_piece.hh b/util/file_piece.hh
@@ -33,7 +33,7 @@ class LineIterator {
   public:
     LineIterator() : backing_(NULL) {}
 
-    explicit LineIterator(FilePiece &f) : backing_(&f) {
+    explicit LineIterator(FilePiece &f, char delim = '\n') : backing_(&f), delim_(delim) {
       ++*this;
     }
 
@@ -55,6 +55,7 @@ class LineIterator {
   private:
     FilePiece *backing_;
     StringPiece line_;
+    char delim_;
 };
 
 // Memory backing the returned StringPiece may vanish on the next call.