Java Code Examples for org.springframework.hateoas.Resource#add()

@Override
public Resource<Ad> process(Resource<Ad> resource) {
    Ad ad = resource.getContent();
    if (hasAccessToModify(ad)) {
        Ad.Status status = ad.getStatus();
        if (status == Ad.Status.NEW) {
            resource.add(entityLinks.linkForSingleResource(Ad.class, ad.getId()).withRel("update"));
            resource.add(entityLinks.linkForSingleResource(Ad.class, ad.getId()).withRel("deletion"));
            resource.add(linkTo(methodOn(AdResourceController.class).publish(ad.getId(), null)).withRel("publishing"));
        }
        if (status == Ad.Status.PUBLISHED) {
            resource.add(linkTo(methodOn(AdResourceController.class).expire(ad.getId(), null)).withRel("expiration"));
        }
    }
    return resource;
}
 

@PostMapping(path = "/some/{someId}/other/{otherId}")
public ResponseEntity<Resource<TestDataHolder>> doSomething(@PathVariable String someId,
                                                            @PathVariable Integer otherId,
                                                            @RequestHeader("X-Custom-Header") String customHeader,
                                                            @RequestBody TestDataHolder testDataHolder) {
    testDataHolder.setId(UUID.randomUUID().toString());
    Resource<TestDataHolder> resource = new Resource<>(testDataHolder);
    resource.add(linkTo(methodOn(TestController.class).doSomething(someId, otherId, null, null)).withSelfRel());
    resource.add(linkTo(methodOn(TestController.class).doSomething(someId, otherId, null, null)).withRel("multiple"));
    resource.add(linkTo(methodOn(TestController.class).doSomething(someId, otherId, null, null)).withRel("multiple"));
    return ResponseEntity
            .ok()
            .header("X-Custom-Header", customHeader)
            .body(resource);
}
 

/**
 * Person creation.
 * @return Person
 */
@RequestMapping(value = "/", method = RequestMethod.POST)
@ResponseBody public ResponseEntity<?> savePerson(@RequestBody final Person person) {
  final Person persistedPerson = repository.save(person);
  final Resource<Person> resource = new Resource<Person>(persistedPerson);
  resource.add(
      linkTo(methodOn(PersonController.class).getPerson(persistedPerson.getId())).withSelfRel()
  );
  return ResponseEntity
      .status(HttpStatus.CREATED)
      .contentType(MediaType.APPLICATION_JSON)
      .body(resource);
}
 

/**
 * Person retriever.
 * @return Person
 */
@RequestMapping(value = "/{personId}", method = RequestMethod.GET)
@ResponseBody public ResponseEntity<?> getPerson(@PathVariable final Long personId) {
  final Person person = repository.findOne(personId);
  if (person == null) {
    return ResponseEntity.notFound().build();
  }
  final Resource<Person> resource = new Resource<Person>(person);
  resource.add(linkTo(methodOn(PersonController.class).getPerson(personId)).withSelfRel());

  return ResponseEntity.ok(resource);
}
 

private static Resource<Object> getResource(String result,Link link){
	ObjectMapper mapper = new ObjectMapper();
	ObjectNode onode = mapper.createObjectNode();
	onode.putPOJO("health", HttpStatus.OK);
	onode.putPOJO("status", HttpStatus.OK.value());
	onode.putPOJO("data", result);
	Resource<Object> resource = new Resource<Object>(onode);
	resource.add(link);
	return resource;
}
 

@Override
public Resource<Person> process(Resource<Person> resource) {
	String id = Long.toString(resource.getContent().getId());
	UriComponents uriComponents = ServletUriComponentsBuilder.fromCurrentContextPath()
			.path("/people/{id}/photo").buildAndExpand(id);
	String uri = uriComponents.toUriString();
	resource.add(new Link(uri, "photo"));
	return resource;
}
 

/**
 * Renders the given {@link Receipt} including links to the associated
 * {@link Order} as well as a self link in case the {@link Receipt} is still
 * available.
 * 
 * @param receipt
 * @return
 */
private HttpEntity<Resource<Receipt>> createReceiptResponse(Receipt receipt) {

	Order order = receipt.getOrder();

	Resource<Receipt> resource = new Resource<Receipt>(receipt);
	resource.add(entityLinks.linkToSingleResource(order));

	if (!order.isTaken()) {
		resource.add(entityLinks.linkForSingleResource(order).slash("receipt").withSelfRel());
	}

	return new ResponseEntity<Resource<Receipt>>(resource, HttpStatus.OK);
}
 

@Override
public Resource<Task> toResource(Task task) {
    Resource<Task> resource = new Resource<>(task);

    resource.add(getTaskLinkBuilder(task).withSelfRel());

    return resource;
}
 

@GetMapping(path = "/users/{name}/todos/{id}")
public Resource<Todo> retrieveTodo(@PathVariable String name, @PathVariable int id) {
	Todo todo = todoService.retrieveTodo(id);
	if (todo == null) {
		throw new TodoNotFoundException("Todo Not Found");
	}
	
	Resource<com.mastering.spring.springboot.bean.Todo> todoResource = new Resource<com.mastering.spring.springboot.bean.Todo>(todo);
	ControllerLinkBuilder linkTo = linkTo(methodOn(this.getClass()).retrieveTodos(name));
	todoResource.add(linkTo.withRel("parent"));
	return todoResource;
}
 

@GetMapping(path = "/users/{name}/todos/{id}")
public Resource<Todo> retrieveTodo(@PathVariable String name, @PathVariable int id) {
	Todo todo = todoService.retrieveTodo(id);
	if (todo == null) {
		throw new TodoNotFoundException("Todo Not Found");
	}
	
	Resource<com.mastering.spring.springboot.bean.Todo> todoResource = new Resource<com.mastering.spring.springboot.bean.Todo>(todo);
	ControllerLinkBuilder linkTo = linkTo(methodOn(this.getClass()).retrieveTodos(name));
	todoResource.add(linkTo.withRel("parent"));
	return todoResource;
}
 

