com.ociweb.pronghorn.TelemetryTestTool Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of pronghorn Show documentation

Primary dependency for any project using the Pronghorn framework

The newest version!

package com.ociweb.pronghorn;

import com.ociweb.pronghorn.pipe.Pipe;
import com.ociweb.pronghorn.pipe.RawDataSchema;
import com.ociweb.pronghorn.stage.route.ReplicatorStage;
import com.ociweb.pronghorn.stage.scheduling.GraphManager;
import com.ociweb.pronghorn.stage.scheduling.StageScheduler;
import com.ociweb.pronghorn.stage.test.PipeCleanerStage;

public class TelemetryTestTool {

	////////////////////////////////////////////////////////////
	//To run this from the command prompt:
	// mvn exec:java -Dexec.addResourcesToClasspth=true -Dexec.mainClass="com.ociweb.pronghorn.TelemetryTestTool"
	///////////////////////////////////////////////////////////
	
	public static void main(String[] args) {
	
			launch(8092, 14, 7);			
			launch(8093, 50, 2);
			launch(8094, 4, 13);
					
	}

	private static void launch(int port, int width, int height) {
		GraphManager gm = new GraphManager();
		GraphManager.addDefaultNota(gm, GraphManager.SCHEDULE_RATE, 500_000);

		Pipe output = RawDataSchema.instance.newPipe(8, 8);
		ExampleProducerStage producer = new ExampleProducerStage(gm, output);
					
		int c = Math.min(1, height);
		int b = height-c;
		
		int i = width;
		Pipe[] targets = new Pipe[i];
		while (--i>=0) {
			targets[i] = new Pipe(output.config().grow2x());
			Pipe temp = null;
			Pipe prev = targets[i];
			
			int k = b;
			while (--k>=0) {
				//only grow some of these because this will take up too much memory
				temp = new Pipe(k<5 ? prev.config().grow2x() : prev.config());
				//slow replicator so it batches
				GraphManager.addNota(gm, GraphManager.SCHEDULE_RATE, 10_000_000, 
				new BatchingStage(gm, .90, prev, temp) );
				prev = temp;
			}
							
			
			int j = c; //replicators
			while (--j>=0) {
				temp = new Pipe(prev.config().grow2x());
				//slow replicator so it batches
				GraphManager.addNota(gm, GraphManager.SCHEDULE_RATE, 10_000_000, 
				new ReplicatorStage(gm, prev, temp) );
				prev = temp;
			}
			GraphManager.addNota(gm, GraphManager.SCHEDULE_RATE, 5_000_000, 
		    new PipeCleanerStage(gm, temp) );
		}			
		//slow replicator so it batches
		GraphManager.addNota(gm, GraphManager.SCHEDULE_RATE, 10_000_000, 
		new ReplicatorStage<>(gm, output, targets) );
		gm.enableTelemetry(port);

		StageScheduler.defaultScheduler(gm).startup();
	}

}