org.jgroups.util.GenerateProfilingScript Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jgroups
Reliable cluster communication toolkit
There is a newer version: 5.3.13.Final
package org.jgroups.util;

import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.stream.Stream;

/**
 * Takes a list of methods to profile and generates the Byteman script from them. See ./conf/scripts/profiled-methods.txt
 * as an example
 * @author Bela Ban
 * @since  5.2.13
 */
public class GenerateProfilingScript {
    protected int num_rules_generated;

    protected void generate(String input_file, String output_file) throws IOException {
        try(OutputStream out=new FileOutputStream(output_file); Stream lines=Files.lines(Path.of(input_file))) {
            out.write(PREFIX.getBytes());
            lines.filter(l -> !l.trim().isEmpty())
              .forEach(l -> generateRule(l, out));
        }
    }

    protected void generateRule(String method, OutputStream out) {
        int index=method.lastIndexOf('.');
        if(index < 0)
            throw new IllegalArgumentException(String.format("method name not found in %s", method));
        String classname=method.substring(0, index).trim(), method_name=method.substring(index+1).trim();
        try {
            Class clazz=Util.loadClass(classname, (Class)null);
            boolean is_interface=clazz.isInterface();
            String start_rule=String.format(START_RULE, clazz.getSimpleName() + "." + method_name,
                                            is_interface? "INTERFACE" : "CLASS",
                                            classname, method_name, clazz.getSimpleName() + "." + method_name);
            String stop_rule=String.format(STOP_RULE, clazz.getSimpleName() + "." + method_name,
                                           is_interface? "INTERFACE" : "CLASS",
                                           classname, method_name, clazz.getSimpleName() + "." + method_name);

            out.write(start_rule.getBytes());
            out.write(stop_rule.getBytes());
        }
        catch(Exception e) {
            throw new RuntimeException(e);
        }
    }

    public static void main(String[] args) throws IOException {
        if(args.length != 2) {
            System.err.printf("%s  \n", GenerateProfilingScript.class.getSimpleName());
            return;
        }
        String input=args[0], output=args[1];
        GenerateProfilingScript tmp=new GenerateProfilingScript();
        tmp.generate(input, output);
    }



    protected static final String PREFIX="# Byteman script to profile individual methods\n" +
      "\n" +
      "\n" +
      "RULE DiagnosticHandler creation\n" +
      "CLASS ^TP\n" +
      "HELPER org.jgroups.util.ProfilingHelper\n" +
      "METHOD handleConnect()\n" +
      "AT ENTRY\n" +
      "BIND tp=$this, diag=tp.getDiagnosticsHandler();\n" +
      "IF TRUE\n" +
      "   DO diagCreated(diag);\n" +
      "ENDRULE\n\n";


    protected static final String START_RULE="RULE %s start\n" +
      "%s %s\n" +
      "HELPER org.jgroups.util.ProfilingHelper\n" +
      "METHOD %s\n" +
      "COMPILE\n" +
      "AT ENTRY\n" +
      "IF TRUE\n" +
      "   DO start(\"%s\");\n" +
      "ENDRULE\n\n";

    protected static final String STOP_RULE="RULE %s stop\n" +
      "%s %s\n" +
      "HELPER org.jgroups.util.ProfilingHelper\n" +
      "METHOD %s\n" +
      "COMPILE\n" +
      "AT EXIT\n" +
      "IF TRUE\n" +
      "   DO stop(\"%s\");\n" +
      "ENDRULE\n\n";
}