diff --git a/src/main/resources/vm/java/domain.java.vm b/src/main/resources/vm/java/domain.java.vm
index 86fab07..3253229 100644
--- a/src/main/resources/vm/java/domain.java.vm
+++ b/src/main/resources/vm/java/domain.java.vm
@@ -1,8 +1,8 @@
 package ${packageName}.domain;
 
 
-
-
+import com.baomidou.mybatisplus.annotation.IdType;
+import com.baomidou.mybatisplus.annotation.TableId;
 import lombok.AllArgsConstructor;
 import lombok.Data;
 import lombok.EqualsAndHashCode;
@@ -35,7 +35,7 @@ import lombok.Data;
 @SuperBuilder
 @NoArgsConstructor
 @AllArgsConstructor
-@TableName(value = "" ,autoResultMap = true)
+@TableName(value = "${tableName}" ,autoResultMap = true)
 public class ${ClassName} extends ${Entity}
 {
     private static final long serialVersionUID = 1L;
@@ -59,6 +59,9 @@ public class ${ClassName} extends ${Entity}
     @Excel(name = "${comment}")
 #end
 #end
+    #if($parentheseIndex == 0)
+        @TableId(value = "${column.javaField}",type = IdType.AUTO)
+    #end
     private $column.javaType $column.javaField;
 
 #end