selenium之多线程启动grid分布式测试框架封装(四)

package com.lingfeng.utils;

import java.net.MalformedURLException;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Set;

import org.dom4j.DocumentException;
import org.openqa.selenium.WebDriver;

import com.lingfeng.staticPage.RemoteServerOpenHomepage;
import com.lingfeng.utils.RemoteServerInit;

/**
 * 此类提供启动所有远程服务端浏览器的方法
 * @author 凌风
 *
 */
public class LaunchAllRemoteBrowsers {
	private static String driverName;
	/**
	 * 此方法为浏览器启动入口，启用了此方法就意味着调起了所有远程服务端的浏览器
	 * @return Map<String,Thread> 包含浏览器名称和所运行的所有线程。key为浏览器名称，value是运行的线程。
	 */
	public static Map<String,Thread> launch(){

		//存储所有正在运行的浏览器名称和线程
		Map<String,Thread> threadMap = new HashMap<String, Thread>();
		
		try {
			//初始化浏览器配置信息
			RemoteServerInit.launchBrowser();
			//获取初始化后的所有浏览器启动的driver
			Map<String,WebDriver>  driverMap = RemoteServerInit.getAllLaunchDriverMap();
			//获取启动浏览器的driver以及name
			Set<Entry<String, WebDriver>> driverInfoSet = driverMap.entrySet();
						
			Iterator<Entry<String, WebDriver>> driverIterator = driverInfoSet.iterator();
			while(driverIterator.hasNext()){				
				Thread currentThread = launchThread(driverIterator);
				threadMap.put(driverName, currentThread);			
			}
			return threadMap;
		} catch (MalformedURLException e) {
			e.printStackTrace();
		} catch (DocumentException e) {
			e.printStackTrace();
		}
		return threadMap;
	}

	/**
	 * 多线程调用，启动一个线程
	 * @param driverIterator
	 * @return Thread 当前启动的线程
	 */
	private static Thread launchThread(Iterator<Entry<String, WebDriver>> driverIterator) {
				
		Entry<String, WebDriver> driverEntry = driverIterator.next();
		driverName = driverEntry.getKey();
		WebDriver webdriver = driverEntry.getValue();
		RemoteServerOpenHomepage homepage= new RemoteServerOpenHomepage(webdriver,driverName);
		Thread t = new Thread(homepage);
		t.start();
		return t;
		
	}
}

package com.lingfeng.start;

import java.util.Iterator;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Set;

import com.lingfeng.utils.LaunchAllRemoteBrowsers;
import com.lingfeng.utils.RemoteThreadStatusMonitor;

public class StartTest {
	

	
	public static void start(){
		
		Map<String, Thread> allRunningThreads = LaunchAllRemoteBrowsers.launch();
		Set<Entry<String,Thread>> RunningThreadsSet = allRunningThreads.entrySet();
		Iterator<Entry<String,Thread>> it = RunningThreadsSet.iterator();

		int threadCount=0; //记录启动线程个数		
		while(it.hasNext() ){
			threadCount++;
			Entry<String, Thread> entry = it.next();
			//启动监控器线程，监控运行线程的状态。
			RemoteThreadStatusMonitor c = new RemoteThreadStatusMonitor(entry.getKey(),entry.getValue());
			Thread t = new Thread(c);
			t.start();
		}
		//消亡线程个数等于了启动线程的个数则jvm退出。
		while(RemoteThreadStatusMonitor.getDeadThreadCount()!=threadCount){
			//必须加此线程休眠方法，不然主这个线程结束不了。  
			try {
				Thread.sleep(1000);
			} catch (InterruptedException e) {
				e.printStackTrace();
			}					
		}   		
	}
	}

		//消亡线程个数等于了启动线程的个数则jvm退出。
		while(RemoteThreadStatusMonitor.getDeadThreadCount()!=threadCount){
			//必须加此线程休眠方法，不然主这个线程结束不了。  
			try {
				Thread.sleep(1000);
			} catch (InterruptedException e) {
				e.printStackTrace();
			}					
		}   

package com.lingfeng.junit;

import org.junit.Test;

import com.lingfeng.start.StartTest;

public class JunitTest {
	@Test
	public void testStart(){
		StartTest.start();
	}
	

}