AttachmentHelper附件管理工具类

快来打我* 2022-11-14 11:49 144阅读 0赞

package com.hhh.platform.advisors.attachment;
    
    import java.io.File;
    import java.io.FileInputStream;
    import java.io.FileOutputStream;
    import java.io.IOException;
    import java.io.InputStream;
    import java.io.OutputStream;
    
    import org.apache.commons.logging.Log;
    import org.apache.commons.logging.LogFactory;
    
    public class AttachmentHelper {
    	private static Log log = LogFactory.getLog(AttachmentHelper.class);
    	public static String SEP = "/";
    
    	public static void copyStream(InputStream input, OutputStream output) throws Exception {
    		byte[] buffer = new byte[1024 * 4];
    		int n = 0;
    		try {
    			while ((n = input.read(buffer)) != -1) {
    				output.write(buffer, 0, n);
    			}
    		} catch (IOException e) {
    			int code = (int) System.currentTimeMillis();
    			log.error("复制文件流出错(" + code + ")", e);
    			throw new Exception("复制文件流出错", e);
    		}
    	}
    
    	public static void writeStream(String str, OutputStream output) throws Exception {
    		try {
    			output.write(str.getBytes());
    		} catch (IOException e) {
    			int code = (int) System.currentTimeMillis();
    			log.error("复制文件流出错(" + code + ")", e);
    			throw new Exception("复制文件流出错", e);
    		}
    	}
    
    	/**
    	 * 
    	 * @param dir
    	 * @param fileName
    	 * @param inputStream
    	 *            是字符串或是 inputStream?
    	 * @throws Exception
    	 */
    	public static void saveFile(String dir, String fileName, InputStream inputStream) throws Exception {
    		FileOutputStream out = null;
    		try {
    			File file = new File(dir + AttachmentHelper.SEP + fileName);
    			out = new FileOutputStream(file);
    			copyStream(inputStream, out);
    		} catch (Exception e) {
    			int code = (int) System.currentTimeMillis();
    			log.error("上传文件异常(" + code + ")", e);
    			throw new Exception("上传文件异常！", e);
    		} finally {
    			if (inputStream != null) {
    				inputStream.close();
    			}
    			if (out != null) {
    				out.close();
    			}
    
    		}
    
    	}
    
    	/**
    	 * 获取目录下的所有文件名称
    	 * 
    	 * @param dir
    	 * @return
    	 * @throws Exception
    	 */
    	public static String[] listFiles(String dir) throws Exception {
    		// 定位目录
    		String[] fileArray = null;
    		try {
    			File directory = new File(dir);
    			if (!directory.exists()) {
    				return null;
    			} else {
    				File[] files = directory.listFiles();
    				fileArray = new String[files.length];
    				if (fileArray.length > 0) {
    					for (int i = 0; i < files.length; i++) {
    						fileArray[i] = files[i].getName();
    					}
    				}
    				return fileArray;
    			}
    		} catch (Exception e) {
    			int code = (int) System.currentTimeMillis();
    			log.error("获取目录下所有文件异常(" + code + ")", e);
    			throw new Exception("获取目录下所有文件异常！", e);
    		}
    
    	}
    
    	/**
    	 * 获取目录下某个文件
    	 * 
    	 * @param dir
    	 * @param fileName
    	 * @return
    	 * @throws Exception
    	 */
    	public static InputStream getFileAsStream(String dir, String fileName) throws Exception {
    		try {
    			FileInputStream stream = new FileInputStream(dir + AttachmentHelper.SEP + fileName);
    			return stream;
    		} catch (Exception e) {
    			int code = (int) System.currentTimeMillis();
    			log.error("读取文件异常(" + code + ")", e);
    			throw new Exception("读取文件异常！", e);
    		}
    
    	}
    
    	/**
    	 * 删除指定目录下文件
    	 * 
    	 * @param dir
    	 * @param fileName
    	 * @throws Exception
    	 */
    	public static void deleteFile(String dir, String fileName) throws Exception {
    		try {
    			File file = new File(dir + AttachmentHelper.SEP + fileName);
    			if (file.exists()) {
    				file.delete();
    			}
    		} catch (Exception e) {
    			int code = (int) System.currentTimeMillis();
    			log.error("删除文件异常(" + code + ")", e);
    			throw new Exception("附件不存在或删除失败！", e);
    		}
    	}
    
    	/**
    	 * 删除指定目录
    	 * 
    	 * @param dir_path
    	 * @param fileName
    	 * @throws Exception
    	 */
    	public static void deleteDir(String dir_path) throws Exception {
    		try {
    			File directory = new File(dir_path);
    			if (directory.exists()) {
    				if (directory.isDirectory()) {
    					File[] files = directory.listFiles();
    					if (files != null && files.length > 0) {
    						for (int i = 0; i < files.length; i++) {
    							files[i].delete();
    						}
    					}
    				}
    				directory.delete();
    			}
    		} catch (Exception e) {
    			int code = (int) System.currentTimeMillis();
    			log.error("删除目录异常(" + code + ")", e);
    			throw new Exception("删除目录失败！", e);
    		}
    	}
    
    }