修改javaagent包结构为java项目默认包结构

Author: 丁建水

src/main/java/com/wenshuo/agent/Agent.java

@@ -0,0 +1,27 @@
+/*
+ * @(#)Agent.java	2015-7-24 上午09:49:34
+ * javaagent
+ * Copyright 2015 wenshuo, Inc. All rights reserved.
+ * wenshuo PROPRIETARY/CONFIDENTIAL. Use is subject to license terms.
+ */
+package com.wenshuo.agent;
+import java.lang.instrument.Instrumentation;
+import com.wenshuo.agent.log.ExecuteLogUtils;
+import com.wenshuo.agent.transformer.AgentLogClassFileTransformer;
+
+/**
+ * Agent
+ *
+ * @author dingjsh
+ */
+public class Agent {
+
+    public static void premain(String agentArs, Instrumentation inst) {
+        System.out.println("javaagent启动成功，将自动记录方法的执行次数和时间");
+        // 初始化配置
+        ConfigUtils.initProperties(agentArs);
+        ExecuteLogUtils.init();
+        inst.addTransformer(new AgentLogClassFileTransformer());
+    }
+
+}

src/main/java/com/wenshuo/agent/AgentUtils.java

@@ -0,0 +1,180 @@
+/*
+ * @(#)AgentUtils.java  2015-7-27 下午05:26:24
+ * javaagent
+ * Copyright 2015 wenshuo, Inc. All rights reserved.
+ * wenshuo PROPRIETARY/CONFIDENTIAL. Use is subject to license terms.
+ */
+package com.wenshuo.agent;
+
+import java.io.ByteArrayOutputStream;
+import java.io.File;
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.OutputStream;
+import java.io.Writer;
+
+/**
+ * AgentUtils
+ *
+ * @author dingjsh
+ * @time 2015-7-27下午05:26:24
+ */
+public class AgentUtils {
+
+    /**
+     * <p>
+     * Checks if a String is whitespace, empty ("") or null.
+     * </p>
+     *
+     * <pre>
+     * StringUtils.isBlank(null)      = true
+     * StringUtils.isBlank("")        = true
+     * StringUtils.isBlank(" ")       = true
+     * StringUtils.isBlank("bob")     = false
+     * StringUtils.isBlank("  bob  ") = false
+     * </pre>
+     *
+     * @param str the String to check, may be null
+     * @return <code>true</code> if the String is null, empty or whitespace
+     * @since 2.0
+     */
+    public static boolean isBlank(String str) {
+        int strLen;
+        if (str == null || (strLen = str.length()) == 0) {
+            return true;
+        }
+        for (int i = 0; i < strLen; i++) {
+            if (!Character.isWhitespace(str.charAt(i))) {
+                return false;
+            }
+        }
+        return true;
+    }
+
+    /**
+     * <p>
+     * Checks if a String is not empty (""), not null and not whitespace only.
+     * </p>
+     *
+     * <pre>
+     * StringUtils.isNotBlank(null)      = false
+     * StringUtils.isNotBlank("")        = false
+     * StringUtils.isNotBlank(" ")       = false
+     * StringUtils.isNotBlank("bob")     = true
+     * StringUtils.isNotBlank("  bob  ") = true
+     * </pre>
+     *
+     * @param str the String to check, may be null
+     * @return <code>true</code> if the String is not empty and not null and not
+     * whitespace
+     * @since 2.0
+     */
+    static boolean isNotBlank(String str) {
+        return !isBlank(str);
+    }
+
+    /**
+     * Unconditionally close an <code>InputStream</code>.
+     * <p>
+     * Equivalent to {@link InputStream#close()}, except any exceptions will be
+     * ignored. This is typically used in finally blocks.
+     *
+     * @param input the InputStream to close, may be null or already closed
+     */
+    static void closeQuietly(InputStream input) {
+        try {
+            if (input != null) {
+                input.close();
+            }
+        } catch (IOException ioe) {
+            // ignore
+        }
+    }
+
+    public static int copy(InputStream input, OutputStream output)
+            throws IOException {
+        long count = copyLarge(input, output);
+        if (count > Integer.MAX_VALUE) {
+            return -1;
+        }
+        return (int) count;
+    }
+
+    private static long copyLarge(InputStream input, OutputStream output)
+            throws IOException {
+        byte[] buffer = new byte[4096];
+        long count = 0;
+        int n;
+        while (-1 != (n = input.read(buffer))) {
+            output.write(buffer, 0, n);
+            count += n;
+        }
+        return count;
+    }
+
+    // read toByteArray
+    //-----------------------------------------------------------------------
+
+    /**
+     * Get the contents of an <code>InputStream</code> as a <code>byte[]</code>.
+     * <p>
+     * This method buffers the input internally, so there is no need to use a
+     * <code>BufferedInputStream</code>.
+     *
+     * @param input the <code>InputStream</code> to read from
+     * @return the requested byte array
+     * @throws NullPointerException if the input is null
+     * @throws IOException          if an I/O error occurs
+     */
+    public static byte[] toByteArray(InputStream input) throws IOException {
+        ByteArrayOutputStream output = new ByteArrayOutputStream();
+        copy(input, output);
+        return output.toByteArray();
+    }
+
+    /**
+     * Unconditionally close a <code>Writer</code>.
+     * <p>
+     * Equivalent to {@link Writer#close()}, except any exceptions will be ignored.
+     * This is typically used in finally blocks.
+     *
+     * @param output the Writer to close, may be null or already closed
+     */
+    public static void closeQuietly(Writer output) {
+        try {
+            if (output != null) {
+                output.close();
+            }
+        } catch (IOException ioe) {
+            // ignore
+        }
+    }
+
+    /**
+     * Makes a directory, including any necessary but nonexistent parent
+     * directories. If there already exists a file with specified name or
+     * the directory cannot be created then an exception is thrown.
+     *
+     * @param directory directory to create, must not be <code>null</code>
+     * @throws NullPointerException if the directory is <code>null</code>
+     * @throws IOException          if the directory cannot be created
+     */
+    public static void forceMkdir(File directory) throws IOException {
+        if (directory.exists()) {
+            if (directory.isFile()) {
+                String message =
+                        "File "
+                                + directory
+                                + " exists and is "
+                                + "not a directory. Unable to create directory.";
+                throw new IOException(message);
+            }
+        } else {
+            if (!directory.mkdirs()) {
+                String message =
+                        "Unable to create directory " + directory;
+                throw new IOException(message);
+            }
+        }
+    }
+}

