001/* 002 * Copyright 2002-2018 the original author or authors. 003 * 004 * Licensed under the Apache License, Version 2.0 (the "License"); 005 * you may not use this file except in compliance with the License. 006 * You may obtain a copy of the License at 007 * 008 * https://www.apache.org/licenses/LICENSE-2.0 009 * 010 * Unless required by applicable law or agreed to in writing, software 011 * distributed under the License is distributed on an "AS IS" BASIS, 012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 013 * See the License for the specific language governing permissions and 014 * limitations under the License. 015 */ 016package org.springframework.http.server.reactive; 017 018import java.util.function.Supplier; 019 020import org.reactivestreams.Publisher; 021import reactor.core.publisher.Mono; 022 023import org.springframework.core.io.buffer.DataBuffer; 024import org.springframework.core.io.buffer.DataBufferFactory; 025import org.springframework.http.HttpHeaders; 026import org.springframework.http.HttpStatus; 027import org.springframework.http.ResponseCookie; 028import org.springframework.lang.Nullable; 029import org.springframework.util.Assert; 030import org.springframework.util.MultiValueMap; 031 032/** 033 * Wraps another {@link ServerHttpResponse} and delegates all methods to it. 034 * Sub-classes can override specific methods selectively. 035 * 036 * @author Rossen Stoyanchev 037 * @since 5.0 038 */ 039public class ServerHttpResponseDecorator implements ServerHttpResponse { 040 041 private final ServerHttpResponse delegate; 042 043 044 public ServerHttpResponseDecorator(ServerHttpResponse delegate) { 045 Assert.notNull(delegate, "Delegate is required"); 046 this.delegate = delegate; 047 } 048 049 050 public ServerHttpResponse getDelegate() { 051 return this.delegate; 052 } 053 054 055 // ServerHttpResponse delegation methods... 056 057 @Override 058 public boolean setStatusCode(@Nullable HttpStatus status) { 059 return getDelegate().setStatusCode(status); 060 } 061 062 @Override 063 public HttpStatus getStatusCode() { 064 return getDelegate().getStatusCode(); 065 } 066 067 @Override 068 public HttpHeaders getHeaders() { 069 return getDelegate().getHeaders(); 070 } 071 072 @Override 073 public MultiValueMap<String, ResponseCookie> getCookies() { 074 return getDelegate().getCookies(); 075 } 076 077 @Override 078 public void addCookie(ResponseCookie cookie) { 079 getDelegate().addCookie(cookie); 080 } 081 082 @Override 083 public DataBufferFactory bufferFactory() { 084 return getDelegate().bufferFactory(); 085 } 086 087 @Override 088 public void beforeCommit(Supplier<? extends Mono<Void>> action) { 089 getDelegate().beforeCommit(action); 090 } 091 092 @Override 093 public boolean isCommitted() { 094 return getDelegate().isCommitted(); 095 } 096 097 @Override 098 public Mono<Void> writeWith(Publisher<? extends DataBuffer> body) { 099 return getDelegate().writeWith(body); 100 } 101 102 @Override 103 public Mono<Void> writeAndFlushWith(Publisher<? extends Publisher<? extends DataBuffer>> body) { 104 return getDelegate().writeAndFlushWith(body); 105 } 106 107 @Override 108 public Mono<Void> setComplete() { 109 return getDelegate().setComplete(); 110 } 111 112 113 @Override 114 public String toString() { 115 return getClass().getSimpleName() + " [delegate=" + getDelegate() + "]"; 116 } 117 118}