queryrunnable.java

来自「mysql集群」· Java 代码 · 共 76 行

/**
 * 
 */
package com.meidusa.amoeba.aladdin.handler;

import java.util.concurrent.CountDownLatch;

import org.apache.log4j.Logger;

import com.meidusa.amoeba.aladdin.io.ResultPacket;
import com.meidusa.amoeba.mysql.net.MysqlClientConnection;
import com.meidusa.amoeba.net.MessageHandler;
import com.meidusa.amoeba.net.poolable.PoolableObject;

public abstract class QueryRunnable implements MessageHandlerRunner {
	private static Logger logger = Logger.getLogger(QueryRunnable.class);
	private PoolableObject conn;
	protected Object parameter;
	protected CountDownLatch latch;
	protected ResultPacket packet;
	protected String query;
	protected MysqlClientConnection source;

	public QueryRunnable(CountDownLatch latch, PoolableObject conn,
			String query, Object parameter, ResultPacket packet) {
		this.conn = conn;
		this.parameter = parameter;
		this.packet = packet;
		this.query = query;
		this.latch = latch;
	}

	public void init(MessageHandler handler) {
		this.source = ((CommandMessageHandler)handler).source;
	}

	protected static boolean isSelect(String query) {
		char ch = query.trim().charAt(0);
		if (ch == 's' || ch == 'S') {
			return true;
		} else {
			return false;
		}
	}

	/**
	 * PoolableObject 将在end session 中返回到pool中
	 * 
	 * @param conn
	 */
	protected abstract void doRun(PoolableObject conn);

	public void run() {
		try {
			try {
				doRun(conn);
			} catch (Exception e) {
				logger.error("run query error:", e);
			}
		} finally {
			if(latch != null){
				latch.countDown();
			}
			reset();
		}
	}
	
	public void reset(){
		source = null;
		query= null;
		packet = null;
		latch = null;
		parameter = null;
		conn = null;
	}
}