检查插件是否被添加

pagehelper · Dec 28, 2023 · b6e029e · b6e029e
1 parent 6440439
commit b6e029e
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/src/main/java/com/github/pagehelper/PageInterceptor.java b/src/main/java/com/github/pagehelper/PageInterceptor.java
@@ -248,7 +248,9 @@ private Long count(Executor executor, MappedStatement ms, Object parameter,
 
     @Override
     public Object plugin(Object target) {
-        return Plugin.wrap(target, this);
+        Object wrapObjet = Plugin.wrap(target, this);
+        PageMethod.isAddToInterceptors = true;
+        return wrapObjet;
     }
 
     @Override

diff --git a/src/main/java/com/github/pagehelper/page/PageMethod.java b/src/main/java/com/github/pagehelper/page/PageMethod.java
@@ -26,6 +26,7 @@
 
 import com.github.pagehelper.ISelect;
 import com.github.pagehelper.Page;
+import com.github.pagehelper.PageException;
 import com.github.pagehelper.util.PageObjectUtil;
 
 import java.util.Properties;
@@ -36,6 +37,7 @@
  * @author liuzh
  */
 public abstract class PageMethod {
+    public static volatile boolean           isAddToInterceptors = false;
     protected static final ThreadLocal<Page> LOCAL_PAGE    = new ThreadLocal<Page>();
     protected static       boolean           DEFAULT_COUNT = true;
 
@@ -45,6 +47,9 @@ public abstract class PageMethod {
      * @param page
      */
     public static void setLocalPage(Page page) {
+        if(!isAddToInterceptors){
+            throw new PageException("page helper not added to interceptors!");
+        }
         LOCAL_PAGE.set(page);
     }