cn.wandersnail.common.IOUtils Maven / Gradle / Ivy
package cn.wandersnail.common;
import java.io.BufferedReader;
import java.io.ByteArrayOutputStream;
import java.io.Closeable;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.List;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
public class IOUtils {
/**
* 关闭一个或多个流对象
*
* @param closeables 可关闭的流对象列表
*/
public static void close(Closeable... closeables) throws IOException {
if (closeables != null) {
for (Closeable closeable : closeables) {
if (closeable != null) {
closeable.close();
}
}
}
}
/**
* 关闭一个或多个流对象
*
* @param closeables 可关闭的流对象列表
*/
public static void closeQuietly(Closeable... closeables) {
try {
close(closeables);
} catch (IOException ignore) {
}
}
/**
* 使用默认编码格式从输入流中获取字符串
*
* @param in 输入流
*/
public static String getString(InputStream in) throws IOException {
return getString(in, null);
}
/**
* 从输入流中获取字符串
*
* @param in 输入流，不会主动关闭
* @param enc 返回的字符串采用的编码格式, 如果为null则使用平台默认的编码格式
*/
public static String getString(InputStream in, String enc) throws IOException {
ByteArrayOutputStream out = new ByteArrayOutputStream();
byte[] buf = new byte[10240];
int len;
while ((len = in.read(buf)) != -1) {
out.write(buf, 0, len);
}
if (enc == null) {
return out.toString();
}
return out.toString(enc);
}
/**
* 从输入流中获取符合指定正则表达式的字符串
*
* @param in 输入流，不会主动关闭
* @param regex 用来筛选字符串的正则表达式
* @return 筛选反的字符串集合
*/
public static List getStrings(InputStream in, String regex) throws IOException {
return getStrings(in, regex, null);
}
/**
* 从输入流中获取符合指定正则表达式的字符串
*
* @param in 输入流，不会主动关闭
* @param regex 用来筛选字符串的正则表达式
* @param enc 用来解析输入流数据的字符集
* @return 筛选反的字符串集合
*/
public static List getStrings(InputStream in, String regex, String enc) throws IOException {
List strs = new ArrayList<>();
BufferedReader bfIn;
if (enc == null) {
bfIn = new BufferedReader(new InputStreamReader(in));
} else {
bfIn = new BufferedReader(new InputStreamReader(in, enc));
}
String line;
Pattern p = Pattern.compile(regex);
while ((line = bfIn.readLine()) != null) {
Matcher m = p.matcher(line);
while (m.find()) {
strs.add(m.group());
}
}
return strs;
}
}