queuetakeworker.java

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

/**
 * Created at Nov 20, 2008
 */
package com.jdev.net.data;

import java.io.InputStream;
import java.io.OutputStream;

import com.jdev.net.queue.MessageQueue;

/**
 * <p>Title: QueueTakeWorker</p>
 * <p>Description: </p>
 * @author Lawrence
 * @version 1.0
 */
public class QueueTakeWorker implements QueueWorker {

	private MessageQueue messageQueue = null;

	public QueueTakeWorker(MessageQueue queue) {
		this.messageQueue = queue;
	}

	/* (non-Javadoc)
	 * @see com.jdev.net.data.QueueWorker#run(int, com.jdev.net.data.Linkable)
	 */
	public void run(int msgType, Linkable object) throws Exception {
		InputStream bin = null;

		OutputStream outputStream = object.getOutputStream();
		if (msgType == REQUEST) {
			bin = (InputStream) messageQueue.removeReqFirst();
		} else if (msgType == RESPONSE) {
			bin = (InputStream) messageQueue.removeResFirst();

		}
		object.setInputStream(bin);
		bin.close();
	}

}