xref: /illumos-gate/usr/src/lib/libdtrace_jni/java/docs/examples/TestTarget.java (revision 20a7641f9918de8574b8b3b47dbe35c4bfc78df1)
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