@GetMapping(path = "/users/{name}/todos/{id}")
public Resource<Todo> retrieveTodo(@PathVariable String name, @PathVariable int id) {
	Todo todo = todoService.retrieveTodo(id);
	if (todo == null) {
		throw new TodoNotFoundException("Todo Not Found");
	}
	
	Resource<com.mastering.spring.springboot.bean.Todo> todoResource = new Resource<com.mastering.spring.springboot.bean.Todo>(todo);
	ControllerLinkBuilder linkTo = linkTo(methodOn(this.getClass()).retrieveTodos(name));
	todoResource.add(linkTo.withRel("parent"));
	return todoResource;
}
 

@GetMapping(path = "/users/{name}/todos/{id}")
public Resource<Todo> retrieveTodo(@PathVariable String name, @PathVariable int id) {
	Todo todo = todoService.retrieveTodo(id);
	if (todo == null) {
		throw new TodoNotFoundException("Todo Not Found");
	}

	Resource<Todo> todoResource = new Resource<com.mastering.spring.springboot.bean.Todo>(todo);
	ControllerLinkBuilder linkTo = linkTo(methodOn(this.getClass()).retrieveTodos(name));
	todoResource.add(linkTo.withRel("parent"));
	return todoResource;
}
 

@Override
public Resource<Bookmark> toResource(Bookmark entity) {
    Resource<Bookmark> resource = new Resource<>(entity,
            linkTo(methodOn(BookmarkController.class).getBookmark(entity.getUuid())).withSelfRel());
    if(request.isUserInRole("ADMIN")){
        resource.add(
                linkTo(methodOn(BookmarkController.class).getBookmark(entity.getUuid())).withRel("update"),
                linkTo(methodOn(BookmarkController.class).getBookmark(entity.getUuid())).withRel("delete")
        );
    }
    return resource;
}
 

@GetMapping(path = "/jpa/users/{id}")
public Resource<User> retrieveUser(@PathVariable int id) {
	final User user = userJPAService.findOne(id);
	if (Objects.isNull(user)) {
		throw new UserNotFoundException("There is no user with id: " + id);
	}

	final Resource<User> resource = new Resource<>(user);
	final ControllerLinkBuilder linkTo = linkTo(methodOn(this.getClass()).retrieveAllUsers());
	resource.add(linkTo.withRel("all-users"));

	return resource;
}
 

@Override
public Resource<Person> process(Resource<Person> resource) {
	String id = Long.toString(resource.getContent().getId());
	UriComponents uriComponents = ServletUriComponentsBuilder.fromCurrentContextPath()
			.path("/people/{id}/photo").buildAndExpand(id);
	String uri = uriComponents.toUriString();
	resource.add(new Link(uri, "photo"));
	return resource;
}
 

@GetMapping(path = "/users/{name}/todos/{id}")
public Resource<Todo> retrieveTodo(@PathVariable String name, @PathVariable int id) {
	Todo todo = todoService.retrieveTodo(id);
	if (todo == null) {
		throw new TodoNotFoundException("Todo Not Found");
	}

	Resource<Todo> todoResource = new Resource<com.mastering.spring.springboot.bean.Todo>(todo);
	ControllerLinkBuilder linkTo = linkTo(methodOn(this.getClass()).retrieveTodos(name));
	todoResource.add(linkTo.withRel("parent"));
	return todoResource;
}
 

@GetMapping(path = "/users/{name}/todos/{id}")
public Resource<Todo> retrieveTodo(@PathVariable String name, @PathVariable int id) {
	Todo todo = todoService.retrieveTodo(id);
	if (todo == null) {
		throw new TodoNotFoundException("Todo Not Found");
	}

	Resource<Todo> todoResource = new Resource<com.mastering.spring.springboot.bean.Todo>(todo);
	ControllerLinkBuilder linkTo = linkTo(methodOn(this.getClass()).retrieveTodos(name));
	todoResource.add(linkTo.withRel("parent"));
	return todoResource;
}
 

@GetMapping("/jpa/users/{id}")
public Resource<User> retrieveUser(@PathVariable int id) {
	Optional<User> user = userRepository.findById(id);

	if (!user.isPresent())
		throw new UserNotFoundException("id-" + id);

	// "all-users", SERVER_PATH + "/users"
	// retrieveAllUsers
	Resource<User> resource = new Resource<User>(user.get());

	ControllerLinkBuilder linkTo = linkTo(methodOn(this.getClass()).retrieveAllUsers());

	resource.add(linkTo.withRel("all-users"));

	// HATEOAS

	return resource;
}
 

@GetMapping("/users/{id}")
public Resource<User> retrieveUser(@PathVariable int id) {
	User user = service.findOne(id);
	
	if(user==null)
		throw new UserNotFoundException("id-"+ id);
	
	
	//"all-users", SERVER_PATH + "/users"
	//retrieveAllUsers
	Resource<User> resource = new Resource<User>(user);
	
	ControllerLinkBuilder linkTo = 
			linkTo(methodOn(this.getClass()).retrieveAllUsers());
	
	resource.add(linkTo.withRel("all-users"));
	
	//HATEOAS
	
	return resource;
}
 

@Override
public Resource<ProductAdded> process(Resource<ProductAdded> resource) {

	ProductAdded content = resource.getContent();

	resource.add(entityLinks.linkToSingleResource(Product.class, content.getProduct().getId()));

	return resource;
}