refactor: 优化EnumInFragmentBuilder逻辑

Author: zhou-hao

hsweb-easy-orm-rdb/src/main/java/org/hswebframework/ezorm/rdb/metadata/dialect/Dialect.java

@@ -4,6 +4,8 @@
 import org.hswebframework.ezorm.rdb.metadata.DataType;
 import org.hswebframework.ezorm.rdb.metadata.RDBColumnMetadata;
 import org.hswebframework.ezorm.rdb.metadata.RDBFeatureType;
+import org.hswebframework.ezorm.rdb.operator.builder.fragments.SqlFragments;
+import org.hswebframework.ezorm.rdb.operator.builder.fragments.term.EnumInFragmentBuilder;
 import org.hswebframework.ezorm.rdb.supports.h2.H2Dialect;
 import org.hswebframework.ezorm.rdb.supports.mssql.SqlServerDialect;
 import org.hswebframework.ezorm.rdb.supports.mysql.MysqlDialect;
@@ -76,6 +78,16 @@ default String buildColumnFullName(String tableName, String columnName, boolean
         return StringUtils.concat(tableName, ".", getQuoteStart(), changeCase && isColumnToUpperCase() ? columnName.toUpperCase() : columnName, getQuoteEnd());
     }
 
+    /**
+     * 位运算AND操作,用于枚举类型的查询等操作
+     *
+     * @param column 列名
+     * @param value  位值
+     * @return SQL片段
+     */
+    default SqlFragments bitAnd(String column, long value) {
+        return SqlFragments.of(column, "&", String.valueOf(value));
+    }
 
     Dialect MYSQL = new MysqlDialect();
     Dialect ORACLE = new OracleDialect();

hsweb-easy-orm-rdb/src/main/java/org/hswebframework/ezorm/rdb/operator/builder/fragments/term/EnumInFragmentBuilder.java

@@ -73,8 +73,12 @@ public static EnumInFragmentBuilder of(Dialect dialect) {
         if (dialect instanceof OracleDialect) {
             return OracleEnumInFragmentBuilder.in;
         }
-
-        throw new UnsupportedOperationException("unsupported db type :" + dialect);
+        return new EnumInFragmentBuilder(false) {
+            @Override
+            public SqlFragments bitAnd(String column, long value) {
+                return dialect.bitAnd(column,value);
+            }
+        };
     }
 
     public static EnumInFragmentBuilder ofNot(Dialect dialect) {
@@ -90,7 +94,11 @@ public static EnumInFragmentBuilder ofNot(Dialect dialect) {
         if (dialect instanceof OracleDialect) {
             return OracleEnumInFragmentBuilder.notIn;
         }
-
-        throw new UnsupportedOperationException("unsupported db type :" + dialect.getType());
+        return new EnumInFragmentBuilder(true) {
+            @Override
+            public SqlFragments bitAnd(String column, long value) {
+                return dialect.bitAnd(column,value);
+            }
+        };
     }
 }

hsweb-easy-orm-rdb/src/main/java/org/hswebframework/ezorm/rdb/supports/mysql/MysqlDialect.java

@@ -4,6 +4,7 @@
 import org.hswebframework.ezorm.rdb.metadata.JdbcDataType;
 import org.hswebframework.ezorm.rdb.metadata.dialect.DefaultDialect;
 import org.hswebframework.ezorm.core.utils.StringUtils;
+import org.hswebframework.ezorm.rdb.operator.builder.fragments.SqlFragments;
 
 import java.sql.Date;
 import java.sql.JDBCType;

hsweb-easy-orm-rdb/src/main/java/org/hswebframework/ezorm/rdb/supports/oracle/OracleDialect.java

@@ -4,10 +4,14 @@
 import org.hswebframework.ezorm.rdb.metadata.JdbcDataType;
 import org.hswebframework.ezorm.rdb.metadata.dialect.DefaultDialect;
 import org.hswebframework.ezorm.core.utils.StringUtils;
+import org.hswebframework.ezorm.rdb.operator.builder.fragments.SimpleSqlFragments;
+import org.hswebframework.ezorm.rdb.operator.builder.fragments.SqlFragments;
 
 import java.math.BigDecimal;
 import java.sql.Date;
 import java.sql.JDBCType;
+import java.util.Arrays;
+import java.util.Collections;
 
 public class OracleDialect extends DefaultDialect {
     public OracleDialect() {
@@ -81,4 +85,12 @@ public String getId() {
     public String getName() {
         return "Oracle";
     }
+
+    @Override
+    public SqlFragments bitAnd(String column, long value) {
+        return SimpleSqlFragments.of(
+            Arrays.asList("BITAND(", column, ",", "?)"),
+            Collections.singletonList(value)
+        );
+    }
 }