From d8d3c13e641f02600f849f532cca2e9aa5422f0f Mon Sep 17 00:00:00 2001
From: Robot <luo-zhan@qq.com>
Date: Thu, 18 Jan 2024 22:05:09 +0800
Subject: [PATCH] =?UTF-8?q?refactor:=20=E8=87=AA=E5=8A=A8=E5=88=A4?=
 =?UTF-8?q?=E6=96=AD=E7=B1=BB=E5=9E=8B=EF=BC=8C=E7=9C=81=E7=95=A5=E5=AD=90?=
 =?UTF-8?q?=E7=B1=BB=E5=AE=9E=E7=8E=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../annotation/AbstractInsertListener.java    | 20 ++++++++++++++-----
 .../annotation/AbstractUpdateListener.java    | 20 ++++++++++++++-----
 2 files changed, 30 insertions(+), 10 deletions(-)

diff --git a/mybatis-flex-annotation/src/main/java/com/mybatisflex/annotation/AbstractInsertListener.java b/mybatis-flex-annotation/src/main/java/com/mybatisflex/annotation/AbstractInsertListener.java
index 23b6a006..8b35bd78 100644
--- a/mybatis-flex-annotation/src/main/java/com/mybatisflex/annotation/AbstractInsertListener.java
+++ b/mybatis-flex-annotation/src/main/java/com/mybatisflex/annotation/AbstractInsertListener.java
@@ -16,20 +16,31 @@
 
 package com.mybatisflex.annotation;
 
+import java.lang.reflect.ParameterizedType;
+import java.lang.reflect.Type;
+
 /**
  * 类型支持 insert 监听器。
  *
  * @author snow
+ * @author robot.luo
  * @since 2023/4/28
  */
 public abstract class AbstractInsertListener<T> implements InsertListener {
 
     /**
-     * 该监听器支持的 entity 类型。
-     *
-     * @return type
+     * 支持的类型
      */
-    public abstract Class<T> supportType();
+    private final Class<T> supportType;
+
+    @SuppressWarnings("unchecked")
+    protected AbstractInsertListener() {
+        Type[] params = ((ParameterizedType) getClass().getGenericSuperclass()).getActualTypeArguments();
+        if (params.length == 0) {
+            throw new IllegalArgumentException(this.getClass().getSimpleName() + "继承AbstractInsertListener请指定泛型");
+        }
+        supportType = (Class<T>) params[0];
+    }
 
     /**
      * 新增操作的前置操作。
@@ -41,7 +52,6 @@ public abstract class AbstractInsertListener<T> implements InsertListener {
     @Override
     @SuppressWarnings("unchecked")
     public void onInsert(Object entity) {
-        Class<T> supportType = supportType();
         if (supportType.isInstance(entity)) {
             T object = (T) entity;
             doInsert(object);
diff --git a/mybatis-flex-annotation/src/main/java/com/mybatisflex/annotation/AbstractUpdateListener.java b/mybatis-flex-annotation/src/main/java/com/mybatisflex/annotation/AbstractUpdateListener.java
index 18117c80..780aa80a 100644
--- a/mybatis-flex-annotation/src/main/java/com/mybatisflex/annotation/AbstractUpdateListener.java
+++ b/mybatis-flex-annotation/src/main/java/com/mybatisflex/annotation/AbstractUpdateListener.java
@@ -16,20 +16,31 @@
 
 package com.mybatisflex.annotation;
 
+import java.lang.reflect.ParameterizedType;
+import java.lang.reflect.Type;
+
 /**
  * 类型支持 update 监听器。
  *
  * @author snow
+ * @author robot.luo
  * @since 2023/4/28
  */
 public abstract class AbstractUpdateListener<T> implements UpdateListener {
 
     /**
-     * 该监听器支持的 entity 类型。
-     *
-     * @return type
+     * 支持的类型
      */
-    public abstract Class<T> supportType();
+    private final Class<T> supportType;
+
+    @SuppressWarnings("unchecked")
+    protected AbstractUpdateListener() {
+        Type[] params = ((ParameterizedType) getClass().getGenericSuperclass()).getActualTypeArguments();
+        if (params.length == 0) {
+            throw new IllegalArgumentException(this.getClass().getSimpleName() + "继承AbstractUpdateListener请指定泛型");
+        }
+        supportType = (Class<T>) params[0];
+    }
 
     /**
      * 更新操作的前置操作。
@@ -41,7 +52,6 @@ public abstract class AbstractUpdateListener<T> implements UpdateListener {
     @Override
     @SuppressWarnings("unchecked")
     public void onUpdate(Object entity) {
-        Class<T> supportType = supportType();
         if (supportType.isInstance(entity)) {
             T object = (T) entity;
             doUpdate(object);