diff --git a/processor/src/main/java/org/mapstruct/ap/model/PropertyMapping.java b/processor/src/main/java/org/mapstruct/ap/model/PropertyMapping.java index cbda3550b..53a8f5d11 100644 --- a/processor/src/main/java/org/mapstruct/ap/model/PropertyMapping.java +++ b/processor/src/main/java/org/mapstruct/ap/model/PropertyMapping.java @@ -253,7 +253,7 @@ public class PropertyMapping extends ModelElement { } Assignment factoryMethod = ctx.getMappingResolver().getFactoryMethod( method, targetType, null, null ); - result = new UpdateWrapper( result, method.getThrownTypes(), targetType, factoryMethod ); + result = new UpdateWrapper( result, method.getThrownTypes(), factoryMethod ); } else { result = new SetterWrapper( result, method.getThrownTypes() ); @@ -579,8 +579,7 @@ public class PropertyMapping extends ModelElement { } Assignment factoryMethod = ctx.getMappingResolver().getFactoryMethod( method, targetType, null, null ); - assignment = - new UpdateWrapper( assignment, method.getThrownTypes(), targetType, factoryMethod ); + assignment = new UpdateWrapper( assignment, method.getThrownTypes(), factoryMethod ); } else { assignment = new SetterWrapper( assignment, method.getThrownTypes() ); diff --git a/processor/src/main/java/org/mapstruct/ap/model/assignment/UpdateWrapper.java b/processor/src/main/java/org/mapstruct/ap/model/assignment/UpdateWrapper.java index a31e78a60..b4ab7b6c5 100644 --- a/processor/src/main/java/org/mapstruct/ap/model/assignment/UpdateWrapper.java +++ b/processor/src/main/java/org/mapstruct/ap/model/assignment/UpdateWrapper.java @@ -19,9 +19,7 @@ package org.mapstruct.ap.model.assignment; import java.util.ArrayList; -import java.util.HashSet; import java.util.List; -import java.util.Set; import org.mapstruct.ap.model.common.Type; /** @@ -32,14 +30,12 @@ import org.mapstruct.ap.model.common.Type; public class UpdateWrapper extends AssignmentWrapper { private final List exceptionTypesToExclude; - private final Type targetType; private final Assignment factoryMethod; - public UpdateWrapper(Assignment decoratedAssignment, List exceptionTypesToExclude, Type targetType, - Assignment factoryMethod ) { + public UpdateWrapper(Assignment decoratedAssignment, List exceptionTypesToExclude, + Assignment factoryMethod ) { super( decoratedAssignment ); this.exceptionTypesToExclude = exceptionTypesToExclude; - this.targetType = targetType; this.factoryMethod = factoryMethod; } @@ -57,14 +53,6 @@ public class UpdateWrapper extends AssignmentWrapper { return result; } - @Override - public Set getImportTypes() { - Set imported = new HashSet(); - imported.addAll( super.getImportTypes() ); - imported.add( targetType ); /* is a new target type */ - return imported; - } - public Assignment getFactoryMethod() { return factoryMethod; }