com.mirth.connect.connectors.dimse.DICOMMessageReceiver.java Source code

Introduction

Here is the source code for com.mirth.connect.connectors.dimse.DICOMMessageReceiver.java

Source

/*
 * Copyright (c) Mirth Corporation. All rights reserved.
 * http://www.mirthcorp.com
 *
 * The software in this package is published under the terms of the MPL
 * license a copy of which has been included with this distribution in
 * the LICENSE.txt file.
 */

package com.mirth.connect.connectors.dimse;

import java.net.URI;

import org.apache.commons.lang.StringUtils;
import org.apache.log4j.Logger;
import org.dcm4che2.data.UID;
import org.dcm4che2.tool.dcmrcv.DcmRcv;
import org.dcm4che2.tool.dcmrcv.MirthDcmRcv;
import org.mule.config.i18n.Message;
import org.mule.providers.AbstractMessageReceiver;
import org.mule.providers.ConnectException;
import org.mule.umo.UMOComponent;
import org.mule.umo.endpoint.UMOEndpoint;
import org.mule.umo.lifecycle.InitialisationException;
import org.mule.umo.provider.UMOConnector;

import com.mirth.connect.server.controllers.ControllerFactory;
import com.mirth.connect.server.controllers.MonitoringController;
import com.mirth.connect.server.controllers.MonitoringController.ConnectorType;
import com.mirth.connect.server.controllers.MonitoringController.Event;
import com.mirth.connect.server.mule.transformers.JavaScriptPostprocessor;

public class DICOMMessageReceiver extends AbstractMessageReceiver {
    private Logger logger = Logger.getLogger(this.getClass());
    protected DICOMConnector connector;
    private MonitoringController monitoringController = ControllerFactory.getFactory().createMonitoringController();
    private JavaScriptPostprocessor postProcessor = new JavaScriptPostprocessor();
    private ConnectorType connectorType = ConnectorType.LISTENER;
    private DcmRcv dcmrcv;

    public DICOMMessageReceiver(UMOConnector connector, UMOComponent component, UMOEndpoint endpoint)
            throws InitialisationException {
        super(connector, component, endpoint);
        this.connector = (DICOMConnector) connector;
        this.dcmrcv = new MirthDcmRcv(this, postProcessor, endpoint);
    }

    @Override
    public void doConnect() throws ConnectException {
        disposing.set(false);
        URI uri = endpoint.getEndpointURI().getUri();

        try {
            dcmrcv.setPort(uri.getPort());
            dcmrcv.setHostname(uri.getHost());
            dcmrcv.setAEtitle("DCMRCV");

            String[] only_def_ts = { UID.ImplicitVRLittleEndian };
            String[] native_le_ts = { UID.ImplicitVRLittleEndian };
            String[] native_ts = { UID.ImplicitVRLittleEndian };
            String[] non_retired_ts = { UID.ImplicitVRLittleEndian };

            if (StringUtils.isNotBlank(connector.getDest())) {
                dcmrcv.setDestination(connector.getDest());
            }

            if (connector.isDefts()) {
                dcmrcv.setTransferSyntax(only_def_ts);
            } else if (connector.isNativeData()) {
                if (connector.isBigendian()) {
                    dcmrcv.setTransferSyntax(native_ts);
                } else {
                    dcmrcv.setTransferSyntax(native_le_ts);
                }
            } else if (connector.isBigendian()) {
                dcmrcv.setTransferSyntax(non_retired_ts);
            }

            if (StringUtils.isNotBlank(connector.getApplicationEntity())) {
                dcmrcv.setAEtitle(connector.getApplicationEntity());
            }

            if (connector.getReaper() != 10) {
                dcmrcv.setAssociationReaperPeriod(connector.getReaper());
            }

            if (connector.getIdleto() != 60) {
                dcmrcv.setIdleTimeout(connector.getIdleto());
            }

            if (connector.getRequestto() != 5) {
                dcmrcv.setRequestTimeout(connector.getRequestto());
            }

            if (connector.getReleaseto() != 5) {
                dcmrcv.setReleaseTimeout(connector.getReleaseto());
            }

            if (connector.getSoclosedelay() != 50) {
                dcmrcv.setSocketCloseDelay(connector.getSoclosedelay());
            }

            if (connector.getRspdelay() > 0) {
                dcmrcv.setDimseRspDelay(connector.getRspdelay());
            }

            if (connector.getRcvpdulen() != 16) {
                dcmrcv.setMaxPDULengthReceive(connector.getRcvpdulen());
            }

            if (connector.getSndpdulen() != 16) {
                dcmrcv.setMaxPDULengthSend(connector.getSndpdulen());
            }

            if (connector.getSosndbuf() > 0) {
                dcmrcv.setSendBufferSize(connector.getSosndbuf());
            }

            if (connector.getSorcvbuf() > 0) {
                dcmrcv.setReceiveBufferSize(connector.getSorcvbuf());
            }

            if (connector.getBufsize() != 1) {
                dcmrcv.setFileBufferSize(connector.getBufsize());
            }

            dcmrcv.setPackPDV(connector.isPdv1());
            dcmrcv.setTcpNoDelay(!connector.isTcpdelay());

            if (connector.getAsync() > 0) {
                dcmrcv.setMaxOpsPerformed(connector.getAsync());
            }

            dcmrcv.initTransferCapability();

            // connection tls settings

            if (!StringUtils.equals(connector.getTls(), "notls")) {
                if (connector.getTls().equals("without")) {
                    dcmrcv.setTlsWithoutEncyrption();
                } else if (connector.getTls().equals("3des")) {
                    dcmrcv.setTls3DES_EDE_CBC();
                } else if (connector.getTls().equals("aes")) {
                    dcmrcv.setTlsAES_128_CBC();
                }

                if (StringUtils.isNotBlank(connector.getTruststore())) {
                    dcmrcv.setTrustStoreURL(connector.getTruststore());
                }

                if (StringUtils.isNotBlank(connector.getTruststorepw())) {
                    dcmrcv.setTrustStorePassword(connector.getTruststorepw());
                }

                if (StringUtils.isNotBlank(connector.getKeypw())) {
                    dcmrcv.setKeyPassword(connector.getKeypw());
                }

                if (StringUtils.isNotBlank(connector.getKeystore())) {
                    dcmrcv.setKeyStoreURL(connector.getKeystore());
                }

                if (StringUtils.isNotBlank(connector.getKeystorepw())) {
                    dcmrcv.setKeyStorePassword(connector.getKeystorepw());
                }

                dcmrcv.setTlsNeedClientAuth(connector.isNoclientauth());

                if (!connector.isNossl2()) {
                    dcmrcv.setTlsProtocol(new String[] { "TLSv1", "SSLv3" });
                }

                dcmrcv.initTLS();
            }

            // start the DICOM port
            dcmrcv.start();

            monitoringController.updateStatus(this.connector, connectorType, Event.INITIALIZED);
        } catch (Exception e) {
            throw new ConnectException(new Message("DICOM", 1, uri), e, this);
        }
    }

    @Override
    public void doDisconnect() throws ConnectException {
        // this will cause the server thread to quit
    }

    @Override
    public void doDispose() {
        disposing.set(true);

        try {
            dcmrcv.stop();
        } catch (Exception e) {
            logger.error("Unable to close DICOM port.", e);
        } finally {
            monitoringController.updateStatus(connector, connectorType, Event.DISCONNECTED);
        }

        logger.debug("closed DICOM port");
    }
}