package com.meidusa.amoeba.mysql.test.jdbc;


import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.util.Properties;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit;

import com.meidusa.amoeba.util.StringUtil;

public class PerformaceTest {

	/**
	 * @param args
	 * @throws Exception 
	 */
	public static void main(String[] args) throws Exception {
		int threadCount = 1;
		int totalQuery = 1;
		int count = 1;
		if(args.length>=2){
			threadCount = Integer.parseInt(args[0]);
			totalQuery = Integer.parseInt(args[1]);
			if(args.length>2){
				count = Integer.parseInt(args[2]);
			}
		}
		final String ip = System.getProperty("ip","10.241.14.43");
		final String port = System.getProperty("port","3306");
		final String password = null;// = System.getProperty("password","sdfriend");
		final String user = System.getProperty("user","root");
		String sql = System.getProperty("sql","SELECT * FROM snda_relation_ex.t_user t where ptnum_id in (884018121,520206926)");
		if(sql.startsWith("\"")){
			sql = sql.substring(1, sql.length() -1);
		}
		sql = StringUtil.replace(sql, "\\", "");
		final String sqlext = sql;
		System.out.println("query:"+sqlext +" to ip="+ip);

		final int runcount = count;
		Properties props = new Properties(); 

		// ����failover ���ϻָ�����
		//props.put("autoReconnect", "false"); 

		//������ѯ
		//props.put("roundRobinLoadBalance", "false"); 

		props.put("user", "root"); 
		props.put("password", "hello"); 
		//props.put("password", "...."); 
		int testCount = totalQuery;
		Class.forName("com.mysql.jdbc.Driver");
		ThreadPoolExecutor executor = new ThreadPoolExecutor(threadCount, threadCount, Long.MAX_VALUE, TimeUnit.NANOSECONDS,new LinkedBlockingQueue<Runnable>());
		long start = System.currentTimeMillis();
		final ThreadLocal<Connection> threadLocal = new ThreadLocal<Connection>();
		final CountDownLatch latch = new CountDownLatch(testCount);
		for(int i=0;i<testCount;i++){
			executor.execute(new Runnable(){
				
				public void run() {
					Connection conn = threadLocal.get();
					PreparedStatement statment = null;
					ResultSet result = null;
					try{
						if(conn == null){
							conn = DriverManager.getConnection("jdbc:mysql://"+ip+":"+port+"/test?useUnicode=true&characterEncoding=utf8&useServerPrepStmts=true&autoReconnect=true&socketTimeout=100000000&&zeroDateTimeBehavior=convertToNull",user,password);
							threadLocal.set(conn);
						}
						for(int i=0;i<runcount;i++){
							try{
								statment = conn.prepareStatement(sqlext);
								//statment.setLong(1, 1109969745L);
								//statment.setLong(2, 1108391525L);
								//statment.setInt(3, 6);
								result = statment.executeQuery();
								while(result.next()){
									System.out.println(new String(result.getString("PLAYER_NAME").getBytes("ISO8859-1")));
									System.out.println(new String(result.getString("PLAYER_NAME").getBytes("ISO8859-1")));
								}
							}finally{
								if(result != null){
									try{
										result.close();
									}catch(Exception e){}
								}
								
								if(statment != null){
									try{
										statment.close();
									}catch(Exception e){}
								}
							}
						}
					}catch(Exception e){
						e.printStackTrace();
					}finally{
						/*if(conn != null){
							try{
								conn.close();
							}catch(Exception e){}
						}*/
						latch.countDown();
						System.out.println("countDown count:"+latch.getCount());
					}
					
				}
			});
		}
		latch.await();
		long end = System.currentTimeMillis();
		System.out.println("total:"+(end-start));
		executor.shutdownNow();
	}
}