usr/src/lib/libdtrace_jni/java/docs/examples/TestAPI2.java

   1 /*
   2  * CDDL HEADER START
   3  *
   4  * The contents of this file are subject to the terms of the
   5  * Common Development and Distribution License (the "License").
   6  * You may not use this file except in compliance with the License.
   7  *
   8  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
   9  * or http://www.opensolaris.org/os/licensing.
  10  * See the License for the specific language governing permissions
  11  * and limitations under the License.
  12  *
  13  * When distributing Covered Code, include this CDDL HEADER in each
  14  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
  15  * If applicable, add the following below this CDDL HEADER, with the
  16  * fields enclosed by brackets "[]" replaced with your own identifying
  17  * information: Portions Copyright [yyyy] [name of copyright owner]
  18  *
  19  * CDDL HEADER END
  20  */
  21
  22 /*
  23  * Copyright 2006 Sun Microsystems, Inc.  All rights reserved.
  24  * Use is subject to license terms.
  25  *
  26  * ident        "%Z%%M% %I%     %E% SMI"
  27  */
  28
  29 import org.opensolaris.os.dtrace.*;
  30 import java.io.File;
  31
  32 public class TestAPI2 {
  33     public static void
  34     main(String[] args)
  35     {
  36         if (args.length < 1) {
  37             System.err.println("Usage: java TestAPI2 <script> " +
  38                     "[ macroargs... ]");
  39             System.exit(2);
  40         }
  41
  42         File file = new File(args[0]);
  43         String[] macroArgs = new String[args.length - 1];
  44         System.arraycopy(args, 1, macroArgs, 0, (args.length - 1));
  45
  46         Consumer consumer = new LocalConsumer();
  47         consumer.addConsumerListener(new ConsumerAdapter() {
  48             public void dataReceived(DataEvent e) {
  49                 // System.out.println(e.getProbeData());
  50                 ProbeData data = e.getProbeData();
  51                 java.util.List < Record > records = data.getRecords();
  52                 for (Record r : records) {
  53                     if (r instanceof ExitRecord) {
  54                     } else {
  55                         System.out.println(r);
  56                     }
  57                 }
  58             }
  59         });
  60
  61         try {
  62             consumer.open();
  63             consumer.compile(file, macroArgs);
  64             consumer.enable();
  65             consumer.go();
  66
  67             Aggregate a;
  68             do {
  69                 Thread.sleep(1000);
  70                 a = consumer.getAggregate();
  71                 if (!a.asMap().isEmpty()) {
  72                     System.out.println(a);
  73                 }
  74             } while (consumer.isRunning());
  75         } catch (Exception e) {
  76             e.printStackTrace();
  77             System.exit(1);
  78         }
  79     }
  80 }