com.neotys.neoload.model.v3.binding.serializer.sla.SlaThresholdHelper Maven / Gradle / Ivy

Go to download
package com.neotys.neoload.model.v3.binding.serializer.sla;

import java.io.IOException;
import java.util.Arrays;
import java.util.List;

import com.neotys.neoload.model.v3.binding.serializer.DefaultErrorListener;
import org.antlr.v4.runtime.CharStreams;
import org.antlr.v4.runtime.CommonTokenStream;

import com.neotys.neoload.model.v3.binding.serializer.sla.SlaThresholdParser.ThresholdContext;
import com.neotys.neoload.model.v3.project.sla.SlaThreshold;

public final class SlaThresholdHelper {

	private SlaThresholdHelper() {
		super();
	}

	public static SlaThreshold convertToThreshold(final String input) throws IOException {
		// Normalyze threshold
		final String thresholdAsText = (input != null) ? input : "";
		
		// Manages the errors
		final DefaultErrorListener thresholdErrorListener = new DefaultErrorListener();
		
		// Lexer
		final SlaThresholdLexer lexer = new SlaThresholdLexer(CharStreams.fromString(thresholdAsText));
	    lexer.removeErrorListeners();
	    lexer.addErrorListener(thresholdErrorListener);
	    // Tokens
	    final CommonTokenStream tokens = new CommonTokenStream(lexer);
	    // Parser
	    final SlaThresholdParser parser = new SlaThresholdParser(tokens);
	    parser.removeErrorListeners();
	    parser.addErrorListener(thresholdErrorListener);
	    // Context
	    final ThresholdContext context;
	    try {
	    	context = parser.threshold();
	    }
	    catch (final Exception e) {
	    	throw newIOException(thresholdAsText, Arrays.asList(e.getMessage()));
		}
	    
	    // Throw the errors if necessary
	    final List errors = thresholdErrorListener.getErrors();
	    if (!errors.isEmpty()) {
	    	throw newIOException(thresholdAsText, errors);
	    }
	    
	    // Threshold visitor
	    final DefaultSlaThresholdVisitor visitor = new DefaultSlaThresholdVisitor();
	    return visitor.visit(context);
	}
	
	private static IOException newIOException(final String threshold, final List errors) {
		final StringBuilder message = new StringBuilder();
   		message.append(threshold);
   		message.append(" is not a valid threshold: ");
   		for (String error : errors) {
			message.append(System.lineSeparator());
			message.append(error);
		}
   		return new IOException(message.toString());
	}
}