jdk-24/test/micro/org/openjdk/bench/java/io/DataOutputStreamBench.java
Shaojin Wen b42fbf43df 8339699: Optimize DataOutputStream writeUTF
Reviewed-by: liach, bpb
2024-10-04 22:35:03 +00:00

92 lines
3.1 KiB
Java

/*
* Copyright (c) 2024, Alibaba Group Holding Limited. 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 Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
* or visit www.oracle.com if you need additional information or have any
* questions.
*/
package org.openjdk.bench.java.io;
import org.openjdk.jmh.annotations.*;
import org.openjdk.jmh.infra.Blackhole;
import java.io.*;
import java.nio.charset.StandardCharsets;
import java.util.HexFormat;
import java.util.concurrent.TimeUnit;
@BenchmarkMode(Mode.AverageTime)
@OutputTimeUnit(TimeUnit.MICROSECONDS)
@Fork(2)
@Measurement(iterations = 6, time = 1)
@Warmup(iterations = 4, time = 2)
@State(Scope.Thread)
public class DataOutputStreamBench {
@Param({"ascii", "utf8_2_bytes", "utf8_3_bytes", "emoji"})
public String charType;
ByteArrayOutputStream bytesOutput;
DataOutputStream dataOutput;
ObjectOutputStream objectOutput;
String[] strings;
@Setup(Level.Trial)
public void setup() throws Exception {
byte[] bytes = HexFormat.of().parseHex(
switch (charType) {
case "ascii" -> "78";
case "utf8_2_bytes" -> "c2a9";
case "utf8_3_bytes" -> "e6b8a9";
case "emoji" -> "e29da3efb88f";
default -> throw new IllegalArgumentException("bad charType: " + charType);
}
);
String s = new String(bytes, 0, bytes.length, StandardCharsets.UTF_8);
strings = new String[128];
for (int i = 0; i < strings.length; i++) {
strings[i] = "A".repeat(i).concat(s.repeat(i));
}
bytesOutput = new ByteArrayOutputStream(1024 * 64);
dataOutput = new DataOutputStream(bytesOutput);
objectOutput = new ObjectOutputStream(bytesOutput);
}
@Benchmark
public void dataOutwriteUTF(Blackhole bh) throws Exception {
bytesOutput.reset();
for (var s : strings) {
dataOutput.writeUTF(s);
}
dataOutput.flush();
bh.consume(bytesOutput.size());
}
@Benchmark
public void objectWriteUTF(Blackhole bh) throws Exception {
bytesOutput.reset();
for (var s : strings) {
objectOutput.writeUTF(s);
}
objectOutput.flush();
bh.consume(bytesOutput.size());
}
}