Skip to content

Commit

Permalink
Add ObjectFactory generator
Browse files Browse the repository at this point in the history 
SUpport constructor class mapping

git-svn-id: https://orika.googlecode.com/svn/trunk@238 f4a4e40d-1d15-de9b-c3e8-c5ea35f5a4c4
  • Loading branch information
@elaatifi
elaatifi committed Nov 10, 2011
1 parent 0849540 commit f5b4096
Show file tree
Hide file tree
Showing 12 changed files with 694 additions and 155 deletions.
5 changes: 5 additions & 0 deletions core/pom.xml
View file
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,11 @@
<artifactId>javassist</artifactId>
</dependency>

<dependency>
<groupId>com.thoughtworks.paranamer</groupId>
<artifactId>paranamer</artifactId>
</dependency>

<dependency>
<groupId>junit</groupId>
<artifactId>junit</artifactId>
Expand Down
5 changes: 5 additions & 0 deletions core/src/main/java/ma/glasnost/orika/MapperFactory.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -56,4 +56,9 @@ public interface MapperFactory {
void build();

Set<ClassMap<Object, Object>> lookupUsedClassMap(MapperKey mapperKey);

<A, B> ClassMap<A, B> getClassMap(MapperKey mapperKey);

Set<Class<Object>> lookupMappedClasses(Class<Object> clazz);

}
10 changes: 10 additions & 0 deletions core/src/main/java/ma/glasnost/orika/constructor/PickingConstructorStrategy.java
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
package ma.glasnost.orika.constructor;

import java.lang.reflect.Constructor;

import ma.glasnost.orika.metadata.ClassMap;

public interface PickingConstructorStrategy {

<T, A, B> Constructor<T> pick(ClassMap<A, B> classMap, Class<T> sourceClass);
}
20 changes: 20 additions & 0 deletions core/src/main/java/ma/glasnost/orika/constructor/SimplePickingConstructorStrategy.java
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,20 @@
package ma.glasnost.orika.constructor;

import java.lang.reflect.Constructor;

import ma.glasnost.orika.metadata.ClassMap;

