import com.google.gson.JsonObject;

import java.util.ArrayList;
import java.util.List;

public class TestClientPool {
 protected TestClientRunnable testClientRunnable;
 protected List<Thread> threadList;
 protected volatile boolean running;
 protected volatile boolean init;
 protected String url;
 protected RequestMethod method;
 protected long waitTime;
 protected JsonObject data;
 protected List<TestClientRunnable> runnableList ;
 protected int count;
 protected long totalTime;
 protected int failCount;
 protected long longestTime = 0;
 protected long shortestTime = 0x7fffffffffffffffL;
 public TestClientPool(String url, RequestMethod method, long waitTime){
 this(url, method, waitTime, new JsonObject());
 }
 public TestClientPool(String url, RequestMethod method, long waitTime, JsonObject data){
 this.url = url;
 this.method = method;
 this.waitTime = waitTime;
 this.data = data;
 running = false;
 }

 public boolean init(int threadCount){
 if (running){
 return false;
 }
 threadList = new ArrayList<>();
 runnableList = new ArrayList<>();
 for(int i = 0 ; i < threadCount ; ++i){
 runnableList.add(new TestClientRunnable(url, method, waitTime, data));
 threadList.add(new Thread(runnableList.get(i), "WebTesterThread-" + i));
 }
 init = true;
 return true;
 }

 public boolean start(){
 if (running || !init){
 return false;
 }
 count = 0;
 long wait = waitTime/threadList.size();
 for(int i = 0 ; i != threadList.size() ; ++i){
 threadList.get(i).setDaemon(true);
 threadList.get(i).start();
 try {
 Thread.sleep(wait);
 }
 catch (InterruptedException e){

 }
 }
 running = true;
 return true;
 }

 public boolean stop(){
 if (!running || !init){
 return false;
 }
 for(int i = 0 ; i != threadList.size() ; ++i){
 threadList.get(i).interrupt();
 count += runnableList.get(i).getCount();
 failCount += runnableList.get(i).getFailCount();
 totalTime += runnableList.get(i).getTotalTime();
 longestTime = runnableList.get(i).getLongestTime() > longestTime ? runnableList.get(i).getLongestTime() : longestTime;
 shortestTime = runnableList.get(i).getShortestTime() < shortestTime ? runnableList.get(i).getShortestTime() : shortestTime;
 }
 threadList = null;
 running = false;
 init = false;
 return true;
 }

 public int getCount(){
 return count;
 }

 public int getFailCount(){
 return failCount;
 }

 public long getTotalTime(){
 return totalTime;
 }

 public long getShortestTime() {
 return shortestTime;
 }

 public long getLongestTime() {
 return longestTime;
 }
}