xref: /illumos-gate/usr/src/lib/libdtrace_jni/java/docs/examples/TestTarget.java (revision 812316658691ecf37b845d7385ab5145a4a68ff8)
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 TestTarget {
33      public static void
34      main(String[] args)
35      {
36  	if (args.length != 2) {
37  	    System.err.println("Usage: java TestTarget <script> <command>");
38  	    System.exit(2);
39  	}
40  
41  	File file = new File(args[0]);
42  	String command = args[1];
43  
44  	final Consumer consumer = new LocalConsumer();
45  	consumer.addConsumerListener(new ConsumerAdapter() {
46  	    public void dataReceived(DataEvent e) {
47  		System.out.println(e.getProbeData());
48  	    }
49  	    public void consumerStopped(ConsumerEvent e) {
50  		try {
51  		    Aggregate a = consumer.getAggregate();
52  		    for (Aggregation agg : a.asMap().values()) {
53  			for (AggregationRecord rec : agg.asMap().values()) {
54  			    System.out.println(rec.getTuple() + " " +
55  				    rec.getValue());
56  			}
57  		    }
58  		} catch (Exception x) {
59  		    x.printStackTrace();
60  		    System.exit(1);
61  		}
62  		consumer.close();
63  	    }
64  	    public void processStateChanged(ProcessEvent e) {
65  		System.out.println(e.getProcessState());
66  	    }
67  	});
68  
69  	try {
70  	    consumer.open();
71  	    // pid replaces $target variable in D script
72  	    consumer.createProcess(command);
73  	    consumer.compile(file);
74  	    consumer.enable();
75  	    consumer.go();
76  	} catch (Exception e) {
77  	    e.printStackTrace();
78  	    System.exit(1);
79  	}
80      }
81  }
82