jdk/test/java/rmi/reliability/benchmark/bench/serial/Bytes.java
changeset 2 90ce3da70b43
child 309 bda219d843f6
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/jdk/test/java/rmi/reliability/benchmark/bench/serial/Bytes.java	Sat Dec 01 00:00:00 2007 +0000
@@ -0,0 +1,82 @@
+/* 
+ * Copyright 1999 Sun Microsystems, Inc.  All Rights Reserved.
+ * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
+ *
+ * This code is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU General Public License version 2 only, as
+ * published by the Free Software Foundation.
+ *
+ * This code is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
+ * version 2 for more details (a copy is included in the LICENSE file that
+ * accompanied this code).
+ *
+ * You should have received a copy of the GNU General Public License version
+ * 2 along with this work; if not, write to the Free Software Foundation,
+ * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
+ *
+ * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
+ * CA 95054 USA or visit www.sun.com if you need additional information or
+ * have any questions.
+ */
+
+/*
+ *
+ */
+
+package bench.serial;
+
+import bench.Benchmark;
+import java.io.ObjectInputStream;
+import java.io.ObjectOutputStream;
+
+/**
+ * Benchmark for testing speed of byte reads/writes.
+ */
+public class Bytes implements Benchmark {
+    
+    /**
+     * Write and read byte values to/from a stream.  The benchmark is run in
+     * batches: each "batch" consists of a fixed number of read/write cycles,
+     * and the stream is flushed (and underlying stream buffer cleared) in
+     * between each batch.
+     * Arguments: <# batches> <# cycles per batch>
+     */
+    public long run(String[] args) throws Exception {
+	int nbatches = Integer.parseInt(args[0]);
+	int ncycles = Integer.parseInt(args[1]);
+	StreamBuffer sbuf = new StreamBuffer();
+	ObjectOutputStream oout = 
+	    new ObjectOutputStream(sbuf.getOutputStream());
+	ObjectInputStream oin =
+	    new ObjectInputStream(sbuf.getInputStream());
+	
+	doReps(oout, oin, sbuf, 1, ncycles);	// warmup
+
+	long start = System.currentTimeMillis();
+	doReps(oout, oin, sbuf, nbatches, ncycles);
+        return System.currentTimeMillis() - start;
+    }
+    
+    /**
+     * Run benchmark for given number of batches, with given number of cycles
+     * for each batch.
+     */
+    void doReps(ObjectOutputStream oout, ObjectInputStream oin,
+	        StreamBuffer sbuf, int nbatches, int ncycles)
+	throws Exception
+    {
+	for (int i = 0; i < nbatches; i++) {
+	    sbuf.reset();
+	    for (int j = 0; j < ncycles; j++) {
+		oout.writeByte(0);
+	    }
+	    oout.flush();
+	    for (int j = 0; j < ncycles; j++) {
+		oin.readByte();
+	    }
+	}
+    }
+}
+