diff --git a/mybatis-flex-core/src/main/java/com/mybatisflex/core/query/SqlOperators.java b/mybatis-flex-core/src/main/java/com/mybatisflex/core/query/SqlOperators.java index fbb8fe00..9e67b3e1 100644 --- a/mybatis-flex-core/src/main/java/com/mybatisflex/core/query/SqlOperators.java +++ b/mybatis-flex-core/src/main/java/com/mybatisflex/core/query/SqlOperators.java @@ -36,17 +36,13 @@ public class SqlOperators extends HashMap { return new SqlOperators(); } - public static SqlOperators of( - LambdaGetter getter, SqlOperator operator - ) { + public static SqlOperators of( LambdaGetter getter, SqlOperator operator ) { SqlOperators map = new SqlOperators(1); map.put(LambdaUtil.getFieldName(getter), operator); return map; } - public static SqlOperators of( - String fieldName, SqlOperator operator - ) { + public static SqlOperators of(String fieldName, SqlOperator operator) { SqlOperators map = new SqlOperators(1); map.put(fieldName, operator); return map; @@ -64,8 +60,14 @@ public class SqlOperators extends HashMap { return this; } - public SqlOperators set(String fieldName, SqlOperator operator) { + public SqlOperators set(String fieldName, SqlOperator operator) { this.put(fieldName, operator); return this; } + + public SqlOperators set(QueryColumn column, SqlOperator operator) { + this.put(column.getName(), operator); + return this; + } + }