Merge pull request #13 from mfeng-ya/fn_bug_fix

fixed #mybatis-mapper/mapper/issues/108
mybatis-mapper · May 20, 2024 · 0da0d50 · 0da0d50
2 parents 9dff28c + b4394ad
commit 0da0d50
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/src/main/java/io/mybatis/mapper/fn/Fn.java b/src/main/java/io/mybatis/mapper/fn/Fn.java
@@ -184,6 +184,18 @@ public R apply(T t) {
       return fn.apply(t);
     }
 
+    @Override
+    public boolean equals(Object o) {
+      if (this == o) return true;
+      if (o == null || getClass() != o.getClass()) return false;
+      FnImpl<?, ?> fnImpl = (FnImpl<?, ?>) o;
+      return Objects.equals(entityClass, fnImpl.entityClass) && Objects.equals(fn, fnImpl.fn);
+    }
+
+    @Override
+    public int hashCode() {
+      return Objects.hash(entityClass, fn);
+    }
   }
 
   /**

diff --git a/src/test/java/io/mybatis/mapper/fn/FnTest.java b/src/test/java/io/mybatis/mapper/fn/FnTest.java
@@ -73,6 +73,11 @@ public void testMemoryOverflow() throws InterruptedException {
       Assert.assertEquals(columnSize + 2, Fn.FN_COLUMN_MAP.size());
       Assert.assertEquals(fieldSize + 2, Fn.FN_CLASS_FIELD_MAP.size());
     }
+    for (int i = 0; i < 100; i++) {
+      Fn.of(User.class, User::getUserName);
+      Assert.assertEquals(columnSize + 3, Fn.FN_COLUMN_MAP.size());
+      Assert.assertEquals(fieldSize + 3, Fn.FN_CLASS_FIELD_MAP.size());
+    }
   }
 
   public static class BaseId {