Examples with Nullable org.springframework.lang.Nullable used on opensource projects

Example 1 with Nullable

use of org.springframework.lang.Nullable in project spring-data-mongodb by spring-projects.

the class MongoTemplate method count.

/*
	 * (non-Javadoc)
	 * @see org.springframework.data.mongodb.core.MongoOperations#count(org.springframework.data.mongodb.core.query.Query, java.lang.Class, java.lang.String)
	 */
public long count(Query query, @Nullable Class<?> entityClass, String collectionName) {
    Assert.notNull(query, "Query must not be null!");
    Assert.hasText(collectionName, "Collection name must not be null or empty!");
    CountOptions options = new CountOptions();
    query.getCollation().map(Collation::toMongoCollation).ifPresent(options::collation);
    Document document = queryMapper.getMappedObject(query.getQueryObject(), Optional.ofNullable(entityClass).map(it -> mappingContext.getPersistentEntity(entityClass)));
    return execute(collectionName, collection -> collection.count(document, options));
}

Example 2 with Nullable

use of org.springframework.lang.Nullable in project spring-data-mongodb by spring-projects.

the class DocumentAccessor method get.

/**
 * Returns the value the given {@link MongoPersistentProperty} refers to. By default this will be a direct field but
 * the method will also transparently resolve nested values the {@link MongoPersistentProperty} might refer to through
 * a path expression in the field name metadata.
 *
 * @param property must not be {@literal null}.
 * @return
 */
@Nullable
public Object get(MongoPersistentProperty property) {
    String fieldName = property.getFieldName();
    if (!fieldName.contains(".")) {
        return BsonUtils.asMap(this.document).get(fieldName);
    }
    Iterator<String> parts = Arrays.asList(fieldName.split("\\.")).iterator();
    Map<String, Object> source = BsonUtils.asMap(this.document);
    Object result = null;
    while (source != null && parts.hasNext()) {
        result = source.get(parts.next());
        if (parts.hasNext()) {
            source = getAsMap(result);
        }
    }
    return result;
}

Example 3 with Nullable

use of org.springframework.lang.Nullable in project spring-data-mongodb by spring-projects.

the class MappingMongoConverter method read.

@Nullable
private <S extends Object> S read(final MongoPersistentEntity<S> entity, final Document bson, final ObjectPath path) {
    DefaultSpELExpressionEvaluator evaluator = new DefaultSpELExpressionEvaluator(bson, spELContext);
    ParameterValueProvider<MongoPersistentProperty> provider = getParameterProvider(entity, bson, evaluator, path);
    EntityInstantiator instantiator = instantiators.getInstantiatorFor(entity);
    S instance = instantiator.createInstance(entity, provider);
    PersistentPropertyAccessor accessor = new ConvertingPropertyAccessor(entity.getPropertyAccessor(instance), conversionService);
    MongoPersistentProperty idProperty = entity.getIdProperty();
    DocumentAccessor documentAccessor = new DocumentAccessor(bson);
    // make sure id property is set before all other properties
    Object idValue = null;
    if (idProperty != null && documentAccessor.hasValue(idProperty)) {
        idValue = readIdValue(path, evaluator, idProperty, documentAccessor);
        accessor.setProperty(idProperty, idValue);
    }
    ObjectPath currentPath = path.push(instance, entity, idValue != null ? bson.get(idProperty.getFieldName()) : null);
    MongoDbPropertyValueProvider valueProvider = new MongoDbPropertyValueProvider(documentAccessor, evaluator, currentPath);
    DbRefResolverCallback callback = new DefaultDbRefResolverCallback(bson, currentPath, evaluator, MappingMongoConverter.this);
    readProperties(entity, accessor, idProperty, documentAccessor, valueProvider, callback);
    return instance;
}

Example 4 with Nullable

use of org.springframework.lang.Nullable in project spring-data-mongodb by spring-projects.

the class MappingMongoConverter method read.

