From 3a3109e7dcb621cb5a1b1ab565449eece808d49b Mon Sep 17 00:00:00 2001
From: Mike Dirolf <mike@10gen.com>
Date: Wed, 4 Mar 2009 11:07:22 -0500
Subject: [PATCH] enough encoders to do bson_benchmark

---
 bin/bson_benchmark.rb  |  2 +-
 ext/cbson/cbson.c      | 51 ++++++++++++++++++++++++++++++++++++++++++
 lib/mongo/util/bson.rb |  2 +-
 3 files changed, 53 insertions(+), 2 deletions(-)

diff --git a/bin/bson_benchmark.rb b/bin/bson_benchmark.rb
index 840a6ff..2e54711 100644
--- a/bin/bson_benchmark.rb
+++ b/bin/bson_benchmark.rb
@@ -44,7 +44,7 @@ TEST_CASES = [{},
                 "a bool" => false
               },
               {
-                "this"=> 5,
+                "this" => 5,
                 "is" => {"a" => true},
                 "big" => [true, 5.5],
                 "object" => nil
diff --git a/ext/cbson/cbson.c b/ext/cbson/cbson.c
index b16a873..6bbe521 100644
--- a/ext/cbson/cbson.c
+++ b/ext/cbson/cbson.c
@@ -11,6 +11,9 @@ typedef struct {
 } bson_buffer;
 
 static char zero = 0;
+static char one = 1;
+
+static void write_doc(bson_buffer* buffer, VALUE hash);
 
 static bson_buffer* buffer_new(void) {
     bson_buffer* buffer;
@@ -78,6 +81,54 @@ static int write_element(VALUE key, VALUE value, VALUE extra) {
     bson_buffer* buffer = (bson_buffer*)extra;
 
     switch(TYPE(value)) {
+    case T_FIXNUM:
+        write_name_and_type(buffer, key, 0x10);
+        int int_value = FIX2INT(value);
+        buffer_write_bytes(buffer, (char*)&int_value, 4);
+        break;
+    case T_TRUE:
+        write_name_and_type(buffer, key, 0x08);
+        buffer_write_bytes(buffer, &one, 1);
+        break;
+    case T_FALSE:
+        write_name_and_type(buffer, key, 0x08);
+        buffer_write_bytes(buffer, &zero, 1);
+        break;
+    case T_FLOAT:
+        write_name_and_type(buffer, key, 0x01);
+        double d = NUM2DBL(value);
+        buffer_write_bytes(buffer, (char*)&d, 8);
+        break;
+    case T_NIL:
+        write_name_and_type(buffer, key, 0x0A);
+        break;
+    case T_HASH:
+        write_name_and_type(buffer, key, 0x03);
+        write_doc(buffer, value);
+        break;
+    case T_ARRAY:
+        write_name_and_type(buffer, key, 0x04);
+        int start_position = buffer->position;
+
+        // save space for length
+        int length_location = buffer_save_bytes(buffer, 4);
+
+        int items = RARRAY_LEN(value);
+        VALUE* values = RARRAY_PTR(value);
+        int i;
+        for(i = 0; i < items; i++) {
+            char* name;
+            asprintf(&name, "%d", i);
+            VALUE key = rb_str_new2(name);
+            write_element(key, values[i], (VALUE)buffer);
+            free(name);
+        }
+
+        // write null byte and fill in length
+        buffer_write_bytes(buffer, &zero, 1);
+        int obj_length = buffer->position - start_position;
+        memcpy(buffer->buffer + length_location, &obj_length, 4);
+        break;
     case T_STRING:
         write_name_and_type(buffer, key, 0x02);
         int length = RSTRING(value)->len + 1;
diff --git a/lib/mongo/util/bson.rb b/lib/mongo/util/bson.rb
index 82bc835..adc7bba 100644
--- a/lib/mongo/util/bson.rb
+++ b/lib/mongo/util/bson.rb
@@ -134,7 +134,7 @@ class BSON
   def deserialize(buf=nil, parent=nil)
     # If buf is nil, use @buf, assumed to contain already-serialized BSON.
     # This is only true during testing.
-    @buf = ByteBuffer.new(buf.to_a) if buf
+    @buf = ByteBuffer.new(buf) if buf
     @buf.rewind
     @buf.get_int                # eat message size
     doc = OrderedHash.new