New methods for loading and unloading encoding & decoding hooks in more detail

Brendan W. McAdams · Brendan W. McAdams · commit c86c378e048a · 2010-11-09T12:25:22.000-05:00
diff --git a/src/main/org/bson/BSON.java b/src/main/org/bson/BSON.java
@@ -150,7 +150,7 @@ private static void _warnUnsupportedRegex( String flag ) {
     // --- (en|de)coding hooks -----
 
     public static void addEncodingHook( Class c , Transformer t ){
-        _anyHooks = true;
+        _encodeHooks = true;
         List<Transformer> l = _encodingHooks.get( c );
         if ( l == null ){
             l = new Vector<Transformer>();
@@ -160,7 +160,7 @@ public static void addEncodingHook( Class c , Transformer t ){
     }
     
     public static void addDecodingHook( Class c , Transformer t ){
-        _anyHooks = true;
+        _decodeHooks = true;
         List<Transformer> l = _decodingHooks.get( c );
         if ( l == null ){
             l = new Vector<Transformer>();
@@ -170,7 +170,7 @@ public static void addDecodingHook( Class c , Transformer t ){
     }
 
     public static Object applyEncodingHooks( Object o ){
-        if ( ! _anyHooks )
+        if ( ! _anyHooks() )
             return o;
 
         if ( _encodingHooks.size() == 0 || o == null )
@@ -183,7 +183,7 @@ public static Object applyEncodingHooks( Object o ){
     }
 
     public static Object applyDecodingHooks( Object o ){
-        if ( ! _anyHooks || o == null )
+        if ( ! _anyHooks() || o == null )
             return o;
 
         List<Transformer> l = _decodingHooks.get( o.getClass() );
@@ -193,14 +193,80 @@ public static Object applyDecodingHooks( Object o ){
         return o;
     }
 
+   /**
+     * Returns the encoding hook(s) associated with the specified class
+     * 
+     */
+    public static List<Transformer> getEncodingHooks( Class c ){
+        return _encodingHooks.get( c );
+    }
 
-    public static void clearAllHooks(){
-        _anyHooks = false;
+    /**
+     * Clears *all* encoding hooks.
+     */
+    public static void clearEncodingHooks(){
+        _encodeHooks = false;
         _encodingHooks.clear();
+    }
+
+    /** 
+     * Remove all encoding hooks for a specific class.
+     */
+    public static void removeEncodingHooks( Class c ){
+        _encodingHooks.remove( c );
+    }
+
+    /** 
+     * Remove a specific encoding hook for a specific class.
+     */
+    public static void removeEncodingHook( Class c , Transformer t ){
+        getEncodingHooks( c ).remove( t );
+    }
+
+   /**
+     * Returns the decoding hook(s) associated with the specific class
+     */
+    public static List<Transformer> getDecodingHooks( Class c ){
+        return _decodingHooks.get( c );
+    }
+
+    /**
+     * Clears *all* decoding hooks.
+     */
+    public static void clearDecodingHooks(){
+        _decodeHooks = false;
         _decodingHooks.clear();
     }
 
-    private static boolean _anyHooks = false;
+    /** 
+     * Remove all decoding hooks for a specific class.
+     */
+    public static void removeDecodingHooks( Class c ){
+        _decodingHooks.remove( c );
+    }
+
+    /** 
+     * Remove a specific encoding hook for a specific class.
+     */
+    public static void removeDecodingHook( Class c , Transformer t ){
+        getDecodingHooks( c ).remove( t );
+    }
+
+
+    public static void clearAllHooks(){
+        clearEncodingHooks();
+        clearDecodingHooks();
+    }
+
+    /**
+     * Returns true if any encoding or decoding hooks are loaded.
+     */
+    private static boolean _anyHooks(){
+        return _encodeHooks | _decodeHooks;
+    }
+
+    private static boolean _encodeHooks = false;
+    private static boolean _decodeHooks = false;
     static ClassMap<List<Transformer>> _encodingHooks = 
 	new ClassMap<List<Transformer>>();
         
