From 7f384ebd51991de75577891a397374db16e27264 Mon Sep 17 00:00:00 2001
From: Ryan McGeary <ryanongit@mcgeary.org>
Date: Tue, 3 May 2011 11:54:18 -0400
Subject: [PATCH] Refactored two code paths of GridIO#gets into well-named
 methods

---
 lib/mongo/gridfs/grid_io.rb | 80 ++++++++++++++++++++-----------------
 1 file changed, 44 insertions(+), 36 deletions(-)

diff --git a/lib/mongo/gridfs/grid_io.rb b/lib/mongo/gridfs/grid_io.rb
index 4e2349f..e7020c0 100644
--- a/lib/mongo/gridfs/grid_io.rb
+++ b/lib/mongo/gridfs/grid_io.rb
@@ -208,43 +208,9 @@ module Mongo
       elsif separator.is_a?(Integer)
         read_length(separator)
       elsif separator.length > 1
-        result = ''
-        len = 0
-        match_idx = 0
-        match_num = separator.length - 1
-        to_match = separator[match_idx].chr
-        if length
-          matcher = lambda {|idx, num| idx < num && len < length }
-        else
-          matcher = lambda {|idx, num| idx < num}
-        end
-        while matcher.call(match_idx, match_num) && char = getc
-          result << char
-          len += 1
-          if char == to_match
-            while match_idx < match_num do
-              match_idx += 1
-              to_match = separator[match_idx].chr
-              char = getc
-              result << char
-              if char != to_match
-                match_idx = 0
-                to_match = separator[match_idx].chr
-                break
-              end
-            end
-          end
-        end
-        result.length > 0 ? result : nil
+        read_to_string(separator, length)
       else
-        result = ''
-        len = 0
-        while char = getc
-          result << char
-          len += 1
-          break if char == separator || (length ? len >= length : false)
-        end
-        result.length > 0 ? result : nil
+        read_to_character(separator, length)
       end
     end
 
@@ -359,6 +325,48 @@ module Mongo
       buf
     end
 
+    def read_to_character(character="\n", length=nil)
+      result = ''
+      len = 0
+      while char = getc
+        result << char
+        len += 1
+        break if char == character || (length ? len >= length : false)
+      end
+      result.length > 0 ? result : nil
+    end
+
+    def read_to_string(string="\n", length=nil)
+      result = ''
+      len = 0
+      match_idx = 0
+      match_num = string.length - 1
+      to_match = string[match_idx].chr
+      if length
+        matcher = lambda {|idx, num| idx < num && len < length }
+      else
+        matcher = lambda {|idx, num| idx < num}
+      end
+      while matcher.call(match_idx, match_num) && char = getc
+        result << char
+        len += 1
+        if char == to_match
+          while match_idx < match_num do
+            match_idx += 1
+            to_match = string[match_idx].chr
+            char = getc
+            result << char
+            if char != to_match
+              match_idx = 0
+              to_match = string[match_idx].chr
+              break
+            end
+          end
+        end
+      end
+      result.length > 0 ? result : nil
+    end
+
     def cache_chunk_data
       @current_chunk_data = @current_chunk['data'].to_s
       if @current_chunk_data.respond_to?(:force_encoding)