fix schema

Author: xlorne

springboot-starter-persistence/src/main/java/com/codingapi/springboot/persistence/property/SchemaProperty.java

@@ -26,9 +26,9 @@ public SchemaProperty(Class<?> domainClazz) {
 
 
     public List<BeanProperty> getProperties(boolean hasId) {
-        if(hasId){
+        if (hasId) {
             return properties;
-        }else{
+        } else {
             return properties.stream()
                     .filter(beanProperty -> !beanProperty.getName().equals("id"))
                     .collect(Collectors.toList());
@@ -46,4 +46,13 @@ public boolean hasIdValue(Object domain) {
     public void setIdValue(Object domain, Number key) {
         idBeanProperty.setIdValue(domain, key);
     }
+
+    public boolean hasIdProperty() {
+        for (BeanProperty property : properties) {
+            if (property.getName().equals("id")) {
+                return true;
+            }
+        }
+        return false;
+    }
 }

springboot-starter-persistence/src/main/java/com/codingapi/springboot/persistence/register/DomainClassRegister.java

@@ -16,8 +16,13 @@ private DomainClassRegister() {
         this.classes = new ArrayList<>();
     }
 
-    public void register(Class<?> domainClass) {
+    public DomainClassRegister register(Class<?> domainClass) {
         this.classes.add(domainClass);
+        return this;
+    }
+
+    public void register(Class<?>... domainClasses) {
+        this.classes.addAll(List.of(domainClasses));
     }
 
     public boolean supports(Class<?> domainClass) {

springboot-starter-persistence/src/main/java/com/codingapi/springboot/persistence/schema/Schema.java

@@ -17,8 +17,14 @@ public Schema(Class<?> domainClass) {
         log.info("Schema init:{}", domainClass);
         this.domainClass = domainClass;
         this.schemaProperty = new SchemaProperty(domainClass);
+        this.check();
     }
 
+    public void check() {
+        if (!this.schemaProperty.hasIdProperty()) {
+            throw new RuntimeException("schema id property not found");
+        }
+    }
 
     public abstract BuildSchema buildSchema();
 

springboot-starter-persistence/src/test/java/com/example/demo/register/DomainRegister.java

@@ -1,7 +1,7 @@
 package com.example.demo.register;
 
-import com.example.demo.domain.Demo;
 import com.codingapi.springboot.persistence.register.DomainClassRegister;
+import com.example.demo.domain.Demo;
 import org.springframework.beans.factory.InitializingBean;
 import org.springframework.stereotype.Component;