io.muserver.rest.JaxRSResponse Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of mu-server Show documentation
Show all versions of mu-server Show documentation
A simple but powerful web server framework
package io.muserver.rest;
import io.muserver.HeaderNames;
import io.muserver.Headers;
import javax.ws.rs.core.*;
import java.lang.annotation.Annotation;
import java.net.URI;
import java.util.*;
import java.util.stream.Stream;
import static java.util.Arrays.asList;
import static java.util.stream.Collectors.toMap;
class JaxRSResponse extends Response {
static {
MuRuntimeDelegate.ensureSet();
}
private final Headers headers;
private final StatusType status;
private final Object entity;
private final MediaType type;
private final NewCookie[] cookies;
JaxRSResponse(StatusType status, Headers headers, Object entity, MediaType type, NewCookie[] cookies) {
this.status = status;
this.headers = headers;
this.entity = entity;
this.type = type;
this.cookies = cookies;
}
@Override
public int getStatus() {
return status.getStatusCode();
}
@Override
public StatusType getStatusInfo() {
return status;
}
@Override
public Object getEntity() {
return entity;
}
@Override
public T readEntity(Class entityType) {
throw NotImplementedException.willNot();
}
@Override
public T readEntity(GenericType entityType) {
throw NotImplementedException.willNot();
}
@Override
public T readEntity(Class entityType, Annotation[] annotations) {
throw NotImplementedException.willNot();
}
@Override
public T readEntity(GenericType entityType, Annotation[] annotations) {
throw NotImplementedException.willNot();
}
@Override
public boolean hasEntity() {
return entity != null;
}
@Override
public boolean bufferEntity() {
throw NotImplementedException.willNot();
}
@Override
public void close() {
throw NotImplementedException.notYet();
}
@Override
public MediaType getMediaType() {
return type;
}
@Override
public Locale getLanguage() {
throw NotImplementedException.notYet();
}
@Override
public int getLength() {
throw NotImplementedException.notYet();
}
@Override
public Set getAllowedMethods() {
throw NotImplementedException.notYet();
}
@Override
public Map getCookies() {
return Stream.of(cookies).collect(toMap(Cookie::getName, c -> c));
}
@Override
public EntityTag getEntityTag() {
throw NotImplementedException.notYet();
}
@Override
public Date getDate() {
throw NotImplementedException.notYet();
}
@Override
public Date getLastModified() {
throw NotImplementedException.notYet();
}
@Override
public URI getLocation() {
throw NotImplementedException.notYet();
}
@Override
public Set getLinks() {
throw NotImplementedException.notYet();
}
@Override
public boolean hasLink(String relation) {
throw NotImplementedException.notYet();
}
@Override
public Link getLink(String relation) {
throw NotImplementedException.notYet();
}
@Override
public Link.Builder getLinkBuilder(String relation) {
throw NotImplementedException.notYet();
}
@Override
public MultivaluedMap getMetadata() {
MultivaluedMap map = new MultivaluedHashMap<>();
for (String name : headers.names()) {
map.addAll(name, headers.getAll(name));
}
return map;
}
Headers getMuHeaders() {
return this.headers;
}
@Override
public MultivaluedMap getStringHeaders() {
return muHeadersToJax(headers);
}
static MultivaluedMap muHeadersToJax(Headers headers) {
MultivaluedMap map = new MultivaluedHashMap<>();
for (String name : headers.names()) {
map.addAll(name, headers.getAll(name));
}
return map;
}
static MultivaluedMap muHeadersToJaxObj(Headers headers) {
MultivaluedMap map = new MultivaluedHashMap<>();
for (String name : headers.names()) {
map.addAll(name, headers.getAll(name));
}
return map;
}
@Override
public String getHeaderString(String name) {
return headers.get(name);
}
public static class Builder extends Response.ResponseBuilder {
static {
MuRuntimeDelegate.ensureSet();
}
private final Headers headers = new Headers();
private final List linkHeaders = new ArrayList<>();
private StatusType status;
private Object entity;
private Annotation[] annotations;
private NewCookie[] cookies = new NewCookie[0];
private MediaType type;
private List variants = new ArrayList<>();
@Override
public Response build() {
for (Link linkHeader : linkHeaders) {
headers.add(HeaderNames.LINK, linkHeader.toString());
}
MediaType typeToUse = this.type;
if (typeToUse == null) {
String ct = headers.get(HeaderNames.CONTENT_TYPE);
if (ct != null) {
typeToUse = MediaType.valueOf(ct);
}
} else {
headers.set(HeaderNames.CONTENT_TYPE, typeToUse.toString());
}
return new JaxRSResponse(status, headers, entity, typeToUse, cookies);
}
@Override
public ResponseBuilder clone() {
throw NotImplementedException.notYet();
}
@Override
public ResponseBuilder status(int code) {
return status(code, null);
}
@Override
public ResponseBuilder status(int code, String reasonPhrase) {
if (code < 100 || code > 599) {
throw new IllegalArgumentException("Status must be between 100 and 599, but was " + code);
}
this.status = Status.fromStatusCode(code);
if (this.status == null || reasonPhrase != null) {
this.status = new CustomStatus(Status.Family.familyOf(code), code, reasonPhrase);
}
return this;
}
@Override
public ResponseBuilder entity(Object entity) {
this.entity = entity;
this.annotations = null;
return this;
}
@Override
public ResponseBuilder entity(Object entity, Annotation[] annotations) {
this.entity = entity;
this.annotations = annotations;
return this;
}
@Override
public ResponseBuilder allow(String... methods) {
if (methods == null || (methods.length == 1 && methods[0] == null)) {
return allow((Set) null);
} else {
return allow(new HashSet<>(Arrays.asList(methods)));
}
}
@Override
public ResponseBuilder allow(Set methods) {
if (methods == null) {
return header(HttpHeaders.ALLOW, null);
}
StringBuilder allow = new StringBuilder();
for (String m : methods) {
append(allow, true, m);
}
return header(HttpHeaders.ALLOW, allow.toString());
}
private void append(StringBuilder sb, boolean v, String s) {
if (v) {
if (sb.length() > 0) {
sb.append(',');
}
sb.append(s);
}
}
@Override
public ResponseBuilder cacheControl(CacheControl cacheControl) {
return header(HeaderNames.CACHE_CONTROL, cacheControl.toString());
}
@Override
public ResponseBuilder encoding(String encoding) {
return header(HeaderNames.CONTENT_ENCODING, encoding);
}
private ResponseBuilder header(CharSequence name, Object value) {
if (value instanceof Iterable) {
((Iterable)value).forEach(v -> header(name, v));
} else {
if (value == null) {
headers.remove(name);
} else {
headers.add(name, value);
}
}
return this;
}
@Override
public ResponseBuilder header(String name, Object value) {
return header((CharSequence) name, value);
}
@Override
public ResponseBuilder replaceAll(MultivaluedMap headers) {
this.headers.clear();
for (Map.Entry> entry : headers.entrySet()) {
this.headers.add(entry.getKey(), entry.getValue());
}
return this;
}
@Override
public ResponseBuilder language(String language) {
return header(HeaderNames.CONTENT_LANGUAGE, language);
}
@Override
public ResponseBuilder language(Locale language) {
return language(language.toLanguageTag());
}
@Override
public ResponseBuilder type(MediaType type) {
this.type = type;
return this;
}
@Override
public ResponseBuilder type(String type) {
if (type == null) {
this.type = null;
return this;
}
return type(MediaType.valueOf(type));
}
@Override
public ResponseBuilder variant(Variant variant) {
this.variants.add(variant);
return this;
}
@Override
public ResponseBuilder contentLocation(URI location) {
return header(HeaderNames.CONTENT_LOCATION, location);
}
@Override
public ResponseBuilder cookie(NewCookie... cookies) {
this.cookies = cookies;
return this;
}
@Override
public ResponseBuilder expires(Date expires) {
return header(HeaderNames.EXPIRES, expires);
}
@Override
public ResponseBuilder lastModified(Date lastModified) {
return header(HeaderNames.LAST_MODIFIED, lastModified);
}
@Override
public ResponseBuilder location(URI location) {
return header(HeaderNames.LOCATION, location);
}
@Override
public ResponseBuilder tag(EntityTag tag) {
return tag(tag.toString());
}
@Override
public ResponseBuilder tag(String tag) {
return header(HeaderNames.ETAG, tag);
}
@Override
public ResponseBuilder variants(Variant... variants) {
return variants(asList(variants));
}
@Override
public ResponseBuilder variants(List variants) {
if (variants == null) {
this.variants.clear();
} else {
this.variants.addAll(variants);
}
return this;
}
@Override
public ResponseBuilder links(Link... links) {
if (links == null) {
linkHeaders.clear();
} else {
linkHeaders.addAll(asList(links));
}
return this;
}
@Override
public ResponseBuilder link(URI uri, String rel) {
Link link = Link.fromUri(uri).rel(rel).build();
linkHeaders.add(link);
return this;
}
@Override
public ResponseBuilder link(String uri, String rel) {
return link(URI.create(uri), rel);
}
}
private static class CustomStatus implements StatusType {
private final String reason;
private final Status.Family family;
private final int code;
private CustomStatus(Status.Family family, int code, String reason) {
this.reason = reason;
this.family = family;
this.code = code;
}
@Override
public int getStatusCode() {
return code;
}
@Override
public Status.Family getFamily() {
return family;
}
@Override
public String getReasonPhrase() {
return reason;
}
}
}