1 /*
   2  * Copyright (c) 1999, 2013, Oracle and/or its affiliates. All rights reserved.
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * This code is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License version 2 only, as
   7  * published by the Free Software Foundation.
   8  *
   9  * This code is distributed in the hope that it will be useful, but WITHOUT
  10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  12  * version 2 for more details (a copy is included in the LICENSE file that
  13  * accompanied this code).
  14  *
  15  * You should have received a copy of the GNU General Public License version
  16  * 2 along with this work; if not, write to the Free Software Foundation,
  17  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  18  *
  19  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  20  * or visit www.oracle.com if you need additional information or have any
  21  * questions.
  22  */
  23 
  24 /*
  25  * @test
  26  * @summary The RMI Serialization benchmark test. This java class is used to run
  27  *          the test under JTREG.
  28  * @library ../../
  29  * @build bench.BenchInfo bench.HtmlReporter bench.Util bench.Benchmark 
  30  * @build bench.Reporter bench.XmlReporter bench.ConfigFormatException 
  31  * @build bench.Harness bench.TextReporter
  32  * @build bench.serial.BooleanArrays bench.serial.Booleans
  33  * @build bench.serial.ByteArrays bench.serial.Bytes bench.serial.CharArrays
  34  * @build bench.serial.Chars bench.serial.ClassDesc bench.serial.Cons
  35  * @build bench.serial.CustomDefaultObjTrees bench.serial.CustomObjTrees
  36  * @build bench.serial.DoubleArrays bench.serial.Doubles
  37  * @build bench.serial.ExternObjTrees bench.serial.FloatArrays
  38  * @build bench.serial.Floats bench.serial.GetPutFieldTrees
  39  * @build bench.serial.IntArrays bench.serial.Ints bench.serial.LongArrays
  40  * @build bench.serial.Longs bench.serial.Main bench.serial.ObjArrays
  41  * @build bench.serial.ObjTrees bench.serial.ProxyArrays
  42  * @build bench.serial.ProxyClassDesc bench.serial.RepeatObjs
  43  * @build bench.serial.ReplaceTrees bench.serial.ShortArrays
  44  * @build bench.serial.Shorts bench.serial.SmallObjTrees
  45  * @build bench.serial.StreamBuffer bench.serial.Strings
  46  * @run main/othervm/timeout=1800 bench.serial.Main -c jtreg-config
  47  * @author Mike Warres, Nigel Daley
  48  */
  49 
  50 package bench.serial;
  51 
  52 import bench.ConfigFormatException;
  53 import bench.Harness;
  54 import bench.HtmlReporter;
  55 import bench.Reporter;
  56 import bench.TextReporter;
  57 import bench.XmlReporter;
  58 import java.io.File;
  59 import java.io.FileInputStream;
  60 import java.io.FileNotFoundException;
  61 import java.io.FileOutputStream;
  62 import java.io.InputStream;
  63 import java.io.IOException;
  64 import java.io.OutputStream;
  65 import java.io.PrintStream;
  66 import java.util.Timer;
  67 import java.util.TimerTask;
  68 
  69 /**
  70  * Object serialization benchmark mainline.
  71  */
  72 public class Main {
  73 
  74     static final String CONFFILE = "config";
  75     static final String VERSION = "1.3";
  76     static final String TEST_SRC_PATH = System.getProperty("test.src") + File.separator;
  77 
  78     static final int TEXT = 0;
  79     static final int HTML = 1;
  80     static final int XML = 2;
  81 
  82     static boolean verbose;
  83     static boolean list;
  84     static boolean exitOnTimer;
  85     static int testDurationSeconds;
  86     static volatile boolean exitRequested;
  87     static Timer timer;
  88     static int format = TEXT;
  89     static InputStream confstr;
  90     static OutputStream repstr;
  91     static Harness harness;
  92     static Reporter reporter;
  93 
  94     /**
  95      * Print help message.
  96      */
  97     static void usage() {
  98         PrintStream p = System.err;
  99         p.println("\nUsage: java -jar serialbench.jar [-options]");
 100         p.println("\nwhere options are:");
 101         p.println("  -h              print this message");
 102         p.println("  -v              verbose mode");
 103         p.println("  -l              list configuration file");
 104         p.println("  -t <num hours>  repeat benchmarks for specified number of hours");
 105         p.println("  -o <file>       specify output file");
 106         p.println("  -c <file>       specify (non-default) configuration file");
 107         p.println("  -html           format output as html (default is text)");
 108         p.println("  -xml            format output as xml");
 109     }
 110 
 111     /**
 112      * Throw RuntimeException that wrap message.
 113      *
 114      * @param mesg a message will be wrapped in the RuntimeException.
 115      */
 116     static void die(String mesg) {
 117         throw new RuntimeException(mesg);

 118     }
 119 
 120     /**
 121      * Mainline parses command line, then hands off to benchmark harness.
 122      * 
 123      * @param args
 124      */
 125     public static void main(String[] args) {
 126         parseArgs(args);
 127         setupStreams();
 128         if (list) {
 129             listConfig();
 130         } else {
 131             setupHarness();
 132             setupReporter();
 133             if (exitOnTimer) {
 134                 setupTimer(testDurationSeconds);
 135                 do {
 136                     runBenchmarks();
 137                 } while (!exitRequested);



 138             } else {
 139                 runBenchmarks();

 140             }
 141         }
 142     }
 143 
 144     /**
 145      * Parse command-line arguments.
 146      */
 147     static void parseArgs(String[] args) {
 148         for (int i = 0; i < args.length; i++) {
 149             switch (args[i]) {
 150                 case "-h":
 151                     usage();
 152                     System.exit(0);
 153                     break;
 154                 case "-v":
 155                     verbose = true;
 156                     break;
 157                 case "-l":
 158                     list = true;
 159                     break;
 160                 case "-t":
 161                     if (++i >= args.length)
 162                         die("Error: no timeout value specified");
 163                     try {
 164                         exitOnTimer = true;
 165                         testDurationSeconds = Integer.parseInt(args[i]) * 3600;
 166                     } catch (NumberFormatException e) {
 167                         die("Error: unable to determine timeout value");
 168                     }   break;
 169                 case "-o":
 170                     if (++i >= args.length)
 171                         die("Error: no output file specified");
 172                     try {
 173                         repstr = new FileOutputStream(args[i]);
 174                     } catch (FileNotFoundException e) {
 175                         die("Error: unable to open \"" + args[i] + "\"");
 176                     }   break;
 177                 case "-c":
 178                     if (++i >= args.length)
 179                         die("Error: no config file specified");
 180                     String confFileName = TEST_SRC_PATH + args[i];
 181                     try {
 182                         confstr = new FileInputStream(confFileName);
 183                     } catch (FileNotFoundException e) {
 184                         die("Error: unable to open \"" + confFileName + "\"");
 185                     }   break;
 186                 case "-html":
 187                     if (format != TEXT)
 188                         die("Error: conflicting formats");
 189                     format = HTML;
 190                     break;
 191                 case "-xml":
 192                     if (format != TEXT)
 193                         die("Error: conflicting formats");
 194                     format = XML;
 195                     break;
 196                 default:
 197                     usage();
 198                     die("Illegal option: \"" + args[i] + "\"");
 199             }
 200         }
 201     }
 202 
 203     /**
 204      * Set up configuration file and report streams, if not set already.
 205      */
 206     static void setupStreams() {
 207         if (repstr == null)
 208             repstr = System.out;
 209         if (confstr == null)
 210             confstr = Main.class.getResourceAsStream(TEST_SRC_PATH + CONFFILE);
 211         if (confstr == null)
 212             die("Error: unable to find default config file");
 213     }
 214 
 215     /**
 216      * Print contents of configuration file to selected output stream.
 217      */
 218     static void listConfig() {
 219         try {
 220             byte[] buf = new byte[256];
 221             int len;
 222             while ((len = confstr.read(buf)) != -1)
 223                 repstr.write(buf, 0, len);
 224         } catch (IOException e) {
 225             die("Error: failed to list config file");
 226         }
 227     }
 228 
 229     /**
 230      * Set up the timer to end the test.
 231      *
 232      * @param delay the amount of delay, in seconds, before requesting
 233      * the process exit
 234      */
 235     static void setupTimer(int delay) {
 236         timer = new Timer(true);
 237         timer.schedule(
 238             new TimerTask() {
 239                 @Override
 240                 public void run() {
 241                     exitRequested = true;
 242                 }
 243             },
 244             delay * 1000);
 245     }
 246 
 247     /**
 248      * Set up benchmark harness.
 249      */
 250     static void setupHarness() {
 251         try {
 252             harness = new Harness(confstr);
 253         } catch (ConfigFormatException e) {
 254             String errmsg = e.getMessage();
 255             if (errmsg != null) {
 256                 die("Error parsing config file: " + errmsg);
 257             } else {
 258                 die("Error: illegal config file syntax");
 259             }
 260         } catch (IOException e) {
 261             die("Error: failed to read config file");
 262         }
 263     }
 264 
 265     /**
 266      * Setup benchmark reporter.
 267      */
 268     static void setupReporter() {
 269         String title = "Object Serialization Benchmark, v" + VERSION;
 270         switch (format) {
 271             case TEXT:
 272                 reporter = new TextReporter(repstr, title);
 273                 break;
 274 
 275             case HTML:
 276                 reporter = new HtmlReporter(repstr, title);
 277                 break;
 278 
 279             case XML:
 280                 reporter = new XmlReporter(repstr, title);
 281                 break;
 282 
 283             default:
 284                 die("Error: unrecognized format type");
 285         }
 286     }
 287 
 288     /**
 289      * Run benchmarks.
 290      */
 291     static void runBenchmarks() {
 292         harness.runBenchmarks(reporter, verbose);
 293     }
 294 }
--- EOF ---