org.voovan.tools.TEnv Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of voovan-framework Show documentation
Show all versions of voovan-framework Show documentation
Voovan is a java framwork and it not depends on any third-party framework.
package org.voovan.tools;
import com.sun.tools.attach.AgentInitializationException;
import com.sun.tools.attach.AgentLoadException;
import com.sun.tools.attach.AttachNotSupportedException;
import com.sun.tools.attach.VirtualMachine;
import org.voovan.tools.log.Logger;
import org.voovan.tools.reflect.TReflect;
import java.io.File;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.lang.instrument.Instrumentation;
import java.lang.management.ManagementFactory;
import java.util.ArrayList;
import java.util.List;
import java.util.Vector;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.TimeoutException;
import java.util.function.BooleanSupplier;
import java.util.function.Supplier;
import java.util.jar.JarEntry;
import java.util.jar.JarFile;
import java.util.regex.Matcher;
/**
* 系统环境相关
*
* @author helyho
*
* Voovan Framework.
* WebSite: https://github.com/helyho/Voovan
* Licence: Apache v2 License
*/
public class TEnv {
public static Float JDK_VERSION = Float.valueOf(System.getProperty("java.vm.specification.version"));
public static String OS_NAME = System.getProperty("os.name").toUpperCase();
public static Thread MAIN_THREAD = getMainThread();
public static volatile boolean IS_SHUTDOWN = false;
public static Vector SHUT_DOWN_HOOKS = new Vector();
static {
Runtime.getRuntime().addShutdownHook(new Thread() {
@Override
public void run() {
TEnv.IS_SHUTDOWN = true;
for(BooleanSupplier supplier : SHUT_DOWN_HOOKS.toArray(new BooleanSupplier[0])){
try {
if(supplier!=null) {
supplier.getAsBoolean();
}
} catch (Throwable e) {
System.out.println("Run shutdown hooks failed." + e.getMessage() + "\r\n" + TEnv.getStackElementsMessage(e.getStackTrace()));
}
}
}
});
}
public static void addShutDownHook(BooleanSupplier supplier){
SHUT_DOWN_HOOKS.add(supplier);
}
public static Instrumentation instrumentation;
//环境名称, 用于多环境条件下的配置文件读取
private static String ENV_NAME = null;
/**
* 获取环境的配置名
* @return 环境的配置名
*/
public static String getEnvName(){
return ENV_NAME;
}
/**
* 设置环境的配置名
* @param envName 环境的配置名
*/
public static void setEnvName(String envName){
ENV_NAME = envName;
}
/**
* 获取当前进程 PID
* @return 当前进程 ID
*/
public static long getCurrentPID(){
return Long.parseLong(ManagementFactory.getRuntimeMXBean().getName().split("@")[0]);
}
/**
* 构造一个系统进程
* @param command 命令行
* @param env 环境变量参数
* @param workDir 工作目录
* @return Process 对象
* @throws IOException IO 异常
*/
public static Process createSysProcess(String command, String[] env, File workDir) throws IOException {
Runtime runTime = Runtime.getRuntime();
if(workDir==null || workDir.exists()) {
return runTime.exec(command, env, workDir);
}
return null;
}
/**
* 构造一个系统进程
* @param command 命令行
* @param env 环境变量参数
* @param workDir 工作目录
* @return Process 对象
* @throws IOException IO 异常
*/
public static Process createSysProcess(String command, String[] env, String workDir) throws IOException {
return createSysProcess(command, env, (workDir==null ? null :new File(workDir)) );
}
/**
* 休眠函数
*
* @param sleepTime 休眠时间
* @return true: 超时结束, false: sleep 被打断
*/
public static boolean sleep(int sleepTime) {
try {
Thread.sleep(sleepTime);
return true;
} catch (InterruptedException e) {
return false;
}
}
/**
* 休眠函数
* @param timeUnit 休眠时间单位
* @param sleepTime 休眠时间
* @return true: 超时结束, false: sleep 被打断
*/
public static boolean sleep(TimeUnit timeUnit, int sleepTime) {
try {
timeUnit.sleep(sleepTime);
return true;
} catch (InterruptedException e) {
return false;
}
}
/**
* 获取当前栈信息
*
* @return 当前栈信息
*/
public static StackTraceElement[] getStackElements() {
Throwable ex = new Throwable();
return ex.getStackTrace();
}
/**
* 判断指定的 Class 是否在当前的线程栈中
* @param clazzName 类对象
* @param method 方法名
* @return true: 在当前的线程栈中, false: 不在当前的线程栈中
*/
public static boolean classInCurrentStack(String clazzName, String method){
for(StackTraceElement stackTraceElement : getStackElements()){
if(clazzName!=null && stackTraceElement.getClassName().contains(clazzName)){
if(method == null || stackTraceElement.getMethodName().equals(method)){
return true;
}
} else {
continue;
}
}
return false;
}
/**
* 获取当前栈信息
* 会自动过滤掉栈里的第一行,即当前类的信息
*
* @return 当前栈信息
*/
public static String getStackMessage(){
StringBuilder stackInfo = new StringBuilder();
Throwable ex = new Throwable();
int row = 0;
for (StackTraceElement stackTraceElement : ex.getStackTrace()) {
if(row!=0){
stackInfo.append(stackTraceElement.toString());
stackInfo.append("\r\n");
}
row++;
}
return stackInfo.toString();
}
/**
* 获取当前栈信息
* @param stackTraceElements 栈信息对象数组
* @return 当前栈信息
*/
public static String getStackElementsMessage(StackTraceElement[] stackTraceElements){
StringBuilder stackInfo = new StringBuilder();
for (StackTraceElement stackTraceElement : stackTraceElements) {
stackInfo.append(stackTraceElement.toString());
stackInfo.append("\r\n");
}
return stackInfo.toString().trim();
}
/**
* 获取JVM中的所有线程
* @return 线程对象数组
*/
public static Thread[] getThreads(){
ThreadGroup group = Thread.currentThread().getThreadGroup().getParent();
int estimatedSize = group.activeCount() * 2;
Thread[] slackList = new Thread[estimatedSize];
int actualSize = group.enumerate(slackList);
Thread[] list = new Thread[actualSize];
System.arraycopy(slackList, 0, list, 0, actualSize);
return list;
}
/**
* 获取进程的主线程
* @return 进程的主线程
*/
public static Thread getMainThread(){
for(Thread thread: TEnv.getThreads()){
if(thread.getId()==1){
return thread;
}
}
return null;
}
/**
* 从当前进程的ClassPath中寻找 Class
* @param pattern 确认匹配的正则表达式
* @param filters 过滤的 class, 满足这些条件的 class 才会被搜索到(注解,接口,继承的类)
* @return 匹配到的 class 集合
* @throws IOException IO 异常
*/
public static List searchClassInEnv(String pattern, Class[] filters) throws IOException {
String userDir = System.getProperty("user.dir");
List classPaths = getClassPath();
ArrayList clazzes = new ArrayList();
for(String classPath : classPaths){
if(TString.isNullOrEmpty(classPath)){
continue;
}
File classPathFile = new File(classPath);
if(classPathFile.exists() && classPathFile.isDirectory()){
clazzes.addAll( getDirectorClass(classPathFile, pattern, filters));
} else if(classPathFile.exists() && classPathFile.isFile() && classPathFile.getName().endsWith(".jar")) {
clazzes.addAll( getJarClass(classPathFile, pattern, filters) );
}
}
return clazzes;
}
/**
* 从指定 File 对象寻找 Class
* @param rootfile 文件目录 File 对象
* @param pattern 确认匹配的正则表达式
* @param filters 过滤的 class, 满足这些条件的 class 才会被搜索到(注解,接口,继承的类)
* @return 匹配到的 class 集合
* @throws IOException IO 异常
*/
public static List getDirectorClass(File rootfile, String pattern, Class[] filters) throws IOException {
if(pattern!=null) {
pattern = pattern.replace(".", File.separator);
}
ArrayList result = new ArrayList();
List files = TFile.scanFile(rootfile, pattern);
for(File file : files){
String fileName = file.getCanonicalPath();
if("class".equals(TFile.getFileExtension(fileName))) {
//如果是内部类则跳过
if(TString.regexMatch(fileName,"G\\$\\d\\.class")>0){
continue;
}
fileName = fileName.replace(rootfile.getCanonicalPath() + File.separator, "");
try {
Class clazz = resourceToClass(fileName);
if(TReflect.classChecker(clazz, filters)) {
result.add(clazz);
}
} catch (ClassNotFoundException e) {
Logger.warn("Try to load class["+fileName+"] failed",e);
}
}
}
return result;
}
/**
* 从指定jar 文件中寻找 Class
* @param jarFile jar 文件 File 对象
* @param pattern 确认匹配的正则表达式
* @param filters 过滤的 class, 满足这些条件的 class 才会被搜索到(注解,接口,继承的类)
* @return 匹配到的 class
* @throws IOException IO 异常
*/
public static List getJarClass(File jarFile, String pattern, Class[] filters) throws IOException {
if(pattern!=null) {
pattern = pattern.replace(".", File.separator);
}
ArrayList result = new ArrayList();
List jarEntrys = TFile.scanJar(jarFile, pattern);
for(JarEntry jarEntry : jarEntrys){
String fileName = jarEntry.getName();
if("class".equals(TFile.getFileExtension(fileName))) {
//如果是内部类则跳过
if (TString.regexMatch(fileName, "\\$\\d\\.class") > 0) {
continue;
}
try {
Class clazz = resourceToClass(fileName);
if(TReflect.classChecker(clazz, filters)) {
result.add(clazz);
}
} catch (Throwable e) {
fileName = null;
}
}
}
return result;
}
/**
* 读取 Class 的字节码
* @param clazz Class 对象
* @return 字节码
*/
public static byte[] loadClassBytes(Class clazz) {
InputStream inputStream = null;
try {
String classLocation = getClassLocation(clazz);
String classPathName = TEnv.classToResource(clazz);
if (classLocation.endsWith("jar")) {
return TZip.loadFileFromZip(classLocation, classPathName);
} else {
return TFile.loadFileFromSysPath(classLocation+classPathName);
}
} catch (Exception e){
Logger.error("Load class bytes by " + clazz.getCanonicalName() + "error", e);
return null;
}
}
/**
* 获取 Class 在物理设备上的文件位置
* @param clazz Class 对象
* @return 在物理设备上的文件位置
*/
public static String getClassLocation(Class clazz){
if(clazz.getProtectionDomain().getCodeSource().getLocation()!=null) {
return clazz.getProtectionDomain().getCodeSource().getLocation().getFile();
} else {
return null;
}
}
/**
* 获取 Class 的修改时间
* @param clazz Class 对象
* @return 修改时间, 返回: -1 文件不存在 / 文件不是 Class 文件 / IO 异常
*/
public static long getClassModifyTime(Class clazz){
String location = getClassLocation(clazz);
if(location==null){
return -1;
}
String classNamePath = TEnv.classToResource(clazz);
try {
if(location.endsWith(".jar")) {
try(JarFile jarFile = new JarFile(location)) {
JarEntry jarEntry = jarFile.getJarEntry(classNamePath);
if(jarEntry!=null) {
return jarEntry.getTime();
}else{
return -1;
}
}
} else if (location.endsWith(File.separator)) {
File classFile = new File(location+classNamePath);
if(classFile!=null && classFile.exists()) {
return classFile.lastModified();
}else{
return -1;
}
} else {
return -1;
}
}catch (IOException e){
return -1;
}
}
/**
* 获取类JVM 的 Class Path
* 因部分 ide 会自动增加全部的 jvm 的 classpath, 所以这里会自动剔除 classpath 中 jvm 的 classPath
* @return 获得用户的类加载路径
*/
public static List getClassPath(){
ArrayList userClassPath = new ArrayList();
String javaHome = System.getProperty("java.home");
//去除 javahome 的最后一个路径节点,扩大搜索范文
javaHome = javaHome.replaceAll("\\/[a-zA-z0-9\\_\\$]*$","");
String[] classPaths = System.getProperty("java.class.path").split(File.pathSeparator);
for(String classPath : classPaths){
if(!classPath.startsWith(javaHome)){
userClassPath.add(classPath);
}
}
return userClassPath;
}
/**
* 将class 转换成 资源资源文件路径
* @param clazz Class对象
* @return 资源文件路径
*/
public static String classToResource(Class clazz){
String classNamePath = clazz.getName();
if(clazz.isMemberClass()) {
classNamePath = TString.fastReplaceAll(classNamePath, "\\$.*$", "");
}
return TString.fastReplaceAll(classNamePath, "\\.", File.separator)+".class";
}
/**
* 将资源文件路径 转换成 Class
* @param resourcePath 资源资源文件路径
* @return Class对象
* @throws ClassNotFoundException 类未找到异常
*/
public static Class resourceToClass(String resourcePath) throws ClassNotFoundException {
String className = null;
if(resourcePath.startsWith(File.separator)){
resourcePath = TString.removePrefix(resourcePath);
}
className = TString.fastReplaceAll(resourcePath, "\\$.*\\.class$", ".class");
className = TString.fastReplaceAll(className, ".class$", "");
className = TString.fastReplaceAll(className, Matcher.quoteReplacement(File.separator), ".");
try {
return Class.forName(className);
}catch (Exception ex) {
throw new ClassNotFoundException("load and define class " + className + " failed");
}
}
/**
* 返回当前 jvm 的 JAVA_HOME 参数
* @return 当前 jvm 的 JAVA_HOME 参数
*/
public static String getJavaHome(){
String sysLibPath = System.getProperty("sun.boot.library.path");
return sysLibPath.substring(0, sysLibPath.indexOf("/jre/lib"));
}
/**
* 查找 AgentJar 文件
* @return AgentJar 文件
*/
private static File findAgentJar(){
List agentJars = new ArrayList();
for(String classPath : System.getProperty("java.class.path").split(File.pathSeparator)) {
if(TString.regexMatch(classPath, "((dd[\\.\\-](\\d\\.?)*.*?)|(voovan-((framework)|(common))[\\.\\-].*?)).?jar$")!=0){
agentJars.add(new File(classPath));
}
}
if(agentJars.size() == 0) {
agentJars = TFile.scanFile(new File(TFile.getContextPath()), "((dd[\\.\\-](\\d\\.?)*.*?)|(voovan-((framework)|(common))[\\.\\-].*?)).?jar$");
}
File agentJar = null;
for (File jarFile : agentJars) {
if(agentJar == null){
agentJar = jarFile;
}
if(agentJar.lastModified() < jarFile.lastModified()){
agentJar = jarFile;
}
}
return agentJar;
}
/**
* 附加 Agentjar 到目标地址
* @param agentJarPath AgentJar 文件
* @throws IOException IO 异常
* @throws AttachNotSupportedException 附加指定进程失败
* @throws AgentLoadException Agent 加载异常
* @throws AgentInitializationException Agent 初始化异常
* @return Instrumentation 对象
*/
public static Instrumentation agentAttach(String agentJarPath) throws IOException, AttachNotSupportedException, AgentLoadException, AgentInitializationException {
if(TEnv.instrumentation==null) {
if (agentJarPath == null) {
File agentJar = findAgentJar();
if (agentJar != null && agentJar.exists()) {
agentJarPath = agentJar.getAbsolutePath();
Logger.info("[System] Choose an agent jar file: " + agentJarPath);
} else {
throw new FileNotFoundException("The agent jar file not found");
}
}
try {
VirtualMachine vm = VirtualMachine.attach(Long.toString(TEnv.getCurrentPID()));
vm.loadAgent(agentJarPath);
TEnv.instrumentation = DynamicAgent.getInstrumentation();
vm.detach();
return TEnv.instrumentation;
} catch (IOException e) {
if (e.getMessage().contains("allowAttachSelf")) {
e = new IOException("You should add java command arguments: -Djdk.attach.allowAttachSelf=true with java command.");
}
throw e;
}
} else {
return TEnv.instrumentation;
}
}
/**
* 等待函数
* @param waitTime 等待时间
* @param supplier 满足条件时一直等待, 如果该方法返回 true 一直等待, false 达到预期退出等待
* @throws TimeoutException 超时异常
*/
public static void wait(int waitTime, Supplier supplier) throws TimeoutException {
long start = System.currentTimeMillis();
while(true){
if(supplier.get()){
TEnv.sleep(1);
} else {
return;
}
if(System.currentTimeMillis()-start>=waitTime){
throw new TimeoutException("TEnv.wait time out");
}
}
}
/**
* 等待函数
* @param supplier 满足条件时一直等待, 如果该方法返回 true 一直等待, false 达到预期退出等待
*/
public static void wait(Supplier supplier) {
try {
wait(Integer.MAX_VALUE, supplier);
} catch (TimeoutException e) {
e.printStackTrace();
}
}
/**
* 性能测试方法
* @param supplier 执行器
* @return 执行时间
*/
public static List © 2015 - 2024 Weber Informatics LLC | Privacy Policy