resender.java

用java实现的

/*
* LumaQQ - Java QQ Client
*
* Copyright (C) 2004 luma <[e-mail]>
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*/
package edu.tsinghua.lumaqq.qq;

import java.util.LinkedList;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

import edu.tsinghua.lumaqq.qq.events.QQEvent;
import edu.tsinghua.lumaqq.qq.packets.InPacket;
import edu.tsinghua.lumaqq.qq.packets.OutPacket;


/**
 * 这个线程执行重发任务
 *
 * @author 马若劼
 */
public class Resender extends Thread {
    // Log对象
    protected static Log log = LogFactory.getLog(Resender.class);
    // 接口类
    private IPort port;
    // 停止标志
    private volatile boolean stop;
    // 检查应答包是否到达的类
    private PacketMonitor monitor;
    // 超时队列
    private LinkedList timeoutQueue;
    
    /**
     * @param port 接口类
     */
    public Resender(IPort port) {
        this.port = port;
        stop = false;
        monitor = PacketMonitor.getInstance();
        timeoutQueue = new LinkedList();
        setName("Resender");
        setDaemon(true);
    }
    
    /**
     * 添加一个包到超时队列
     * @param packet
     */
    public synchronized void add(OutPacket packet) {
        timeoutQueue.addLast(packet);
    }
    
    /**
     * 得到超时队列的第一个包，不把它从队列中删除
     * @return 超时队列的第一个包，如果没有，返回null
     */
    public synchronized OutPacket get() {
        if(timeoutQueue.size() == 0) return null;
        return (OutPacket)timeoutQueue.getFirst();
    }
    
    /**
     * 得到超时队列的第一个包，并把它从队列中删除
     * @return 超时队列的第一个包，如果没有，返回null
     */
    public synchronized OutPacket remove() {
        if(timeoutQueue.size() == 0) return null;
        return (OutPacket)timeoutQueue.removeFirst();
    }
    
    /**
     * 删除ack对应的请求包
     * @param ack
     */
    public synchronized void remove(InPacket ack) {
        int hash = ack.hashCode();
        int size = timeoutQueue.size();
        for(int i = 0; i < size; i++) {
            OutPacket packet = (OutPacket)timeoutQueue.get(i);
            if(packet.hashCode() == hash) {
                timeoutQueue.remove(i);
                break;
            }
        }
    }
    
    /**
     * 得到下一个包的超时时间
     * @return 下一个包的超时时间，如果队列为空，返回一个固定值
     */
    private long getTimeoutLeft() {
        OutPacket packet = get();
        if(packet == null)
            return QQ.QQ_SENDQUEUE_TIMEOUT;
        else
            return packet.getTimeout() - System.currentTimeMillis();
    }
    
    /* (non-Javadoc)
     * @see java.lang.Runnable#run()
     */
    public void run() {
        log.debug("重发线程已经启动");
    	OutPacket packet = null;
        while(!isStop()) {
            synchronized(this) {
	        	long t = getTimeoutLeft();
	        	while(t > 0) {
	                // 等待t时间，然后再去查看是否有了包
	                try {
	                    this.wait(t);
	                } catch (InterruptedException e) {
	                }
	                // 如果停止标志为true，跳出
	                if(stop) break;
	                t = getTimeoutLeft();
	            }                 
            }
        	if(isStop()) break;
        	packet = remove();
        	
            // 发送
            if(packet != null && !monitor.check(packet, false)) {
                if(packet.needResend()) {
                	// 重发次数未到最大，重发
	                port.add(packet);
                } else {
                	// 触发操作超时事件
	            	fireOperationTimeOutEvent(packet);
                }   
            }
        }
        log.debug("重发线程正常退出");
    }    
    
    /**
     * @param stop The stop to set.
     */
    public synchronized void setStop(boolean stop) {
	    this.stop = stop;
	    this.notify();
    }
    
