simple entity return

build.gradle

@@ -13,7 +13,7 @@ dependencyManagement {
     }
 }
 group = 'com.nucleodb'
-version = '3.5.2'
+version = '3.5.3'
 
 repositories {
     mavenCentral()

src/main/java/com/nucleodb/spring/NDBRepositoryFactory.java

@@ -3,14 +3,18 @@
 import com.fasterxml.jackson.core.JsonProcessingException;
 import com.fasterxml.jackson.databind.ObjectMapper;
 import com.nucleodb.library.NucleoDB;
+import com.nucleodb.library.database.tables.table.DataEntry;
 import com.nucleodb.library.database.utils.Serializer;
 import com.nucleodb.spring.impl.NDBConnectionRepositoryImpl;
 import com.nucleodb.spring.impl.NDBDataEntryRepositoryImpl;
+import com.nucleodb.spring.mapping.NDBEntityInformation;
+import com.nucleodb.spring.query.MappingNDBEntityInformation;
 import com.nucleodb.spring.query.QueryParser;
 import com.nucleodb.spring.query.exec.NDBDataEntryRepositoryQuery;
 import com.nucleodb.spring.types.NDBConnRepository;
 import com.nucleodb.spring.types.NDBDataRepository;
 import org.springframework.context.ApplicationEventPublisher;
+import org.springframework.data.mapping.context.MappingContext;
 import org.springframework.data.projection.ProjectionFactory;
 import org.springframework.data.repository.core.EntityInformation;
 import org.springframework.data.repository.core.NamedQueries;
@@ -54,9 +58,8 @@ public NDBRepositoryFactory(NucleoDB nucleoDB, ApplicationEventPublisher publish
   }
 
   @Override
-  public <T, ID> EntityInformation<T, ID> getEntityInformation(Class<T> domainClass) {
-    //return new MappingNDBEntityInformation();
-    return null;
+  public <T, ID> EntityInformation<T, ID> getEntityInformation(Class<T> entityClass) {
+    return new NDBEntityInformation<>(entityClass, (Class<ID>) String.class);
   }
 
   @Override

src/main/java/com/nucleodb/spring/mapping/NDBEntityInformation.java

@@ -0,0 +1,47 @@
+package com.nucleodb.spring.mapping;
+
+import org.springframework.data.repository.core.EntityInformation;
+
+import java.lang.reflect.Field;
+
+public class NDBEntityInformation<T, ID> implements EntityInformation<T, ID> {
+
+    private final Class<T> entityClass;
+    private final Class<ID> idClass;
+
+    public NDBEntityInformation(Class<T> entityClass, Class<ID> idClass) {
+        this.entityClass = entityClass;
+        this.idClass = idClass;
+    }
+
+    @Override
+    public boolean isNew(T entity) {
+        // Assuming that if the ID is null, the entity is new
+        return getId(entity) == null;
+    }
+
+    @Override
+    public ID getId(T entity) {
+        try {
+            Field idField = entityClass.getDeclaredField("id");
+            if(idField == null) {
+                idField = entityClass.getDeclaredField("uuid");
+            }
+            idField.setAccessible(true);
+            return (ID) idField.get(entity);
+        } catch (NoSuchFieldException | IllegalAccessException e) {
+            throw new IllegalStateException("Could not retrieve id field", e);
+        }
+    }
+
+
+    @Override
+    public Class<ID> getIdType() {
+        return idClass;
+    }
+
+    @Override
+    public Class<T> getJavaType() {
+        return entityClass;
+    }
+}