com.github.mkopylec.charon.forwarding.HttpResponseMapper Maven / Gradle / Ivy

Go to download
package com.github.mkopylec.charon.forwarding;

import java.io.IOException;

import javax.servlet.http.HttpServletResponse;

import org.springframework.http.ResponseEntity;

class HttpResponseMapper {

    void map(ResponseEntity responseEntity, HttpServletResponse response) throws IOException {
        setStatus(responseEntity, response);
        setHeaders(responseEntity, response);
        setBody(responseEntity, response);
    }

    private void setStatus(ResponseEntity responseEntity, HttpServletResponse response) {
        response.setStatus(responseEntity.getStatusCodeValue());
    }

    private void setHeaders(ResponseEntity responseEntity, HttpServletResponse response) {
        responseEntity.getHeaders().forEach((name, values) -> values.forEach(value -> response.addHeader(name, value)));
    }

    private void setBody(ResponseEntity responseEntity, HttpServletResponse response) throws IOException {
        if (responseEntity.getBody() != null) {
            response.getOutputStream().write(responseEntity.getBody());
        }
    }
}