refs the api of bookTest

refs the api of bookTest

Bench4Q-Agent/src/main/java/org/bench4q/agent/api/TestController.java

@@ -61,7 +61,7 @@ public class TestController {
 	@RequestMapping(value = "/bookTest/{poolSize}", method = {
 			RequestMethod.GET, RequestMethod.POST })
 	@ResponseBody
-	public RunScenarioResultModel bookTest(@PathVariable int poolSize) {
+	public RunScenarioResultModel bookTest(@PathVariable int poolSize, @RequestParam long startTime) {
 		try {
 			UUID runId = UUID.randomUUID();
 			this.getScenarioEngine().addRunningTestWithoutScenario(runId,

Bench4Q-Agent/src/main/java/org/bench4q/agent/scenario/engine/ScenarioContext.java

@@ -130,8 +130,9 @@ public class ScenarioContext implements Observer {
 				this.getExecutor().getActiveCount());
 	}
 
-	public void initTasks() {
-		for (int i = 0; i < this.getExecutor().getCorePoolSize(); i++) {
+	public void initTasks(int currentLoad) {
+		this.updatePopulation(currentLoad);
+		for (int i = 0; i < currentLoad; i++) {
 			addTask();
 		}
 	}

Bench4Q-Agent/src/main/java/org/bench4q/agent/scenario/engine/ScenarioEngine.java

@@ -15,7 +15,7 @@ import org.springframework.stereotype.Component;
 @Component
 public class ScenarioEngine implements Observer {
 	private Map<UUID, ScenarioContext> runningTests;
-	private Logger logger = Logger.getLogger(ScenarioEngine.class);
+	private final Logger logger = Logger.getLogger(ScenarioEngine.class);
 	private PluginManager pluginManager;
 
 	public ScenarioEngine() {
@@ -72,9 +72,20 @@ public class ScenarioEngine implements Observer{
 			logger.error("The context required is null");
 			return false;
 		}
-		scenarioContext.initTasks();
-		new Supervisor(scenarioContext).start();;
+		try {
+			int currentLoad = scenarioContext
+					.getScenario()
+					.getSchedule()
+					.loadFor(
+							System.currentTimeMillis()
+									- scenarioContext.getStartDate().getTime());
+			scenarioContext.initTasks(currentLoad);
+			new Supervisor(scenarioContext).start();
 			return true;
+		} catch (Exception e) {
+			this.logger.info(e, e);
+			return false;
+		}
 	}
 
 	@Override
@@ -86,5 +97,4 @@ public class ScenarioEngine implements Observer{
 		this.getRunningTests().remove(testId);
 	}
 
-	
 }

Bench4Q-Agent/src/main/java/org/bench4q/agent/scenario/engine/Schedule.java

@@ -101,6 +101,19 @@ public class Schedule extends Observable {
 		throw new Bench4QRunTimeException("Should not come to this place");
 	}
 
+	/**
+	 * 
+	 * @param relativeTime, relativeTime from beginDate
+	 * @return
+	 */
+	public int loadFor(long relativeTime){
+		Segment segment = getSegment(relativeTime);
+		if (segment == null) {
+			throw new IllegalArgumentException();
+		}
+		return segment.loadFor(relativeTime);
+	}
+	
 	public static class Segment {
 		private final Point start;
 		private final Point end;