Java对mongodb-GridFS的增加，删除，查找操作

package com.amarsoft.lab.mongo;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.net.UnknownHostException;

import org.apache.commons.codec.digest.DigestUtils;
import org.junit.Assert;
import org.junit.Test;

import com.mongodb.BasicDBObject;
import com.mongodb.DB;
import com.mongodb.DBObject;
import com.mongodb.Mongo;
import com.mongodb.gridfs.GridFS;
import com.mongodb.gridfs.GridFSDBFile;
import com.mongodb.gridfs.GridFSInputFile;

public class MyGridFsTest {
	private String host = "127.0.0.1";
	private int port = 27017;
	private String dbName = "demogridfs";

	/**
	 * 上传文件测试
	 * @throws UnknownHostException
	 * @throws FileNotFoundException
	 */
	@Test
	public void putFileTest() throws UnknownHostException,
			FileNotFoundException {
		Mongo connection = new Mongo(host, port);
		DB db = connection.getDB(dbName);
		GridFS gridFs = new GridFS(db,"myfs");

		String fileName = "/Users/yangsong/win-share/ams/ref/api-docs/JDK_API_1_6_zh_CN.CHM";
		String name = "JDK_API_1_6_zh_CN";

		InputStream in = new FileInputStream(fileName);
		GridFSInputFile gridFSInputFile = gridFs.createFile(in);
		gridFSInputFile.put("fcu", "tsir");
		gridFSInputFile.setFilename(name);
		gridFSInputFile.save();
		
	}
	
	/**
	 * 根据文件名查找文件
	 * @throws IOException
	 */
	@Test
	public void testFindFile() throws IOException{
		Mongo connection = new Mongo(host, port);
		DB db = connection.getDB(dbName);
		GridFS gridFs = new GridFS(db);

		//按名称查找文件
		String fileName = "/Users/yangsong/Documents/tmp/JDK_API_1_6_zh_CN-out.chm";
		String name = "JDK_API_1_6_zh_CN";
		GridFSDBFile gfsFile = getByFileName(name,gridFs);
		File outFile = new File(fileName);
//		if(!outFile.exists())outFile.createNewFile();
		gfsFile.writeTo(outFile);
		
		/**
		 * 测试删除文件
		 */
		DBObject query = new BasicDBObject("filename", fileName);
		gridFs.remove(query);
		gfsFile = getByFileName(name,gridFs);
		Assert.assertNull(gfsFile);
	}
	
	/**
	 * 根据MD5码查找文件
	 * @throws IOException
	 */
	@Test
	public void testFindFileByMd5() throws IOException{
		Mongo connection = new Mongo(host, port);
		DB db = connection.getDB(dbName);
		GridFS gridFs = new GridFS(db);
		
		String fileName = "/Users/yangsong/Documents/tmp/JDK_API_1_6_zh_CN-out.chm";
		GridFSDBFile gfsFile = getByMd5("df03c9cf37c5fbc5a7894e04297127d7",gridFs);
		File outFile = new File(fileName);
		gfsFile.writeTo(outFile);
		System.out.println("content-type:"+gfsFile.getContentType());
	}
	
	/**
	 * 计算文件的MD5码，看与数据库中存储的是否一致
	 * @throws FileNotFoundException
	 * @throws IOException
	 */
	@Test
	public void testCalcFileMd5() throws FileNotFoundException, IOException{
		String fileName = "/Users/yangsong/win-share/ams/ref/api-docs/JDK_API_1_6_zh_CN.CHM";
		File file = new File(fileName);
		String md5 = DigestUtils.md5Hex(new FileInputStream(file)); 
		System.out.println(md5);
	}

	public GridFSDBFile getById(Object id, GridFS fs) {
		DBObject query = new BasicDBObject("_id", id);
		GridFSDBFile gridFSDBFile = fs.findOne(query);
		return gridFSDBFile;
	}

	public GridFSDBFile getByFileName(String fileName, GridFS fs) {
		DBObject query = new BasicDBObject("filename", fileName);
		GridFSDBFile gridFSDBFile = fs.findOne(query);
		return gridFSDBFile;
	}
	public GridFSDBFile getByMd5(String md5, GridFS fs) {
		DBObject query = new BasicDBObject("md5", md5);
		GridFSDBFile gridFSDBFile = fs.findOne(query);
		return gridFSDBFile;
	}

}