From d9dbd4a92cebcb3043c975806da4133609c69a00 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=8E=8B=E5=B8=85?= <1474983351@qq.com>
Date: Mon, 30 Sep 2024 09:03:54 +0800
Subject: [PATCH] =?UTF-8?q?test:=20=E6=B5=8B=E8=AF=95=E6=95=B0=E7=BB=84/?=
 =?UTF-8?q?=E9=9B=86=E5=90=88=E4=B8=BA=20null=20=E6=88=96=E8=80=85?=
 =?UTF-8?q?=E7=A9=BA=E6=97=B6=E7=9A=84=E8=A1=8C=E4=B8=BA=E3=80=82?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../java/com/mybatisflex/test/Account.java    |  3 +-
 .../mybatisflex/test/AccountNativeTest.java   | 58 +++++++++++++++++++
 2 files changed, 60 insertions(+), 1 deletion(-)

diff --git a/mybatis-flex-test/mybatis-flex-native-test/src/main/java/com/mybatisflex/test/Account.java b/mybatis-flex-test/mybatis-flex-native-test/src/main/java/com/mybatisflex/test/Account.java
index 03228d8d..88608a4f 100644
--- a/mybatis-flex-test/mybatis-flex-native-test/src/main/java/com/mybatisflex/test/Account.java
+++ b/mybatis-flex-test/mybatis-flex-native-test/src/main/java/com/mybatisflex/test/Account.java
@@ -54,7 +54,8 @@ public class Account extends BaseEntity implements Serializable, AgeAware {
 //    @Column(isLogicDelete = true)
     private Boolean isDelete;
 
-    @Column(version = true)
+    // @Column(version = true)
+    @Column(ignore = true)
     private Integer version;
 
     private List<Article> articles;
diff --git a/mybatis-flex-test/mybatis-flex-native-test/src/test/java/com/mybatisflex/test/AccountNativeTest.java b/mybatis-flex-test/mybatis-flex-native-test/src/test/java/com/mybatisflex/test/AccountNativeTest.java
index ca3326ec..d94a417e 100644
--- a/mybatis-flex-test/mybatis-flex-native-test/src/test/java/com/mybatisflex/test/AccountNativeTest.java
+++ b/mybatis-flex-test/mybatis-flex-native-test/src/test/java/com/mybatisflex/test/AccountNativeTest.java
@@ -42,11 +42,16 @@ import org.springframework.jdbc.datasource.embedded.EmbeddedDatabase;
 import org.springframework.jdbc.datasource.embedded.EmbeddedDatabaseBuilder;
 import org.springframework.jdbc.datasource.embedded.EmbeddedDatabaseType;
 
+import java.lang.reflect.Array;
+import java.util.Arrays;
+import java.util.Collection;
 import java.util.List;
 import java.util.Objects;
 
 import static com.mybatisflex.test.table.AccountTableDef.ACCOUNT;
 import static com.mybatisflex.test.table.ArticleTableDef.ARTICLE;
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertTrue;
 
 
 public class AccountNativeTest implements WithAssertions {
@@ -91,6 +96,59 @@ public class AccountNativeTest implements WithAssertions {
         DataSourceKey.clear();
     }
 
+    @Test
+    public void testWhereIn() {
+        QueryWrapper queryWrapper;
+        // IN null 数组会先执行 QueryColumnBehavior.setIgnoreFunction 判断是否忽略这个条件
+        // 默认会忽略掉 null 值的条件，所以结果 IN 语句不会被拼接到 SQL 中，会查出所有数据！
+        assertFalse(accountMapper.selectAll().isEmpty());
+        queryWrapper = QueryWrapper.create()
+            .select()
+            .from(ACCOUNT)
+            .where(ACCOUNT.ID.in((Object[]) null));
+        assertFalse(accountMapper.selectListByQuery(queryWrapper).isEmpty());
+        // IN 空数组直接拼接到 SQL 语句：WHERE id IN ()
+        // 这样应该不会查出所有数据，具体要看各家数据库如何处理的了
+        queryWrapper = QueryWrapper.create()
+            .select()
+            .from(ACCOUNT)
+            .where(ACCOUNT.ID.in());
+        assertTrue(accountMapper.selectListByQuery(queryWrapper).isEmpty());
+        // 对于空数组/集合的处理，统一放到 QueryColumnBehavior.setIgnoreFunction 中
+        // 默认是数据库处理 WHERE id IN () 的情况，也可以放到代码当中，比如抛出异常
+        QueryColumnBehavior.setIgnoreFunction(value -> {
+            if (value == null) {
+                return true;
+            }
+            if (value.getClass().isArray() && Array.getLength(value) == 0) {
+                throw new IllegalArgumentException("EMPTY ARRAY!");
+            }
+            if (value instanceof Collection && ((Collection<?>) value).isEmpty()) {
+                throw new IllegalArgumentException("EMPTY COLLECTION!");
+            }
+            return false;
+        });
+        try {
+            QueryWrapper.create()
+                .select()
+                .from(ACCOUNT)
+                .where(ACCOUNT.ID.in());
+        } catch (IllegalArgumentException e) {
+            // ...
+        } finally {
+            QueryColumnBehavior.setIgnoreFunction(QueryColumnBehavior.IGNORE_NULL);
+        }
+
+        // 条件忽略测试
+        queryWrapper = QueryWrapper.create()
+            .select()
+            .from(ACCOUNT)
+            .where(ACCOUNT.ID.in(1, 2, 3))
+            .and(ACCOUNT.ID.in(new Object[]{4, 5, 6}, false))
+            .and(ACCOUNT.ID.in(Arrays.asList(7, 8, 9), () -> Boolean.FALSE));
+        assertFalse(accountMapper.selectListByQuery(queryWrapper).isEmpty());
+    }
+
     @Test
     public void testSelect() {
         QueryColumnBehavior.setIgnoreFunction(QueryColumnBehavior.IGNORE_NONE);