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

#1895 Refactor class org.mapstruct.ap.internal.util.Filters (#1896)

Browse Source
This commit is contained in:
Andrei Arlou 2019-09-14 03:57:42 +03:00 committed by GitHub
parent 3868735da7
commit 2043506179
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 34 additions and 61 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

93
processor/src/main/java/org/mapstruct/ap/internal/util/Filters.java
View File

@ -7,6 +7,7 @@ package org.mapstruct.ap.internal.util;
import java.util.LinkedList; import java.util.LinkedList;
import java.util.List; import java.util.List;
import java.util.stream.Collectors;
import javax.lang.model.element.Element; import javax.lang.model.element.Element;
import javax.lang.model.element.ExecutableElement; import javax.lang.model.element.ExecutableElement;
import javax.lang.model.element.VariableElement; import javax.lang.model.element.VariableElement;
@ -43,72 +44,37 @@ public class Filters {
} }
public List<Accessor> getterMethodsIn(List<ExecutableElement> elements) { public List<Accessor> getterMethodsIn(List<ExecutableElement> elements) {
List<Accessor> getterMethods = new LinkedList<>(); return elements.stream()
.filter( accessorNaming::isGetterMethod )
for ( ExecutableElement method : elements ) { .map( method -> new ExecutableElementAccessor( method, getReturnType( method ), GETTER ) )
if ( accessorNaming.isGetterMethod( method ) ) { .collect( Collectors.toCollection( LinkedList::new ) );
getterMethods.add( new ExecutableElementAccessor( method, getReturnType( method ), GETTER ) );
}
}
return getterMethods;
}
public List<Accessor> fieldsIn(List<VariableElement> accessors) {
List<Accessor> fieldAccessors = new LinkedList<>();
for ( VariableElement accessor : accessors ) {
if ( Fields.isFieldAccessor( accessor ) ) {
fieldAccessors.add( new VariableElementAccessor( accessor ) );
}
}
return fieldAccessors;
}
public List<Accessor> presenceCheckMethodsIn(List<ExecutableElement> elements) {
List<Accessor> presenceCheckMethods = new LinkedList<>();
for ( ExecutableElement method : elements ) {
if ( accessorNaming.isPresenceCheckMethod( method ) ) {
presenceCheckMethods.add( new ExecutableElementAccessor(
method,
getReturnType( method ),
PRESENCE_CHECKER
) );
}
}
return presenceCheckMethods;
}
public List<Accessor> setterMethodsIn(List<ExecutableElement> elements) {
List<Accessor> setterMethods = new LinkedList<>();
for ( ExecutableElement method : elements ) {
if ( accessorNaming.isSetterMethod( method ) ) {
setterMethods.add( new ExecutableElementAccessor( method, getFirstParameter( method ), SETTER ) );
}
}
return setterMethods;
}
public List<Accessor> adderMethodsIn( List<ExecutableElement> elements) {
List<Accessor> adderMethods = new LinkedList<>();
for ( ExecutableElement method : elements ) {
if ( accessorNaming.isAdderMethod( method ) ) {
adderMethods.add( new ExecutableElementAccessor( method, getFirstParameter( method ), ADDER ) );
}
}
return adderMethods;
} }
private TypeMirror getReturnType(ExecutableElement executableElement) { private TypeMirror getReturnType(ExecutableElement executableElement) {
return getWithinContext( executableElement ).getReturnType(); return getWithinContext( executableElement ).getReturnType();
} }
public List<Accessor> fieldsIn(List<VariableElement> accessors) {
return accessors.stream()
.filter( Fields::isFieldAccessor )
.map( VariableElementAccessor::new )
.collect( Collectors.toCollection( LinkedList::new ) );
}
public List<Accessor> presenceCheckMethodsIn(List<ExecutableElement> elements) {
return elements.stream()
.filter( accessorNaming::isPresenceCheckMethod )
.map( method -> new ExecutableElementAccessor( method, getReturnType( method ), PRESENCE_CHECKER ) )
.collect( Collectors.toCollection( LinkedList::new ) );
}
public List<Accessor> setterMethodsIn(List<ExecutableElement> elements) {
return elements.stream()
.filter( accessorNaming::isSetterMethod )
.map( method -> new ExecutableElementAccessor( method, getFirstParameter( method ), SETTER ) )
.collect( Collectors.toCollection( LinkedList::new ) );
}
private TypeMirror getFirstParameter(ExecutableElement executableElement) { private TypeMirror getFirstParameter(ExecutableElement executableElement) {
return first( getWithinContext( executableElement ).getParameterTypes() ); return first( getWithinContext( executableElement ).getParameterTypes() );
} }
@ -116,4 +82,11 @@ public class Filters {
private ExecutableType getWithinContext( ExecutableElement executableElement ) { private ExecutableType getWithinContext( ExecutableElement executableElement ) {
return (ExecutableType) typeUtils.asMemberOf( (DeclaredType) typeMirror, executableElement ); return (ExecutableType) typeUtils.asMemberOf( (DeclaredType) typeMirror, executableElement );
} }
public List<Accessor> adderMethodsIn(List<ExecutableElement> elements) {
return elements.stream()
.filter( accessorNaming::isAdderMethod )
.map( method -> new ExecutableElementAccessor( method, getFirstParameter( method ), ADDER ) )
.collect( Collectors.toCollection( LinkedList::new ) );
}
} }