1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
| 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<String,Object> 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<Map<String,Object>> query(String table,Map<String,Object> params,int start, int size){
List<Map<String,Object>> result=new ArrayList<Map<String,Object>>();
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<String,Object> 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<String,Object> params,Map<String,Object> 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<1){
result=false;
logger.info("mongodb更新数据失败,database name:" + database + ",table:" + table + ",params:" + params+",updatedata:"+updateData);
}
}catch(Exception e){
logger.error("mongodb修改数据出错,database name:" + database + ",table:" + table + ",params:" + params+",updatedata:"+updateData+",update errmsg:"+e.getMessage());
result=false;
}
return result;
}
protected boolean deleleOne(String table,Map<String,Object> params){
boolean result=true;
try{
BasicDBObject conditions=new BasicDBObject(params);
DB db=this.getMongoDB(database);
DBCollection tableObj=db.getCollection(table);
int count=tableObj.remove(conditions).getN();
if(count<1){
result=false;
logger.info("从mongodb删除数据失败,database name:" + database + ",table:" + table + ",params:" + params);
}
}catch(Exception e){
logger.error("从mongodb删除数据出错,database name:" + database + ",table:" + table + ",params:" + params+",deleteone errmsg:"+e.getMessage());
result=false;
}
return result;
}
/**
* 获取主键
* @return
*/
protected String getId(){
return ObjectId.get().toString();
}
protected String getDatabase() {
return database;
}
protected void setDatabase(String database) {
this.database = database;
}
@SuppressWarnings("unchecked")
private void processQueryResult(DBObject item,List<Map<String,Object>> result){
if(null==item){
return;
}
for(String key:item.keySet()){
Object value=item.get(key);
if(value instanceof Date){
Date dateValue=(Date)value;
String dateStr=dateFormat.format(dateValue);
item.put(key,dateStr);
}
}
result.add(item.toMap());
}
}
|