MySQLConsistencyHelper.java

/*
 * Copyright (c) 2013, OpenCloudDB/MyCAT and/or its affiliates. All rights reserved.
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 *
 * This code is free software;Designed and Developed mainly by many Chinese
 * opensource volunteers. you can redistribute it and/or modify it under the
 * terms of the GNU General Public License version 2 only, as published by the
 * Free Software Foundation.
 *
 * This code 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
 * version 2 for more details (a copy is included in the LICENSE file that
 * accompanied this code).
 *
 * You should have received a copy of the GNU General Public License version
 * 2 along with this work; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * Any questions about this component can be directed to it's project Web address
 * https://code.google.com/p/opencloudb/.
 *
 */
package io.mycat.backend.heartbeat;

import io.mycat.server.interceptor.impl.GlobalTableUtil;
import io.mycat.sqlengine.SQLJob;
import io.mycat.sqlengine.SQLQueryResult;
import io.mycat.sqlengine.SQLQueryResultListener;

import java.util.Map;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicInteger;

import org.apache.commons.lang.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.alibaba.fastjson.JSON;

/**
 * @author digdeep@126.com
 */
public class MySQLConsistencyHelper implements SQLQueryResultListener<SQLQueryResult<Map<String, String>>> {
	private static final Logger LOGGER = LoggerFactory.getLogger(MySQLConsistencyHelper.class);
    private MySQLConsistencyChecker heartbeat;
    private volatile SQLJob sqlJob;
    private int RETRY_TIMES = 5;
    private AtomicInteger retryTime = new AtomicInteger();
    
    public MySQLConsistencyHelper(MySQLConsistencyChecker heartbeat, SQLJob sqlJob) {
    	this.heartbeat = heartbeat;
        this.sqlJob = sqlJob;
        this.retryTime.set(RETRY_TIMES);
    }
    
    public MySQLConsistencyHelper(MySQLConsistencyChecker heartbeat, 
    								SQLJob sqlJob, int retryTime) {
    	this.heartbeat = heartbeat;
        this.sqlJob = sqlJob;
        if(retryTime > 0 && retryTime < 10)
        	this.retryTime.set(retryTime);
        else
        	this.retryTime.set(RETRY_TIMES);
    }

    @Override
    public void onResult(SQLQueryResult<Map<String, String>> result) {
    	// {"dataNode":"db2","result":{"max_timestamp":"1450423751170"},"success":true}
    	// {"dataNode":"db3","result":{"count(*)":"1"},"success":true}
    	LOGGER.debug("resultresultresultresult:" + JSON.toJSONString(result));
    	Map<String, String> rowMap = null;
    	String count = null; String innerCol = null;
    	String maxTimestamp = null;
    	if(result != null)
    		rowMap = result.getResult();
    	
    	if(rowMap != null){
    		maxTimestamp = rowMap.get(GlobalTableUtil.MAX_COLUMN);
    		count = rowMap.get(GlobalTableUtil.COUNT_COLUMN);
    		innerCol = rowMap.get(GlobalTableUtil.INNER_COLUMN);
    		if((rowMap.containsKey(GlobalTableUtil.MAX_COLUMN) && StringUtils.isNotBlank(maxTimestamp))
    				|| (rowMap.containsKey(GlobalTableUtil.COUNT_COLUMN) && StringUtils.isNotBlank(count))
    				|| (rowMap.containsKey(GlobalTableUtil.INNER_COLUMN) && StringUtils.isNotBlank(innerCol))){
    			heartbeat.setResult(result);
    			return;
    		}else{
    			if(this.retryTime.get() > 0){
					try {
						TimeUnit.MICROSECONDS.sleep(10);
					} catch (InterruptedException e) {
					}
					this.retryTime.decrementAndGet();
    				this.sqlJob.run();
    				return;
				}
    			heartbeat.setResult(result);
    			return;
    		}
    	}else{
    		if(this.retryTime.get() > 0){
    			try {
    				TimeUnit.MICROSECONDS.sleep(3);
				} catch (InterruptedException e) {
				}
				this.retryTime.decrementAndGet();
				this.sqlJob.run();
				return;
			}
    		heartbeat.setResult(result);
			return;
    	}
    }

    public void close(String msg) {
        SQLJob curJob = sqlJob;
        if (curJob != null && !curJob.isFinished()) {
            curJob.teminate(msg);
            sqlJob = null;
        }
    }
    public MySQLConsistencyChecker getHeartbeat() {
        return heartbeat;
    }
	public SQLJob getSqlJob() {
		return sqlJob;
	}
	public void setSqlJob(SQLJob sqlJob) {
		this.sqlJob = sqlJob;
	}
    
//		if(count == null){
//    		LOGGER.warn(heartbeat.getCountSQL() + " execute failed in db: "
//        			+ result.getDataNode() +" during global table consistency heartbeat.");
//    	}
//    	if(maxTimestamp == null){
//    		LOGGER.warn(heartbeat.getMaxSQL() + " execute failed in db: "
//        			+ result.getDataNode() +" during global table consistency heartbeat.");
//    	}
//    	return;
//	}
}