Spring/spring-hateoas-examples
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Use Links API.

Browse Source
This commit is contained in:
Greg Turnquist
2020-04-13 10:37:56 -05:00
parent 287995aa1a
commit fa2ac888db
1 changed files with 3 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
hypermedia/src/main/java/org/springframework/hateoas/examples/EmployeeController.java
View File

@@ -17,14 +17,12 @@ package org.springframework.hateoas.examples;
import static org.springframework.hateoas.server.mvc.WebMvcLinkBuilder.*;
import java.util.List;
import java.util.stream.Collectors;
import java.util.stream.StreamSupport;
import org.springframework.hateoas.CollectionModel;
import org.springframework.hateoas.EntityModel;
import org.springframework.hateoas.IanaLinkRelations;
import org.springframework.hateoas.Link;
import org.springframework.hateoas.Links;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
@@ -88,16 +86,8 @@ class EmployeeController {
CollectionModel<EntityModel<Employee>> collectionModel = assembler
.toCollectionModel(repository.findByManagerId(id));
List<Link> newLinks = collectionModel.getLinks().stream() //
.map(link -> {
// Replace the "self" link so it links back to here
if (link.getRel() == IanaLinkRelations.SELF) {
return linkTo(methodOn(EmployeeController.class).findEmployees(id)).withSelfRel();
} else {
return link;
}
}) //
.collect(Collectors.toList());
Links newLinks = collectionModel.getLinks().merge(Links.MergeMode.REPLACE_BY_REL,
linkTo(methodOn(EmployeeController.class).findEmployees(id)).withSelfRel());
return ResponseEntity.ok(new CollectionModel<>(collectionModel.getContent(), newLinks));
}