mirror of
https://github.com/mapstruct/mapstruct.git
synced 2025-07-12 00:00:08 +08:00
#209, #412 Adding support for mapping static inner classes and enums by adding import declarations for static inner types also if declared in the same package
This commit is contained in:
parent
f778d6c5ad
commit
2ebfd04fe9
@ -40,6 +40,8 @@ import org.mapstruct.ap.version.VersionInformation;
|
|||||||
*/
|
*/
|
||||||
public abstract class GeneratedType extends ModelElement {
|
public abstract class GeneratedType extends ModelElement {
|
||||||
|
|
||||||
|
private static final String JAVA_LANG_PACKAGE = "java.lang";
|
||||||
|
|
||||||
private final String packageName;
|
private final String packageName;
|
||||||
private final String name;
|
private final String name;
|
||||||
private final String superClassName;
|
private final String superClassName;
|
||||||
@ -170,11 +172,7 @@ public abstract class GeneratedType extends ModelElement {
|
|||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
if ( typeToAdd.isImported() &&
|
if ( needsImportDeclaration( typeToAdd ) ) {
|
||||||
typeToAdd.getPackageName() != null &&
|
|
||||||
!typeToAdd.getPackageName().equals( packageName ) &&
|
|
||||||
!typeToAdd.getPackageName().startsWith( "java.lang" ) ) {
|
|
||||||
|
|
||||||
if ( typeToAdd.isArrayType() ) {
|
if ( typeToAdd.isArrayType() ) {
|
||||||
collection.add( typeToAdd.getComponentType() );
|
collection.add( typeToAdd.getComponentType() );
|
||||||
}
|
}
|
||||||
@ -187,4 +185,26 @@ public abstract class GeneratedType extends ModelElement {
|
|||||||
addWithDependents( collection, type );
|
addWithDependents( collection, type );
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private boolean needsImportDeclaration(Type typeToAdd) {
|
||||||
|
if ( !typeToAdd.isImported() ) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
if ( typeToAdd.getPackageName() == null ) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
if ( typeToAdd.getPackageName().startsWith( JAVA_LANG_PACKAGE ) ) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
if ( typeToAdd.getPackageName().equals( packageName ) ) {
|
||||||
|
if ( !typeToAdd.getTypeElement().getNestingKind().isNested() ) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
@ -0,0 +1,92 @@
|
|||||||
|
/**
|
||||||
|
* Copyright 2012-2015 Gunnar Morling (http://www.gunnarmorling.de/)
|
||||||
|
* and/or other contributors as indicated by the @authors tag. See the
|
||||||
|
* copyright.txt file in the distribution for a full listing of all
|
||||||
|
* contributors.
|
||||||
|
*
|
||||||
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
* you may not use this file except in compliance with the License.
|
||||||
|
* You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing, software
|
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
* See the License for the specific language governing permissions and
|
||||||
|
* limitations under the License.
|
||||||
|
*/
|
||||||
|
package org.mapstruct.ap.test.imports;
|
||||||
|
|
||||||
|
import static org.fest.assertions.Assertions.assertThat;
|
||||||
|
|
||||||
|
import org.junit.Rule;
|
||||||
|
import org.junit.Test;
|
||||||
|
import org.junit.runner.RunWith;
|
||||||
|
import org.mapstruct.ap.test.imports.innerclasses.BeanFacade;
|
||||||
|
import org.mapstruct.ap.test.imports.innerclasses.BeanWithInnerEnum;
|
||||||
|
import org.mapstruct.ap.test.imports.innerclasses.BeanWithInnerEnum.InnerEnum;
|
||||||
|
import org.mapstruct.ap.test.imports.innerclasses.BeanWithInnerEnumMapper;
|
||||||
|
import org.mapstruct.ap.test.imports.innerclasses.InnerClassMapper;
|
||||||
|
import org.mapstruct.ap.test.imports.innerclasses.SourceWithInnerClass;
|
||||||
|
import org.mapstruct.ap.test.imports.innerclasses.SourceWithInnerClass.SourceInnerClass;
|
||||||
|
import org.mapstruct.ap.test.imports.innerclasses.TargetWithInnerClass;
|
||||||
|
import org.mapstruct.ap.test.imports.innerclasses.TargetWithInnerClass.TargetInnerClass;
|
||||||
|
import org.mapstruct.ap.testutil.IssueKey;
|
||||||
|
import org.mapstruct.ap.testutil.WithClasses;
|
||||||
|
import org.mapstruct.ap.testutil.runner.AnnotationProcessorTestRunner;
|
||||||
|
import org.mapstruct.ap.testutil.runner.GeneratedSource;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Test for generating a mapper which references nested types (static inner classes).
|
||||||
|
*
|
||||||
|
* @author Ewald Volkert
|
||||||
|
*/
|
||||||
|
@WithClasses({
|
||||||
|
SourceWithInnerClass.class, TargetWithInnerClass.class, InnerClassMapper.class, //
|
||||||
|
BeanFacade.class, BeanWithInnerEnum.class, BeanWithInnerEnumMapper.class
|
||||||
|
})
|
||||||
|
@RunWith(AnnotationProcessorTestRunner.class)
|
||||||
|
public class InnerClassesImportsTest {
|
||||||
|
|
||||||
|
private final GeneratedSource generatedSource = new GeneratedSource();
|
||||||
|
|
||||||
|
@Rule
|
||||||
|
public GeneratedSource getGeneratedSource() {
|
||||||
|
return generatedSource;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
@IssueKey( "412" )
|
||||||
|
public void mapperRequiresInnerClassImports() {
|
||||||
|
SourceWithInnerClass source = new SourceWithInnerClass();
|
||||||
|
source.setInnerClassMember( new SourceInnerClass( 412 ) );
|
||||||
|
|
||||||
|
TargetWithInnerClass target = InnerClassMapper.INSTANCE.sourceToTarget( source );
|
||||||
|
|
||||||
|
assertThat( target ).isNotNull();
|
||||||
|
assertThat( target.getInnerClassMember().getValue() ).isEqualTo( 412 );
|
||||||
|
generatedSource.forMapper( InnerClassMapper.class ).containsImportFor( SourceInnerClass.class );
|
||||||
|
generatedSource.forMapper( InnerClassMapper.class ).containsImportFor( TargetInnerClass.class );
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
@IssueKey( "209" )
|
||||||
|
public void mapperRequiresInnerEnumImports() {
|
||||||
|
BeanWithInnerEnum source = new BeanWithInnerEnum();
|
||||||
|
source.setTest( "whatever" );
|
||||||
|
source.setInnerEnum( InnerEnum.A );
|
||||||
|
|
||||||
|
BeanFacade target = BeanWithInnerEnumMapper.INSTANCE.toFacade( source );
|
||||||
|
|
||||||
|
assertThat( target ).isNotNull();
|
||||||
|
assertThat( target.getInnerEnum() ).isEqualTo( "A" );
|
||||||
|
|
||||||
|
BeanWithInnerEnum sourceAgain = BeanWithInnerEnumMapper.INSTANCE.fromFacade( target );
|
||||||
|
|
||||||
|
assertThat( sourceAgain ).isNotNull();
|
||||||
|
assertThat( sourceAgain.getInnerEnum() ).isEqualTo( InnerEnum.A );
|
||||||
|
|
||||||
|
generatedSource.forMapper( BeanWithInnerEnumMapper.class ).containsImportFor( InnerEnum.class );
|
||||||
|
}
|
||||||
|
}
|
@ -0,0 +1,41 @@
|
|||||||
|
/**
|
||||||
|
* Copyright 2012-2015 Gunnar Morling (http://www.gunnarmorling.de/)
|
||||||
|
* and/or other contributors as indicated by the @authors tag. See the
|
||||||
|
* copyright.txt file in the distribution for a full listing of all
|
||||||
|
* contributors.
|
||||||
|
*
|
||||||
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
* you may not use this file except in compliance with the License.
|
||||||
|
* You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing, software
|
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
* See the License for the specific language governing permissions and
|
||||||
|
* limitations under the License.
|
||||||
|
*/
|
||||||
|
package org.mapstruct.ap.test.imports.innerclasses;
|
||||||
|
|
||||||
|
public class BeanFacade {
|
||||||
|
|
||||||
|
private String test;
|
||||||
|
private String innerEnum;
|
||||||
|
|
||||||
|
public String getTest() {
|
||||||
|
return test;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setTest(String test) {
|
||||||
|
this.test = test;
|
||||||
|
}
|
||||||
|
|
||||||
|
public String getInnerEnum() {
|
||||||
|
return innerEnum;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setInnerEnum(String innerEnum) {
|
||||||
|
this.innerEnum = innerEnum;
|
||||||
|
}
|
||||||
|
}
|
@ -0,0 +1,45 @@
|
|||||||
|
/**
|
||||||
|
* Copyright 2012-2015 Gunnar Morling (http://www.gunnarmorling.de/)
|
||||||
|
* and/or other contributors as indicated by the @authors tag. See the
|
||||||
|
* copyright.txt file in the distribution for a full listing of all
|
||||||
|
* contributors.
|
||||||
|
*
|
||||||
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
* you may not use this file except in compliance with the License.
|
||||||
|
* You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing, software
|
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
* See the License for the specific language governing permissions and
|
||||||
|
* limitations under the License.
|
||||||
|
*/
|
||||||
|
package org.mapstruct.ap.test.imports.innerclasses;
|
||||||
|
|
||||||
|
public class BeanWithInnerEnum {
|
||||||
|
|
||||||
|
private String test;
|
||||||
|
private InnerEnum innerEnum;
|
||||||
|
|
||||||
|
public enum InnerEnum {
|
||||||
|
A, B;
|
||||||
|
}
|
||||||
|
|
||||||
|
public String getTest() {
|
||||||
|
return test;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setTest(String test) {
|
||||||
|
this.test = test;
|
||||||
|
}
|
||||||
|
|
||||||
|
public InnerEnum getInnerEnum() {
|
||||||
|
return innerEnum;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setInnerEnum(InnerEnum innerEnum) {
|
||||||
|
this.innerEnum = innerEnum;
|
||||||
|
}
|
||||||
|
}
|
@ -0,0 +1,32 @@
|
|||||||
|
/**
|
||||||
|
* Copyright 2012-2015 Gunnar Morling (http://www.gunnarmorling.de/)
|
||||||
|
* and/or other contributors as indicated by the @authors tag. See the
|
||||||
|
* copyright.txt file in the distribution for a full listing of all
|
||||||
|
* contributors.
|
||||||
|
*
|
||||||
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
* you may not use this file except in compliance with the License.
|
||||||
|
* You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing, software
|
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
* See the License for the specific language governing permissions and
|
||||||
|
* limitations under the License.
|
||||||
|
*/
|
||||||
|
package org.mapstruct.ap.test.imports.innerclasses;
|
||||||
|
|
||||||
|
import org.mapstruct.Mapper;
|
||||||
|
import org.mapstruct.factory.Mappers;
|
||||||
|
|
||||||
|
@Mapper
|
||||||
|
public interface BeanWithInnerEnumMapper {
|
||||||
|
|
||||||
|
BeanWithInnerEnumMapper INSTANCE = Mappers.getMapper( BeanWithInnerEnumMapper.class );
|
||||||
|
|
||||||
|
BeanWithInnerEnum fromFacade(BeanFacade beanFacade);
|
||||||
|
|
||||||
|
BeanFacade toFacade(BeanWithInnerEnum beanWithInnerEnum);
|
||||||
|
}
|
@ -0,0 +1,35 @@
|
|||||||
|
/**
|
||||||
|
* Copyright 2012-2015 Gunnar Morling (http://www.gunnarmorling.de/)
|
||||||
|
* and/or other contributors as indicated by the @authors tag. See the
|
||||||
|
* copyright.txt file in the distribution for a full listing of all
|
||||||
|
* contributors.
|
||||||
|
*
|
||||||
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
* you may not use this file except in compliance with the License.
|
||||||
|
* You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing, software
|
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
* See the License for the specific language governing permissions and
|
||||||
|
* limitations under the License.
|
||||||
|
*/
|
||||||
|
package org.mapstruct.ap.test.imports.innerclasses;
|
||||||
|
|
||||||
|
import org.mapstruct.Mapper;
|
||||||
|
import org.mapstruct.ap.test.imports.innerclasses.SourceWithInnerClass.SourceInnerClass;
|
||||||
|
import org.mapstruct.ap.test.imports.innerclasses.TargetWithInnerClass.TargetInnerClass;
|
||||||
|
import org.mapstruct.factory.Mappers;
|
||||||
|
|
||||||
|
@Mapper
|
||||||
|
public interface InnerClassMapper {
|
||||||
|
|
||||||
|
InnerClassMapper INSTANCE = Mappers.getMapper( InnerClassMapper.class );
|
||||||
|
|
||||||
|
TargetWithInnerClass sourceToTarget(SourceWithInnerClass source);
|
||||||
|
|
||||||
|
TargetInnerClass innerSourceToInnerTarget(SourceInnerClass source);
|
||||||
|
|
||||||
|
}
|
@ -0,0 +1,52 @@
|
|||||||
|
/**
|
||||||
|
* Copyright 2012-2015 Gunnar Morling (http://www.gunnarmorling.de/)
|
||||||
|
* and/or other contributors as indicated by the @authors tag. See the
|
||||||
|
* copyright.txt file in the distribution for a full listing of all
|
||||||
|
* contributors.
|
||||||
|
*
|
||||||
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
* you may not use this file except in compliance with the License.
|
||||||
|
* You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing, software
|
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
* See the License for the specific language governing permissions and
|
||||||
|
* limitations under the License.
|
||||||
|
*/
|
||||||
|
package org.mapstruct.ap.test.imports.innerclasses;
|
||||||
|
|
||||||
|
public class SourceWithInnerClass {
|
||||||
|
|
||||||
|
private SourceInnerClass innerClassMember;
|
||||||
|
|
||||||
|
public SourceInnerClass getInnerClassMember() {
|
||||||
|
return innerClassMember;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setInnerClassMember(SourceInnerClass innerClassMember) {
|
||||||
|
this.innerClassMember = innerClassMember;
|
||||||
|
}
|
||||||
|
|
||||||
|
public static class SourceInnerClass {
|
||||||
|
private int value;
|
||||||
|
|
||||||
|
public SourceInnerClass() {
|
||||||
|
}
|
||||||
|
|
||||||
|
public SourceInnerClass(int value) {
|
||||||
|
this.value = value;
|
||||||
|
}
|
||||||
|
|
||||||
|
public int getValue() {
|
||||||
|
return value;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setValue(int value) {
|
||||||
|
this.value = value;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
@ -0,0 +1,45 @@
|
|||||||
|
/**
|
||||||
|
* Copyright 2012-2015 Gunnar Morling (http://www.gunnarmorling.de/)
|
||||||
|
* and/or other contributors as indicated by the @authors tag. See the
|
||||||
|
* copyright.txt file in the distribution for a full listing of all
|
||||||
|
* contributors.
|
||||||
|
*
|
||||||
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
* you may not use this file except in compliance with the License.
|
||||||
|
* You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing, software
|
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
* See the License for the specific language governing permissions and
|
||||||
|
* limitations under the License.
|
||||||
|
*/
|
||||||
|
package org.mapstruct.ap.test.imports.innerclasses;
|
||||||
|
|
||||||
|
public class TargetWithInnerClass {
|
||||||
|
|
||||||
|
private TargetInnerClass innerClassMember;
|
||||||
|
|
||||||
|
public TargetInnerClass getInnerClassMember() {
|
||||||
|
return innerClassMember;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setInnerClassMember(TargetInnerClass innerClassMember) {
|
||||||
|
this.innerClassMember = innerClassMember;
|
||||||
|
}
|
||||||
|
|
||||||
|
public static class TargetInnerClass {
|
||||||
|
private int value;
|
||||||
|
|
||||||
|
public int getValue() {
|
||||||
|
return value;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setValue(int value) {
|
||||||
|
this.value = value;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
@ -63,7 +63,8 @@ public class JavaFileAssert extends FileAssert {
|
|||||||
* @param importedClass the class expected to be imported in this Java file
|
* @param importedClass the class expected to be imported in this Java file
|
||||||
*/
|
*/
|
||||||
public void containsImportFor(Class<?> importedClass) {
|
public void containsImportFor(Class<?> importedClass) {
|
||||||
content().contains( "import " + importedClass.getName() + ";" );
|
|
||||||
|
content().contains( getClassImportDeclaration( importedClass ) );
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@ -72,6 +73,23 @@ public class JavaFileAssert extends FileAssert {
|
|||||||
* @param importedClass the class expected not to be imported in this Java file
|
* @param importedClass the class expected not to be imported in this Java file
|
||||||
*/
|
*/
|
||||||
public void containsNoImportFor(Class<?> importedClass) {
|
public void containsNoImportFor(Class<?> importedClass) {
|
||||||
content().doesNotContain( "import " + importedClass.getName() + ";" );
|
content().doesNotContain( getClassImportDeclaration( importedClass ) );
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Build a class import declaration string.
|
||||||
|
*
|
||||||
|
* @param importedClass
|
||||||
|
* @return
|
||||||
|
*/
|
||||||
|
private String getClassImportDeclaration(Class<?> importedClass) {
|
||||||
|
String classname = importedClass.getName();
|
||||||
|
if ( importedClass.isMemberClass() ) {
|
||||||
|
// Member-Class name: a.b.Outer$Inner
|
||||||
|
// Import declaration: import a.b.Outer.Inner
|
||||||
|
classname = classname.replace( '$', '.' );
|
||||||
|
}
|
||||||
|
|
||||||
|
return "import " + classname + ";";
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user