增加@keyproperty，暂时先不提供使用。

abel533 · abel533 · commit 63af87f90d60 · 2015-08-26T23:47:47.000+08:00
diff --git a/src/main/java/tk/mybatis/mapper/annotation/KeyProperty.java b/src/main/java/tk/mybatis/mapper/annotation/KeyProperty.java
@@ -0,0 +1,16 @@
+package tk.mybatis.mapper.annotation;
+
+import java.lang.annotation.ElementType;
+import java.lang.annotation.Retention;
+import java.lang.annotation.RetentionPolicy;
+import java.lang.annotation.Target;
+
+/**
+ * keyProperty注解，标记的字段会使用JDBC方式回写自增结果
+ *
+ * @author liuzh
+ */
+@Target(ElementType.FIELD)
+@Retention(RetentionPolicy.RUNTIME)
+public @interface KeyProperty {
+}
diff --git a/src/main/java/tk/mybatis/mapper/entity/EntityColumn.java b/src/main/java/tk/mybatis/mapper/entity/EntityColumn.java
@@ -14,6 +14,7 @@ public class EntityColumn {
     private boolean id = false;
     private boolean uuid = false;
     private boolean identity = false;
+    private boolean key = false;
     private String generator;
     private String orderBy;
 
@@ -104,6 +105,14 @@ public void setOrderBy(String orderBy) {
         this.orderBy = orderBy;
     }
 
+    public boolean isKey() {
+        return key;
+    }
+
+    public void setKey(boolean key) {
+        this.key = key;
+    }
+
     @Override
     public boolean equals(Object o) {
         if (this == o) return true;
diff --git a/src/main/java/tk/mybatis/mapper/mapperhelper/EntityHelper.java b/src/main/java/tk/mybatis/mapper/mapperhelper/EntityHelper.java
@@ -24,6 +24,7 @@
 
 package tk.mybatis.mapper.mapperhelper;
 
+import tk.mybatis.mapper.annotation.KeyProperty;
 import tk.mybatis.mapper.annotation.NameStyle;
 import tk.mybatis.mapper.code.IdentityDialect;
 import tk.mybatis.mapper.code.Style;
@@ -209,10 +210,12 @@ public static synchronized void initEntityNameMap(Class<?> entityClass, Style st
             if (field.isAnnotationPresent(Transient.class)) {
                 continue;
             }
+            //Id
             EntityColumn entityColumn = new EntityColumn(entityTable);
             if (field.isAnnotationPresent(Id.class)) {
                 entityColumn.setId(true);
             }
+            //Column
             String columnName = null;
             if (field.isAnnotationPresent(Column.class)) {
                 Column column = field.getAnnotation(Column.class);
@@ -224,7 +227,7 @@ public static synchronized void initEntityNameMap(Class<?> entityClass, Style st
             entityColumn.setProperty(field.getName());
             entityColumn.setColumn(columnName);
             entityColumn.setJavaType(field.getType());
-            //order by
+            //OrderBy
             if (field.isAnnotationPresent(OrderBy.class)) {
                 OrderBy orderBy = field.getAnnotation(OrderBy.class);
                 if (orderBy.value().equals("")) {
@@ -233,6 +236,10 @@ public static synchronized void initEntityNameMap(Class<?> entityClass, Style st
                     entityColumn.setOrderBy(orderBy.value());
                 }
             }
+            //KeyProperty
+            if (field.isAnnotationPresent(KeyProperty.class)) {
+                entityColumn.setKey(true);
+            }
             //主键策略 - Oracle序列，MySql自动增长，UUID
             if (field.isAnnotationPresent(SequenceGenerator.class)) {
                 SequenceGenerator sequenceGenerator = field.getAnnotation(SequenceGenerator.class);
