jdk-24/test/micro/org/openjdk/bench/javax/xml/AbstractXMLMicro.java
Mikael Vidstedt 3318314c7c 8237589: Fix copyright header formatting
Reviewed-by: jjg, serb, iris, jwilhelm
2020-01-21 14:23:01 -08:00

100 lines
3.3 KiB
Java

/*
* Copyright (c) 2014, Oracle and/or its affiliates. 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.javax.xml;
import org.openjdk.jmh.annotations.Param;
import org.openjdk.jmh.annotations.Scope;
import org.openjdk.jmh.annotations.State;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.URISyntaxException;
import java.util.concurrent.ConcurrentHashMap;
/**
* Abstract base class with functionality and constants used by the XML micros.
*/
@State(Scope.Benchmark)
public abstract class AbstractXMLMicro {
public static final String BUILDIMPL = "build-impl.xml";
public static final String LOGCOMP = "log_comp.xml";
public static final String MESSAGE12 = "message_12.xml";
public static final String MSGATTACH = "msgAttach.xml";
public static final String REZ = "reZ003vExc23082309.xml";
protected static final ConcurrentHashMap<String, byte[]> byteCache = new ConcurrentHashMap<>();
@Param({BUILDIMPL,LOGCOMP,MESSAGE12,MSGATTACH,REZ})
protected String doc;
/**
* Gets a given InputStream as a byte-array.
*
* @param is stream to read from
* @return byte-array
* @throws IOException if things go crazy-crazy
*/
private static byte[] getBytes(InputStream is) throws IOException {
ByteArrayOutputStream baos = new ByteArrayOutputStream();
byte[] b = new byte[1024];
int available = is.available();
while (available > 0) {
int read = Math.min(b.length, available);
int actuallyRead = is.read(b, 0, read);
baos.write(b, 0, actuallyRead);
available = is.available();
}
is.close();
byte array[] = baos.toByteArray();
baos.close();
return array;
}
/**
* Gets a given resource as a byte-array.
*
* @param name resource to fetch
* @return byte-array
* @throws IOException if things go crazy-crazy
* @throws URISyntaxException if resource given doesn't match syntax
*/
protected byte[] getFileBytesFromResource(String name) throws IOException, URISyntaxException {
byte[] bytes = byteCache.get(name);
if (bytes == null) {
bytes = getBytes(this.getClass().getResourceAsStream("/"
+ AbstractXMLMicro.class.getPackage().getName().replace(".", "/")
+ "/" + name));
byteCache.put(name, bytes);
}
return bytes;
}
}