[JAVA-312]: com.mongodb.util.JSON.serialize does not support UUID

Author: agirbal

src/main/com/mongodb/util/JSON.java

@@ -193,6 +193,14 @@ public static void serialize( Object o , StringBuilder buf ){
             return;
         }
 
+        if ( o instanceof UUID ){
+            UUID uuid = (UUID)o;
+            BasicDBObject temp = new BasicDBObject();
+            temp.put( "$uuid" , uuid.toString() );
+            serialize( temp, buf );
+            return;
+        }
+
         if ( o instanceof CodeWScope ){
             CodeWScope c = (CodeWScope)o;
 

src/main/com/mongodb/util/JSONCallback.java

@@ -87,6 +87,13 @@ public Object objectDone(){
 		} else {
 		    setRoot(o);
 		}
+	    } else if ( b.containsField( "$uuid" ) ) {
+                o = UUID.fromString((String)b.get("$uuid"));
+		if (!isStackEmpty()) {
+		    cur().put( name, o );
+		} else {
+		    setRoot(o);
+		}
 	    }
 	}
         return o;

src/test/com/mongodb/JavaClientTest.java

@@ -750,7 +750,7 @@ public void testBadKey(){
     public void testAllTypes(){
         DBCollection c = _db.getCollectionFromString( "foo" );
         c.drop();
-        String json = "{ 'str' : 'asdfasd' , 'long' : 5 , 'float' : 0.4 , 'bool' : false , 'date' : { '$date' : '2011-05-18T18:56:00Z'} , 'pat' : { '$regex' : '.*' , '$options' : ''} , 'oid' : { '$oid' : '4d83ab3ea39562db9c1ae2ae'} , 'ref' : { '$ref' : 'test.test' , '$id' : { '$oid' : '4d83ab59a39562db9c1ae2af'}} , 'code' : { '$code' : 'asdfdsa'} , 'codews' : { '$code' : 'ggggg' , '$scope' : { }} , 'ts' : { '$ts' : 1300474885 , '$inc' : 10} , 'null' :  null }";
+        String json = "{ 'str' : 'asdfasd' , 'long' : 5 , 'float' : 0.4 , 'bool' : false , 'date' : { '$date' : '2011-05-18T18:56:00Z'} , 'pat' : { '$regex' : '.*' , '$options' : ''} , 'oid' : { '$oid' : '4d83ab3ea39562db9c1ae2ae'} , 'ref' : { '$ref' : 'test.test' , '$id' : { '$oid' : '4d83ab59a39562db9c1ae2af'}} , 'code' : { '$code' : 'asdfdsa'} , 'codews' : { '$code' : 'ggggg' , '$scope' : { }} , 'ts' : { '$ts' : 1300474885 , '$inc' : 10} , 'null' :  null, 'uuid' : { '$uuid' : '60f65152-6d4a-4f11-9c9b-590b575da7b5' }}";
         BasicDBObject a = (BasicDBObject) JSON.parse(json);
         c.insert(a);
         DBObject b = c.findOne();

src/test/com/mongodb/util/JSONTest.java

@@ -304,7 +304,7 @@ public void testDate() {
 
     @org.testng.annotations.Test
     public void testJSONEncoding() throws ParseException {
-        String json = "{ 'str' : 'asdfasd' , 'long' : 5 , 'float' : 0.4 , 'bool' : false , 'date' : { '$date' : '2011-05-18T18:56:00Z'} , 'pat' : { '$regex' : '.*' , '$options' : ''} , 'oid' : { '$oid' : '4d83ab3ea39562db9c1ae2ae'} , 'ref' : { '$ref' : 'test.test' , '$id' : { '$oid' : '4d83ab59a39562db9c1ae2af'}} , 'code' : { '$code' : 'asdfdsa'} , 'codews' : { '$code' : 'ggggg' , '$scope' : { }} , 'ts' : { '$ts' : 1300474885 , '$inc' : 10} , 'null' :  null }";
+        String json = "{ 'str' : 'asdfasd' , 'long' : 5 , 'float' : 0.4 , 'bool' : false , 'date' : { '$date' : '2011-05-18T18:56:00Z'} , 'pat' : { '$regex' : '.*' , '$options' : ''} , 'oid' : { '$oid' : '4d83ab3ea39562db9c1ae2ae'} , 'ref' : { '$ref' : 'test.test' , '$id' : { '$oid' : '4d83ab59a39562db9c1ae2af'}} , 'code' : { '$code' : 'asdfdsa'} , 'codews' : { '$code' : 'ggggg' , '$scope' : { }} , 'ts' : { '$ts' : 1300474885 , '$inc' : 10} , 'null' :  null, 'uuid' : { '$uuid' : '60f65152-6d4a-4f11-9c9b-590b575da7b5' }}";
         BasicDBObject a = (BasicDBObject) JSON.parse(json);
         assert (a.get("str").equals("asdfasd"));
         assert (a.get("long").equals(5L));
@@ -325,6 +325,7 @@ public void testJSONEncoding() throws ParseException {
         assert (a.get("code").equals(new Code("asdfdsa")));
         assert (a.get("codews").equals(new CodeWScope("ggggg", new BasicBSONObject())));
         assert (a.get("ts").equals(new BSONTimestamp(1300474885, 10)));
+        assert (a.get("uuid").equals(UUID.fromString("60f65152-6d4a-4f11-9c9b-590b575da7b5")));
         String json2 = JSON.serialize(a);
         BasicDBObject b = (BasicDBObject) JSON.parse(json2);
         a.equals(b);