org.beifengtz.jvmm.server.controller.ExecuteController Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jvmm-server Show documentation
Provides access to operating system, process, thread and other information during Java runtime.
There is a newer version: 2.4.3
package org.beifengtz.jvmm.server.controller;

import com.google.gson.JsonArray;
import org.beifengtz.jvmm.common.tuple.Pair;
import org.beifengtz.jvmm.common.util.CommonUtil;
import org.beifengtz.jvmm.convey.annotation.HttpController;
import org.beifengtz.jvmm.convey.annotation.HttpRequest;
import org.beifengtz.jvmm.convey.annotation.JvmmController;
import org.beifengtz.jvmm.convey.annotation.JvmmMapping;
import org.beifengtz.jvmm.convey.annotation.RequestBody;
import org.beifengtz.jvmm.convey.annotation.RequestParam;
import org.beifengtz.jvmm.convey.enums.GlobalType;
import org.beifengtz.jvmm.convey.enums.Method;
import org.beifengtz.jvmm.core.JvmmFactory;
import org.beifengtz.jvmm.core.entity.result.JpsResult;
import org.beifengtz.jvmm.server.ServerContext;

import java.lang.reflect.InvocationTargetException;
import java.util.List;

import static org.beifengtz.jvmm.server.ServerBootstrap.AGENT_BOOT_CLASS;

/**
 * 
 * Description: TODO
 * 
 * 
 * Created in 5:55 下午 2021/5/30
 *
 * @author beifengtz
 */
@JvmmController
@HttpController
public class ExecuteController {

    @JvmmMapping(typeEnum = GlobalType.JVMM_TYPE_EXECUTE_GC)
    @HttpRequest("/execute/gc")
    public String gc() {
        JvmmFactory.getExecutor().gc();
        return ServerContext.STATUS_OK;
    }

    @JvmmMapping(typeEnum = GlobalType.JVMM_TYPE_EXECUTE_SET_CLASSLOADING_VERBOSE)
    @HttpRequest("/execute/set_classloading_verbose")
    public String setClassLoadingVerbose(@RequestParam boolean verbose) {
        JvmmFactory.getExecutor().setClassLoadingVerbose(verbose);
        return ServerContext.STATUS_OK;
    }

    @JvmmMapping(typeEnum = GlobalType.JVMM_TYPE_EXECUTE_SET_MEMORY_VERBOSE)
    @HttpRequest("/execute/set_memory_verbose")
    public String setMemoryVerbose(@RequestParam boolean verbose) {
        JvmmFactory.getExecutor().setMemoryVerbose(verbose);
        return ServerContext.STATUS_OK;
    }

    @JvmmMapping(typeEnum = GlobalType.JVMM_TYPE_EXECUTE_SET_THREAD_CPU_TIME_ENABLED)
    @HttpRequest("/execute/set_thread_cpu_time_enabled")
    public String setThreadCpuTimeEnabled(@RequestParam boolean verbose) {
        JvmmFactory.getExecutor().setThreadCpuTimeEnabled(verbose);
        return ServerContext.STATUS_OK;
    }

    @JvmmMapping(typeEnum = GlobalType.JVMM_TYPE_EXECUTE_SET_THREAD_CONTENTION_MONITOR_ENABLED)
    @HttpRequest("/execute/set_thread_contention_monitor_enabled")
    public String setThreadContentionMonitoringEnabled(@RequestParam boolean verbose) {
        JvmmFactory.getExecutor().setThreadContentionMonitoringEnabled(verbose);
        return ServerContext.STATUS_OK;
    }

    @JvmmMapping(typeEnum = GlobalType.JVMM_TYPE_EXECUTE_RESET_PEAK_THREAD_COUNT)
    @HttpRequest("/execute/reset_peak_thread_count")
    public String resetPeakThreadCount() {
        JvmmFactory.getExecutor().resetPeakThreadCount();
        return ServerContext.STATUS_OK;
    }

    @JvmmMapping(typeEnum = GlobalType.JVMM_TYPE_EXECUTE_JAVA_PROCESS)
    @HttpRequest("/execute/jps")
    public JsonArray listJavaProcess() throws Exception {
        Pair, String> pair = JvmmFactory.getExecutor().listJavaProcess();
        if (pair.getRight() == null) {
            JsonArray result = new JsonArray();
            pair.getLeft().forEach(o -> result.add(o.toJson()));
            return result;
        } else {
            throw new Exception(pair.getRight());
        }
    }

    @JvmmMapping(typeEnum = GlobalType.JVMM_TYPE_EXECUTE_JVM_TOOL)
    @HttpRequest(value = "/execute/jvm_tool", method = Method.POST)
    public Object jvmTool(@RequestBody String command) throws Exception {
        Pair, Boolean> pair = JvmmFactory.getExecutor().executeJvmTools(command);
        if (pair.getRight()) {
            return pair.getLeft();
        } else {
            return CommonUtil.join("\n", pair.getLeft());
        }
    }

    @JvmmMapping(typeEnum = GlobalType.JVMM_TYPE_EXECUTE_JAD)
    @HttpRequest("/execute/jad")
    public String jad(@RequestParam String className, @RequestParam String methodName) throws Throwable {
        try {
            Class bootClazz = Thread.currentThread().getContextClassLoader().loadClass(AGENT_BOOT_CLASS);
            return (String) bootClazz.getMethod("jad", String.class, String.class).invoke(null, className, methodName);
        } catch (InvocationTargetException e) {
            throw e.getTargetException();
        }
    }
}