    /**
     * 触发超时事件
     * @param p
     */
    private void fireOperationTimeOutEvent(OutPacket packet) {
    	QQEvent e = null;
        switch(packet.getCommand()) {
        	case QQ.QQ_CMD_SEND_IM:
        		e = new QQEvent(packet);
        		e.type = QQEvent.QQ_OPERATION_TIMEOUT;
        		e.operation = QQ.QQ_CMD_SEND_IM;
        		break;
        	case QQ.QQ_CMD_KEEP_ALIVE:
        		// 说明连接已经丢失
        		e = new QQEvent(packet);
        		e.type = QQEvent.QQ_CONNECTION_LOST;
        		break;
        	case QQ.QQ_CMD_LOGIN:
        		// 这种情况说明登陆超时
        		e = new QQEvent(packet);
        		e.type = QQEvent.QQ_OPERATION_TIMEOUT;
        		e.operation = QQ.QQ_CMD_LOGIN;
        		break;
        	case QQ.QQ_CMD_GET_USER_INFO:
        		// 用户更新好友信息超时
        		e = new QQEvent(packet);
        		e.type = QQEvent.QQ_OPERATION_TIMEOUT;
        		e.operation = QQ.QQ_CMD_GET_USER_INFO;
        		break;
        	case QQ.QQ_CMD_MODIFY_INFO:
        	    // 更新自己的信息的超时
        	    e = new QQEvent(packet);
        		e.type = QQEvent.QQ_OPERATION_TIMEOUT;
        		e.operation = QQ.QQ_CMD_MODIFY_INFO;
        		break;
        	case QQ.QQ_CMD_DELETE_FRIEND:
        	    // 删除好友超时
        	    e = new QQEvent(packet);
        		e.type = QQEvent.QQ_OPERATION_TIMEOUT;
        		e.operation = QQ.QQ_CMD_DELETE_FRIEND;
        		break;
        	case QQ.QQ_CMD_ADD_FRIEND:
        	    // 添加好友超时
        	    e = new QQEvent(packet);
        		e.type = QQEvent.QQ_OPERATION_TIMEOUT;
        		e.operation = QQ.QQ_CMD_ADD_FRIEND;
        		break;
        	case QQ.QQ_CMD_ADD_FRIEND_AUTH:
        	    // 发送认证信息超时
        	    e = new QQEvent(packet);
        		e.type = QQEvent.QQ_OPERATION_TIMEOUT;
        		e.operation = QQ.QQ_CMD_ADD_FRIEND_AUTH;
        		break;
        	case QQ.QQ_CMD_GET_FRIEND_LIST:
        	    // 得到好友列表超时
        	    e = new QQEvent(packet);
        		e.type = QQEvent.QQ_OPERATION_TIMEOUT;
        		e.operation = QQ.QQ_CMD_GET_FRIEND_LIST;
        		break;
        	case QQ.QQ_CMD_GET_FRIEND_ONLINE:
        	    // 得到好友在线列表超时
        	    e = new QQEvent(packet);
        		e.type = QQEvent.QQ_OPERATION_TIMEOUT;
        		e.operation = QQ.QQ_CMD_GET_FRIEND_ONLINE;
        		break;
        	case QQ.QQ_CMD_REMOVE_SELF:
        		// 删除自己失败
        		e = new QQEvent(packet);
        		e.type = QQEvent.QQ_OPERATION_TIMEOUT;
        		e.operation = QQ.QQ_CMD_REMOVE_SELF;
        		break;
        	case QQ.QQ_CMD_SEARCH_USER:
        		// 搜索在线用户超时
        		e = new QQEvent(packet);
        		e.type = QQEvent.QQ_OPERATION_TIMEOUT;
        		e.operation = QQ.QQ_CMD_SEARCH_USER;
        		break;
        	case QQ.QQ_CMD_GROUP_NAME_OP:
        		// 上传分组名称超时
        		e = new QQEvent(packet);
        		e.type = QQEvent.QQ_OPERATION_TIMEOUT;
        		e.operation = QQ.QQ_CMD_GROUP_NAME_OP;
        		break;
        	case QQ.QQ_CMD_UPLOAD_GROUP_FRIEND:
        		// 上传分组名称超时
        		e = new QQEvent(packet);
        		e.type = QQEvent.QQ_OPERATION_TIMEOUT;
        		e.operation = QQ.QQ_CMD_UPLOAD_GROUP_FRIEND;
        		break;
        	case QQ.QQ_CMD_DOWNLOAD_GROUP_FRIEND:
        		// 上传分组名称超时
        		e = new QQEvent(packet);
        		e.type = QQEvent.QQ_OPERATION_TIMEOUT;
        		e.operation = QQ.QQ_CMD_DOWNLOAD_GROUP_FRIEND;
        		break;
        	case QQ.QQ_CMD_FRIEND_REMARK_OP:
        		// 好友备注信息操作超时
        		e = new QQEvent(packet);
        		e.type = QQEvent.QQ_OPERATION_TIMEOUT;
        		e.operation = QQ.QQ_CMD_FRIEND_REMARK_OP;
        		break;
        	case QQ.QQ_CMD_CLUSTER_CMD:
        		// 群操作超时
        		e = new QQEvent(packet);
        		e.type = QQEvent.QQ_OPERATION_TIMEOUT;
        		e.operation = QQ.QQ_CMD_CLUSTER_CMD;
        		break;
        }
        // 触发事件
        if(e != null)
            port.fireQQEvent(e);
    }
    
    /**
     * @return Returns the stop.
     */
    public synchronized boolean isStop() {
        return stop;
    }
}