package cn.saiz.drkms.task.crack.utils; 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 java.util.ArrayList; import java.util.HashMap; import java.util.List; import java.util.Map; import org.csource.common.NameValuePair; import org.csource.fastdfs.ClientGlobal; import org.csource.fastdfs.FileInfo; import org.csource.fastdfs.StorageClient; import org.csource.fastdfs.StorageClient1; import org.csource.fastdfs.StorageServer; import org.csource.fastdfs.TrackerClient; import org.csource.fastdfs.TrackerServer; import org.slf4j.Logger; import org.slf4j.LoggerFactory; public class FastDFSUtil { public static Logger log = LoggerFactory.getLogger(FastDFSUtil.class); static { try { String path = System.getProperty("user.dir") + "/fdfs_client.conf"; File file = new File(path); if(file == null || !file.exists()){ path = FastDFSUtil.class.getClassLoader() .getResource("fdfs_client.conf").getPath(); file = new File(path); if(file == null || !file.exists()){ path = Thread.currentThread().getContextClassLoader(). getResource("fdfs_client.conf").getPath(); } } // 读取配置文件 ClientGlobal.init(path); } catch (Exception e) { log.error("Init fastdfs error:{}",e.getMessage()); throw new RuntimeException(e); } } public static StorageClient1 getClient() throws IOException { return new StorageClient1(new TrackerClient().getConnection(), null); } public static StorageClient1 getClient(TrackerServer trackerServer,StorageServer storageServer){ return new StorageClient1(trackerServer, storageServer); } public static TrackerServer getTrackerServer() throws IOException{ return new TrackerClient().getConnection(); } public static void releaseServerResource(TrackerServer trackerServer,StorageServer storageServer) throws IOException{ if(trackerServer != null) trackerServer.close(); if(storageServer != null) storageServer.close(); } ///////////////////上传文件//////////////// public static String uploadFile(String filePath) throws IOException { return uploadFile(null, new File(filePath)); } public static String uploadFile(String groupName, String filePath) throws IOException { return uploadFile(groupName, new File(filePath)); } public static String uploadFile(String groupName, File file) throws IOException { if (file == null) return null; return uploadFile(groupName, new FileInputStream(file), file.getName(), file.length()); } public static String uploadFile(File file) throws IOException { return uploadFile(null, file); } public static String uploadFile(InputStream input,String fileName, long fileLength) throws IOException { return uploadFile(null, input, fileName, fileLength); } public static String uploadFile(String groupName, InputStream inStream, String uploadFileName, long fileLength) throws IOException { return uploadByteFile(groupName, uploadFileName, fileLength, getFileBuffer(inStream, fileLength)); } public static String uploadByteFile(String groupName, String fileName, long fileLength, byte[] fileBuff) throws IOException { String fileId = null; String fileExtName = null; if (fileName.contains(".")) { fileExtName = fileName.substring(fileName .lastIndexOf(".") + 1); } else { log.warn("Fail to upload file, because the format of filename is illegal."); return fileId; } // 建立连接 TrackerServer trackerServer = getTrackerServer(); StorageServer storageServer = null; StorageClient1 client = getClient(trackerServer, storageServer); // 设置元信息 NameValuePair[] metaList = new NameValuePair[3]; metaList[0] = new NameValuePair("fileName", fileName); metaList[1] = new NameValuePair("fileExtName", fileExtName); metaList[2] = new NameValuePair("fileLength", String.valueOf(fileLength)); // 上传文件 try { fileId = client.upload_file1(groupName, fileBuff, fileExtName, metaList); log.info("upload success. file id is: " + fileId); } catch (Exception e) { log.warn("Upload file "" + fileName + ""fails"); throw new RuntimeException(e); } finally { releaseServerResource(trackerServer, storageServer); } return fileId; } //////////////////上传文件END///////////////// /////////////////下载文件//////////////////// public static byte[] downloadFile(String groupName, String filepath) throws Exception { if(filepath == null || "".equals(filepath)) return null; TrackerServer trackerServer = null; StorageServer storageServer = null; try { TrackerClient tracker = new TrackerClient(); trackerServer = tracker.getConnection(); StorageClient storageClient = new StorageClient(trackerServer, storageServer); // StorageClient1 storageClient1 = new StorageClient1(trackerServer, // storageServer); return storageClient.download_file(groupName, filepath); } finally { releaseServerResource(trackerServer, storageServer); } } public static byte[] downloadFile(String fileId) throws Exception { TrackerServer trackerServer = null; StorageServer storageServer = null; try { trackerServer = getTrackerServer(); StorageClient1 storageClient1 = getClient(trackerServer, storageServer); return storageClient1.download_file1(fileId); } finally { releaseServerResource(trackerServer, storageServer); } } public static void downloadFile(String fileId, OutputStream out) throws Exception { if(fileId != null && out != null){ try { byte[] b = downloadFile(fileId); out.write(b); out.flush(); } finally { out.close(); } } } public static void downloadFile(String groupName, String filepath, OutputStream out) throws Exception { if(filepath != null && out != null){ try { byte[] b = downloadFile(groupName, filepath); out.write(b); out.flush(); } finally { out.close(); } } } public static void downloadFile(String groupName, String filepath, File descFile) throws Exception { OutputStream out = null; try { out = new FileOutputStream(descFile); downloadFile(groupName, filepath, out); } finally { if(out != null) out.close(); } } public static List<Map<String,String>> getFileMate(String groupName, String filepath) throws Exception { if(filepath == null) return null; TrackerServer trackerServer = getTrackerServer(); StorageServer storageServer = null; StorageClient storageClient = getClient(trackerServer, storageServer); NameValuePair nvps[] = storageClient.get_metadata(groupName, filepath); List<Map<String,String>> list = new ArrayList<Map<String,String>>(); for (NameValuePair nvp : nvps) { Map<String,String> map = new HashMap<String,String>(); map.put(nvp.getName(), nvp.getValue()); list.add(map); } return list; } /** * 删除文件 * @return 0:删除成功 其他数值失败 */ public static int deleteFile(String groupName, String filepath) throws Exception { if(filepath == null) return -1; TrackerServer trackerServer = getTrackerServer(); StorageServer storageServer = null; StorageClient storageClient = new StorageClient(trackerServer, storageServer); return storageClient.delete_file(groupName, filepath); } /** * 删除文件 * * @param groupName * @param filepath * @throws Exception */ public static int deleteFile(String fileId) throws Exception { if(fileId == null) return -1; TrackerServer trackerServer = null; StorageServer storageServer = null; try { trackerServer = getTrackerServer(); StorageClient1 storageClient1 = getClient(trackerServer, storageServer); return storageClient1.delete_file1(fileId); } finally { releaseServerResource(trackerServer, storageServer); } } /** * 通过fileID查询上传的文件信息 * * @param fileId * eg:group1/M00/00/00/wKi3glS_XEaAVL3DAAwdkYUdoP8278.gif * @return * @throws Exception */ public static FileInfo getFileInfo(String fileId) throws Exception { if (fileId == null) return null; TrackerServer trackerServer = getTrackerServer(); StorageServer storageServer = null; try { StorageClient1 client = new StorageClient1(trackerServer, storageServer); return client.get_file_info1(fileId); } finally { releaseServerResource(trackerServer, storageServer); } } private static byte[] getFileBuffer(InputStream inStream, long fileLength) throws IOException { byte[] buffer = new byte[256 * 1024]; byte[] fileBuffer = new byte[(int) fileLength]; int count = 0; int length = 0; while ((length = inStream.read(buffer)) != -1) { for (int i = 0; i < length; ++i) { fileBuffer[count + i] = buffer[i]; } count += length; } return fileBuffer; } public static void main(String[] args) throws Exception { List<Map<String, String>> fileMate = getFileMate("group1","M00/98/73/CgMEwVh5M62AQnf-ABa36R1PUuY660.pdf"); System.out.println(fileMate.toString()); } }