io.leopard.data.mongo.MongoImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of leopard-mongo Show documentation
MongoDB操作，基于mongo-java-driver封装，简化MongoDB操作.
There is a newer version: 0.9.12
package io.leopard.data.mongo;

import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Set;

import com.mongodb.BasicDBList;
import com.mongodb.BasicDBObject;
import com.mongodb.DB;
import com.mongodb.DBCollection;
import com.mongodb.DBCursor;
import com.mongodb.DBObject;
import com.mongodb.MongoClient;
import com.mongodb.MongoClientOptions;
import com.mongodb.ServerAddress;
import com.mongodb.WriteResult;
import com.mongodb.gridfs.GridFS;

public class MongoImpl implements Mongo {

	private String server;
	private String database;
	private String collectionName;

	private String username;
	private String password;

	private MongoClient client;
	private DBCollection collection;
	private DB db;

	public MongoImpl() {

	}

	public MongoImpl(String server, String database, String collectionName) {
		this.server = server;
		this.database = database;
		this.collectionName = collectionName;
	}

	public String getServer() {
		return server;
	}

	public void setServer(String server) {
		this.server = server;
	}

	public String getDatabase() {
		return database;
	}

	public void setDatabase(String database) {
		this.database = database;
	}

	public String getCollectionName() {
		return collectionName;
	}

	public void setCollectionName(String collectionName) {
		this.collectionName = collectionName;
	}

	public String getUsername() {
		return username;
	}

	public void setUsername(String username) {
		this.username = username;
	}

	public String getPassword() {
		return password;
	}

	public void setPassword(String password) {
		this.password = password;
	}

	@SuppressWarnings("deprecation")
	public void init() {
		String[] list = server.split(":");
		String host = list[0];
		int port = Integer.parseInt(list[1]);
		int connectTimeout = 1000 * 60;
		MongoClientOptions options = new MongoClientOptions.Builder().connectTimeout(connectTimeout).build();
		client = new MongoClient(new ServerAddress(host, port), options);
		this.db = client.getDB(this.database);
		if (username != null && username.length() > 0) {
			if (password != null && password.length() > 0) {
				db.addUser(username, password.toCharArray());
			}
		}
		this.collection = db.getCollection(collectionName);
	}

	public void destroy() {
		this.client.close();
	}

	@Override
	public boolean insert(Object bean) {
		List fieldList = listFields(bean);
		DBObject obj = new BasicDBObject();
		for (Field field : fieldList) {
			String name = field.getName();
			field.setAccessible(true);

			Object value;
			try {
				value = field.get(bean);
			}
			catch (Exception e) {
				throw new RuntimeException(e.getMessage(), e);
			}
			obj.put(name, value);
		}

		this.collection.insert(obj);
		return true;
	}

	public static List listFields(Object bean) {
		List list = new ArrayList();
		Class currentClazz = bean.getClass();
		while (true) {
			Field[] fields = currentClazz.getDeclaredFields();
			for (Field field : fields) {
				list.add(field);
			}
			if (currentClazz.getSuperclass() == null) {
				break;
			}
			currentClazz = currentClazz.getSuperclass();
		}
		return list;
	}

	@Override
	public  T query() {
		return null;
	}

	@Override
	public boolean drop() {
		this.collection.drop();
		return true;
	}

	@Override
	public  List find(IFind find, Class clazz) {
		DBCursor cursor = find.find(collection);
		return this.toBeanList(cursor, clazz);
	}

	@Override
	public  List find(DBObject ref, Class clazz, int start, int size) {
		DBCursor cursor = this.collection.find(ref).skip(start).limit(size);
		return this.toBeanList(cursor, clazz);
	}

	public  List toBeanList(DBCursor cursor, Class clazz) {
		List list = new ArrayList();
		while (cursor.hasNext()) {
			DBObject user = cursor.next();
			T bean;
			try {
				bean = clazz.newInstance();
			}
			catch (Exception e) {
				throw new RuntimeException(e.getMessage(), e);
			}
			@SuppressWarnings({ "unchecked" })
			Map map = user.toMap();
			Set> set = map.entrySet();

			for (Entry entry : set) {
				String fieldName = entry.getKey();
				Object value = entry.getValue();
				if ("_id".equals(fieldName)) {
					continue;
				}
				else if ("location".equals(fieldName)) {
					BasicDBList dbList = (BasicDBList) value;
					double x = (Double) dbList.get(0);
					double y = (Double) dbList.get(1);
					double[] geo = new double[] { x, y };
					value = geo;
				}

				try {
					Field field = clazz.getDeclaredField(fieldName);
					field.setAccessible(true);
					field.set(bean, value);
				}
				catch (Exception e) {
					System.out.println("value:" + value + " " + value.getClass().getName());
					throw new RuntimeException(e.getMessage(), e);
				}
				// System.out.println("entry:" + entry.getKey());
			}

			list.add(bean);
			// System.out.println(user.toString());
		}

		return list;
	}

	@Override
	public boolean truncate() {
		WriteResult resulst = this.collection.remove(new BasicDBObject());
		return resulst.getN() > 0;
	}

	@Override
	public boolean remove(DBObject ref) {
		WriteResult resulst = this.collection.remove(ref);
		return resulst.getN() > 0;
	}

	@Override
	public boolean remove(String fieldName, Object value) {
		return this.remove(new BasicDBObject(fieldName, value));
	}

	@Override
	public boolean update(DBObject q, DBObject o) {
		WriteResult resulst = this.collection.update(q, o);
		// System.out.println("getN:" + resulst.getN());
		return resulst.getN() > 0;
	}

	@Override
	public GridFS getGridFS(String type) {
		return new GridFS(db, "fs");
	}

}