my-fastdfs-client

/** Copyright (C) 2012 Happy Fish / YuQing * My FastDFS Java Client may be copied only under the terms of the GNU Lesser * General Public License (LGPL). * Please visit the FastDFS Home Page http://www.csource.org/ for more detail. */ package org.csource.myfastdfs; import java.io.FileNotFoundException; import java.io.IOException; import org.csource.common.MyException; import org.csource.fastdfs.TrackerServer; import org.csource.fastdfs.StorageServer; import org.csource.fastdfs.UploadCallback; import org.csource.fastdfs.DownloadCallback; import org.csource.fastdfs.FileInfo; import org.csource.fastdfs.StorageClient1; import org.csource.fastdht.ServerGroup; import org.csource.fastdht.FastDHTClient; import org.csource.fastdht.KeyInfo; /** My FastDFS client * @author Happy Fish / YuQing * @version Version 1.00 */ public class MyFastDFSClient { public static final String MY_CLIENT_FILE_ID_KEY_NAME = "fdfs_fid"; protected StorageClient1 storageClient1; protected FastDHTClient fdhtClient; protected String fdhtNamespace; protected int status; //last error code /** * constructor * @param fdhtNamespace the FastDHT namespace */ public MyFastDFSClient(String fdhtNamespace) throws MyException { this.storageClient1 = new StorageClient1(); this.fdhtClient = new FastDHTClient(true); if (fdhtNamespace == null || fdhtNamespace.length() == 0) { throw new MyException("FastDHT namespace must be set!"); } this.fdhtNamespace = fdhtNamespace; } /** * constructor * @param trackerServer the FastDFS tracker server, can be null * @param storageServer the FastDFS storage server, can be null * @param serverGroup the FastDHT group info * @param fdhtNamespace the FastDHT namespace */ public MyFastDFSClient(TrackerServer trackerServer, StorageServer storageServer, ServerGroup serverGroup, String fdhtNamespace) throws MyException { this.storageClient1 = new StorageClient1(trackerServer, storageServer); this.fdhtClient = new FastDHTClient(serverGroup); if (fdhtNamespace == null || fdhtNamespace.length() == 0) { throw new MyException("FastDHT namespace must be set!"); } this.fdhtNamespace = fdhtNamespace; } /** * load parameters from FastDFS and FastDHT config file * @param fdfsConfigFilename the FastDFS config filename * @param fdhtConfigFilename the FastDHT config filename */ public static void init(String fdfsConfigFilename, String fdhtConfigFilename) throws FileNotFoundException, IOException, MyException { org.csource.fastdfs.ClientGlobal.init(fdfsConfigFilename); org.csource.fastdht.ClientGlobal.init(fdhtConfigFilename); } /** * @return the error code of last call */ public int getErrorCode() { return this.status; } /** * close connections */ public void close() { this.fdhtClient.close(); } /** * check FastDFS file id not exist * @param keyInfo the FastDHT key object * @return true for not exist, otherwise false */ protected boolean check_fdfs_file_id_not_exist(KeyInfo keyInfo) throws IOException, MyException { String fdfs_file_id = this.fdhtClient.get(keyInfo); this.status = this.fdhtClient.getErrorCode(); if (fdfs_file_id != null || this.status == 0) { this.status = 17; //EEXIST throw new MyException("file id: " + new String(keyInfo.getObjectId(), org.csource.fastdht.ClientGlobal.g_charset) + " already exist"); } return (this.status == 2); } /** * save FastDFS file id to FastDHT * @param keyInfo the FastDHT key object * @param fdfs_file_id the FastDFS file id * @return 0 for success, != 0 for error (error no) */ protected int set_fdfs_file_id(KeyInfo keyInfo, String fdfs_file_id) throws IOException, MyException { try { if ((this.status=this.fdhtClient.set(keyInfo, fdfs_file_id)) != 0) { this.storageClient1.delete_file1(fdfs_file_id); //rollback } } catch(Exception ex) { this.status = 5; ex.printStackTrace(); this.storageClient1.delete_file1(fdfs_file_id); //rollback } return this.status; } /** * get FastDFS file id * @param my_file_id the file id specified by application * @return FastDFS file id for success, return null for fail */ public String get_fdfs_file_id(String my_file_id) throws IOException, MyException { KeyInfo keyInfo = new KeyInfo(this.fdhtNamespace, my_file_id, MY_CLIENT_FILE_ID_KEY_NAME); String fdfs_file_id = this.fdhtClient.get(keyInfo); this.status = this.fdhtClient.getErrorCode(); return fdfs_file_id; } /** * upload file to storage server (by file name) * @param my_file_id the file id specified by application * @param local_filename local filename to upload * @param file_ext_name file ext name, do not include dot(.), null to extract ext name from the local filename * @return 0 for success, != 0 for error (error no) */ public int upload_file(String my_file_id, String local_filename, String file_ext_name) throws IOException, MyException { final String group_name = ""; return this.upload_file(my_file_id, group_name, local_filename, file_ext_name); } /** * upload file to storage server (by file name) * @param my_file_id the file id specified by application * @param group_name the group name to upload file to, can be empty * @param local_filename local filename to upload * @param file_ext_name file ext name, do not include dot(.), null to extract ext name from the local filename * @return 0 for success, != 0 for error (error no) */ public int upload_file(String my_file_id, String group_name, String local_filename, String file_ext_name) throws IOException, MyException { KeyInfo keyInfo = new KeyInfo(this.fdhtNamespace, my_file_id, MY_CLIENT_FILE_ID_KEY_NAME); if (!this.check_fdfs_file_id_not_exist(keyInfo)) { return this.status; } String fdfs_file_id = this.storageClient1.upload_file1(group_name, local_filename, file_ext_name, null); this.status = this.storageClient1.getErrorCode(); if (fdfs_file_id == null) { return this.status; } return this.set_fdfs_file_id(keyInfo, fdfs_file_id); } /** * upload file to storage server (by file buffer) * @param my_file_id the file id specified by application * @param file_buff the file content / buffer to upload * @param file_ext_name file ext name, do not include dot(.), null to extract ext name from the local filename * @return 0 for success, != 0 for error (error no) */ public int upload_file(String my_file_id, byte[] file_buff, String file_ext_name) throws IOException, MyException { final String group_name = ""; return this.upload_file(my_file_id, group_name, file_buff, file_ext_name); } /** * upload file to storage server (by file buffer) * @param my_file_id the file id specified by application * @param group_name the group name to upload file to, can be empty * @param file_buff the file content / buffer to upload * @param file_ext_name file ext name, do not include dot(.), null to extract ext name from the local filename * @return 0 for success, != 0 for error (error no) */ public int upload_file(String my_file_id, String group_name, byte[] file_buff, String file_ext_name) throws IOException, MyException { KeyInfo keyInfo = new KeyInfo(this.fdhtNamespace, my_file_id, MY_CLIENT_FILE_ID_KEY_NAME); if (!this.check_fdfs_file_id_not_exist(keyInfo)) { return this.status; } String fdfs_file_id = this.storageClient1.upload_file1(group_name, file_buff, file_ext_name, null); this.status = this.storageClient1.getErrorCode(); if (fdfs_file_id == null) { return this.status; } return this.set_fdfs_file_id(keyInfo, fdfs_file_id); } /** * upload file to storage server (by callback object) * @param my_file_id the file id specified by application * @param file_size the file size * @param callback the write data callback object * @param file_ext_name file ext name, do not include dot(.), null to extract ext name from the local filename * @return 0 for success, != 0 for error (error no) */ public int upload_file(Stri