third_codes/mapstruct
1
0
Fork 0
mirror of https://github.com/mapstruct/mapstruct.git synced 2025-07-12 00:00:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

#34 Accessing annotation in a safer way

Browse Source
This commit is contained in:
Gunnar Morling 2014-01-09 21:45:01 +01:00
parent a7be616ee9
commit 217562d52b
2 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
processor/src/main/java/org/mapstruct/ap/model/DefaultMapperReference.java
View File

@ -37,10 +37,10 @@ public class DefaultMapperReference extends AbstractModelElement implements Mapp
private final boolean isAnnotatedMapper; private final boolean isAnnotatedMapper;
private final Set<Type> importTypes; private final Set<Type> importTypes;
public DefaultMapperReference(Type type, TypeFactory typeFactory) { public DefaultMapperReference(Type type, boolean isAnnotatedMapper, TypeFactory typeFactory) {
this.type = type; this.type = type;
isAnnotatedMapper = type.isAnnotatedWith( "org.mapstruct.Mapper" ); this.isAnnotatedMapper = isAnnotatedMapper;
importTypes = Collections.asSet( type ); importTypes = Collections.asSet( type );
if ( isAnnotatedMapper() ) { if ( isAnnotatedMapper() ) {

8
processor/src/main/java/org/mapstruct/ap/processor/MapperCreationProcessor.java
View File

@ -148,7 +148,13 @@ public class MapperCreationProcessor implements ModelElementProcessor<List<Metho
MapperPrism mapperPrism = MapperPrism.getInstanceOn( element ); MapperPrism mapperPrism = MapperPrism.getInstanceOn( element );
for ( TypeMirror usedMapper : mapperPrism.uses() ) { for ( TypeMirror usedMapper : mapperPrism.uses() ) {
mapperReferences.add( new DefaultMapperReference( typeFactory.getType( usedMapper ), typeFactory ) ); mapperReferences.add(
new DefaultMapperReference(
typeFactory.getType( usedMapper ),
MapperPrism.getInstanceOn( typeUtils.asElement( usedMapper ) ) != null,
typeFactory
)
);
} }
return mapperReferences; return mapperReferences;