flag support

kristina · kristina · commit 246c6916dac4 · 2009-01-10T20:55:09.000-05:00
diff --git a/src/main/com/mongodb/Bytes.java b/src/main/com/mongodb/Bytes.java
@@ -4,6 +4,8 @@
 
 import java.nio.*;
 import java.nio.charset.*;
+import java.util.regex.Pattern;
+import java.util.Arrays;
 
 /**
  * <type><name>0<data>
@@ -106,15 +108,67 @@ public static boolean cameFromDB( DBObject o ){
     }
 
     public static int patternFlags( String flags ){
-        if ( flags.length() == 0 )
-            return 0;
-        throw new RuntimeException( "can't convert flags yet" );    
+        flags = flags.toLowerCase();
+        Flag f[] = Flag.values();
+        int fint = 0;
+        int count = 0;
+
+        for( Flag flag : f ) {
+            if( flags.indexOf( flag.flagChar ) >= 0 ) {
+                fint |= flag.javaFlag;
+                count++;
+                if( flag.unsupported != null )
+                    _warnUnsupported( flag.unsupported );
+            }
+        }
+
+        if( count < flags.length() )
+            throw new RuntimeException( "some flags could not be converted: "+flags );
+
+        return fint;
     }
 
     public static String patternFlags( int flags ){
-        if ( flags == 0 )
-            return "";
-        throw new RuntimeException( "can't convert flags yet" );
+        Flag f[] = Flag.values();
+        byte b[] = new byte[ f.length ];
+        int count = 0;
+
+        for( Flag flag : f ) {
+            if( ( flags & flag.javaFlag ) > 0 ) {
+                b[ count++ ] = (byte)flag.flagChar;
+                flags -= flag.javaFlag;
+            }
+        }
+
+        if( flags > 0 )
+            throw new RuntimeException( "some flags could not be recognized." );
+
+        Arrays.sort( b );
+        return new String( b, 0, count );
+    }
+
+    private static enum Flag { 
+        CANON_EQ( Pattern.CANON_EQ, 'c', "PATTERN.CANON_EQ" ),
+        DOTALL( Pattern.DOTALL, 'd', "Pattern.DOTALL" ),
+        CASE_INSENSITIVE( Pattern.CASE_INSENSITIVE, 'i', null ),
+        UNIX_LINES( Pattern.UNIX_LINES, 'l', "Pattern.UNIX_LINES" ),
+        MULTILINE(Pattern.MULTILINE, 'm', null ),
+        LITERAL( Pattern.LITERAL, 't', "Pattern.LITERAL" ),
+        UNICODE_CASE( Pattern.UNICODE_CASE, 'u', "Pattern.UNICODE_CASE" ),
+        COMMENTS( Pattern.COMMENTS, 'x', null );
+
+        public final int javaFlag;
+        public final char flagChar;
+        public final String unsupported;
+        Flag( int f, char ch, String u ) {
+            javaFlag = f;
+            flagChar = ch;
+            unsupported = u;
+        }
+    }
+
+    private static void _warnUnsupported( String flag ) {
+        System.out.println( "flag " + flag + " not supported by db." );
     }
     
     static final String NO_REF_HACK = "_____nodbref_____";
diff --git a/src/test/com/mongodb/ByteTest.java b/src/test/com/mongodb/ByteTest.java
@@ -346,12 +346,24 @@ public void testPatternFlags() {
         threw = false;
 
         try {
-            Bytes.patternFlags( 6 );
+            Bytes.patternFlags( 257 );
         }
         catch( RuntimeException e ) {
             threw = true;
         }
         assertEquals( threw, true );
+
+        Pattern lotsoflags = Pattern.compile( "foo", Pattern.CANON_EQ & 
+                                              Pattern.DOTALL &
+                                              Pattern.CASE_INSENSITIVE &
+                                              Pattern.UNIX_LINES &
+                                              Pattern.MULTILINE &
+                                              Pattern.LITERAL &
+                                              Pattern.UNICODE_CASE &
+                                              Pattern.COMMENTS );
+
+        int check = Bytes.patternFlags( Bytes.patternFlags( lotsoflags.flags() ) );
+        assertEquals( lotsoflags.flags(), check );
     }
 
     @Test(groups = {"basic"})
@@ -373,7 +385,6 @@ public void testPattern() {
 
         encoder.done();
         decoder.done();
-
     }
 
     final DBBase _db;
