java_concurrency1_fix.java

/**
 * Manages statistic updates for a Runner in a load test.
 */
public class WebRunnerStatsSender
{
	private final Map<String, VariableGroup> resourceToVariableGroup = new HashMap<>();
	private final Clock clock;

	public WebRunnerStatsSender( Clock clock )
	{
		this.clock = clock;
	}

	/**
	 * Register a resource. This is done once per resource in the beginning of a test.
	 */
	public void addResource( String resource )
	{
		resourceToVariableGroup.put( resource, new VariableGroup( resource ) );
	}

	/**
	 * Updates the time taken statistic with a new sample.
	 * This is done once per response (i.e. typically several times per second throughout the test).
	 */
	public void updateTimeTaken( String resource, long timeTaken )
	{
		long timeStamp = clock.millis();

		VariableGroup resorceVariables = getVariablesFor( resource );
		if( resorceVariables != null )
		{
			resorceVariables.timeTakenVariable.update( timeStamp, timeTaken );
		}
	}
	
	private VariableGroup getVariablesFor( String resource )
	{
		return resourceToVariableGroup.get( resource );
	}
}





////////////////////////////////////////////////////////////////////////





/**
 * Manages statistic updates for a Runner in a load test.
 */
public class WebRunnerStatsSender
{
	private volatile ImmutableMap<String, VariableGroup> resourceToVariableGroup = ImmutableMap.of();
	private final Clock clock;

	public WebRunnerStatsSender( Clock clock )
	{
		this.clock = clock;
	}

	/**
	 * Register a resource. This is done once per resource in the beginning of a test.
	 */
	public void addResources( Iterable<String> resources )
	{
		ImmutableMap.Builder<String, VariableGroup> mapBuilder = ImmutableMap.builder();
		mapBuilder.putAll( resourceToVariableGroup );
		for( String resource : resources )
		{
			mapBuilder.put( resource, new VariableGroup( resource ) );
		}
		resourceToVariableGroup = mapBuilder.build();
	}

	/**
	 * Updates the time taken statistic with a new sample.
	 * This is done once per response (i.e. typically several times per second throughout the test).
	 */
	public void updateTimeTaken( String resource, long timeTaken )
	{
		long timeStamp = clock.millis();

		VariableGroup resorceVariables = getVariablesFor( resource );
		if( resorceVariables != null )
		{
			resorceVariables.timeTakenVariable.update( timeStamp, timeTaken );
		}
	}
	
	private VariableGroup getVariablesFor( String resource )
	{
		return resourceToVariableGroup.get( resource );
	}
}