edu.harvard.i2b2.fr.ws.FileRepositoryService.java Source code

Introduction

Here is the source code for edu.harvard.i2b2.fr.ws.FileRepositoryService.java

Source

/*
 * Copyright (c) 2006-2012 Massachusetts General Hospital 
 * All rights reserved. This program and the accompanying materials 
 * are made available under the terms of the i2b2 Software License v2.1 
 * which accompanies this distribution. 
 * 
 * Contributors: 
 *     Rajesh Kuttan
 *     Mike Mendis
 *     Bill Wang
 */
package edu.harvard.i2b2.fr.ws;

import java.io.StringReader;

import javax.activation.DataHandler;
import javax.activation.FileDataSource;
import javax.xml.stream.XMLInputFactory;
import javax.xml.stream.XMLStreamException;
import javax.xml.stream.XMLStreamReader;

import org.apache.axiom.om.OMAbstractFactory;
import org.apache.axiom.om.OMElement;
import org.apache.axiom.om.OMFactory;
import org.apache.axiom.om.OMNamespace;
import org.apache.axiom.om.impl.builder.StAXOMBuilder;
import org.apache.axis2.context.MessageContext;
import org.apache.axis2.context.OperationContext;
import org.apache.axis2.wsdl.WSDLConstants;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

import edu.harvard.i2b2.common.exception.I2B2Exception;
import edu.harvard.i2b2.fr.delegate.RecvfileRequestHandler;
import edu.harvard.i2b2.fr.delegate.LoaderQueryRequestDelegate;
import edu.harvard.i2b2.fr.delegate.SendfileRequestHandler;

public class FileRepositoryService {
    protected final Log log = LogFactory.getLog(getClass());

    public OMElement sendfileRequest(OMElement request) {
        LoaderQueryRequestDelegate queryDelegate = new LoaderQueryRequestDelegate();
        OMElement responseElement = null;
        try {
            String requestXml = request.toString();
            if (requestXml.indexOf("<soapenv:Body>") > -1) {
                requestXml = requestXml.substring(requestXml.indexOf("<soapenv:Body>") + 14,
                        requestXml.indexOf("</soapenv:Body>"));
            }

            SendfileRequestHandler handler = new SendfileRequestHandler(requestXml);
            String response = queryDelegate.handleRequest(requestXml, handler);

            responseElement = buildOMElementFromString(response, "");

        } catch (XMLStreamException e) {
            log.error("xml stream exception", e);
        } catch (I2B2Exception e) {
            log.error("i2b2 exception", e);
        } catch (Throwable e) {
            log.error("Throwable", e);
        }
        return responseElement;
    }

    public OMElement recvfileRequest(OMElement request) {
        LoaderQueryRequestDelegate queryDelegate = new LoaderQueryRequestDelegate();
        OMElement responseElement = null;

        FileDataSource fileDataSource;
        DataHandler fileDataHandler;

        try {
            String requestXml = request.toString();
            if (requestXml.indexOf("<soapenv:Body>") > -1) {
                requestXml = requestXml.substring(requestXml.indexOf("<soapenv:Body>") + 14,
                        requestXml.indexOf("</soapenv:Body>"));
            }
            RecvfileRequestHandler handler = new RecvfileRequestHandler(requestXml);
            String response = queryDelegate.handleRequest(requestXml, handler);

            String filename = handler.getFilename();
            // We can obtain the request (incoming) MessageContext as follows
            MessageContext inMessageContext = MessageContext.getCurrentMessageContext();
            // We can obtain the operation context from the request message
            // context
            OperationContext operationContext = inMessageContext.getOperationContext();
            MessageContext outMessageContext = operationContext
                    .getMessageContext(WSDLConstants.MESSAGE_LABEL_OUT_VALUE);
            outMessageContext.setDoingSwA(true);
            outMessageContext.setDoingREST(false);
            if (!filename.equals("")) {
                fileDataSource = new FileDataSource(filename);
                fileDataHandler = new DataHandler(fileDataSource);
                // use requested filename as content ID
                outMessageContext.addAttachment(handler.getRequestedFilename(), fileDataHandler);
                outMessageContext.setDoingMTOM(false);
                outMessageContext.setDoingSwA(true);
                responseElement = buildOMElementFromString(response, fileDataHandler.getName());
            } else {
                log.error("where did the file go? ");
            }
        } catch (XMLStreamException e) {
            log.error("xml stream exception", e);
        } catch (I2B2Exception e) {
            log.error("i2b2 exception", e);
        } catch (Throwable e) {
            log.error("Throwable", e);
        }
        return responseElement;
    }

    private OMElement buildOMElementFromString(String xmlString, String contentID) throws XMLStreamException {
        OMElement returnElement = null;

        try {
            StringReader strReader = new StringReader(xmlString);
            XMLInputFactory xif = XMLInputFactory.newInstance();
            XMLStreamReader reader = xif.createXMLStreamReader(strReader);

            StAXOMBuilder builder = new StAXOMBuilder(reader);
            returnElement = builder.getDocumentElement();

            OMFactory factory = OMAbstractFactory.getOMFactory();
            OMNamespace omNs = factory.createOMNamespace("http://www.i2b2.org/xsd", "swa");

            OMElement fileElement = factory.createOMElement("file", omNs, returnElement);
            fileElement.addAttribute("href", contentID, null);
        } catch (XMLStreamException ex) {
            log.error("Error while converting FR response PDO to OMElement", ex);
            throw ex;
        }

        return returnElement;
    }

    public OMElement getFile(OMElement getFileElement) throws I2B2Exception {
        return recvfileRequest(getFileElement);
    }

    public OMElement putFile(OMElement putFileElement) throws I2B2Exception {
        return sendfileRequest(putFileElement);
    }
}