fr.ouestfrance.querydsl.postgrest.ResponseUtils Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of querydsl-postgrest-resttemplate-adapter Show documentation

There is a newer version: 1.4.5

Show newest version

package fr.ouestfrance.querydsl.postgrest;

import fr.ouestfrance.querydsl.postgrest.model.BulkResponse;
import fr.ouestfrance.querydsl.postgrest.model.HeaderRange;
import lombok.AccessLevel;
import lombok.NoArgsConstructor;
import org.springframework.http.HttpHeaders;
import org.springframework.http.ResponseEntity;

import java.util.List;
import java.util.Optional;

/**
 * Utility class that helps to transform response
 */
@NoArgsConstructor(access = AccessLevel.PRIVATE)
public final class ResponseUtils {

    /**
     * Transform a ResponseEntity of list to bulkResponse
     *
     * @param response response entity
     * @param       Type of response
     * @return BulkResponse
     */
    public static  BulkResponse toBulkResponse(ResponseEntity> response) {
        return Optional.ofNullable(response)
                .map(x -> {
                    Optional count = getCount(x.getHeaders());
                    return new BulkResponse<>(x.getBody(), count.map(HeaderRange::getCount).orElse(0L), count.map(HeaderRange::getTotalElements).orElse(0L));
                })
                .orElse(new BulkResponse<>(List.of(), 0L, 0L));
    }

    /**
     * Extract Range headers
     *
     * @param headers headers where Content-Range is
     * @return range object
     */
    public static Optional getCount(HttpHeaders headers) {
        return Optional.ofNullable(headers.get("Content-Range"))
                .flatMap(x -> x.stream().findFirst())
                .map(HeaderRange::of);
    }
}