Expose registerDefaultTransformer in CrudHelper

antelopesystems · Aug 16, 2021 · 0b4169e · 0b4169e
1 parent 79f07d0
commit 0b4169e
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 8 deletions.
diff --git a/...ramework-core/src/main/java/com/antelopesystem/crudframework/crud/handler/CrudHelper.java b/...ramework-core/src/main/java/com/antelopesystem/crudframework/crud/handler/CrudHelper.java
@@ -6,6 +6,7 @@
 import com.antelopesystem.crudframework.crud.decorator.ObjectDecorator;
 import com.antelopesystem.crudframework.crud.hooks.interfaces.CRUDHooks;
 import com.antelopesystem.crudframework.crud.model.EntityMetadataDTO;
+import com.antelopesystem.crudframework.fieldmapper.transformer.base.FieldTransformer;
 import com.antelopesystem.crudframework.model.BaseCrudEntity;
 import com.antelopesystem.crudframework.modelfilter.DynamicModelFilter;
 import com.antelopesystem.crudframework.modelfilter.FilterField;
@@ -63,4 +64,8 @@ public interface CrudHelper {
 	<Entity> Long getTotalFromPagingCache(Class<Entity> entityClazz, DynamicModelFilter filter);
 
 	<ID extends Serializable, Entity extends BaseCrudEntity<ID>> CrudDao getCrudDaoForEntity(Class<Entity> entityClazz);
+
+	void registerDefaultTransformer(FieldTransformer transformer);
+
+	void registerDefaultTransformer(FieldTransformer transformer, Class<?> fromType, Class<?> toType);
 }
diff --git a/...work-core/src/main/java/com/antelopesystem/crudframework/crud/handler/CrudHelperImpl.java b/...work-core/src/main/java/com/antelopesystem/crudframework/crud/handler/CrudHelperImpl.java
@@ -451,6 +451,16 @@ public <ID extends Serializable, Entity extends BaseCrudEntity<ID>> CrudDao getC
 		});
 	}
 
+	@Override
+	public void registerDefaultTransformer(FieldTransformer transformer) {
+		fieldMapper.registerDefaultTransformer(transformer);
+	}
+
+	@Override
+	public void registerDefaultTransformer(FieldTransformer transformer, Class<?> fromType, Class<?> toType) {
+		fieldMapper.registerDefaultTransformer(transformer, fromType, toType);
+	}
+
 	private void initFieldMapper() {
 		for(Map.Entry<String, FieldTransformer> entry : fieldTransformers.entrySet()) {
 			fieldMapper.registerTransformer(entry.getKey(), entry.getValue());

diff --git a/...k-fieldmapper/src/main/java/com/antelopesystem/crudframework/fieldmapper/FieldMapper.java b/...k-fieldmapper/src/main/java/com/antelopesystem/crudframework/fieldmapper/FieldMapper.java
@@ -53,15 +53,13 @@ public void registerDefaultTransformer(FieldTransformer transformer) {
 	}
 
 	public void registerDefaultTransformer(FieldTransformer transformer, Class<?> fromType, Class<?> toType) {
-		if(transformer.isDefault()) {
-			Pair key = new Pair<>(fromType, toType);
-			FieldTransformer existing = defaultTransformers.get(key);
-			if(existing != null) {
-				throw new IllegalStateException("Cannot register default transformer for pair [ " + key + " ] - already registered by [ " + existing.getClass().getName() + " ]");
-			}
-
-			defaultTransformers.put(key, transformer);
+		Pair key = new Pair<>(fromType, toType);
+		FieldTransformer existing = defaultTransformers.get(key);
+		if(existing != null) {
+			throw new IllegalStateException("Cannot register default transformer for pair [ " + key + " ] - already registered by [ " + existing.getClass().getName() + " ]");
 		}
+
+		defaultTransformers.put(key, transformer);
 	}
 
 	public <T> T processMappedFields(Object object, Class<T> toClazz) {