su.nlq.prometheus.jmx.agent.CollectorAgent Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jmx-prometheus-agent Show documentation

Pipe for JMX MBeans to Prometheus

There is a newer version: 0.8.0

package su.nlq.prometheus.jmx.agent;

import io.prometheus.client.hotspot.DefaultExports;
import org.jetbrains.annotations.NotNull;
import org.kohsuke.args4j.Argument;
import su.nlq.prometheus.jmx.arguments.Arguments;
import su.nlq.prometheus.jmx.http.CollectorServer;
import su.nlq.prometheus.jmx.http.ExpositionFormat;
import su.nlq.prometheus.jmx.http.ServerParameters;

import java.lang.instrument.Instrumentation;

public enum CollectorAgent {
  ;

  public static void premain(@NotNull String argument, @NotNull Instrumentation instrumentation) {
    Arguments.of(argument.split(":"), new AgentArguments()).ifPresent(arguments -> {
      CollectorServer.start(arguments);
      DefaultExports.initialize();
    });
  }

  private static final class AgentArguments extends ServerParameters {

    @Argument(index = 0, metaVar = "FORMAT", usage = "exposition format", required = false)
    private @NotNull ExpositionFormat format = ExpositionFormat.Text;

    @Argument(index = 1, metaVar = "HOST", usage = "host", required = false)
    private @NotNull String host = "";

    @Argument(index = 2, metaVar = "PORT", usage = "port", required = true)
    private int port = 0;

    @Argument(index = 3, metaVar = "FILE", usage = "configuration xml file", required = true)
    private @NotNull String config = "";

    @Override
    protected @NotNull String getConfig() {
      return config;
    }

    @Override
    protected int getPort() {
      return port;
    }

    @Override
    protected @NotNull String getHost() {
      return host;
    }

    @Override
    protected @NotNull ExpositionFormat getFormat() {
      return format;
    }
  }
}