All Downloads are FREE. Search and download functionalities are using the official Maven repository.
Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
org.graylog.events.contentpack.facade.EventDefinitionFacade 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 {
private static final Logger LOG = LoggerFactory.getLogger(EventDefinitionFacade.class);
private final ObjectMapper objectMapper;
private final EventDefinitionHandler eventDefinitionHandler;
private final DBJobDefinitionService jobDefinitionService;
private final DBEventDefinitionService eventDefinitionService;
private final Set pluginMetaData;
@Inject
public EventDefinitionFacade(ObjectMapper objectMapper,
EventDefinitionHandler eventDefinitionHandler,
Set pluginMetaData,
DBJobDefinitionService jobDefinitionService,
DBEventDefinitionService eventDefinitionService) {
this.objectMapper = objectMapper;
this.pluginMetaData = pluginMetaData;
this.eventDefinitionHandler = eventDefinitionHandler;
this.jobDefinitionService = jobDefinitionService;
this.eventDefinitionService = eventDefinitionService;
}
@VisibleForTesting
private Entity exportNativeEntity(EventDefinitionDto eventDefinition, EntityDescriptorIds entityDescriptorIds) {
// Presence of a job definition means that the event definition should be scheduled
final Optional jobDefinition = jobDefinitionService.getByConfigField(EventProcessorExecutionJob.Config.FIELD_EVENT_DEFINITION_ID, eventDefinition.id());
final EventDefinitionEntity entity = eventDefinition.toContentPackEntity(entityDescriptorIds)
.toBuilder()
.isScheduled(ValueReference.of(jobDefinition.isPresent()))
.build();
final JsonNode data = objectMapper.convertValue(entity, JsonNode.class);
return EntityV1.builder()
.id(ModelId.of(entityDescriptorIds.getOrThrow(eventDefinition.id(), ModelTypes.EVENT_DEFINITION_V1)))
.type(ModelTypes.EVENT_DEFINITION_V1)
.constraints(versionConstraints(eventDefinition))
.data(data)
.build();
}
@Override
public Optional exportEntity(EntityDescriptor entityDescriptor, EntityDescriptorIds entityDescriptorIds) {
final ModelId modelId = entityDescriptor.id();
final Optional eventDefinition = eventDefinitionService.get(modelId.id());
if (!eventDefinition.isPresent()) {
LOG.debug("Couldn't find event definition {}", entityDescriptor);
return Optional.empty();
}
return Optional.of(exportNativeEntity(eventDefinition.get(), entityDescriptorIds));
}
private ImmutableSet versionConstraints(EventDefinitionDto eventDefinitionDto) {
final String packageName = eventDefinitionDto.config().getContentPackPluginPackage();
return pluginMetaData.stream()
.filter(metaData -> packageName.equals(metaData.getClass().getCanonicalName()))
.map(PluginVersionConstraint::of)
.collect(ImmutableSet.toImmutableSet());
}
@Override
public NativeEntity createNativeEntity(Entity entity,
Map parameters,
Map nativeEntities,
String username) {
if (entity instanceof EntityV1) {
return decode((EntityV1) entity, parameters, nativeEntities);
} else {
throw new IllegalArgumentException("Unsupported entity version: " + entity.getClass());
}
}
private NativeEntity decode(EntityV1 entity,
Map parameters,
Map nativeEntities) {
final EventDefinitionEntity eventDefinitionEntity = objectMapper.convertValue(entity.data(),
EventDefinitionEntity.class);
final EventDefinitionDto eventDefinition = eventDefinitionEntity.toNativeEntity(parameters, nativeEntities);
final EventDefinitionDto savedDto;
if (eventDefinitionEntity.isScheduled().asBoolean(parameters)) {
savedDto = eventDefinitionHandler.create(eventDefinition);
} else {
savedDto = eventDefinitionHandler.createWithoutSchedule(eventDefinition);
}
return NativeEntity.create(entity.id(), savedDto.id(), ModelTypes.EVENT_DEFINITION_V1, savedDto.title(), savedDto);
}
@Override
public Optional> loadNativeEntity(NativeEntityDescriptor nativeEntityDescriptor) {
final Optional eventDefinition = eventDefinitionService.get(nativeEntityDescriptor.id().id());
return eventDefinition.map(eventDefinitionDto ->
NativeEntity.create(nativeEntityDescriptor, eventDefinitionDto));
}
@Override
public void delete(EventDefinitionDto nativeEntity) {
eventDefinitionHandler.delete(nativeEntity.id());
}
@Override
public EntityExcerpt createExcerpt(EventDefinitionDto nativeEntity) {
return EntityExcerpt.builder()
.id(ModelId.of(nativeEntity.id()))
.type(ModelTypes.EVENT_DEFINITION_V1)
.title(nativeEntity.title())
.build();
}
@Override
public Set listEntityExcerpts() {
return eventDefinitionService.streamAll()
.map(this::createExcerpt)
.collect(Collectors.toSet());
}
@Override
public Graph resolveNativeEntity(EntityDescriptor entityDescriptor) {
final MutableGraph mutableGraph = GraphBuilder.directed().build();
mutableGraph.addNode(entityDescriptor);
final ModelId modelId = entityDescriptor.id();
final Optional eventDefinition = eventDefinitionService.get(modelId.id());
if (eventDefinition.isPresent()) {
eventDefinition.get().resolveNativeEntity(entityDescriptor, mutableGraph);
} else {
LOG.debug("Couldn't find event definition {}", entityDescriptor);
}
return ImmutableGraph.copyOf(mutableGraph);
}
@Override
public Graph resolveForInstallation(Entity entity, Map parameters, Map entities) {
if (entity instanceof EntityV1) {
return resolveForInstallationV1((EntityV1) entity, parameters, entities);
} else {
throw new IllegalArgumentException("Unsupported entity version: " + entity.getClass());
}
}
private Graph resolveForInstallationV1(EntityV1 entity, Map parameters, Map entities) {
final MutableGraph graph = GraphBuilder.directed().build();
graph.addNode(entity);
final EventDefinitionEntity eventDefinition = objectMapper.convertValue(entity.data(), EventDefinitionEntity.class);
eventDefinition.resolveForInstallation(entity, parameters, entities, graph);
return ImmutableGraph.copyOf(graph);
}
}