src/main/java/com/wenshuo/agent/ConfigConsts.java

@@ -0,0 +1,39 @@
+/*
+ * @(#)ConfigConsts.java	2015-7-27 下午06:06:23
+ * javaagent
+ * Copyright 2015 wenshuo, Inc. All rights reserved.
+ * wenshuo PROPRIETARY/CONFIDENTIAL. Use is subject to license terms.
+ */
+package com.wenshuo.agent;
+
+/**
+ * ConfigConsts
+ * 
+ * @author dingjsh
+ * @time 2015-7-27下午06:06:23
+ */
+public interface ConfigConsts {
+
+    String EXCLUDE_PACKAGE_DEFAULT = "agent.exclude.package.default";
+
+    String EXCLUDE_PACKAGE = "agent.exclude.package";
+
+    String INCLUDE_PACKAGE = "agent.include.package";
+
+    String LOG_FILE = "agent.log.file";
+
+    String LOG_INTERVAL_SECONDS = "agent.log.interval.seconds";
+
+    String EXCLUDE_CLASS_REGEX = "agent.exclude.class.regex";
+
+    String EXCLUDE_CLASS_REGEX_DEFAULT = "agent.exclude.class.regex.default";
+
+    String LOG_AVG_EXECUTE_TIME = "agent.log.avg.execute.time";
+    
+    String POJO_MONITOR_OPEN = "agent.pojo.monitor.open";
+    
+    String LOG_TIME_NANO = "agent.log.nano";
+
+    int DEFAULT_LOG_INTERVAL = 600;
+
+}