com.ksc.http.StaxResponseHandler Maven / Gradle / Ivy
The newest version!
/*
* Copyright 2010-2016 ksyun.com, Inc. or its affiliates. All Rights Reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License").
* You may not use this file except in compliance with the License.
* A copy of the License is located at
*
* http://ksyun.com/apache2.0
*
* or in the "license" file accompanying this file. This file 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 com.ksc.http;
import java.io.ByteArrayInputStream;
import java.io.InputStream;
import java.util.Map;
import javax.xml.stream.XMLEventReader;
import javax.xml.stream.XMLInputFactory;
import javax.xml.stream.XMLStreamException;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import com.ksc.KscWebServiceResponse;
import com.ksc.ResponseMetadata;
import com.ksc.transform.StaxUnmarshallerContext;
import com.ksc.transform.Unmarshaller;
import com.ksc.transform.VoidStaxUnmarshaller;
import com.ksc.util.StringUtils;
/**
* Default implementation of HttpResponseHandler that handles a successful
* response from an KSC service and unmarshalls the result using a StAX
* unmarshaller.
*
* @param
* Indicates the type being unmarshalled by this response handler.
*/
// TODOFIX
public class StaxResponseHandler implements HttpResponseHandler> {
/** The StAX unmarshaller to use when handling the response */
private Unmarshaller responseUnmarshaller;
/** Shared logger for profiling information */
private static final Log log = LogFactory.getLog("com.ksc.request");
/** Shared factory for creating XML event readers */
private static final XMLInputFactory xmlInputFactory = XMLInputFactory.newInstance();
/**
* Constructs a new response handler that will use the specified StAX
* unmarshaller to unmarshall the service response and uses the specified
* response element path to find the root of the business data in the
* service's response.
*
* @param responseUnmarshaller
* The StAX unmarshaller to use on the response.
*/
public StaxResponseHandler(Unmarshaller responseUnmarshaller) {
this.responseUnmarshaller = responseUnmarshaller;
/*
* Even if the invoked operation just returns null, we still need an
* unmarshaller to run so we can pull out response metadata.
*
* We might want to pass this in through the client class so that we
* don't have to do this check here.
*/
if (this.responseUnmarshaller == null) {
this.responseUnmarshaller = new VoidStaxUnmarshaller();
}
}
/**
* @see com.ksc.http.HttpResponseHandler#handle(com.ksc.http.HttpResponse)
*/
public KscWebServiceResponse handle(HttpResponse response) throws Exception {
log.trace("Parsing service response XML");
InputStream content = response.getContent();
if (content == null) {
content = new ByteArrayInputStream("".getBytes(StringUtils.UTF8));
}
XMLEventReader eventReader;
synchronized (xmlInputFactory) {
eventReader = xmlInputFactory.createXMLEventReader(content);
}
try {
KscWebServiceResponse kscResponse = new KscWebServiceResponse();
StaxUnmarshallerContext unmarshallerContext = new StaxUnmarshallerContext(eventReader, response.getHeaders());
unmarshallerContext.registerMetadataExpression("ResponseMetadata/RequestId", 2, ResponseMetadata.KSC_REQUEST_ID);
unmarshallerContext.registerMetadataExpression("requestId", 2, ResponseMetadata.KSC_REQUEST_ID);
registerAdditionalMetadataExpressions(unmarshallerContext);
T result = responseUnmarshaller.unmarshall(unmarshallerContext);
kscResponse.setResult(result);
Map metadata = unmarshallerContext.getMetadata();
Map responseHeaders = response.getHeaders();
if (responseHeaders != null) {
if (responseHeaders.get(X_KSC_REQUEST_ID_HEADER) != null) {
metadata.put(ResponseMetadata.KSC_REQUEST_ID,
responseHeaders.get(X_KSC_REQUEST_ID_HEADER));
}
}
kscResponse.setResponseMetadata(new ResponseMetadata(metadata));
log.trace("Done parsing service response");
return kscResponse;
} finally {
try {
eventReader.close();
} catch (XMLStreamException e) {
log.warn("Error closing xml parser", e);
}
}
}
/**
* Hook for subclasses to override in order to collect additional metadata
* from service responses.
*
* @param unmarshallerContext
* The unmarshaller context used to configure a service's response
* data.
*/
protected void registerAdditionalMetadataExpressions(StaxUnmarshallerContext unmarshallerContext) {}
/**
* Since this response handler completely consumes all the data from the
* underlying HTTP connection during the handle method, we don't need to
* keep the HTTP connection open.
*
* @see com.ksc.http.HttpResponseHandler#needsConnectionLeftOpen()
*/
public boolean needsConnectionLeftOpen() {
return false;
}
}