io.inverno.mod.http.server.internal.http2.Http2ResponseHeaders Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of inverno-http-server Show documentation
Show all versions of inverno-http-server Show documentation
Inverno HTTP 1.x/2 server module
/*
* Copyright 2020 Jeremy KUHN
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package io.inverno.mod.http.server.internal.http2;
import io.inverno.mod.base.converter.ObjectConverter;
import io.inverno.mod.http.base.InboundSetCookies;
import io.inverno.mod.http.base.OutboundResponseHeaders;
import io.inverno.mod.http.base.OutboundSetCookies;
import io.inverno.mod.http.base.Parameter;
import io.inverno.mod.http.base.Status;
import io.inverno.mod.http.base.header.Header;
import io.inverno.mod.http.base.header.HeaderService;
import io.inverno.mod.http.base.header.Headers;
import io.inverno.mod.http.base.internal.GenericParameter;
import io.inverno.mod.http.server.internal.GenericResponseCookies;
import io.inverno.mod.http.server.internal.InternalResponseHeaders;
import io.netty.handler.codec.http2.DefaultHttp2Headers;
import io.netty.handler.codec.http2.Http2Headers;
import io.netty.handler.codec.http2.Http2Headers.PseudoHeaderName;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Optional;
import java.util.Set;
import java.util.function.Consumer;
import java.util.stream.Collectors;
/**
*
* HTTP/2 {@link InternalResponseHeaders} implementation.
*
*
* @author Jeremy Kuhn
* @since 1.0
*
* @see InternalResponseHeaders
*/
class Http2ResponseHeaders implements InternalResponseHeaders {
private final HeaderService headerService;
private final ObjectConverter parameterConverter;
private final Http2Headers underlyingHeaders;
private GenericResponseCookies responseCookies;
private boolean written;
/**
*
* Creates HTTP/2 response headers.
*
*
* @param headerService the header service
* @param parameterConverter a string object converter
*/
public Http2ResponseHeaders(HeaderService headerService, ObjectConverter parameterConverter) {
this.headerService = headerService;
this.parameterConverter = parameterConverter;
this.underlyingHeaders = new DefaultHttp2Headers();
this.underlyingHeaders.set(PseudoHeaderName.STATUS.value(), "200");
}
/**
*
* Returns the underlying headers.
*
*
* @return the underlying headers
*/
Http2Headers getUnderlyingHeaders() {
return this.underlyingHeaders;
}
@Override
public void setWritten(boolean written) {
this.written = written;
}
@Override
public boolean isWritten() {
return this.written;
}
@Override
public Http2ResponseHeaders status(Status status) {
return this.status(status.getCode());
}
@Override
public Http2ResponseHeaders status(int status) {
this.underlyingHeaders.setInt(PseudoHeaderName.STATUS.value(), status);
return this;
}
@Override
public Status getStatus() {
return Status.valueOf(this.getStatusCode());
}
@Override
public int getStatusCode() {
return this.underlyingHeaders.getInt(PseudoHeaderName.STATUS.value());
}
@Override
public Http2ResponseHeaders contentType(String contentType) {
this.underlyingHeaders.set(Headers.NAME_CONTENT_TYPE, contentType);
return this;
}
@Override
public String getContentType() {
return this.underlyingHeaders.get(Headers.NAME_CONTENT_TYPE).toString();
}
@Override
public Headers.ContentType getContentTypeHeader() {
return this.getHeader(Headers.NAME_CONTENT_TYPE).orElse(null);
}
@Override
public CharSequence getContentTypeCharSequence() {
return this.underlyingHeaders.get(Headers.NAME_CONTENT_TYPE);
}
@Override
public Http2ResponseHeaders contentLength(long contentLength) {
this.underlyingHeaders.setLong(Headers.NAME_CONTENT_LENGTH, contentLength);
return this;
}
@Override
public Long getContentLength() {
return this.underlyingHeaders.getLong(Headers.NAME_CONTENT_LENGTH);
}
@Override
public OutboundResponseHeaders cookies(Consumer cookiesConfigurer) {
if(this.responseCookies == null) {
this.responseCookies = new GenericResponseCookies(this.headerService, this, this.parameterConverter);
}
cookiesConfigurer.accept(this.responseCookies);
return this;
}
@Override
public InboundSetCookies cookies() {
if(this.responseCookies == null) {
this.responseCookies = new GenericResponseCookies(this.headerService, this, this.parameterConverter);
}
return this.responseCookies;
}
@Override
public Http2ResponseHeaders add(CharSequence name, CharSequence value) {
this.underlyingHeaders.add(name, value);
return this;
}
@Override
public Http2ResponseHeaders add(Header... headers) {
for(Header header : headers) {
this.underlyingHeaders.add(header.getHeaderName(), header.getHeaderValue());
}
return this;
}
@Override
public Http2ResponseHeaders set(CharSequence name, CharSequence value) {
this.underlyingHeaders.set(name, value);
return this;
}
@Override
public Http2ResponseHeaders set(Header... headers) {
for(Header header : headers) {
this.underlyingHeaders.set(header.getHeaderName(), header.getHeaderValue());
}
return this;
}
@Override
public Http2ResponseHeaders remove(CharSequence... names) {
for(CharSequence name : names) {
this.underlyingHeaders.remove(name);
}
return this;
}
@Override
public boolean contains(CharSequence name) {
return this.underlyingHeaders.contains(name);
}
@Override
public boolean contains(CharSequence name, CharSequence value) {
return this.underlyingHeaders.contains(name, value, true);
}
@Override
public Set getNames() {
return this.underlyingHeaders.names().stream().map(CharSequence::toString).collect(Collectors.toSet());
}
@Override
public Optional get(CharSequence name) {
return Optional.ofNullable(this.underlyingHeaders.get(name)).map(Object::toString);
}
@Override
public List getAll(CharSequence name) {
return this.underlyingHeaders.getAll(name).stream().map(CharSequence::toString).collect(Collectors.toList());
}
@Override
public List> getAll() {
List> result = new LinkedList<>();
this.underlyingHeaders.forEach(e -> {
result.add(Map.entry(e.getKey().toString(), e.getValue().toString()));
});
return result;
}
@Override
public Optional getHeader(CharSequence name) {
return this.get(name).map(value -> this.headerService.decode(name.toString(), value));
}
@Override
public List getAllHeader(CharSequence name) {
return this.underlyingHeaders.getAll(name).stream().map(value -> this.headerService.decode(name.toString(), value.toString())).collect(Collectors.toList());
}
@Override
public List getAllHeader() {
List result = new LinkedList<>();
this.underlyingHeaders.forEach(e -> {
result.add(this.headerService.decode(e.getKey().toString(), e.getValue().toString()));
});
return result;
}
@Override
public Optional getParameter(CharSequence name) {
return this.get(name).map(value -> new GenericParameter(name.toString(), value, this.parameterConverter));
}
@Override
public List getAllParameter(CharSequence name) {
return this.underlyingHeaders.getAll(name).stream().map(value -> new GenericParameter(name.toString(), value.toString(), this.parameterConverter)).collect(Collectors.toList());
}
@Override
public List getAllParameter() {
List result = new LinkedList<>();
this.underlyingHeaders.forEach(e -> {
result.add(new GenericParameter(e.getKey().toString(), e.getValue().toString(), this.parameterConverter));
});
return result;
}
@Override
public CharSequence getCharSequence(CharSequence name) {
return this.underlyingHeaders.get(name);
}
@Override
public List getAllCharSequence(CharSequence name) {
return this.underlyingHeaders.getAll(name);
}
@Override
public List> getAllCharSequence() {
List> result = new LinkedList<>();
this.underlyingHeaders.forEach(e -> {
result.add(Map.entry(e.getKey(), e.getValue()));
});
return result;
}
}