diff --git a/src/main/java/im/zhaojun/common/model/StorageConfig.java b/src/main/java/im/zhaojun/common/model/StorageConfig.java
index 39585a9..1b3e46f 100644
--- a/src/main/java/im/zhaojun/common/model/StorageConfig.java
+++ b/src/main/java/im/zhaojun/common/model/StorageConfig.java
@@ -1,9 +1,19 @@
 package im.zhaojun.common.model;
 
 import im.zhaojun.common.enums.StorageTypeEnum;
+import lombok.Data;
 
+import javax.persistence.Entity;
+import javax.persistence.GeneratedValue;
+import javax.persistence.GenerationType;
+import javax.persistence.Id;
+
+@Entity
+@Data
 public class StorageConfig {
 
+    @Id
+    @GeneratedValue(strategy = GenerationType.IDENTITY)
     private Integer id;
 
     private StorageTypeEnum type;
diff --git a/src/main/java/im/zhaojun/common/model/SystemConfig.java b/src/main/java/im/zhaojun/common/model/SystemConfig.java
index fa6f4da..f12cd56 100644
--- a/src/main/java/im/zhaojun/common/model/SystemConfig.java
+++ b/src/main/java/im/zhaojun/common/model/SystemConfig.java
@@ -3,10 +3,20 @@ package im.zhaojun.common.model;
 import com.fasterxml.jackson.annotation.JsonIgnore;
 import im.zhaojun.common.enums.StorageTypeEnum;
 import im.zhaojun.common.enums.ViewModeEnum;
+import lombok.Data;
 
+import javax.persistence.Entity;
+import javax.persistence.GeneratedValue;
+import javax.persistence.GenerationType;
+import javax.persistence.Id;
+
+@Entity
+@Data
 public class SystemConfig {
 
     @JsonIgnore
+    @Id
+    @GeneratedValue(strategy = GenerationType.IDENTITY)
     private Integer id;
 
     private String siteName;