All Downloads are FREE. Search and download functionalities are using the official Maven repository.

artoria.context.spring.ContextSupportResponseBodyAdvice Maven / Gradle / Ivy

The newest version!
package artoria.context.spring;

import artoria.bean.BeanHolder;
import artoria.context.support.ServletContext;
import org.springframework.core.MethodParameter;
import org.springframework.http.MediaType;
import org.springframework.http.converter.HttpMessageConverter;
import org.springframework.http.server.ServerHttpRequest;
import org.springframework.http.server.ServerHttpResponse;
import org.springframework.lang.NonNull;
import org.springframework.web.bind.annotation.ControllerAdvice;
import org.springframework.web.servlet.mvc.method.annotation.ResponseBodyAdvice;

import java.util.Map;

/**
 * The response body advice for the servlet context.
 * @author Kahle
 * @see org.springframework.web.servlet.mvc.method.annotation.ResponseBodyAdvice
 */
@ControllerAdvice
public class ContextSupportResponseBodyAdvice implements ResponseBodyAdvice {

    @Override
    public boolean supports(@NonNull MethodParameter returnType,
                            @NonNull Class> converterType) {
        return true;
    }

    @Override
    public Object beforeBodyWrite(Object body,
                                  @NonNull MethodParameter returnType,
                                  @NonNull MediaType selectedContentType,
                                  @NonNull Class> selectedConverterType,
                                  @NonNull ServerHttpRequest request,
                                  @NonNull ServerHttpResponse response) {
        Map beans = BeanHolder.getBeans(ServletContext.class);
        if (beans == null || beans.isEmpty()) { return body; }
        for (Map.Entry entry : beans.entrySet()) {
            ServletContext context = entry.getValue();
            context.setResponseBody(body);
        }
        return body;
    }

}