Set<Object> getPartitions(List<Document> documents) { final HashSet<Object> partitions = Sets.newHashSet(); for (Document document : documents) { final Collection<Object> documentPartitions = document.<Collection<Object>>getField(partitionIdFieldName); if (documentPartitions != null) { partitions.addAll(documentPartitions); } } return partitions; }