From d31d36887e5c50ffdc93960377fa1fc699630a5c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=BC=80=E6=BA=90=E6=B5=B7=E5=93=A5?= <fuhai999@gmail.com>
Date: Fri, 28 Apr 2023 07:53:12 +0800
Subject: [PATCH] rename UpdateEntity.wrap to UpdateEntity.of; close #I6Z7HK

---
 .../mybatisflex/core/util/UpdateEntity.java   | 51 +++++++++++++++++--
 1 file changed, 47 insertions(+), 4 deletions(-)

diff --git a/mybatis-flex-core/src/main/java/com/mybatisflex/core/util/UpdateEntity.java b/mybatis-flex-core/src/main/java/com/mybatisflex/core/util/UpdateEntity.java
index 3cb9176e..f192bced 100644
--- a/mybatis-flex-core/src/main/java/com/mybatisflex/core/util/UpdateEntity.java
+++ b/mybatis-flex-core/src/main/java/com/mybatisflex/core/util/UpdateEntity.java
@@ -16,26 +16,69 @@
 package com.mybatisflex.core.util;
 
 import com.mybatisflex.core.javassist.ModifyAttrsRecordProxyFactory;
+import com.mybatisflex.core.table.IdInfo;
+import com.mybatisflex.core.table.TableInfo;
+import com.mybatisflex.core.table.TableInfoFactory;
 import org.apache.ibatis.reflection.DefaultReflectorFactory;
 import org.apache.ibatis.reflection.Reflector;
 import org.apache.ibatis.reflection.ReflectorFactory;
+import org.apache.ibatis.reflection.invoker.Invoker;
+
+import java.lang.reflect.Array;
+import java.util.List;
 
 public class UpdateEntity {
 
     private static ReflectorFactory reflectorFactory = new DefaultReflectorFactory();
 
 
-
-    public static <T> T wrap(Class<T> clazz) {
+    public static <T> T of(Class<T> clazz) {
         return ModifyAttrsRecordProxyFactory.getInstance().get(clazz);
     }
 
 
+    public static <T> T of(Class<T> clazz, Object id) {
+        T newEntity = ModifyAttrsRecordProxyFactory.getInstance().get(clazz);
+        TableInfo tableInfo = TableInfoFactory.ofEntityClass(clazz);
+        List<IdInfo> primaryKeyList = tableInfo.getPrimaryKeyList();
+        Reflector reflector = reflectorFactory.findForClass(clazz);
+
+        if (primaryKeyList != null && !primaryKeyList.isEmpty()) {
+            for (int i = 0; i < primaryKeyList.size(); i++) {
+                IdInfo idInfo = primaryKeyList.get(i);
+                Object idValue = getIdValue(id, i);
+                Invoker setInvoker = reflector.getSetInvoker(idInfo.getProperty());
+                try {
+                    setInvoker.invoke(newEntity, new Object[]{ConvertUtil.convert(idValue, idInfo.getPropertyType())});
+                } catch (Exception e) {
+                    throw new RuntimeException(e.getMessage(), e);
+                }
+            }
+        }
+        return newEntity;
+    }
+
+
+    private static Object getIdValue(Object id, int index) {
+        if (id == null) {
+            return null;
+        }
+        if (ClassUtil.isArray(id.getClass())) {
+            if (index >= Array.getLength(id)) {
+                return null;
+            } else {
+                return Array.get(id, index);
+            }
+        }
+        //not array
+        return index == 0 ? id : null;
+    }
+
 
     public static <T> T ofNotNull(T entity) {
         Class<?> usefulClass = ClassUtil.getUsefulClass(entity.getClass());
 
-        T newEntity = (T) wrap(usefulClass);
+        T newEntity = (T) of(usefulClass);
 
         Reflector reflector = reflectorFactory.findForClass(usefulClass);
         String[] propertyNames = reflector.getGetablePropertyNames();
@@ -48,7 +91,7 @@ public class UpdateEntity {
                     reflector.getSetInvoker(propertyName).invoke(newEntity, new Object[]{value});
                 }
             } catch (Exception e) {
-                //ignore();
+                //ignore;
             }
         }