pro.taskana.common.rest.assembler.CollectionRepresentationModelAssembler Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of taskana-rest-spring Show documentation

The taskana rest logic.

The newest version!

package pro.taskana.common.rest.assembler;

import java.io.UnsupportedEncodingException;
import java.net.URLDecoder;
import java.util.List;
import java.util.stream.Collectors;
import java.util.stream.StreamSupport;
import org.springframework.hateoas.Link;
import org.springframework.hateoas.RepresentationModel;
import org.springframework.hateoas.server.RepresentationModelAssembler;
import org.springframework.web.servlet.support.ServletUriComponentsBuilder;
import org.springframework.web.util.UriComponentsBuilder;
import pro.taskana.common.rest.models.CollectionRepresentationModel;

public interface CollectionRepresentationModelAssembler<
        T, D extends RepresentationModel, C extends CollectionRepresentationModel>
    extends RepresentationModelAssembler {

  C buildCollectionEntity(List content);

  default C toTaskanaCollectionModel(Iterable entities) {
    return StreamSupport.stream(entities.spliterator(), false)
        .map(this::toModel)
        .collect(
            Collectors.collectingAndThen(
                Collectors.toList(),
                content -> addLinksToCollectionModel(buildCollectionEntity(content))));
  }

  default C addLinksToCollectionModel(C model) {
    final UriComponentsBuilder original = ServletUriComponentsBuilder.fromCurrentRequest();

    try {
      model.add(Link.of(URLDecoder.decode(original.toUriString(), "UTF-8")).withSelfRel());
    } catch (UnsupportedEncodingException e) {
      throw new RuntimeException("UTF-8 encoding not supported. This is unexpected.");
    }
    return model;
  }
}