org.graylog.plugins.pipelineprocessor.db.mongodb.MongoDbRuleService Maven / Gradle / Ivy
/**
* This file is part of Graylog.
*
* Graylog is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* Graylog is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with Graylog. If not, see .
*/
package org.graylog.plugins.pipelineprocessor.db.mongodb;
import com.google.common.collect.ImmutableSet;
import com.mongodb.BasicDBObject;
import com.mongodb.MongoException;
import org.graylog.plugins.pipelineprocessor.db.RuleDao;
import org.graylog.plugins.pipelineprocessor.db.RuleService;
import org.graylog.plugins.pipelineprocessor.events.RulesChangedEvent;
import org.graylog2.bindings.providers.MongoJackObjectMapperProvider;
import org.graylog2.database.MongoConnection;
import org.graylog2.database.NotFoundException;
import org.graylog2.events.ClusterEventBus;
import org.mongojack.DBCursor;
import org.mongojack.DBQuery;
import org.mongojack.DBSort;
import org.mongojack.JacksonDBCollection;
import org.mongojack.WriteResult;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import javax.inject.Inject;
import java.util.Collection;
import java.util.Collections;
/**
* A RuleService backed by a MongoDB collection.
*/
public class MongoDbRuleService implements RuleService {
private static final Logger log = LoggerFactory.getLogger(MongoDbRuleService.class);
private static final String COLLECTION = "pipeline_processor_rules";
private final JacksonDBCollection dbCollection;
private final ClusterEventBus clusterBus;
@Inject
public MongoDbRuleService(MongoConnection mongoConnection,
MongoJackObjectMapperProvider mapper,
ClusterEventBus clusterBus) {
this.dbCollection = JacksonDBCollection.wrap(
mongoConnection.getDatabase().getCollection(COLLECTION),
RuleDao.class,
String.class,
mapper.get());
this.clusterBus = clusterBus;
dbCollection.createIndex(DBSort.asc("title"), new BasicDBObject("unique", true));
}
@Override
public RuleDao save(RuleDao rule) {
final WriteResult save = dbCollection.save(rule);
final RuleDao savedRule = save.getSavedObject();
clusterBus.post(RulesChangedEvent.updatedRuleId(savedRule.id()));
return savedRule;
}
@Override
public RuleDao load(String id) throws NotFoundException {
final RuleDao rule = dbCollection.findOneById(id);
if (rule == null) {
throw new NotFoundException("No rule with id " + id);
}
return rule;
}
@Override
public RuleDao loadByName(String name) throws NotFoundException {
final DBQuery.Query query = DBQuery.is("title", name);
final RuleDao rule = dbCollection.findOne(query);
if (rule == null) {
throw new NotFoundException("No rule with name " + name);
}
return rule;
}
@Override
public Collection loadAll() {
try(DBCursor ruleDaos = dbCollection.find().sort(DBSort.asc("title"))) {
return ImmutableSet.copyOf((Iterable) ruleDaos);
} catch (MongoException e) {
log.error("Unable to load processing rules", e);
return Collections.emptySet();
}
}
@Override
public void delete(String id) {
final WriteResult result = dbCollection.removeById(id);
if (result.getN() != 1) {
log.error("Unable to delete rule {}", id);
}
clusterBus.post(RulesChangedEvent.deletedRuleId(id));
}
@Override
public Collection loadNamed(Collection ruleNames) {
try (DBCursor ruleDaos = dbCollection.find(DBQuery.in("title", ruleNames))) {
return ImmutableSet.copyOf((Iterable) ruleDaos);
} catch (MongoException e) {
log.error("Unable to bulk load rules", e);
return Collections.emptySet();
}
}
}