diff --git a/mybatis-flex-core/src/main/java/com/mybatisflex/core/query/IfFunctionQueryColumn.java b/mybatis-flex-core/src/main/java/com/mybatisflex/core/query/IfFunctionQueryColumn.java new file mode 100644 index 00000000..c11e3f52 --- /dev/null +++ b/mybatis-flex-core/src/main/java/com/mybatisflex/core/query/IfFunctionQueryColumn.java @@ -0,0 +1,74 @@ +/* + * Copyright (c) 2022-2023, Mybatis-Flex (fuhai999@gmail.com). + *
+ * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + *
+ * http://www.apache.org/licenses/LICENSE-2.0 + *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.mybatisflex.core.query;
+
+import com.mybatisflex.core.dialect.IDialect;
+import com.mybatisflex.core.util.ArrayUtil;
+
+import java.util.List;
+
+/**
+ * IF 函数查询列。
+ *
+ * @author 王帅
+ * @since 2023-07-07
+ */
+public class IfFunctionQueryColumn extends QueryColumn implements HasParamsColumn {
+
+ private QueryCondition condition;
+ private QueryColumn trueValue;
+ private QueryColumn falseValue;
+
+ public IfFunctionQueryColumn(QueryCondition condition, QueryColumn trueValue, QueryColumn falseValue) {
+ this.condition = condition;
+ this.trueValue = trueValue;
+ this.falseValue = falseValue;
+ }
+
+ @Override
+ String toConditionSql(List
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.mybatisflex.coretest;
+
+import com.mybatisflex.core.query.QueryWrapper;
+import org.junit.Test;
+
+import static com.mybatisflex.core.query.QueryMethods.*;
+import static com.mybatisflex.coretest.table.AccountTableDef.ACCOUNT;
+
+/**
+ * @author 王帅
+ * @since 2023-07-07
+ */
+public class IfFunctionTest {
+
+ @Test
+ public void test01() {
+ QueryWrapper queryWrapper = QueryWrapper.create()
+ .select(if_(ACCOUNT.AGE.ge(6), ACCOUNT.IS_NORMAL, ACCOUNT.IS_DELETE).as("type"))
+ .from(ACCOUNT)
+ .where(ACCOUNT.ID.eq(1));
+ System.out.println(queryWrapper.toSQL());
+ }
+
+ @Test
+ public void test02() {
+ QueryWrapper queryWrapper = QueryWrapper.create()
+ .select(if_(ACCOUNT.AGE.ge(18), string("成年人"),
+ if_(ACCOUNT.AGE.le(8), string("未上学"), string("已上学"))).as("type"))
+ .from(ACCOUNT)
+ .where(ACCOUNT.ID.eq(1));
+ System.out.println(queryWrapper.toSQL());
+ }
+
+ @Test
+ public void test03() {
+ QueryWrapper queryWrapper = QueryWrapper.create()
+ .select(ifNull(ACCOUNT.ID, number(0)))
+ .from(ACCOUNT)
+ .where(ACCOUNT.ID.eq(1));
+ System.out.println(queryWrapper.toSQL());
+ }
+
+ @Test
+ public void test04() {
+ QueryWrapper queryWrapper = QueryWrapper.create()
+ .select(ifNull(null_(), number(0)))
+ .from(ACCOUNT);
+ System.out.println(queryWrapper.toSQL());
+ }
+
+}
\ No newline at end of file