From 19d67fd1a8a4808c0722f384302e99a6640fbe21 Mon Sep 17 00:00:00 2001
From: yulichang <570810310@qq.com>
Date: Mon, 22 Jul 2024 03:01:08 +0800
Subject: [PATCH] =?UTF-8?q?=E7=AE=80=E5=8C=96=E4=BB=A3=E7=A0=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../github/yulichang/annotation/Table.java    |  8 +++----
 .../com/github/yulichang/apt/OgnlRoot.java    | 22 +++++++++++++++++++
 2 files changed, 26 insertions(+), 4 deletions(-)

diff --git a/mybatis-plus-join-annotation/src/main/java/com/github/yulichang/annotation/Table.java b/mybatis-plus-join-annotation/src/main/java/com/github/yulichang/annotation/Table.java
index 8e7b5a2..afe866a 100644
--- a/mybatis-plus-join-annotation/src/main/java/com/github/yulichang/annotation/Table.java
+++ b/mybatis-plus-join-annotation/src/main/java/com/github/yulichang/annotation/Table.java
@@ -29,10 +29,10 @@ import java.lang.annotation.Target;
  * <p>
  * 举例：
  * <ul>
- *   <li>去掉3长度的后缀并且大写  Ognl#classInfo.className.substring(0, classInfo.className.length() - 3).toUpperCase() </li>
- *   <li>以“APT”结尾 Ognl#classInfo.className + 'APT'</li>
- *   <li>全大写并且以“APT”结尾 Ognl#classInfo.className.toUpperCase() + 'APT' </li>
- *   <li>"PO"结尾替换为“APT”且全大写  Ognl#stringHelper.replaceSuffix(classInfo.className, 'PO', 'APT').toUpperCase() </li>
+ *   <li>去掉3长度的后缀并且大写  Ognl#className.substring(0, className.length() - 3).toUpperCase() </li>
+ *   <li>以“APT”结尾 Ognl#className + 'APT'</li>
+ *   <li>全大写并且以“APT”结尾 Ognl#className.toUpperCase() + 'APT' </li>
+ *   <li>"PO"结尾替换为“APT”且全大写  Ognl#util.replaceSuffix(className, 'PO', 'APT').toUpperCase() </li>
  * </ul>
  * <p>
  * 支持 三元运算 String所有方法 如lastIndexOf subString toUpperCase等 Ognl语法<p>
diff --git a/mybatis-plus-join-core/src/main/java/com/github/yulichang/apt/OgnlRoot.java b/mybatis-plus-join-core/src/main/java/com/github/yulichang/apt/OgnlRoot.java
index 4574998..2999fa2 100644
--- a/mybatis-plus-join-core/src/main/java/com/github/yulichang/apt/OgnlRoot.java
+++ b/mybatis-plus-join-core/src/main/java/com/github/yulichang/apt/OgnlRoot.java
@@ -13,8 +13,14 @@ import java.util.Objects;
 @Getter
 public class OgnlRoot {
 
+    /**
+     * 类名
+     */
     private final String className;
 
+    /**
+     * 包名
+     */
     private final String classPackage;
 
     private final StringHelper util = new StringHelper();
@@ -26,6 +32,22 @@ public class OgnlRoot {
 
     @SuppressWarnings("unused")
     public static final class StringHelper {
+        /**
+         * 移除后缀
+         *
+         * @param str    原字符串
+         * @param suffix 指定后缀
+         */
+        public String removeSuffix(String str, String suffix) {
+            if (isBlank(str) || isBlank(suffix)) {
+                return str;
+            }
+            if (str.endsWith(suffix)) {
+                return str.substring(0, str.length() - suffix.length());
+            }
+            return str;
+        }
+
         /**
          * 替换后缀
          *