short & byte should be INT JAVA-96

mongodb · Mar 26, 2010 · 263fa95 · 263fa95
1 parent 997e367
commit 263fa95
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/main/com/mongodb/ByteEncoder.java b/src/main/com/mongodb/ByteEncoder.java
@@ -20,6 +20,7 @@
 
 import java.util.*;
 import java.util.regex.*;
+import java.util.concurrent.atomic.*;
 import java.nio.*;
 import java.nio.charset.*;
 import java.lang.reflect.Array;
@@ -380,11 +381,15 @@ protected int putDate( String name , Date d ){
 
     protected int putNumber( String name , Number n ){
         int start = _buf.position();
-	if ( n instanceof Integer ){
+	if ( n instanceof Integer ||
+             n instanceof Short ||
+             n instanceof Byte ||
+             n instanceof AtomicInteger ){
 	    _put( NUMBER_INT , name );
 	    _buf.putInt( n.intValue() );
 	}
-        else if (n instanceof Long ) {
+        else if ( n instanceof Long || 
+                  n instanceof AtomicLong ) {
             _put( NUMBER_LONG , name );
             _buf.putLong( n.longValue() );
         }