Skip to content

Commit 041f1fb

Browse files
committed
code review
1 parent 7c09357 commit 041f1fb

File tree

6 files changed

+16
-11
lines changed

6 files changed

+16
-11
lines changed

springdoc-openapi-data-rest/src/main/java/org/springdoc/data/rest/SpringRepositoryRestResourceProvider.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -230,7 +230,8 @@ else if (handlerMapping instanceof BasePathAwareHandlerMapping) {
230230
.getValue().getBeanType()) || AlpsController.class.equals(requestMappingInfoHandlerMethodEntry
231231
.getValue().getBeanType()))
232232
.collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue, (a1, a2) -> a1));
233-
findControllers(routerOperationList, handlerMethodMapFiltered, resourceMetadata, null, openAPI);
233+
dataRestRepository.setControllerType(ControllerType.GENERAL);
234+
findControllers(routerOperationList, handlerMethodMapFiltered, resourceMetadata, dataRestRepository, openAPI);
234235
}
235236
}
236237
}

springdoc-openapi-data-rest/src/main/java/org/springdoc/data/rest/core/ControllerType.java

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,5 +43,9 @@ public enum ControllerType {
4343
/**
4444
*PROPERTY controller type.
4545
*/
46-
PROPERTY
46+
PROPERTY,
47+
/**
48+
*GENERAL controller type.
49+
*/
50+
GENERAL
4751
}

springdoc-openapi-data-rest/src/main/java/org/springdoc/data/rest/core/DataRestOperationService.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -124,7 +124,7 @@ public Operation buildOperation(HandlerMethod handlerMethod, DataRestRepository
124124
Operation operation = null;
125125
if (ControllerType.ENTITY.equals(controllerType)
126126
|| ControllerType.PROPERTY.equals(controllerType)
127-
|| ControllerType.SCHEMA.equals(controllerType)) {
127+
|| ControllerType.SCHEMA.equals(controllerType) || ControllerType.GENERAL.equals(controllerType)) {
128128
operation = buildEntityOperation(handlerMethod, dataRestRepository,
129129
openAPI, requestMethod, operationPath, methodAttributes, resourceMetadata);
130130
}
@@ -151,7 +151,7 @@ private Operation buildEntityOperation(HandlerMethod handlerMethod, DataRestRepo
151151
OpenAPI openAPI, RequestMethod requestMethod, String operationPath, MethodAttributes methodAttributes,
152152
ResourceMetadata resourceMetadata) {
153153
Class<?> domainType = null;
154-
if (dataRestRepository != null)
154+
if (!ControllerType.GENERAL.equals(dataRestRepository.getControllerType()))
155155
domainType = dataRestRepository.getDomainType();
156156
Operation operation = initOperation(handlerMethod, domainType, requestMethod);
157157
dataRestRequestService.buildParameters(domainType, openAPI, handlerMethod, requestMethod, methodAttributes, operation, resourceMetadata);
@@ -312,7 +312,7 @@ private void addOperationDescription(Operation operation, RequestMethod requestM
312312
*/
313313
private String createDescription( String action, String entity, DataRestRepository dataRestRepository) {
314314
String description;
315-
if (dataRestRepository != null && ControllerType.PROPERTY.equals(dataRestRepository.getControllerType()))
315+
if (ControllerType.PROPERTY.equals(dataRestRepository.getControllerType()))
316316
description = action + dataRestRepository.getPropertyType().getSimpleName().toLowerCase()+ "-by-"+ entity +"-Id";
317317
else
318318
description = action + entity;

springdoc-openapi-data-rest/src/main/java/org/springdoc/data/rest/core/DataRestResponseService.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -214,7 +214,7 @@ private Type getType(MethodParameter methodParameterReturn, Class<?> domainType,
214214
Type returnType = ReturnTypeParser.resolveType(methodParameterReturn.getGenericParameterType(), methodParameterReturn.getContainingClass());
215215
Class returnedEntityType = domainType;
216216

217-
if (dataRestRepository != null && ControllerType.PROPERTY.equals(dataRestRepository.getControllerType()))
217+
if (ControllerType.PROPERTY.equals(dataRestRepository.getControllerType()))
218218
returnedEntityType = dataRestRepository.getPropertyType();
219219

220220
if (returnType instanceof ParameterizedType) {
@@ -268,10 +268,10 @@ else if ((CollectionModel.class.equals(parameterizedType.getRawType())
268268
* @return the class
269269
*/
270270
private Class findType(RequestMethod requestMethod, DataRestRepository dataRestRepository) {
271-
if (dataRestRepository != null && ControllerType.ENTITY.equals(dataRestRepository.getControllerType())
271+
if (ControllerType.ENTITY.equals(dataRestRepository.getControllerType())
272272
&& Arrays.stream(requestMethodsEntityModel).anyMatch(requestMethod::equals))
273273
return EntityModel.class;
274-
else if (dataRestRepository != null && ControllerType.PROPERTY.equals(dataRestRepository.getControllerType())) {
274+
else if (ControllerType.PROPERTY.equals(dataRestRepository.getControllerType())) {
275275
if (dataRestRepository.isCollectionLike())
276276
return CollectionModel.class;
277277
else if (dataRestRepository.isMap())

springdoc-openapi-data-rest/src/main/java/org/springdoc/data/rest/core/DataRestRouterOperationService.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -113,7 +113,7 @@ public void buildEntityRouterOperationList(List<RouterOperation> routerOperation
113113
Map<RequestMappingInfo, HandlerMethod> handlerMethodMap, ResourceMetadata resourceMetadata,
114114
DataRestRepository dataRestRepository, OpenAPI openAPI) {
115115
String path = resourceMetadata.getPath().toString();
116-
ControllerType controllerType = (dataRestRepository == null) ? ControllerType.SCHEMA : dataRestRepository.getControllerType();
116+
ControllerType controllerType = dataRestRepository.getControllerType();
117117
for (Entry<RequestMappingInfo, HandlerMethod> entry : handlerMethodMap.entrySet()) {
118118
buildRouterOperationList(routerOperationList, resourceMetadata, dataRestRepository, openAPI, path, entry, null, controllerType, null);
119119
}
@@ -180,7 +180,7 @@ private void buildRouterOperationList(List<RouterOperation> routerOperationList,
180180
if (patternsRequestCondition != null) {
181181
Set<String> patterns = patternsRequestCondition.getPatterns();
182182
Map<String, String> regexMap = new LinkedHashMap<>();
183-
String relationName = (dataRestRepository != null) ? dataRestRepository.getRelationName() : null;
183+
String relationName = dataRestRepository.getRelationName();
184184
String operationPath = calculateOperationPath(path, subPath, patterns, regexMap, controllerType, relationName);
185185
buildRouterOperation(routerOperationList, dataRestRepository, openAPI, methodResourceMapping,
186186
handlerMethod, requestMethod, resourceMetadata, operationPath, controllerType);

springdoc-openapi-data-rest/src/main/java/org/springdoc/data/rest/core/DataRestTagsService.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -98,7 +98,7 @@ private void buildTags(Operation operation, HandlerMethod handlerMethod,
9898
tagName = ProfileController.class.getSimpleName();
9999
operation.addTagsItem(OpenAPIService.splitCamelCase(tagName));
100100
}
101-
else if (dataRestRepository != null && dataRestRepository.getDomainType() != null) {
101+
else {
102102
Class<?> domainType = dataRestRepository.getDomainType();
103103
Set<Tag> tags = new HashSet<>();
104104
Set<String> tagsStr = new HashSet<>();

0 commit comments

Comments
 (0)