ca.uhn.hl7v2.hoh.llp.HohLlpWriter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of hapi-hl7overhttp Show documentation
The newest version!
package ca.uhn.hl7v2.hoh.llp;

import java.io.DataOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.nio.charset.Charset;

import ca.uhn.hl7v2.hoh.api.EncodeException;
import ca.uhn.hl7v2.hoh.encoder.AbstractHl7OverHttpEncoder;
import ca.uhn.hl7v2.hoh.encoder.Hl7OverHttpRequestEncoder;
import ca.uhn.hl7v2.hoh.encoder.Hl7OverHttpResponseEncoder;
import ca.uhn.hl7v2.hoh.util.ServerRoleEnum;
import ca.uhn.hl7v2.llp.HL7Writer;
import ca.uhn.hl7v2.llp.LLPException;

class HohLlpWriter implements HL7Writer {

	private OutputStream myOutputStream;
	private Charset myPreferredCharset;
	private final Hl7OverHttpLowerLayerProtocol myProtocol;
	private Charset myCharsetForNextMessage;

	/**
	 * Constructor
	 */
	public HohLlpWriter(Hl7OverHttpLowerLayerProtocol theProtocol) {
		myProtocol = theProtocol;
	}

	/**
	 * {@inheritDoc}
	 */
	public void close() throws IOException {
		myOutputStream.close();
	}

	OutputStream getOutputStream() {
		return myOutputStream;
	}

	/**
	 * @return the preferredCharset
	 */
	public Charset getPreferredCharset() {
		return myPreferredCharset;
	}

	/**
	 * {@inheritDoc}
	 */
	@Override
	public void setOutputStream(OutputStream theOutputStream) throws IOException {
		myOutputStream = theOutputStream;
	}

	/**
	 * @param thePreferredCharset
	 *            the preferredCharset to set
	 */
	public void setPreferredCharset(Charset thePreferredCharset) {
		myPreferredCharset = thePreferredCharset;
	}

	/**
	 * {@inheritDoc}
	 */
	@Override
	public void writeMessage(String theRawMessage) throws LLPException, IOException {

		AbstractHl7OverHttpEncoder e;
		if (myProtocol.getRole() == ServerRoleEnum.CLIENT) {
			Hl7OverHttpRequestEncoder encoder = new Hl7OverHttpRequestEncoder();
			if (myProtocol.getHost() != null) {
				encoder.setHost(myProtocol.getHost());
			}
			e = encoder;
			if (myProtocol.getAuthorizationClientCallback() != null) {
				e.setUsername(myProtocol.getAuthorizationClientCallback().provideUsername(myProtocol.getUriPath()));
				e.setPassword(myProtocol.getAuthorizationClientCallback().providePassword(myProtocol.getUriPath()));
			}
		} else {
			e = new Hl7OverHttpResponseEncoder();
		}

		if (myProtocol.getRole() == ServerRoleEnum.CLIENT) {
			e.setSigner(myProtocol.getSigner());
		}
		
		e.setMessage(theRawMessage);
		if (myCharsetForNextMessage != null) {
			e.setCharset(myCharsetForNextMessage);
			myCharsetForNextMessage = null;
		} else if (getPreferredCharset() != null) {
			e.setCharset(getPreferredCharset());
		}

		e.setPath(myProtocol.getUriPath());
		DataOutputStream dos = new DataOutputStream(myOutputStream);
		try {
			e.encodeToOutputStream(dos);
		} catch (EncodeException e1) {
			throw new LLPException("Failed to encode message", e1);
		}

		dos.flush();

	}

	void setCharsetForNextMessage(Charset theCharset) {
		myCharsetForNextMessage = theCharset;
	}

}