|
package com.oplog.db;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.Map;
import javax.annotation.Resource;
import org.bson.types.ObjectId;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.data.mongodb.core.MongoTemplate;
import com.mongodb.BasicDBObject;
import com.mongodb.DB;
import com.mongodb.DBCollection;
import com.mongodb.DBCursor;
import com.mongodb.DBObject;
public class MongoBase {
private static final Logger logger = LoggerFactory.getLogger(MongoBase.class);
protected String database;
protected static final SimpleDateFormat dateFormat = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
@Resource(name="mongoTemplate")
private MongoTemplate mongoTemplate;
public MongoBase() {
}
public MongoTemplate getMongoTemplate() {
return mongoTemplate;
}
public void setMongoTemplate(MongoTemplate mongoTemplate) {
this.mongoTemplate = mongoTemplate;
}
/**
* 根据数据库名称获取数据库对象
* @param DBName
* @return
*/
protected DB getMongoDB(String DBName){
return mongoTemplate.getDb();
}
/**
* 插入数据
* @param table
* @param params
* @return
*/
protected boolean insert(String table,Map params ){
boolean result=true;
try {
String uniqueId = getId();
params.put("_id", uniqueId);
BasicDBObject datas = new BasicDBObject(params);
DB db = this.getMongoDB(database);
DBCollection tableObj = db.getCollection(table);
tableObj.insert(datas);
} catch (Exception e) {
logger.error("插入数据到mongodb出错,database name:" + database + ",table:" + table + ",params:" + params+",insert errmsg:"+e.getMessage());
result=false;
}
return result;
}
/**
* 查询数据
* @param table
* @param params
* @param start起始位置
* @param size 查询数量
* @return
*/
protected List query(String table,Map params,int start, int size){
List result=new ArrayList();
try {
BasicDBObject conditions = new BasicDBObject(params);
DB db = this.getMongoDB(database);
DBCollection tableObj = db.getCollection(table);
DBCursor cursor=tableObj.find(conditions).skip(start).limit(size).sort(new BasicDBObject("_id",-1));
while(cursor.hasNext()){
DBObject item=cursor.next();
processQueryResult(item,result);
}
} catch (Exception e) {
logger.error("从mongodb查询数据出错,database name:" + database + ",table:" + table + ",params:" + params+",start:"+start+",size:"+size+",querylist errmsg:"+e.getMessage());
}
return result;
}
protected long queryCount(String table,Map params){
long totalcount=0;
try{
BasicDBObject conditions = new BasicDBObject(params);
DB db = this.getMongoDB(database);
DBCollection tableObj = db.getCollection(table);
totalcount=tableObj.count(conditions);
}catch(Exception e){
logger.error("从mongodb查询数据总数出错,database name:" + database + ",table:" + table + ",params:" + params+",querycount:"+e.getMessage());
}
return totalcount;
}
protected boolean update(String table,Map params,Map updateData){
boolean result=true;
try{
BasicDBObject condtions=new BasicDBObject(params);
BasicDBObject datas=new BasicDBObject(updateData);
DB db=this.getMongoDB(database);
DBCollection tableObj=db.getCollection(table);
int count=tableObj.update(condtions, new BasicDBObject("$set",datas),false,true).getN();
if(count |
|
|