예제 #1
0
  public static void premain(String agentArgs, Instrumentation inst) {
    String[] args = agentArgs.split(" ");

    if (System.getProperty("java.version").contains("1.8"))
      throw new UnsupportedOperationException(
          "Java 8: https://github.com/fommil/lions-share/issues/7");

    String filename = args[0];
    File outFile = new File(filename);
    if (outFile.delete()) out.println("[AGENT] Deleted an existing " + outFile.getAbsolutePath());

    final Long period = Long.parseLong(args[1]);
    checkArgument(period > 0, "period must be greater than zero seconds");

    out.println("[AGENT] Writing allocation data to " + outFile.getAbsolutePath());
    out.println("[AGENT] Taking snapshots every " + period + " seconds");

    Map<String, Long> rates = Maps.newHashMap();
    if (args.length > 2)
      for (String arg : args[2].split(",")) {
        String[] parts = arg.split(":");
        Long sampleRate = Long.parseLong(parts[1]);
        checkArgument(
            !parts[0].contains(" "), parts[0] + " is not a valid type (spaces not allowed)");
        checkArgument(
            !parts[0].contains("."), parts[0] + " is not a valid type (replace dots with slashes)");
        out.println("[AGENT] " + parts[0] + " will be sampled every " + sampleRate + " bytes");
        rates.put(parts[0], sampleRate);
      }

    AllocationInstrumenter.premain(agentArgs, inst);
    final AllocationSampler sampler = new AllocationSampler(rates, rates.keySet());
    AllocationRecorder.addSampler(sampler);

    final AllocationPrinter printer = new AllocationPrinter(sampler, outFile);

    // HACK futile attempt to clear out stats about JVM internals...
    executor.schedule(
        new Runnable() {
          @Override
          public void run() {
            sampler.clear();
            executor.scheduleWithFixedDelay(printer, period, period, SECONDS);
          }
        },
        period,
        SECONDS);

    Runtime.getRuntime()
        .addShutdownHook(
            new Thread() {
              @Override
              public void run() {
                // tries to get logs on shutdown
                printer.run();
              }
            });
  }
예제 #2
0
 protected AllocationWorker() {
   AllocationRecorder.addSampler(
       new Sampler() {
         @Override
         public void sampleAllocation(int arrayCount, String desc, Object newObj, long size) {
           if (recordAllocations) {
             allocationCount++;
             allocationSize += size;
           }
         }
       });
 }
 public static void main(String[] args) throws Exception {
   AllocationRecorder.addSampler(
       new Sampler() {
         public void sampleAllocation(int count, String desc, Object newObj, long size) {
           System.out.println(
               "I just allocated the object "
                   + newObj
                   + " of type "
                   + desc
                   + " whose size is "
                   + size);
           if (count != -1) {
             System.out.println("It's an array of size " + count);
           }
         }
       });
   new String("foo");
 }