@Override public long totalExtractorCount() { final DBObject query = new BasicDBObject(InputImpl.EMBEDDED_EXTRACTORS, new BasicDBObject("$exists", true)); long extractorsCount = 0; try (DBCursor inputs = dbCollection.find(query, new BasicDBObject(InputImpl.EMBEDDED_EXTRACTORS, 1))) { for (DBObject input : inputs) { final BasicDBList extractors = (BasicDBList) input.get(InputImpl.EMBEDDED_EXTRACTORS); extractorsCount += extractors.size(); } } return extractorsCount; }
@Override public Map<String, String> getStaticFields() { if (fields.get(EMBEDDED_STATIC_FIELDS) == null) { return Collections.emptyMap(); } final BasicDBList list = (BasicDBList) fields.get(EMBEDDED_STATIC_FIELDS); final Map<String, String> staticFields = Maps.newHashMapWithExpectedSize(list.size()); for (final Object element : list) { try { final DBObject field = (DBObject) element; staticFields.put((String) field.get(FIELD_STATIC_FIELD_KEY), (String) field.get(FIELD_STATIC_FIELD_VALUE)); } catch (Exception e) { LOG.error("Cannot build static field from persisted data. Skipping.", e); } } return staticFields; }
groupMapping = Maps.newHashMapWithExpectedSize(groupMappingList.size()); for (Object entry : groupMappingList) { final DBObject field = (DBObject) entry;
@Nullable private Version getMongoVersion(DB adminDb) { final CommandResult buildInfoResult = adminDb.command("buildInfo"); if (buildInfoResult.ok()) { final BasicDBList versionArray = (BasicDBList) buildInfoResult.get("versionArray"); if (versionArray == null || versionArray.size() < 3) { LOG.debug("Couldn't retrieve MongoDB version"); return null; } final int majorVersion = (int) versionArray.get(0); final int minorVersion = (int) versionArray.get(1); final int patchVersion = (int) versionArray.get(2); return Version.forIntegers(majorVersion, minorVersion, patchVersion); } else { LOG.debug("Couldn't retrieve MongoDB buildInfo: {}", buildInfoResult.getErrorMessage()); return null; } }
private Object negate(BasicDBObject arg) { BasicDBList list = new BasicDBList(); for (Map.Entry<String, Object> entry : arg.entrySet()) { if (entry.getKey().equals("$or")) { list.add(asDBObject("$nor", entry.getValue())); } else if (entry.getKey().equals("$and")) { BasicDBList list2 = new BasicDBList(); for (Object o : ((BasicDBList) entry.getValue())) { list2.add(negate((BasicDBObject) o)); } list.add(asDBObject("$or", list2)); } else if (entry.getValue() instanceof Pattern) { list.add(asDBObject(entry.getKey(), asDBObject("$not", entry.getValue()))); } else if (entry.getValue() instanceof BasicDBObject) { list.add(negate(entry.getKey(), (BasicDBObject) entry.getValue())); } else { list.add(asDBObject(entry.getKey(), asDBObject("$ne", entry.getValue()))); } } return list.size() == 1 ? list.get(0) : asDBObject("$or", list); }
/** * Copies this instance into a new Object. * * @return a new BasicDBList with the same values as this instance */ public Object copy() { // copy field values into new object BasicDBList newobj = new BasicDBList(); // need to clone the sub obj for (int i = 0; i < size(); ++i) { Object val = get(i); if (val instanceof BasicDBObject) { val = ((BasicDBObject) val).copy(); } else if (val instanceof BasicDBList) { val = ((BasicDBList) val).copy(); } newobj.add(val); } return newobj; }
if (mappedValue instanceof BasicDBList) { final BasicDBList list = (BasicDBList) mappedValue; if (list.size() != 0) { if (!EmbeddedMapper.shouldSaveClassName(extractFirstElement(value), list.get(0), mf)) { for (Object o : list) {
result = (client).executeScript(script); BasicDBList resultList = (BasicDBList) result; Assert.assertEquals(3, resultList.size()); Assert.assertNotNull(result); resultList = (BasicDBList) result; Assert.assertEquals("vivek", resultList.get(0)); Assert.assertEquals(1, resultList.size()); result = (client).executeScript(script); resultList = (BasicDBList) result; Assert.assertEquals(3, resultList.size()); result = (client).executeScript(script); resultList = (BasicDBList) result; Assert.assertEquals(1, resultList.size());
@Override public int getResultCount() { return resultSet.size(); } });
private void flushBatch() { // Create an executor task to process logger.debug("Queuing batch for processing - size : {}, queuelength : {}", currentBatch.size(), batchQueue.size()); executor.submit(new BatchProcessingTask(currentBatch)); currentBatch = new BasicDBList(); } }
@Override public long totalExtractorCount() { final DBObject query = new BasicDBObject(InputImpl.EMBEDDED_EXTRACTORS, new BasicDBObject("$exists", true)); final DBCursor inputs = collection(InputImpl.class).find(query, new BasicDBObject(InputImpl.EMBEDDED_EXTRACTORS, 1)); long extractorsCount = 0; for (DBObject input : inputs) { final BasicDBList extractors = (BasicDBList) input.get(InputImpl.EMBEDDED_EXTRACTORS); extractorsCount += extractors.size(); } return extractorsCount; }
private void executeBatchUpdate(BulkWriteOperation batchOp, BasicDBList fullBatch) { if(batchOp != null){ BulkWriteResult result = batchOp.execute(); logger.debug("Wrote sample batch - sent {} : updated {}", fullBatch.size(), result.getModifiedCount()); } }
private void executeBatchWrite(BulkWriteOperation batchOp, BasicDBList fullBatch) { if(batchOp != null){ BulkWriteResult result = batchOp.execute(); logger.debug("Wrote sample batch - sent {} : inserted {}", fullBatch.size(), result.getInsertedCount()); } } }
static MultiPolygon toMultiPolygon(DBObject obj) { GeometryFactory gf = new GeometryFactory(); BasicDBList list = (BasicDBList) obj.get("coordinates"); Polygon[] polys = new Polygon[list.size()]; for (int i = 0; i < list.size(); i++) { polys[i] = toPolygon((BasicDBList)list.get(i)); } return gf.createMultiPolygon(polys); }
static GeometryCollection toCollection(DBObject obj) { BasicDBList list = (BasicDBList) obj.get("geometries"); Geometry[] geoms = new Geometry[list.size()]; for (int i = 0; i < list.size(); i++) { geoms[i] = toGeometry((DBObject) list.get(i)); } return new GeometryFactory().createGeometryCollection(geoms); }
BasicDBList bookingsList = (BasicDBList) theObj.get("bookings"); for (int i = 0; i < bookingsList.size(); i++) { //this one gets the first document BasicDBObject bookingObj = (BasicDBObject) bookingsList.get(i); //this one gets 'bookings' embedded document, which is also a list List<BasicDBObject> embeddedBookingObjList = (List<BasicDBObject>) bookingObj.get("bookings"); Date startDateTime = embeddedBookingObjList.get(0).getDate("startDateTime"); booking.setStartDateTime(startDateTime); }
static Polygon toPolygon(BasicDBList list) { GeometryFactory gf = new GeometryFactory(); LinearRing shell = gf.createLinearRing(toCoordinates((BasicDBList) list.get(0))); LinearRing[] holes = list.size() > 1 ? new LinearRing[list.size()-1] : null; for (int i = 1; i < list.size(); i++) { holes[i-1] = gf.createLinearRing(toCoordinates((BasicDBList) list.get(i))); } return new GeometryFactory().createPolygon(shell, holes); } }
static MultiLineString toMultiLineString(DBObject obj) { GeometryFactory gf = new GeometryFactory(); BasicDBList list = (BasicDBList) obj.get("coordinates"); LineString[] lines = new LineString[list.size()]; for (int i = 0; i < list.size(); i++) { lines[i] = gf.createLineString(toCoordinates((BasicDBList)list.get(i))); } return gf.createMultiLineString(lines); }
while(cursor.hasNext()) { BasicDBObject acc=(BasicDBObject) cursor.next(); BasicDBList scores = (BasicDBList) acc.get("scores"); for(int i=0;i<scores.size();i++) { BasicDBObject score = (BasicDBObject) scores.get(i); System.out.println(score.get("score")); } }