github-migration/mongo-ruby-driver
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

dbref encoder

Browse Source
This commit is contained in:
Mike Dirolf 2009-03-05 11:45:04 -05:00
parent 9f2305715e
commit 56595af858
1 changed files with 20 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
ext/cbson/cbson.c
View File

@ -170,9 +170,9 @@ static int write_element(VALUE key, VALUE value, VALUE extra) {
case T_SYMBOL: case T_SYMBOL:
write_name_and_type(buffer, key, 0x0E); write_name_and_type(buffer, key, 0x0E);
const char* str_value = rb_id2name(SYM2ID(value)); const char* str_value = rb_id2name(SYM2ID(value));
int str_length = strlen(str_value) + 1; length = strlen(str_value) + 1;
buffer_write_bytes(buffer, (char*)&str_length, 4); buffer_write_bytes(buffer, (char*)&length, 4);
buffer_write_bytes(buffer, str_value, str_length); buffer_write_bytes(buffer, str_value, length);
break; break;
case T_OBJECT: case T_OBJECT:
{ {
@ -207,6 +207,23 @@ static int write_element(VALUE key, VALUE value, VALUE extra) {
} }
break; break;
} }
if (strcmp(cls, "XGen::Mongo::Driver::DBRef") == 0) {
write_name_and_type(buffer, key, 0x0C);
VALUE ns = rb_funcall(value, rb_intern("namespace"), 0);
int length = RSTRING(ns)->len + 1;
buffer_write_bytes(buffer, (char*)&length, 4);
buffer_write_bytes(buffer, RSTRING(ns)->ptr, length - 1);
buffer_write_bytes(buffer, &zero, 1);
VALUE oid_as_array = rb_funcall(rb_funcall(value, rb_intern("object_id"), 0),
rb_intern("to_a"), 0);
for (i = 0; i < 12; i++) {
char byte = (char)FIX2INT(RARRAY(oid_as_array)->ptr[i]);
buffer_write_bytes(buffer, &byte, 1);
}
break;
}
if (strcmp(cls, "XGen::Mongo::Driver::Undefined") == 0) { if (strcmp(cls, "XGen::Mongo::Driver::Undefined") == 0) {
write_name_and_type(buffer, key, 0x06); write_name_and_type(buffer, key, 0x06);
break; break;