@SuppressWarnings("deprecation") @Override public Jongo get(final String db) { return new Jongo(client.getDB(db), mapper); }
@Override public Object getTemplate() { DB db = database.getDB(); return new Jongo(db); }
@Override public Object getTemplate() { DB db = database.getDB(); return new Jongo(db); }
@Provides @Named("Jongo") public Jongo jongo(@Named(MongoModule.MONGO_DB_NAME) String dbName, @Named(MongoModule.MONGO_CLIENT_NAME) MongoClient mongoClient, @Named("Mapper") Mapper mapper) { return new Jongo(mongoClient.getDB(dbName), mapper); } }
@Provides @Named("Jongo") public Jongo jongo(@Named(MongoModule.MONGO_DB_NAME) String dbName, @Named(MongoModule.MONGO_CLIENT_NAME) MongoClient mongoClient, @Named("Mapper") Mapper mapper) { return new Jongo(mongoClient.getDB(dbName), mapper); } }
logger.debug("method with Jongo argument"); return changeSetMethod.invoke(changeLogInstance, jongo != null ? jongo : new Jongo(db)); } else if (changeSetMethod.getParameterTypes().length == 1 && changeSetMethod.getParameterTypes()[0].equals(MongoTemplate.class)) {
logger.debug("method with Jongo argument"); return changeSetMethod.invoke(changeLogInstance, jongo != null ? jongo : new Jongo(db)); } else if (changeSetMethod.getParameterTypes().length == 1 && changeSetMethod.getParameterTypes()[0].equals(MongoTemplate.class)) {
@Override public void cleanUp() { try { localMachines.removeMachine(customizerMachine); MongoClientURI mongoClientURI = new MongoClientURI( checkNotNull(params.get(DB_URI), DB_URI + " param is required")); Jongo jongo = new Jongo(new MongoClient(mongoClientURI).getDB(mongoClientURI.getDatabase())); Stopwatch stopwatch = Stopwatch.createStarted(); jongo.getCollection(given.getCollection()).drop(); System.out.printf("dropped %s -- %s%n", given.getCollection(), stopwatch.stop().toString()); } catch (UnknownHostException e) { throw new RuntimeException(e); } } };
@Override public int _updateByPrimaryKey(Table table) { try { if(table.getParams().containsKey("_id")){ DB db = database.getDB(); Jongo jongo = new Jongo(db); FindOne find = jongo.getCollection(table.getTableName()).findOne(new ObjectId(String.valueOf(table.getParams().get("_id")))); Map<String, Object> map = find.as(Map.class); boolean update = false; if (null != map) { Iterator<String> iter = table.getParams().keySet().iterator(); while (iter.hasNext()) { String key = iter.next(); if (map.containsKey(key) && !"_id".equals(key)) { map.put(key, table.getParams().get(key)); update = true; } } } if(update){ jongo.getCollection(table.getTableName()).save(map); } return 1; } } catch (MongoException e) { LOG.error("mongo update error", e); } return 0; }
@Override public int _insert(Table table) { try { DB db = database.getDB(); Jongo jongo = new Jongo(db); ObjectId newOid = ObjectId.get(); table.getParams().put("_id", newOid); jongo.getCollection(table.getTableName()).save(table.getParams()); table.getParams().put("id", newOid.toString()); } catch (MongoException e) { LOG.error("mongo insert error", e); } return 1; }
@Override public Map<String, Object> _selectByPrimaryKey(Table table) { Map<String, Object> result = null; try { DB db = database.getDB(); Jongo jongo = new Jongo(db); FindOne find = null; if(table.getConditions().containsKey("_id")){ find = jongo.getCollection(table.getTableName()).findOne(new ObjectId(String.valueOf(table.getConditions().get("_id")))); }else{ find = jongo.getCollection(table.getTableName()).findOne(JsonUtils.objToJson(table.getConditions())); } result = find.as(Map.class); result.put("id", result.get("_id").toString()); } catch (MongoException e) { LOG.error("mongo findOne error", e); } return result; }
@Override protected Jongo createInstance() throws Exception { DB db = mongo.getDB(dbName); JacksonMapper.Builder builder = new JacksonMapper.Builder(); builder.enable(DeserializationFeature.ACCEPT_SINGLE_VALUE_AS_ARRAY); if (CollectionUtils.isNotEmpty(serializers) || MapUtils.isNotEmpty(deserializers)) { builder.registerModule(JacksonUtils.createModule(serializers, deserializers)); } return new Jongo(db, builder.build()); }
QueryCriteria queryCriteria = table.getQueryCriteria(); DB db = database.getDB(); Jongo jongo = new Jongo(db); for(Criteria criteria:queryCriteria.getOredCriteria()){ for(Criterion criterion:criteria.getAllCriteria()){
@Override public int _deleteByPrimaryKey(Table table) { try { DB db = database.getDB(); Jongo jongo = new Jongo(db); if(table.getConditions().containsKey("_id")){ jongo.getCollection(table.getTableName()).remove(new ObjectId(String.valueOf(table.getConditions().get("_id")))); }else{ jongo.getCollection(table.getTableName()).remove(JsonUtils.objToJson(table.getConditions())); } } catch (MongoException e) { LOG.error("mongo findOne error", e); } return 1; }
@Override public int _deleteByCriteria(Table table) { Map<String, Object> coditon = new HashMap<String, Object>(); try { QueryCriteria queryCriteria = table.getQueryCriteria(); DB db = database.getDB(); Jongo jongo = new Jongo(db); for(Criteria criteria:queryCriteria.getOredCriteria()){ for(Criterion criterion:criteria.getAllCriteria()){ coditon = buildCriteria(criterion, coditon); } } Find find = jongo.getCollection(queryCriteria.getTable()).find(JsonUtils.objToJson(coditon)); Iterator<Map> iterator = find.as(Map.class).iterator(); while (iterator.hasNext()) { Map<String, Object> map = iterator.next(); if (null != map) { if(map.containsKey("_id")){ jongo.getCollection(table.getTableName()).remove(new ObjectId(String.valueOf(map.get("_id")))); } } } } catch (MongoException e) { LOG.error("mongo delete error", e); } return 1; }
@Override public int _countByCriteria(Table table) { int count = 0; Map<String, Object> coditon = new HashMap<String, Object>(); try { QueryCriteria queryCriteria = table.getQueryCriteria(); DB db = database.getDB(); Jongo jongo = new Jongo(db); for(Criteria criteria:queryCriteria.getOredCriteria()){ for(Criterion criterion:criteria.getAllCriteria()){ coditon = buildCriteria(criterion, coditon); } } long size = jongo.getCollection(queryCriteria.getTable()).count(JsonUtils.objToJson(coditon)); count = (int) size; } catch (MongoException e) { LOG.error("mongo find error", e); } return count; }
public MongoConfig(DataService dataService, String configId, Map<String, String> properties, boolean odataEnable) throws DataServiceFault { super(dataService, configId, DBConstants.DataSourceTypes.MONGODB, properties, odataEnable); String serversParam = properties.get(DBConstants.MongoDB.SERVERS); if (DBUtils.isEmptyString(serversParam)) { throw new DataServiceFault("The data source param '" + DBConstants.MongoDB.SERVERS + "' is required"); } this.servers = serversParam.split(","); String database = properties.get(DBConstants.MongoDB.DATABASE); if (DBUtils.isEmptyString(database)) { throw new DataServiceFault("The data source param '" + DBConstants.MongoDB.DATABASE + "' is required"); } try { this.mongoClientOptions = extractMongoOptions(properties); this.mongoClient = createNewMongo(properties); String writeConcern = properties.get(DBConstants.MongoDB.WRITE_CONCERN); if (!DBUtils.isEmptyString(writeConcern)) { this.getMongoClient().setWriteConcern(WriteConcern.valueOf(writeConcern)); } String readPref = properties.get(DBConstants.MongoDB.READ_PREFERENCE); if (!DBUtils.isEmptyString(readPref)) { this.getMongoClient().setReadPreference(ReadPreference.valueOf(readPref)); } this.getMongoClient().getDatabase(database); this.jongo = new Jongo(this.getMongoClient().getDB(database)); } catch (Exception e) { throw new DataServiceFault(e, DBConstants.FaultCodes.CONNECTION_UNAVAILABLE_ERROR, e.getMessage()); } }
/** * Instantiates a new mongo state service. * * @param params * the params * @throws UnknownHostException * the unknown host exception */ public MongoStateProvider(final ObjectNode params) throws UnknownHostException { final MongoStateConfig config = new MongoStateConfig(params); // initialization of client & jongo final MongoClient client = createClient(config.getHost(), config.getPort()); jongo = new Jongo(client.getDB(config.getDatabase())); collectionName = config.getCollection(); jongo.runCommand("{collMod: '" + collectionName + "', usePowerOf2Sizes : true }"); }
proxyFactory.createProxyFromOriginal(mongoClient.getDatabase(databaseName), MongoDatabase.class), proxyFactory.createProxyFromOriginal(db, DB.class), proxyFactory.createProxyFromOriginal(jongo != null ? jongo : new Jongo(db), Jongo.class) );
public org.jongo.MongoCollection getJongoCollection(String collectionName) { @SuppressWarnings("deprecation") DB db = mongoClient.getDB(this.db); Jongo jongo = new Jongo(db,new JacksonMapper.Builder() .registerModule(new JSR353Module()) .registerModule(new JsonOrgModule()) .registerModule(new DefaultAccessorModule()) .build()); MongoCollection collection = jongo.getCollection(collectionName); return collection; }