Java callback class returns LinkedHashMap
This commit is contained in:
parent
315e608cd4
commit
159c89b8b4
@ -26,7 +26,7 @@ import org.bson.types.*;
|
|||||||
|
|
||||||
public class RubyBSONJavaCallback implements BSONCallback {
|
public class RubyBSONJavaCallback implements BSONCallback {
|
||||||
|
|
||||||
private RubyHash _root;
|
private LinkedHashMap _root;
|
||||||
private RubyModule _rbclsOrderedHash;
|
private RubyModule _rbclsOrderedHash;
|
||||||
private RubyModule _rbclsObjectId;
|
private RubyModule _rbclsObjectId;
|
||||||
private RubyModule _rbclsBinary;
|
private RubyModule _rbclsBinary;
|
||||||
@ -60,17 +60,10 @@ public class RubyBSONJavaCallback implements BSONCallback {
|
|||||||
_nameStack.clear();
|
_nameStack.clear();
|
||||||
}
|
}
|
||||||
|
|
||||||
public RubyHash createHash() {
|
|
||||||
RubyHash h = (RubyHash)JavaEmbedUtils.invokeMethod(_runtime, _rbclsOrderedHash, "new",
|
|
||||||
new Object[] { }, Object.class);
|
|
||||||
|
|
||||||
return h;
|
|
||||||
}
|
|
||||||
|
|
||||||
public Object create( boolean array , List<String> path ){
|
public Object create( boolean array , List<String> path ){
|
||||||
if ( array )
|
if ( array )
|
||||||
return new ArrayList<Object>();
|
return new ArrayList<Object>();
|
||||||
return createHash();
|
return new LinkedHashMap();
|
||||||
}
|
}
|
||||||
|
|
||||||
public void objectStart(){
|
public void objectStart(){
|
||||||
@ -78,7 +71,7 @@ public class RubyBSONJavaCallback implements BSONCallback {
|
|||||||
throw new IllegalStateException( "something is wrong" );
|
throw new IllegalStateException( "something is wrong" );
|
||||||
}
|
}
|
||||||
|
|
||||||
_root = createHash();
|
_root = new LinkedHashMap();
|
||||||
_stack.add(_root);
|
_stack.add(_root);
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -87,34 +80,21 @@ public class RubyBSONJavaCallback implements BSONCallback {
|
|||||||
}
|
}
|
||||||
|
|
||||||
public void objectStart(String key){
|
public void objectStart(String key){
|
||||||
RubyHash hash = createHash();
|
LinkedHashMap hash = new LinkedHashMap();
|
||||||
|
|
||||||
_nameStack.addLast( key );
|
_nameStack.addLast( key );
|
||||||
|
|
||||||
Object lastObject = _stack.getLast();
|
Object lastObject = _stack.getLast();
|
||||||
|
|
||||||
// Yes, this is a bit hacky.
|
// Yes, this is a bit hacky.
|
||||||
if(lastObject instanceof RubyHash) {
|
if(lastObject instanceof LinkedHashMap) {
|
||||||
writeRubyHash(key, (RubyHash)lastObject, (IRubyObject)hash);
|
((LinkedHashMap)hash).put( key, lastObject );
|
||||||
}
|
}
|
||||||
else {
|
else {
|
||||||
((ArrayList)lastObject).add(Integer.parseInt(key), hash);
|
((ArrayList)lastObject).add(Integer.parseInt(key), hash);
|
||||||
//writeRubyArray(key, (RubyArray)lastObject, (IRubyObject)hash);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
_stack.addLast( (RubyObject)hash );
|
_stack.addLast( hash );
|
||||||
}
|
|
||||||
|
|
||||||
public void writeRubyHash(String key, RubyHash hash, Object obj) {
|
|
||||||
RubyString rkey = _runtime.newString(key);
|
|
||||||
JavaEmbedUtils.invokeMethod(_runtime, hash, "[]=",
|
|
||||||
new Object[] { (IRubyObject)rkey, obj }, Object.class);
|
|
||||||
}
|
|
||||||
|
|
||||||
public void writeRubyArray(String key, RubyArray array, IRubyObject obj) {
|
|
||||||
Long rkey = Long.parseLong(key);
|
|
||||||
RubyFixnum index = new RubyFixnum(_runtime, rkey);
|
|
||||||
array.aset((IRubyObject)index, obj);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
public void arrayStart(String key){
|
public void arrayStart(String key){
|
||||||
@ -123,8 +103,8 @@ public class RubyBSONJavaCallback implements BSONCallback {
|
|||||||
Object lastObject = _stack.getLast();
|
Object lastObject = _stack.getLast();
|
||||||
_nameStack.addLast( key );
|
_nameStack.addLast( key );
|
||||||
|
|
||||||
if(lastObject instanceof RubyHash) {
|
if(lastObject instanceof LinkedHashMap) {
|
||||||
writeRubyHash(key, (RubyHash)lastObject, array);
|
((LinkedHashMap)lastObject).put( key, lastObject );
|
||||||
}
|
}
|
||||||
else {
|
else {
|
||||||
((ArrayList)lastObject).add(Integer.parseInt(key), array);
|
((ArrayList)lastObject).add(Integer.parseInt(key), array);
|
||||||
@ -250,6 +230,7 @@ public class RubyBSONJavaCallback implements BSONCallback {
|
|||||||
_put ( name , result );
|
_put ( name , result );
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// TODO: Optimize this. Incredibly slow.
|
||||||
public void gotObjectId( String name , ObjectId id ){
|
public void gotObjectId( String name , ObjectId id ){
|
||||||
IRubyObject arg = (IRubyObject)RubyString.newString(_runtime, id.toString());
|
IRubyObject arg = (IRubyObject)RubyString.newString(_runtime, id.toString());
|
||||||
// //System.out.println(id.toByteArray().length);
|
// //System.out.println(id.toByteArray().length);
|
||||||
@ -266,13 +247,13 @@ public class RubyBSONJavaCallback implements BSONCallback {
|
|||||||
_put( name, (RubyObject)result );
|
_put( name, (RubyObject)result );
|
||||||
}
|
}
|
||||||
|
|
||||||
// TODO: Incredibly annoying to deserialize to a Ruby DBRef. Might just
|
// DBRef has been deprecated in the Java driver. We're going to follow suit.
|
||||||
// stop supporting this altogether in the driver.
|
|
||||||
public void gotDBRef( String name , String ns , ObjectId id ){
|
public void gotDBRef( String name , String ns , ObjectId id ){
|
||||||
// _put( name , new BasicBSONObject( "$ns" , ns ).append( "$id" , id ) );
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// TODO: I know that this is horrible. To be optimized.
|
// TODO: I know that this is horrible. To be optimized.
|
||||||
|
// This conversion can probably be avoided altogether.
|
||||||
|
// Used only for GridFS.
|
||||||
private RubyArray ja2ra( byte[] b ) {
|
private RubyArray ja2ra( byte[] b ) {
|
||||||
RubyArray result = RubyArray.newArray( _runtime, b.length );
|
RubyArray result = RubyArray.newArray( _runtime, b.length );
|
||||||
|
|
||||||
@ -314,9 +295,7 @@ public class RubyBSONJavaCallback implements BSONCallback {
|
|||||||
//a.aset((IRubyObject)index, o);
|
//a.aset((IRubyObject)index, o);
|
||||||
}
|
}
|
||||||
else {
|
else {
|
||||||
RubyString rkey = RubyString.newString(_runtime, name);
|
((LinkedHashMap)current).put( name, o );
|
||||||
JavaEmbedUtils.invokeMethod(_runtime, current, "[]=",
|
|
||||||
new Object[] { (IRubyObject)rkey, o }, Object.class);
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -328,30 +307,14 @@ public class RubyBSONJavaCallback implements BSONCallback {
|
|||||||
return _root;
|
return _root;
|
||||||
}
|
}
|
||||||
|
|
||||||
protected void setRoot(RubyHash o) {
|
// protected void setRoot(LinkedHashMap o) {
|
||||||
_root = o;
|
// _root = o;
|
||||||
}
|
// }
|
||||||
|
//
|
||||||
protected boolean isStackEmpty() {
|
protected boolean isStackEmpty() {
|
||||||
return _stack.size() < 1;
|
return _stack.size() < 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Helper method for checking whether a Ruby hash has a certain key.
|
|
||||||
private boolean _rbHashHasKey(RubyHash hash, String key) {
|
|
||||||
RubyBoolean b = hash.has_key_p( _runtime.newString( key ) );
|
|
||||||
return b == _runtime.getTrue();
|
|
||||||
}
|
|
||||||
|
|
||||||
// Helper method for getting a value from a Ruby hash.
|
|
||||||
private IRubyObject _rbHashGet(RubyHash hash, Object key) {
|
|
||||||
if (key instanceof String) {
|
|
||||||
return hash.op_aref( _runtime.getCurrentContext(), _runtime.newString((String)key) );
|
|
||||||
}
|
|
||||||
else {
|
|
||||||
return hash.op_aref( _runtime.getCurrentContext(), (RubyObject)key );
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
static final HashMap<String, Object> _getRuntimeCache(Ruby runtime) {
|
static final HashMap<String, Object> _getRuntimeCache(Ruby runtime) {
|
||||||
// each JRuby runtime may have different objects for these constants,
|
// each JRuby runtime may have different objects for these constants,
|
||||||
// so cache them separately for each runtime
|
// so cache them separately for each runtime
|
||||||
|
Loading…
Reference in New Issue
Block a user