Flywheel/src/main/java/com/jozufozu/flywheel/util/StringUtil.java

98 lines
2.6 KiB
Java
Raw Normal View History

package com.jozufozu.flywheel.util;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.nio.Buffer;
import java.nio.ByteBuffer;
import java.nio.channels.Channels;
import java.nio.channels.FileChannel;
import java.nio.channels.ReadableByteChannel;
import java.text.DecimalFormat;
import java.text.NumberFormat;
import java.util.Arrays;
import java.util.stream.Collectors;
import org.lwjgl.system.MemoryUtil;
public class StringUtil {
private static final NumberFormat timeFormat = new DecimalFormat("#0.000");
public static String formatTime(long ns) {
if (ns < 1000) {
return ns + " ns";
} else if (ns < 1000000) {
return timeFormat.format(ns / 1000.) + " μs";
} else if (ns < 1000000000) {
return timeFormat.format(ns / 1000000.) + " ms";
} else {
return timeFormat.format(ns / 1000000000.) + " s";
}
}
public static String args(String functionName, Object... args) {
return functionName + '(' + Arrays.stream(args)
.map(Object::toString)
.collect(Collectors.joining(", ")) + ')';
}
public static String trimEnd(String value) {
int len = value.length();
int st = 0;
while ((st < len) && Character.isWhitespace(value.charAt(len - 1))) {
len--;
}
return value.substring(0, len);
}
public static String readToString(InputStream is) {
ByteBuffer bytebuffer = null;
try {
bytebuffer = readToBuffer(is);
int i = bytebuffer.position();
((Buffer) bytebuffer).rewind();
return MemoryUtil.memASCII(bytebuffer, i);
} catch (IOException ignored) {
} finally {
if (bytebuffer != null) {
MemoryUtil.memFree(bytebuffer);
}
}
return null;
}
public static ByteBuffer readToBuffer(InputStream is) throws IOException {
if (is instanceof FileInputStream fileinputstream) {
return readFileInputStream(fileinputstream);
} else {
return readInputStream(is);
}
}
private static ByteBuffer readInputStream(InputStream is) throws IOException {
ByteBuffer bytebuffer = MemoryUtil.memAlloc(8192);
ReadableByteChannel readablebytechannel = Channels.newChannel(is);
while (readablebytechannel.read(bytebuffer) != -1) {
if (bytebuffer.remaining() == 0) {
bytebuffer = MemoryUtil.memRealloc(bytebuffer, bytebuffer.capacity() * 2);
}
}
return bytebuffer;
}
private static ByteBuffer readFileInputStream(FileInputStream fileinputstream) throws IOException {
FileChannel filechannel = fileinputstream.getChannel();
ByteBuffer bytebuffer = MemoryUtil.memAlloc((int) filechannel.size() + 1);
while (filechannel.read(bytebuffer) != -1) {
}
return bytebuffer;
}
}