package org.jelly.util;
import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.FileReader;
import java.io.FileWriter;
import java.io.InputStream;
import java.io.OutputStream;
import java.io.StringReader;
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import org.jelly.code.IndexCode;
import org.jelly.exception.ExecutetimeException;
/**
* 文件经常使用的操作的工具类
* @since 1.0.0
*/
public class FileUtil {
private FileUtil(){}
private static final String LOCK = "_LOCK";
private static final int BUFFER_SIZE = 1024 * 1024 / 2;
/**
* <des> 文件拷贝, 方法调用结束或抛出异常, 都将关闭输入输出流, 输出流flush </des>
* @param in InputStream
* @param out OutputStream
* @since 1.0.0
*/
public static void copyFile(InputStream in, OutputStream out){
try {
byte[] buffer = new byte[BUFFER_SIZE];
int read;
while((read = in.read(buffer)) != IndexCode.EOF.toCode()){
out.write(buffer, 0, read);
}
} catch (Throwable e) {
throw new ExecutetimeException(e);
} finally {
closeStream(in, out);
}
}
/**
* <des> 写出, 方法调用结束或抛出异常, 都将关闭输入输出流, 输出流flush </des>
* @since 1.6.0
*/
public static void write(InputStream in, OutputStream out){
copyFile(in, out);
}
/**
* <des> 字符串内容写出, 方法调用结束或抛出异常, 都将关闭输入输出流, 输出流flush </des>
* @since 1.6.0
*/
public static void write(String text, OutputStream out){
try {
out.write(text.getBytes());
} catch (Throwable e) {
throw new ExecutetimeException(e);
} finally {
closeStream(out);
}
}
/**
* <des> 字符串内容写出到文件, 方法调用结束或抛出异常, 都将关闭输入输出流, 输出流flush </des>
* @since 1.6.0
*/
public static void write(String text, File file){
BufferedReader reader = null;
BufferedWriter writer = null;
try {
reader = new BufferedReader(new StringReader(text));
writer = new BufferedWriter(new FileWriter(file));
char[] buffer = new char[BUFFER_SIZE];
int read;
while((read = reader.read(buffer)) != IndexCode.EOF.toCode()){
writer.write(buffer, 0, read);
}
} catch (Throwable e) {
throw new ExecutetimeException(e);
} finally {
closeStream(reader, writer);
}
}
/**
* <des> 读取获得文件内容 </des>
* @param pathname 文件全路径名称
* @return 读取获得的文件内容
* @since 1.0.0
*/
public static String readFile(String pathname){
return readFile(new File(pathname));
}
/**
* <des> 读取获得文件内容 </des>
* @param file 文件对象
* @return 读取获得的文件内容
* @since 1.0.0
*/
public static String readFile(File file){
String[] contents = readLineFile(file);
StringBuilder builder = new StringBuilder();
for(String content : contents){
builder.append(content);
}
return builder.toString();
}
/**
* <des> 按行读取文件, 文件的每一行作为数组的一个元素 </des>
* @param pathname 文件全路径名称
* @return 数组
* @since 1.0.0
*/
public static String[] readLineFile(String pathname){
return readLineFile(new File(pathname));
}
/**
* <des> 按行读取文件, 文件的每一行作为数组的一个元素 </des>
* @param file 文件对象
* @return 数组
* @since 1.0.0
*/
public static String[] readLineFile(File file){
BufferedReader reader = null;
try {
reader = new BufferedReader(new FileReader(file));
LinkedList<String> list = new LinkedList<String>();
String read;
while((read = reader.readLine()) != null){
list.add(read);
}
return ArrayUtil.asArray(list);
} catch (Throwable e) {
throw new ExecutetimeException(e);
} finally {
closeStream(reader);
}
}
/**
* <des> 获取文件夹下全部的文件对象 </des>
* @param directory 文件文件夹对象
* @return 文件对象列表
* @since 1.0.0
*/
public static List<File> listFiles(File directory){
return listFiles(directory, null);
}
/**
* <des> 获取文件夹下全部由參数指定类型的文件对象 </des>
* @param source 文件文件夹对象
* @param filter 期望得到的文件的后缀名称
* @return 文件对象列表
* @since 1.0.0
*/
public static List<File> listFiles(File source, String filter){
File[] fileList = source.listFiles();
List<File> list = new ArrayList<File>();
filter = filter == null ? null : filter.toLowerCase();
if(fileList != null && fileList.length > 0){
for(File file : fileList){
if(file.isFile()){
add(list, file, filter);
}else if(file.isDirectory()){
list.addAll(listFiles(file, filter));
}
}
}else if(source.isFile()){
add(list, source, filter);
}
return list;
}
/**
* <des> 在參数指定的文件夹中查找文件 </des>
* @param dirpath 文件文件夹全路径名称
* @param filename 查找的文件名
* @return 若查找的到, 则返还该文件对象, 若查找不到, 则返还null
* @since 1.0.0
*/
public static File findFile(String dirpath, String filename){
return findFile(new File(dirpath), filename);
}
/**
* <des> 在參数指定的文件夹中查找文件 </des>
* @param directory 文件文件夹对象
* @param filename 查找的文件名
* @return 若查找的到, 则返还该文件对象, 若查找不到, 则返还null
* @since 1.0.0
*/
public static File findFile(File directory, String filename){
String filter = StringUtil.substringAfterLastWith(filename, ".");
List<File> files = listFiles(directory, filter);
if(files.size() == 0) return null;
for(File file : files){
if(file.getAbsolutePath().endsWith(filename)){
return file;
}
}
return null;
}
/**
* <des> 删除文件或文件夹 </des>
* @param file 文件或文件文件夹对象
* @since 1.0.0
*/
public static void delete(File file){
synchronized (LOCK) {
if(file.isFile()){
file.delete();
}else if(file.isDirectory()){
File[] files = file.listFiles();
for(File item : files){
delete(item);
}
}
file.delete();
}
}
/**
* <des> 创建文件夹 </des>
* @param pathname 文件夹名称
* @return 若文件夹不存在则创建, 若存在则直接返还true
* @since 1.0.0
*/
public static boolean createDir(String pathname){
return createDir(new File(pathname));
}
/**
* <des> 创建文件夹 </des>
* @param dir 文件夹对象
* @return 若文件夹不存在则创建, 若存在则直接返还true
* @since 1.0.0
*/
public static boolean createDir(File dir){
synchronized (LOCK) {
if(!dir.exists()){
return dir.mkdirs();
}
return true;
}
}
/**
* <des> 获取FileInputStream实例 </des>
* @since 1.0.0
*/
public static FileInputStream getFileInputStream(String pathname){
return getFileInputStream(new File(pathname));
}
/**
* <p><des> 获取FileInputStream实例 </des></p>
* @since 1.0.0
*/
public static FileInputStream getFileInputStream(File file){
try {
return new FileInputStream(file);
} catch (FileNotFoundException e) {
throw new ExecutetimeException(e);
}
}
/**
* <p><des> 获取FileOutputStream实例 </des></p>
* @since 1.0.0
*/
public static FileOutputStream getFileOutputStream(String pathname){
return getFileOutputStream(new File(pathname), false);
}
/**
* <p><des> 获取FileOutputStream实例 </des></p>
* @since 1.0.0
*/
public static FileOutputStream getFileOutputStream(String pathname, boolean append){
return getFileOutputStream(new File(pathname), append);
}
/**
* <p><des> 获取FileOutputStream实例 </des></p>
* @since 1.0.0
*/
public static FileOutputStream getFileOutputStream(File file){
return getFileOutputStream(file, false);
}
/**
* <p><des> 获取FileOutputStream实例 </des></p>
* @since 1.0.0
*/
public static FileOutputStream getFileOutputStream(File file, boolean append){
try {
return new FileOutputStream(file, append);
} catch (FileNotFoundException e) {
throw new ExecutetimeException(e);
}
}
// 加入文件到列表
private static void add(List<File> list, File file, String filter){
if(filter == null){
list.add(file);
}else if(file.getAbsolutePath().toLowerCase().endsWith(filter)){
list.add(file);
}
}
}
