Home | History | Annotate | Download | only in examples
      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