FastDFS的配置、部署與API使用解讀(3)以流的方式上傳檔案的客戶端程式碼

鍾超發表於2011-11-18

呼叫的API為:

String[] upload_file(

String group_name,//組名,不指定則可設為null

long file_size,//檔案大小,必須制定

UploadCallback callback,//回撥

String file_ext_name,

NameValuePair[] meta_list

)


	/**
	 * Upload File to DFS, directly transferring java.io.InputStream to java.io.OutStream
	 * @author Poechant
	 * @email zhongchao.ustc@gmail.com
	 * @param fileBuff, file to be uploaded.
	 * @param uploadFileName, the name of the file.
	 * @param fileLength, the length of the file.
	 * @return the file ID in DFS.
	 * @throws IOException 
	 */
	public String[] uploadFileByStream(InputStream inStream, String uploadFileName, long fileLength) throws IOException {
		
		String[] results = null;
		String fileExtName = "";
		if (uploadFileName.contains(".")) {
			fileExtName = uploadFileName.substring(uploadFileName.lastIndexOf(".") + 1);
		} else {
			logger.warn("Fail to upload file, because the format of filename is illegal.");
			return results;
		}
		
		TrackerClient tracker = new TrackerClient();
        TrackerServer trackerServer = tracker.getConnection();
        StorageServer storageServer = null;
        StorageClient1 client = new StorageClient1(trackerServer, storageServer);
        
        NameValuePair[] metaList = new NameValuePair[3];
        metaList[0] = new NameValuePair("fileName", uploadFileName);
        metaList[1] = new NameValuePair("fileExtName", fileExtName);
        metaList[2] = new NameValuePair("fileLength", String.valueOf(fileLength));
        
        try {
        	// results[0]: groupName, results[1]: remoteFilename.
        	results = client.upload_file(null, fileLength, new UploadFileSender(inStream), fileExtName, metaList);
		} catch (Exception e) {
			logger.warn("Upload file \"" + uploadFileName + "\"fails");
		}
        
  		trackerServer.close();
		
		return results;		
	}

其中的UploadFileSender是一個實現了UploadCallback介面的類:

	private static class UploadFileSender implements UploadCallback {
		
		private InputStream inStream;
		
		public UploadFileSender(InputStream inStream) {
			this.inStream = inStream;
		}
		
		public int send(OutputStream out) throws IOException {
			int readBytes;
			while((readBytes = inStream.read()) > 0) {
				out.write(readBytes);
			}
			return 0;
		}
	} 


相關文章