@Nullable
@SuppressWarnings("unchecked")
private <S extends Object> S read(TypeInformation<S> type, @Nullable Bson bson, ObjectPath path) {
    if (null == bson) {
        return null;
    }
    TypeInformation<? extends S> typeToUse = typeMapper.readType(bson, type);
    Class<? extends S> rawType = typeToUse.getType();
    if (conversions.hasCustomReadTarget(bson.getClass(), rawType)) {
        return conversionService.convert(bson, rawType);
    }
    if (DBObject.class.isAssignableFrom(rawType)) {
        return (S) bson;
    }
    if (Document.class.isAssignableFrom(rawType)) {
        return (S) bson;
    }
    if (typeToUse.isCollectionLike() && bson instanceof List) {
        return (S) readCollectionOrArray(typeToUse, (List<?>) bson, path);
    }
    if (typeToUse.isMap()) {
        return (S) readMap(typeToUse, bson, path);
    }
    if (bson instanceof Collection) {
        throw new MappingException(String.format(INCOMPATIBLE_TYPES, bson, BasicDBList.class, typeToUse.getType(), path));
    }
    if (typeToUse.equals(ClassTypeInformation.OBJECT)) {
        return (S) bson;
    }
    // Retrieve persistent entity info
    Document target = bson instanceof BasicDBObject ? new Document((BasicDBObject) bson) : (Document) bson;
    MongoPersistentEntity<?> entity = mappingContext.getPersistentEntity(typeToUse);
    if (entity == null) {
        throw new MappingException(String.format(INVALID_TYPE_TO_READ, target, typeToUse.getType()));
    }
    return read((MongoPersistentEntity<S>) mappingContext.getRequiredPersistentEntity(typeToUse), target, path);
}

Example 5 with Nullable

use of org.springframework.lang.Nullable in project spring-data-mongodb by spring-projects.

the class MappingMongoConverter method convertToMongoType.

/*
	 * (non-Javadoc)
	 * @see org.springframework.data.mongodb.core.convert.MongoWriter#convertToMongoType(java.lang.Object, org.springframework.data.util.TypeInformation)
	 */
@Nullable
@SuppressWarnings("unchecked")
@Override
public Object convertToMongoType(@Nullable Object obj, TypeInformation<?> typeInformation) {
    if (obj == null) {
        return null;
    }
    Optional<Class<?>> target = conversions.getCustomWriteTarget(obj.getClass());
    if (target.isPresent()) {
        return conversionService.convert(obj, target.get());
    }
    if (conversions.isSimpleType(obj.getClass())) {
        // Doesn't need conversion
        return getPotentiallyConvertedSimpleWrite(obj);
    }
    TypeInformation<?> typeHint = typeInformation;
    if (obj instanceof List) {
        return maybeConvertList((List<Object>) obj, typeHint);
    }
    if (obj instanceof Document) {
        Document newValueDocument = new Document();
        for (String vk : ((Document) obj).keySet()) {
            Object o = ((Document) obj).get(vk);
            newValueDocument.put(vk, convertToMongoType(o, typeHint));
        }
        return newValueDocument;
    }
    if (obj instanceof DBObject) {
        Document newValueDbo = new Document();
        for (String vk : ((DBObject) obj).keySet()) {
            Object o = ((DBObject) obj).get(vk);
            newValueDbo.put(vk, convertToMongoType(o, typeHint));
        }
        return newValueDbo;
    }
    if (obj instanceof Map) {
        Document result = new Document();
        for (Map.Entry<Object, Object> entry : ((Map<Object, Object>) obj).entrySet()) {
            result.put(entry.getKey().toString(), convertToMongoType(entry.getValue(), typeHint));
        }
        return result;
    }
    if (obj.getClass().isArray()) {
        return maybeConvertList(Arrays.asList((Object[]) obj), typeHint);
    }
    if (obj instanceof Collection) {
        return maybeConvertList((Collection<?>) obj, typeHint);
    }
    Document newDocument = new Document();
    this.write(obj, newDocument);
    if (typeInformation == null) {
        return removeTypeInfo(newDocument, true);
    }
    if (typeInformation.getType().equals(NestedDocument.class)) {
        return removeTypeInfo(newDocument, false);
    }
    return !obj.getClass().equals(typeInformation.getType()) ? newDocument : removeTypeInfo(newDocument, true);
}