org.apache.wink.client.internal.handlers.ClientResponseImpl Maven / Gradle / Ivy
/*******************************************************************************
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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 org.apache.wink.client.internal.handlers;
import java.io.IOException;
import java.io.InputStream;
import java.lang.reflect.Type;
import javax.ws.rs.WebApplicationException;
import javax.ws.rs.core.HttpHeaders;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.Response;
import javax.ws.rs.core.Response.StatusType;
import javax.ws.rs.core.Response.Status.Family;
import javax.ws.rs.ext.MessageBodyReader;
import org.apache.wink.client.ClientResponse;
import org.apache.wink.client.ClientRuntimeException;
import org.apache.wink.client.EntityType;
import org.apache.wink.client.internal.ClientRuntimeContext;
import org.apache.wink.common.RuntimeContext;
import org.apache.wink.common.internal.i18n.Messages;
import org.apache.wink.common.internal.registry.ProvidersRegistry;
import org.apache.wink.common.internal.runtime.RuntimeContextTLS;
public class ClientResponseImpl extends BaseRequestResponseImpl implements ClientResponse {
private Object entity;
private String message;
private int status;
private Runnable contentConsumer;
public T getEntity(Class type) {
return getEntity(type, type);
}
public T getEntity(EntityType entityType) {
return getEntity(entityType.getRawClass(), entityType.getType());
}
@SuppressWarnings("unchecked")
private T getEntity(Class type, Type genericType) {
if (type.isInstance(entity)) {
return (T)entity;
}
if (entity instanceof InputStream) {
T t = readEntity(type, genericType, (InputStream)entity);
setEntity(t);
return t;
}
if (entity == null) {
return null;
}
throw new ClassCastException(Messages.getMessage("clientCannotConvertEntity", //$NON-NLS-1$
entity.getClass().getName(),
type.getName()));
}
public String getMessage() {
return message;
}
public int getStatusCode() {
return status;
}
public void setEntity(Object entity) {
this.entity = entity;
}
public void setMessage(String message) {
this.message = message;
}
public void setStatusCode(int code) {
this.status = code;
}
@SuppressWarnings("unchecked")
private T readEntity(Class type, Type genericType, InputStream is) {
if (type == null) {
return null;
}
if (ClientResponse.class.equals(type)) {
return (T)this;
}
ProvidersRegistry providersRegistry = getAttribute(ProvidersRegistry.class);
RuntimeContext saved = RuntimeContextTLS.getRuntimeContext();
ClientRuntimeContext runtimeContext = new ClientRuntimeContext(providersRegistry);
RuntimeContextTLS.setRuntimeContext(runtimeContext);
try {
String contentType = getHeaders().getFirst(HttpHeaders.CONTENT_TYPE);
if (contentType == null || contentType.length() == 0) {
contentType = MediaType.APPLICATION_OCTET_STREAM;
}
MediaType contentMediaType = MediaType.valueOf(contentType);
MessageBodyReader reader =
providersRegistry.getMessageBodyReader(type,
genericType,
null,
contentMediaType,
runtimeContext);
if (reader == null) {
throw new RuntimeException(Messages.getMessage("clientNoReaderForTypeAndMediaType", //$NON-NLS-1$
String.valueOf(type),
contentType));
}
T entity = reader.readFrom(type, genericType, null, contentMediaType, getHeaders(), is);
return entity;
} catch (WebApplicationException e) {
throw new ClientRuntimeException(e);
} catch (IOException e) {
throw new ClientRuntimeException(e);
} finally {
RuntimeContextTLS.setRuntimeContext(saved);
consumeContent();
}
}
public void consumeContent() {
if (contentConsumer != null) {
contentConsumer.run();
}
}
public void setContentConsumer(Runnable contentConsumer) {
this.contentConsumer = contentConsumer;
}
public Runnable getContentConsumer() {
return contentConsumer;
}
public StatusType getStatusType() {
StatusType statusType = Response.Status.fromStatusCode(status);
if (statusType == null) {
statusType = new ClientStatusType(status, message);
}
return statusType;
}
static class ClientStatusType implements StatusType {
final int statusCode;
final String reasonPhrase;
public ClientStatusType(int statusCode, String reasonPhrase) {
this.statusCode = statusCode;
this.reasonPhrase = reasonPhrase;
}
public int getStatusCode() {
return statusCode;
}
public String getReasonPhrase() {
return reasonPhrase;
}
public Family getFamily() {
int family = statusCode / 100;
if (family == 1) {
return Family.INFORMATIONAL;
} else if (family == 2) {
return Family.SUCCESSFUL;
} else if (family == 3) {
return Family.REDIRECTION;
} else if (family == 4) {
return Family.CLIENT_ERROR;
} else if (family == 5) {
return Family.SERVER_ERROR;
}
return Family.OTHER;
}
@Override
public int hashCode() {
return statusCode;
}
@Override
public boolean equals(Object obj) {
if (this == obj) {
return true;
}
if (!(obj instanceof StatusType)) {
return false;
}
StatusType other = (StatusType)obj;
if (statusCode != other.getStatusCode()) {
return false;
}
return true;
}
@Override
public String toString() {
return reasonPhrase;
}
}
}