com.navisarv.mop.instrumentation.InstrumentationProcessor Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mop-javac Show documentation

A javac plugin

The newest version!

package com.navisarv.mop.instrumentation;

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

import com.sun.source.tree.MethodTree;
import com.sun.source.tree.Tree;
import com.sun.source.util.TreePathScanner;

public class InstrumentationProcessor extends TreePathScanner {

	private InstrumentationContext context = null;
	private List> availInstrumentations = null;
	@SafeVarargs
	public InstrumentationProcessor(InstrumentationContext instruContext, MOPInstrumentation...instrumentation) {
		this.context = instruContext;
		availInstrumentations = new ArrayList<>();
		if((instrumentation != null) && (instrumentation.length != 0)){
			availInstrumentations.addAll(Arrays.asList(instrumentation));
		}
	}
	
	@Override
	public Void visitMethod(MethodTree method, Void v) {
		MethodInstrumentation methodInstrumentation = (MethodInstrumentation) findInstrumentation(MethodTree.class);
		if(methodInstrumentation != null) {
			methodInstrumentation.instrument(method, context);
		}
		return super.visitMethod(method, v);
	}
	
	private MOPInstrumentation findInstrumentation(Class type){
		return availInstrumentations.stream().filter(instr-> instr.getInstrumentationType() ==  type).findFirst().orElse(null);
	}

	
	
}