23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee/*
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee * CDDL HEADER START
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee *
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee * The contents of this file are subject to the terms of the
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee * Common Development and Distribution License (the "License").
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee * You may not use this file except in compliance with the License.
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee *
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee * or http://www.opensolaris.org/os/licensing.
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee * See the License for the specific language governing permissions
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee * and limitations under the License.
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee *
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee * When distributing Covered Code, include this CDDL HEADER in each
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee * If applicable, add the following below this CDDL HEADER, with the
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee * fields enclosed by brackets "[]" replaced with your own identifying
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee * information: Portions Copyright [yyyy] [name of copyright owner]
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee *
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee * CDDL HEADER END
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee */
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee/*
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee * Copyright 2006 Sun Microsystems, Inc. All rights reserved.
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee * Use is subject to license terms.
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee *
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee * ident "%Z%%M% %I% %E% SMI"
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee */
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomeeimport java.io.File;
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomeeimport java.io.IOException;
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomeeimport java.util.List;
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomeeimport org.opensolaris.os.dtrace.*;
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee/**
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee * Regression for multi-aggregation printa() corner cases.
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee */
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomeepublic class TestMultiAggPrinta {
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee static int exitStatus;
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee // Gets a string representation of the given PrintaRecord minus the
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee // timestamp of the aggregate snapshot, so that the output is
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee // comparable across multiple test runs.
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee static String
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee printaRecordString(PrintaRecord rec)
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee {
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee StringBuffer buf = new StringBuffer();
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee buf.append(PrintaRecord.class.getName());
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee buf.append("[aggregations = ");
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee buf.append(rec.getAggregations());
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee buf.append(", formattedStrings = ");
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee buf.append(rec.getFormattedStrings());
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee buf.append(", tuples = ");
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee buf.append(rec.getTuples());
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee buf.append(", output = ");
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee buf.append(rec.getOutput());
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee buf.append(']');
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee return buf.toString();
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee }
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee static String
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee probeDataString(ProbeData data)
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee {
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee StringBuffer buf = new StringBuffer();
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee buf.append(ProbeData.class.getName());
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee buf.append("[epid = ");
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee buf.append(data.getEnabledProbeID());
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee // Do not include cpu, since it can change across test runs
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee buf.append(", enabledProbeDescription = ");
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee buf.append(data.getEnabledProbeDescription());
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee buf.append(", flow = ");
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee buf.append(data.getFlow());
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee buf.append(", records = ");
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee List <Record> records = data.getRecords();
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee Record record;
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee Object value;
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee buf.append('[');
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee for (int i = 0; i < records.size(); ++i) {
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee if (i > 0) {
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee buf.append(", ");
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee }
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee record = records.get(i);
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee if (record instanceof ValueRecord) {
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee value = ((ValueRecord)record).getValue();
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee if (value instanceof String) {
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee buf.append("\"");
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee buf.append((String)value);
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee buf.append("\"");
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee } else {
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee buf.append(record);
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee }
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee } else if (record instanceof PrintaRecord) {
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee PrintaRecord printa = (PrintaRecord)record;
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee buf.append(printaRecordString(printa));
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee } else {
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee buf.append(record);
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee }
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee }
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee buf.append(']');
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee return buf.toString();
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee }
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee public static void
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee main(String[] args)
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee {
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee if (args.length != 1) {
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee System.err.println("usage: java TestMultiAggPrinta <script>");
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee System.exit(2);
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee }
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee final Consumer consumer = new LocalConsumer();
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee consumer.addConsumerListener(new ConsumerAdapter() {
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee public void dataReceived(DataEvent e) {
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee ProbeData data = e.getProbeData();
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee List <Record> records = data.getRecords();
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee for (Record r : records) {
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee if (r instanceof ExitRecord) {
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee ExitRecord exitRecord = (ExitRecord)r;
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee exitStatus = exitRecord.getStatus();
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee }
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee }
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee System.out.println(probeDataString(e.getProbeData()));
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee }
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee public void consumerStopped(ConsumerEvent e) {
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee consumer.close();
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee System.exit(exitStatus);
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee }
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee });
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee File file = new File(args[0]);
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee try {
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee consumer.open();
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee consumer.compile(file);
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee consumer.enable();
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee consumer.go();
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee } catch (DTraceException e) {
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee e.printStackTrace();
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee System.exit(1);
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee } catch (IOException e) {
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee e.printStackTrace();
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee System.exit(1);
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee }
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee }
23b5c241225a8ade2b6b9f06ebb891ee459e3b02tomee}