public class SimplePickingConstructorStrategy implements PickingConstructorStrategy {

@SuppressWarnings({ "unchecked" })
public <T, A, B> Constructor<T> pick(ClassMap<A, B> classMap, Class<T> sourceClass) {
boolean aToB = classMap.getBType().equals(sourceClass);
// String[] argumentNames = aToB ? classMap.getConstructorB() :
// classMap.getConstructorA();
Class<?> targetClass = aToB ? classMap.getBType() : classMap.getAType();

// TODO to specify
return (Constructor<T>) targetClass.getConstructors()[0];

}
}
182 changes: 132 additions & 50 deletions core/src/main/java/ma/glasnost/orika/impl/CodeSourceBuilder.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -32,11 +32,11 @@ public class CodeSourceBuilder {
private int currentIndent = 1;

public CodeSourceBuilder(int indent) {
this.currentIndent = indent;
this.currentIndent = indent;
}

public CodeSourceBuilder assertType(String var, Class<?> clazz) {
newLine();
newLine();
append("if(!(" + var + " instanceof ").append(clazz.getName()).append(")) ");
begin();
append("throw new IllegalStateException(\"[" + var + "] is not an instance of " + clazz.getName() + " \");");
Expand All @@ -46,27 +46,25 @@ public CodeSourceBuilder assertType(String var, Class<?> clazz) {

public CodeSourceBuilder convert(Property destination, Property source) {

final String getter = getGetter(source);
final String getter = getGetter(source);
final String setter = getSetter(destination);
final Class<?> destinationClass = destination.getType();
return newLine()
.append("destination.%s((%s)mapperFacade.convert(source.%s, %s.class));", setter, destinationClass.getName(), getter,
destinationClass.getName())
.newLine();
return newLine().append("destination.%s((%s)mapperFacade.convert(source.%s, %s.class));", setter, destinationClass.getName(),
getter, destinationClass.getName()).newLine();
}

public CodeSourceBuilder set(Property d, Property s) {

final String getter = getGetter(s);
final String getter = getGetter(s);
final String setter = getSetter(d);
return newLine().append("destination.%s(source.%s);", setter, getter);
}

public CodeSourceBuilder setCollection(Property dp, Property sp, Property ip, Class<?> dc) {
final Class<?> destinationElementClass = dp.getParameterizedType();

if (destinationElementClass==null) {
throw new MappingException("cannot determine runtime type of destination collection " + dc.getName() + "." + dp.getName());
if (destinationElementClass == null) {
throw new MappingException("cannot determine runtime type of destination collection " + dc.getName() + "." + dp.getName());
}

String destinationCollection = "List";
Expand All @@ -82,30 +80,30 @@ public CodeSourceBuilder setCollection(Property dp, Property sp, Property ip, Cl
final String sourceGetter = getGetter(sp);
final String destinationGetter = getGetter(dp);
final String destinationSetter = getSetter(dp);
boolean destinationHasSetter = false;
boolean destinationHasSetter = false;
try {
destinationHasSetter = (dc.getMethod(destinationSetter, dp.getType())!=null);
} catch (Exception e) {
/* ignored: no destination setter available */
}
destinationHasSetter = (dc.getMethod(destinationSetter, dp.getType()) != null);
} catch (Exception e) {
/* ignored: no destination setter available */
}

if (destinationHasSetter) {
newLine()
.append("if (destination.%s == null) ", destinationGetter)
.begin()
.append("destination.%s(%s);", destinationSetter, newStatement)
.end();
}
if (destinationHasSetter) {
newLine().append("if (destination.%s == null) ", destinationGetter)
.begin()
.append("destination.%s(%s);", destinationSetter, newStatement)
.end();
}

newLine().append("destination.%s.clear();", destinationGetter);
newLine().append("destination.%s.addAll(mapperFacade.mapAs%s(source.%s, %s.class, mappingContext));", destinationGetter,
destinationCollection, sourceGetter, destinationElementClass.getName());
if (ip != null) {
if (ip.isCollection()) {
newLine().append("for (java.util.Iterator orikaIterator = destination.%s.iterator(); orikaIterator.hasNext();) ", getGetter(dp));
begin().append("%s orikaCollectionItem = (%s) orikaIterator.next();", dp.getParameterizedType().getName(), dp
.getParameterizedType().getName());
newLine().append("for (java.util.Iterator orikaIterator = destination.%s.iterator(); orikaIterator.hasNext();) ",
getGetter(dp));
begin().append("%s orikaCollectionItem = (%s) orikaIterator.next();", dp.getParameterizedType().getName(),
dp.getParameterizedType().getName());
newLine().append("if (orikaCollectionItem.%s == null) ", getGetter(ip));
begin();
if (ip.isSet()) {
Expand All @@ -123,9 +121,10 @@ public CodeSourceBuilder setCollection(Property dp, Property sp, Property ip, Cl
} else if (ip.isArray()) {
// TODO To implement
} else {
newLine().append("for (java.util.Iterator orikaIterator = destination.%s.iterator(); orikaIterator.hasNext();)", getGetter(dp));
begin().append("%s orikaCollectionItem = (%s) orikaIterator.next();", dp.getParameterizedType().getName(), dp
.getParameterizedType().getName());
newLine().append("for (java.util.Iterator orikaIterator = destination.%s.iterator(); orikaIterator.hasNext();)",
getGetter(dp));
begin().append("%s orikaCollectionItem = (%s) orikaIterator.next();", dp.getParameterizedType().getName(),
dp.getParameterizedType().getName());
newLine().append("orikaCollectionItem.%s(destination);", getSetter(ip));
end();
}
Expand All @@ -135,11 +134,11 @@ public CodeSourceBuilder setCollection(Property dp, Property sp, Property ip, Cl
}

public CodeSourceBuilder newLine() {
out.append("\n");
for(int i = 0; i < currentIndent; ++i) {
out.append("\t");
}
return this;
out.append("\n");
for (int i = 0; i < currentIndent; ++i) {
out.append("\t");
}
return this;
}

public CodeSourceBuilder append(String str, Object... args) {
Expand All @@ -159,7 +158,7 @@ public CodeSourceBuilder then() {
}

public CodeSourceBuilder begin() {
return then();
return then();
}

public CodeSourceBuilder end() {
Expand All @@ -186,7 +185,8 @@ public CodeSourceBuilder setWrapper(Property dp, Property sp) {
final String getter = getGetter(sp);
final String setter = getSetter(dp);

newLine().append("destination.%s(%s.valueOf((%s) source.%s));", setter, dp.getType().getName(), getPrimitiveType(dp.getType()), getter);
newLine().append("destination.%s(%s.valueOf((%s) source.%s));", setter, dp.getType().getName(), getPrimitiveType(dp.getType()),
getter);
return this;
}

Expand All @@ -205,31 +205,31 @@ public CodeSourceBuilder setArray(Property dp, Property sp) {
final String getter = getGetter(sp);
final String setter = getSetter(dp);

newLine().append("%s[] %s = new %s[source.%s.%s];", paramType, dp.getName(), paramType, getter, getSizeCode).append(
"mapperFacade.mapAsArray((Object[])%s, (%s)source.%s, %s.class, mappingContext);", dp.getName(), castSource, getter,
paramType).append("destination.%s(%s);", setter, dp.getName());
newLine().append("%s[] %s = new %s[source.%s.%s];", paramType, dp.getName(), paramType, getter, getSizeCode)
.append("mapperFacade.mapAsArray((Object[])%s, (%s)source.%s, %s.class, mappingContext);", dp.getName(), castSource,
getter, paramType)
.append("destination.%s(%s);", setter, dp.getName());

return this;
}

public CodeSourceBuilder setToEnumeration(Property dp, Property sp) {
final String getter = getGetter(sp);
final String getter = getGetter(sp);
final String setter = getSetter(dp);

newLine().append("destination.%s((%s)Enum.valueOf(%s.class,\"\"+source.%s));",
setter, dp.getType().getName(), dp.getType().getName(), getter);
newLine().append("destination.%s((%s)Enum.valueOf(%s.class,\"\"+source.%s));", setter, dp.getType().getName(),
dp.getType().getName(), getter);
return this;
}


public CodeSourceBuilder setObject(Property dp, Property sp, Property ip) {
final String sourceGetter = getGetter(sp);

final String destinationGetter = getGetter(dp);
final String destinationSetter = getSetter(dp);
newLine().append("if (destination.%s == null) ", destinationGetter);
begin().append("destination.%s((%s)mapperFacade.map(source.%s, %s.class, mappingContext));", destinationSetter, dp.getType().getName(),
sourceGetter, dp.getType().getName());
begin().append("destination.%s((%s)mapperFacade.map(source.%s, %s.class, mappingContext));", destinationSetter,
dp.getType().getName(), sourceGetter, dp.getType().getName());
elze();
append("mapperFacade.map(source.%s, destination.%s, mappingContext);", sourceGetter, destinationGetter);
end();
Expand All @@ -256,11 +256,9 @@ public CodeSourceBuilder setObject(Property dp, Property sp, Property ip) {
return this;
}



public CodeSourceBuilder ifSourceNotNull(Property sp) {

newLine();
newLine();
if (sp.hasPath()) {
final StringBuilder sb = new StringBuilder("source");
int i = 0;
Expand Down Expand Up @@ -293,8 +291,8 @@ public CodeSourceBuilder ifSourceNotNull(Property sp) {
* @return CodeSourceBuilder
*/
public CodeSourceBuilder ifDestinationNull(Property property) {
if (!property.hasPath()) {
if (!property.hasPath()) {
return this;
}

Expand All @@ -316,6 +314,11 @@ public CodeSourceBuilder ifDestinationNull(Property property) {
return this;
}

public CodeSourceBuilder ifSourceInstanceOf(Class<Object> sourceClass) {
append("if(s instanceof %s)", sourceClass.getName());
return this;
}

private String getLongGetter(NestedProperty property) {
final StringBuilder sb = new StringBuilder();
for (final Property p : property.getPath()) {
Expand Down Expand Up @@ -352,4 +355,83 @@ private String getPrimitiveType(Class<?> clazz) {
}
return type;
}

public CodeSourceBuilder declareVar(Class<?> clazz, String var) {
append("\n%s %s = %s;", clazz.getName(), var, clazz.isPrimitive() ? getDefaultPrimtiveValue(clazz) : "null");
return this;
}

public CodeSourceBuilder assignImmutableVar(String var, Property sp) {
append("%s = source.%s;", var, getGetter(sp));
return this;
}

public CodeSourceBuilder assignObjectVar(String var, Property sp, Class<?> targetClass) {
append("%s = (%s) mapperFacade.map(source.%s, %s.class);", var, targetClass.getName(), getGetter(sp), targetClass.getName());
return this;
}

public CodeSourceBuilder assignCollectionVar(String var, Property sp, Property dp) {
final Class<?> destinationElementClass = dp.getParameterizedType();
String destinationCollection = "List";
String newStatement = "new java.util.ArrayList()";
if (List.class.isAssignableFrom(dp.getType())) {
destinationCollection = "List";
newStatement = "new java.util.ArrayList()";
} else if (Set.class.isAssignableFrom(dp.getType())) {
destinationCollection = "Set";
newStatement = "new java.util.HashSet()";
}

final String sourceGetter = getGetter(sp);
final String destinationGetter = getGetter(dp);
final String destinationSetter = getSetter(dp);

append("if (destination.%s == null) {\n", destinationGetter);
append("destination.%s(%s);\n", destinationSetter, newStatement);
append("}\n");
append("destination.%s.clear();\n", destinationGetter);
append("destination.%s.addAll(mapperFacade.mapAs%s(source.%s, %s.class, mappingContext));", destinationGetter,
destinationCollection, sourceGetter, destinationElementClass.getName());
return this;
}

public CodeSourceBuilder assignArrayVar(String var, Property sp, Class<?> targetClass) {
String getter = getGetter(sp);
final String getSizeCode = sp.getType().isArray() ? "length" : "size()";
final String castSource = sp.getType().isArray() ? "Object[]" : "";
append("%s[] %s = new %s[source.%s.%s];", targetClass, var, targetClass.getName(), getter, getSizeCode).append(
"mapperFacade.mapAsArray((Object[])%s, (%s)source.%s, %s.class, mappingContext);", var, castSource, getter,
targetClass.getName());
return this;
}

public CodeSourceBuilder assignPrimtiveToWrapperVar(String var, Property sp, Class<?> targetClass) {
final String getter = getGetter(sp);

append("%s = %s.valueOf((%s) source.%s));\n", var, targetClass.getName(), getPrimitiveType(targetClass), getter);
return this;
}

public CodeSourceBuilder assignWrapperToPrimitiveVar(String var, Property sp, Class<?> targetClass) {
String getter = getGetter(sp);
append("%s = source.%s.%sValue();\n", var, getter, getPrimitiveType(targetClass));
return this;
}

public CodeSourceBuilder assignConvertedVar(String var, Property source, Class<?> targetClass) {
final String getter = getGetter(source);
append("%s = ((%s)mapperFacade.convert(source.%s, %s.class)); \n", var, targetClass.getName(), getter, targetClass.getName());
return this;

}

private String getDefaultPrimtiveValue(Class<?> clazz) {
if (Boolean.TYPE.equals(clazz))
return "false";
else if (Character.TYPE.equals(clazz))
return "'\0'";
else
return "0";
}
}
Loading

0 comments on commit f5b4096

Please sign in to comment.