emailtakeworker.java

来自「短信发送」· Java 代码 · 共 47 行

/**
 * Created at Dec 4, 2008
 */
package com.jdev.app.db.queue;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.InputStream;

import com.jdev.net.data.Linkable;
import com.jdev.net.data.QueueWorker;

/**
 * <p>Title: EmailTakeWorker</p>
 * <p>Description: </p>
 * @author Lawrence
 * @version 1.0
 */
public class EmailTakeWorker implements QueueWorker {
	private EmailQueue messageQueue = null;

	/**
	 * @param messageQueue
	 */
	public EmailTakeWorker(EmailQueue messageQueue) {
		this.messageQueue = messageQueue;
	}

	/* (non-Javadoc)
	 * @see com.jdev.app.db.queue.DbQueueWorker#run(int, com.jdev.net.data.Linkable)
	 */
	public void run(int msgType, Linkable object) throws Exception {
		InputStream bin = null;
		if (msgType == REQUEST) {
			ByteArrayOutputStream out = (ByteArrayOutputStream)messageQueue.removeRecvFirst();
			bin = new ByteArrayInputStream(out.toByteArray());
		} else if (msgType == RESPONSE) {
			ByteArrayOutputStream out = (ByteArrayOutputStream)messageQueue.removeSendFirst();
			bin = new ByteArrayInputStream(out.toByteArray());
		}
		object.setInputStream(bin);
		bin.close();
	}


}