MongoDB客户端工具类

本是古典何须时尚 2022-08-13 18:37 255阅读 0赞

import java.io.File;  
import java.io.IOException;  
import java.io.InputStream;  
import java.net.UnknownHostException;  
import java.util.ArrayList;  
import java.util.Arrays;  
import java.util.Map;  
  
  
import net.sf.json.JSONArray;  
import net.sf.json.JSONObject;  
  
  
import org.apache.log4j.Logger;  
import org.bson.types.ObjectId;  
  
  
import com.ai.paas.util.CiperTools;  
import com.mongodb.BasicDBObject;  
import com.mongodb.DBObject;  
import com.mongodb.MongoClient;  
import com.mongodb.MongoCredential;  
import com.mongodb.ServerAddress;  
import com.mongodb.gridfs.GridFS;  
import com.mongodb.gridfs.GridFSDBFile;  
import com.mongodb.gridfs.GridFSInputFile;  
import com.mongodb.util.JSON;  
  
  
public class MongoDBClient \{  
private static final Logger log = Logger.getLogger(MongoDBClient.class);  
  
  
private static final String IP\_KEY = "ip";  
private static final String PORT\_KEY = "port";  
private static final String FILE\_NAME = "filename";  
  
  
private MongoClient mongo = null;  
  
  
public MongoDBClient(String addr) \{  
try \{  
JSONArray array = JSONArray.fromObject(addr);  
if (array != null && array.size() > 0) \{  
int size = array.size();  
JSONObject json = null;  
ArrayList<ServerAddress> sa = new ArrayList<ServerAddress>();  
for (int i = 0; i < size; i++) \{  
json = (JSONObject) array.get(i);  
sa.add(new ServerAddress(json.getString(IP\_KEY), json  
.getInt(PORT\_KEY)));  
\}  
mongo = new MongoClient(sa);  
  
  
\}  
\} catch (UnknownHostException e) \{  
log.error("", e);  
\}  
\}  
  
  
public MongoDBClient(String addr, String database, String userName,  
String password) \{  
try \{  
JSONArray array = JSONArray.fromObject(addr);  
if (array != null && array.size() > 0) \{  
int size = array.size();  
JSONObject json = null;  
ArrayList<ServerAddress> sa = new ArrayList<ServerAddress>();  
for (int i = 0; i < size; i++) \{  
json = (JSONObject) array.get(i);  
sa.add(new ServerAddress(json.getString(IP\_KEY), json  
.getInt(PORT\_KEY)));  
\}  
String orignPwd = CiperTools.decrypt(password);  
MongoCredential credential = MongoCredential  
.createMongoCRCredential(userName, database,  
orignPwd.toCharArray());  
mongo = new MongoClient(sa, Arrays.asList(credential));  
  
  
\}  
\} catch (UnknownHostException e) \{  
log.error("", e);  
\}  
\}  
  
  
public void insert(String dbName, String collectionName, String content) \{  
BasicDBObject doc = new BasicDBObject();  
doc.put("content", content);  
mongo.getDB(dbName).getCollection(collectionName).insert(doc);  
\}  
  
  
public void insert(String dbName, String collectionName, JSONObject doc) \{  
DBObject dbObj = (DBObject) JSON.parse(doc.toString());  
mongo.getDB(dbName).getCollection(collectionName).insert(dbObj);  
\}  
  
  
public void insertJSON(String dbName, String collectionName, String doc) \{  
DBObject dbObj = (DBObject) JSON.parse(doc);  
mongo.getDB(dbName).getCollection(collectionName).insert(dbObj);  
\}  
  
  
public void insertJSON(String dbName, String collectionName, JSONObject doc) \{  
DBObject dbObj = (DBObject) JSON.parse(doc.toString());  
mongo.getDB(dbName).getCollection(collectionName).insert(dbObj);  
\}  
  
  
@SuppressWarnings("rawtypes")  
public void insert(String dbName, String collectionName, Map docMap) \{  
DBObject dbObj = new BasicDBObject(docMap);  
mongo.getDB(dbName).getCollection(collectionName).insert(dbObj);  
\}  
  
  
public String saveFile(String dbName, byte\[\] byteFile, String fileName,  
String fileType) \{  
GridFS fs = new GridFS(mongo.getDB(dbName));  
GridFSInputFile dbFile = null;  
try \{  
dbFile = fs.createFile(byteFile);  
\} catch (Exception e) \{  
e.printStackTrace();  
\}  
dbFile.setContentType(fileType);  
dbFile.setFilename(fileName);  
dbFile.put("fileName", fileName);  
dbFile.save();  
return dbFile.getId().toString();  
\}  
  
  
public String saveFile(String dbName, String fileName, String fileType) \{  
if (fileName == null) \{  
return null;  
\}  
String name = fileName.substring(fileName.lastIndexOf("/") + 1);  
GridFS fs = new GridFS(mongo.getDB(dbName));  
File file = new File(fileName);  
GridFSInputFile dbFile = null;  
try \{  
dbFile = fs.createFile(file);  
\} catch (IOException e) \{  
e.printStackTrace();  
\}  
dbFile.setContentType(fileType);  
dbFile.setFilename(name);  
dbFile.save();  
return dbFile.getId().toString();  
\}  
  
  
public void deleteFile(String dbName, String fileId) \{  
if (fileId == null) \{  
return;  
\}  
GridFS fs = new GridFS(mongo.getDB(dbName));  
GridFSDBFile dbFile = fs.findOne(new ObjectId(fileId));  
if (dbFile == null) \{  
return;  
\}  
fs.remove(dbFile);  
\}  
  
  
public void deleteFileByName(String dbName, String fileName) \{  
if (fileName == null) \{  
return;  
\}  
GridFS fs = new GridFS(mongo.getDB(dbName));  
GridFSDBFile dbFile = fs  
.findOne(new BasicDBObject(FILE\_NAME, fileName));  
if (dbFile == null) \{  
return;  
\}  
fs.remove(dbFile);  
\}  
  
  
public byte\[\] readFile(String dbName, String fileId) \{  
if (fileId == null) \{  
return null;  
\}  
GridFS fs = new GridFS(mongo.getDB(dbName));  
GridFSDBFile dbFile = fs.findOne(new ObjectId(fileId));  
if (dbFile == null) \{  
return null;  
\}  
InputStream is = null;  
try \{  
int len = (int) dbFile.getLength();  
byte\[\] ret = new byte\[len\];  
is = dbFile.getInputStream();  
int tmp = 0;  
int i = 0;  
while ((tmp = is.read()) != -1 && i < len) \{  
ret\[i\] = (byte) tmp;  
i++;  
\}  
return ret;  
\} catch (IOException e) \{  
e.printStackTrace();  
return null;  
\} finally \{  
if (is != null) \{  
try \{  
is.close();  
\} catch (IOException e) \{  
e.printStackTrace();  
\}  
\}  
\}  
  
  
\}  
  
  
public byte\[\] readFileByName(String dbName, String fileName) \{  
if (fileName == null) \{  
return null;  
\}  
GridFS fs = new GridFS(mongo.getDB(dbName));  
GridFSDBFile dbFile = fs  
.findOne(new BasicDBObject(FILE\_NAME, fileName));  
if (dbFile == null) \{  
return null;  
\}  
InputStream is = null;  
try \{  
int len = (int) dbFile.getLength();  
byte\[\] ret = new byte\[len\];  
is = dbFile.getInputStream();  
int tmp = 0;  
int i = 0;  
while ((tmp = is.read()) != -1 && i < len) \{  
ret\[i\] = (byte) tmp;  
i++;  
\}  
return ret;  
\} catch (IOException e) \{  
e.printStackTrace();  
return null;  
\} finally \{  
if (is != null) \{  
try \{  
is.close();  
\} catch (IOException e) \{  
e.printStackTrace();  
\}  
\}  
\}  
  
  
\}  
  
  
public void readFile(String dbName, String fileId, String localFileName) \{  
if (fileId == null) \{  
return;  
\}  
GridFS fs = new GridFS(mongo.getDB(dbName));  
GridFSDBFile dbFile = fs.findOne(new ObjectId(fileId));  
if (dbFile == null) \{  
return;  
\}  
File file = new File(localFileName);  
try \{  
dbFile.writeTo(file);  
\} catch (IOException e) \{  
e.printStackTrace();  
\}  
  
  
\}  
  
  
public void readFileByName(String dbName, String fileName,  
String localFileName) \{  
if (fileName == null) \{  
return;  
\}  
GridFS fs = new GridFS(mongo.getDB(dbName));  
GridFSDBFile dbFile = fs  
.findOne(new BasicDBObject(FILE\_NAME, fileName));  
if (dbFile == null) \{  
return;  
\}  
File file = new File(localFileName);  
try \{  
dbFile.writeTo(file);  
\} catch (IOException e) \{  
log.error("", e);  
\}  
  
  
\}  
  
  
public void destroyMongoDB() \{  
if (null != mongo) \{  
mongo.close();  
mongo = null;  
\}  
\}  
public String getFileName(String dbName,String fileId) \{  
if (fileId == null) \{  
return null;  
\}  
GridFS fs = new GridFS(mongo.getDB(dbName));  
GridFSDBFile dbFile = fs.findOne(new ObjectId(fileId));  
if (dbFile == null) \{  
return null;  
\}  
return dbFile.getFilename();  
\}  
\}