mongo_ext 0.16 → 0.17

data/ext/cbson/cbson.c

@@ -38,8 +38,11 @@
 
 #include <assert.h>
 #include <math.h>
+#include <unistd.h>
+#include <time.h>
 
 #define INITIAL_BUFFER_SIZE 256
+#define MAX_HOSTNAME_LENGTH 256
 
 static VALUE Binary;
 static VALUE Time;
@@ -49,6 +52,7 @@ static VALUE Code;
 static VALUE RegexpOfHolding;
 static VALUE OrderedHash;
 static VALUE InvalidName;
+static VALUE DigestMD5;
 
 #if HAVE_RUBY_ENCODING_H
 #include "ruby/encoding.h"
@@ -737,8 +741,58 @@ static VALUE method_deserialize(VALUE self, VALUE bson) {
     return elements_to_hash(buffer, remaining);
 }
 
+
+static VALUE fast_pack(VALUE self)
+{
+    VALUE res;
+    long i;
+    char c;
+
+    res = rb_str_buf_new(0);
+
+    for (i = 0; i < RARRAY_LEN(self); i++) {
+        c = FIX2LONG(RARRAY_PTR(self)[i]);
+        rb_str_buf_cat(res, &c, sizeof(char));
+    }
+
+    return res;
+}
+
+
+static VALUE objectid_generate(VALUE self)
+{
+    VALUE oid, digest;
+    char hostname[MAX_HOSTNAME_LENGTH];
+    unsigned char oid_bytes[12];
+    unsigned long t, inc;
+    unsigned short pid;
+    int i;
+
+    t = htonl(time(NULL));
+    MEMCPY(&oid_bytes, &t, unsigned char, 4);
+
+    if (gethostname(&hostname, MAX_HOSTNAME_LENGTH) != 0) {
+        rb_raise(rb_eRuntimeError, "failed to get hostname");
+    }
+    digest = rb_funcall(DigestMD5, rb_intern("digest"), 1, rb_str_new2(hostname));
+    MEMCPY(&oid_bytes[4], RSTRING_PTR(digest), unsigned char, 3);
+
+    pid = htons(getpid());
+    MEMCPY(&oid_bytes[7], &pid, unsigned char, 2);
+
+    inc = htonl(FIX2ULONG(rb_funcall(self, rb_intern("get_inc"), 0)));
+    MEMCPY(&oid_bytes[9], ((unsigned char*)&inc + 1), unsigned char, 3);
+
+    oid = rb_ary_new2(12);
+    for(i = 0; i < 12; i++) {
+        rb_ary_store(oid, i, INT2FIX((unsigned int)oid_bytes[i]));
+    }
+    return oid;
+}
+
+
 void Init_cbson() {
-    VALUE mongo, CBson;
+    VALUE mongo, CBson, Digest;
     Time = rb_const_get(rb_cObject, rb_intern("Time"));
 
     mongo = rb_const_get(rb_cObject, rb_intern("Mongo"));
@@ -760,4 +814,12 @@ void Init_cbson() {
     CBson = rb_define_module("CBson");
     rb_define_module_function(CBson, "serialize", method_serialize, 2);
     rb_define_module_function(CBson, "deserialize", method_deserialize, 1);
+
+    rb_require("digest/md5");
+    Digest = rb_const_get(rb_cObject, rb_intern("Digest"));
+    DigestMD5 = rb_const_get(Digest, rb_intern("MD5"));
+
+    rb_define_method(ObjectID, "generate", objectid_generate, 0);
+
+    rb_define_method(rb_cArray, "fast_pack", fast_pack, 0);
 }

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: mongo_ext
 version: !ruby/object:Gem::Version 
-  version: "0.16"
+  version: "0.17"
 platform: ruby
 authors: 
 - Mike Dirolf
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2009-10-26 00:00:00 -04:00
+date: 2009-11-16 00:00:00 -05:00
 default_executable: 
 dependencies: []
 
@@ -26,7 +26,7 @@ files:
 - mongo-extensions.gemspec
 - ext/cbson/extconf.rb
 - ext/cbson/cbson.c
-has_rdoc: true
+has_rdoc: false
 homepage: http://www.mongodb.org
 licenses: []