扩展

yzg-util-base/src/main/java/com/yanzuoguang/util/base/ObjectHelper.java

@@ -238,10 +238,27 @@ public class ObjectHelper {
         }
     }
 
+    /**
+     * 获取标记
+     *
+     * @param typeCache 类型
+     * @param tag       标记
+     * @param fromName  来源名称
+     * @return
+     */
+    private static MethodField getField(HashMap<String, MethodField> typeCache, String tag, String fromName) {
+        String toName = fromName.toLowerCase().replace("_", "");
+        if (!typeCache.containsKey(toName)) {
+            toName = toName.substring(tag.length());
+            return getField(typeCache, toName);
+        }
+        return typeCache.get(toName);
+    }
+
     /***
      * 获取缓存中的字段
-     * @param typeCache
-     * @param fromName
+     * @param typeCache 类型
+     * @param fromName 来源名称
      * @return
      */
     private static MethodField getField(HashMap<String, MethodField> typeCache, String fromName) {
@@ -297,8 +314,7 @@ public class ObjectHelper {
                 if (method.getParameterTypes().length != 1) {
                     continue;
                 }
-                String toName = methodNameSource.substring("set".length());
-                MethodField obj = getField(typeCache, toName);
+                MethodField obj = getField(typeCache, "set", methodNameSource);
                 if (obj.setMethod == null) {
                     obj.setMethod = method;
                 }
@@ -306,8 +322,7 @@ public class ObjectHelper {
                 if (method.getReturnType() == null) {
                     continue;
                 }
-                String toName = methodNameSource.substring("get".length());
-                MethodField obj = getField(typeCache, toName);
+                MethodField obj = getField(typeCache, "get", methodNameSource);
                 if (obj.getMethod == null) {
                     obj.getMethod = method;
                 }
@@ -315,8 +330,7 @@ public class ObjectHelper {
                 if (method.getReturnType() == null) {
                     continue;
                 }
-                String toName = methodNameSource.substring("is".length());
-                MethodField obj = getField(typeCache, toName);
+                MethodField obj = getField(typeCache, "is", methodNameSource);
                 if (obj.getMethod == null) {
                     obj.getMethod = method;
                 }