package org.ff4j.mongo.store;
import static org.ff4j.mongo.MongoDbConstants.MONGO_SET;
import java.util.HashSet;
/*
* #%L
* ff4j-store-mongodb-v3
* %%
* Copyright (C) 2013 - 2016 FF4J
* %%
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
* #L%
*/
import java.util.LinkedHashMap;
import java.util.Map;
import java.util.Set;
import org.bson.Document;
import org.ff4j.exception.PropertyAlreadyExistException;
import org.ff4j.mongo.MongoDbConstants;
import org.ff4j.mongo.mapper.MongoPropertyMapper;
import org.ff4j.mongo.mapper.PropertyDocumentBuilder;
import org.ff4j.property.Property;
import org.ff4j.property.store.AbstractPropertyStore;
import org.ff4j.utils.Util;
import com.mongodb.MongoClient;
import com.mongodb.client.MongoCollection;
import com.mongodb.client.MongoDatabase;
/**
* PropertyStore based on MongoDB database.
*
* @author Cedrick Lunven (@clunven)</a>
*/
public class PropertyStoreMongo extends AbstractPropertyStore {
/** Property mapper. */
private MongoPropertyMapper PMAPPER = new MongoPropertyMapper();
/** Build fields. */
private static final PropertyDocumentBuilder BUILDER = new PropertyDocumentBuilder();
/** MongoDB collection. */
private MongoCollection<Document> propertiesCollection;
/** Feature collection Name. */
private String collectionName = MongoDbConstants.DEFAULT_PROPERTY_COLLECTION;
/** Database name. */
private String dbName = MongoDbConstants.DEFAULT_DBNAME;
/** Current mongo client. */
private MongoClient mongoClient;
/**
* Parameterized constructor with collection.
*
* @param collection
* the collection to set
*/
public PropertyStoreMongo() {
}
/**
* Parameterized constructor with collection.
*
* @param collection
* the collection to set
*/
public PropertyStoreMongo(MongoClient client) {
this.mongoClient = client;
this.propertiesCollection = getPropertiesCollection();
}
/**
* Parameterized constructor with collection.
*
* @param collection
* the collection to set
*/
public PropertyStoreMongo(MongoClient client, String dbName, String collectionName) {
this.mongoClient = client;
this.collectionName = collectionName;
this.dbName = dbName;
this.propertiesCollection = getPropertiesCollection();
}
/**
* Parameterized constructor with collection.
*
* @param collection
* the collection to set
*/
public PropertyStoreMongo(MongoDatabase db) {
this(db, MongoDbConstants.DEFAULT_PROPERTY_COLLECTION);
}
/**
* Parameterized constructor with collection.
*
* @param collection
* the collection to set
*/
public PropertyStoreMongo(MongoDatabase db, String collectionName) {
this.propertiesCollection = db.getCollection(collectionName);
this.dbName = db.getName();
}
/**
* Parameterized constructor with collection.
*
* @param collection
* the collection to set
*/
public PropertyStoreMongo(MongoCollection<Document> collection) {
this.propertiesCollection = collection;
}
/**
* Parameterized constructor with collection.
*
* @param collection
* the collection to set
*/
public PropertyStoreMongo(MongoCollection<Document> collection, String xmlConfFile) {
this(collection);
importPropertiesFromXmlFile(xmlConfFile);
}
/** {@inheritDoc} */
public boolean existProperty(String name) {
Util.assertHasLength(name);
return 1 == getPropertiesCollection().count(BUILDER.getName(name));
}
/** {@inheritDoc} */
public <T> void createProperty(Property<T> prop) {
if (prop == null) {
throw new IllegalArgumentException("Property cannot be null nor empty");
}
if (existProperty(prop.getName())) {
throw new PropertyAlreadyExistException(prop.getName());
}
getPropertiesCollection().insertOne(PMAPPER.toStore(prop));
}
/** {@inheritDoc} */
public Property<?> readProperty(String name) {
assertPropertyExist(name);
Document object = getPropertiesCollection().find(BUILDER.getName(name)).first();
return PMAPPER.fromStore(object);
}
/** {@inheritDoc} */
public void deleteProperty(String name) {
assertPropertyExist(name);
getPropertiesCollection().deleteOne(BUILDER.getName(name));
}
/** {@inheritDoc} */
public void clear() {
getPropertiesCollection().deleteMany(new Document());
}
/** {@inheritDoc} */
public void updateProperty(String name, String newValue) {
assertPropertyExist(name);
readProperty(name).fromString(newValue);
Document query = BUILDER.getName(name);
Document update = BUILDER.getValue(newValue);
getPropertiesCollection().updateOne(query, new Document(MONGO_SET, update));
}
/** {@inheritDoc} */
public <T> void updateProperty(Property<T> prop) {
Util.assertNotNull(prop);
// Delete
deleteProperty(prop.getName());
// Create
createProperty(prop);
}
/** {@inheritDoc} */
public Map<String, Property<?>> readAllProperties() {
LinkedHashMap<String, Property<?>> mapP = new LinkedHashMap<String, Property<?>>();
for(Document document : getPropertiesCollection().find()) {
Property<?> prop = PMAPPER.fromStore(document);
mapP.put(prop.getName(), prop);
}
return mapP;
}
/** {@inheritDoc} */
public Set<String> listPropertyNames() {
return readAllProperties().keySet();
}
/** {@inheritDoc} */
@Override
public void createSchema() {
if (!mongoClient.getDatabase(dbName)
.listCollectionNames()
.into(new HashSet<String>())
.contains(collectionName)) {
mongoClient.getDatabase(dbName).createCollection(collectionName);
}
propertiesCollection = mongoClient.getDatabase(dbName).getCollection(collectionName);
}
/**
* Getter accessor for attribute 'featuresCollection'.
*
* @return
* current value of 'featuresCollection'
*/
public MongoCollection<Document> getPropertiesCollection() {
if (propertiesCollection == null) {
if (mongoClient != null) {
createSchema();
} else {
throw new IllegalStateException("Cannot initialize Properties collection : no mongo client defined");
}
}
return propertiesCollection;
}
}