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.graylog2.contentpacks.facades.ViewFacade 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(ViewFacade.class);
private final ObjectMapper objectMapper;
private final ViewService viewService;
private final SearchDbService searchDbService;
@Inject
public ViewFacade(ObjectMapper objectMapper,
SearchDbService searchDbService,
ViewService viewService) {
this.objectMapper = objectMapper;
this.searchDbService = searchDbService;
this.viewService = viewService;
}
@Override
public Optional exportEntity(EntityDescriptor entityDescriptor,
EntityDescriptorIds entityDescriptorIds) {
final ModelId modelId = entityDescriptor.id();
final Optional optionalView = viewService.get(modelId.id());
if (optionalView.isPresent()) {
return Optional.of(exportNativeEntity(optionalView.get(), entityDescriptorIds));
}
LOG.debug("Couldn't find view {}", entityDescriptor);
return Optional.empty();
}
private Entity exportNativeEntity(ViewDTO view, EntityDescriptorIds entityDescriptorIds) {
final ViewEntity.Builder viewEntityBuilder = view.toContentPackEntity(entityDescriptorIds);
final Optional optionalSearch = searchDbService.get(view.searchId());
final Search search = optionalSearch.orElseThrow(() ->
new IllegalArgumentException("Search is missing in view " + view.searchId()));
SearchEntity searchEntity = search.toContentPackEntity(entityDescriptorIds);
final ViewEntity viewEntity = viewEntityBuilder.search(searchEntity).build();
final JsonNode data = objectMapper.convertValue(viewEntity, JsonNode.class);
return EntityV1.builder()
.id(ModelId.of(entityDescriptorIds.getOrThrow(EntityDescriptor.create(view.id(), getModelType()))))
.type(getModelType())
.data(data)
.build();
}
public abstract ModelType getModelType();
protected void ensureV1(Entity entity) {
if (!(entity instanceof EntityV1)) {
throw new IllegalArgumentException("Unsupported entity version: " + entity.getClass());
}
}
@Override
public NativeEntity createNativeEntity(Entity entity,
Map parameters,
Map nativeEntities,
String username) {
ensureV1(entity);
return decode((EntityV1) entity, parameters, nativeEntities);
}
protected NativeEntity decode(EntityV1 entityV1,
Map parameters,
Map nativeEntities) {
final ViewEntity viewEntity = objectMapper.convertValue(entityV1.data(), ViewEntity.class);
final Map viewStateMap = new HashMap<>(viewEntity.state().size());
for (Map.Entry entry : viewEntity.state().entrySet()) {
final ViewStateEntity entity = entry.getValue();
viewStateMap.put(entry.getKey(), entity.toNativeEntity(parameters, nativeEntities));
}
final ViewDTO.Builder viewBuilder = viewEntity.toNativeEntity(parameters, nativeEntities);
viewBuilder.state(viewStateMap);
final Search search = viewEntity.search().toNativeEntity(parameters, nativeEntities);
final Search persistedSearch = searchDbService.save(search);
final ViewDTO persistedView = viewService.save(viewBuilder.searchId(persistedSearch.id()).build());
return NativeEntity.create(entityV1.id(), persistedView.id(), getModelType(), persistedView.title(), persistedView);
}
@Override
public Optional> loadNativeEntity(NativeEntityDescriptor nativeEntityDescriptor) {
Optional optionalViewDTO = viewService.get(nativeEntityDescriptor.id().id());
return optionalViewDTO.map(viewDTO -> NativeEntity.create(nativeEntityDescriptor, viewDTO));
}
@Override
public void delete(ViewDTO nativeEntity) {
viewService.delete(nativeEntity.id());
}
@Override
public Set listEntityExcerpts() {
return getNativeViews().map(this::createExcerpt).collect(Collectors.toSet());
}
protected Stream getNativeViews() {
return viewService.streamAll().filter(v -> v.type().equals(this.getDTOType()));
}
@Override
public EntityExcerpt createExcerpt(ViewDTO nativeEntity) {
return EntityExcerpt.builder()
.id(ModelId.of(nativeEntity.id()))
.type(getModelType())
.title(nativeEntity.title())
.build();
}
public abstract ViewDTO.Type getDTOType();
@SuppressWarnings("UnstableApiUsage")
@Override
public Graph resolveNativeEntity(EntityDescriptor entityDescriptor) {
final MutableGraph mutableGraph = GraphBuilder.directed().build();
mutableGraph.addNode(entityDescriptor);
final ModelId modelId = entityDescriptor.id();
final ViewDTO viewDTO = viewService.get(modelId.id()).
orElseThrow(() -> new NoSuchElementException("Could not find view with id " + modelId.id()));
final Search search = searchDbService.get(viewDTO.searchId()).
orElseThrow(() -> new NoSuchElementException("Could not find search with id " + viewDTO.searchId()));
search.usedStreamIds().stream().map(s -> EntityDescriptor.create(s, ModelTypes.STREAM_V1))
.forEach(streamDescriptor -> mutableGraph.putEdge(entityDescriptor, streamDescriptor));
return ImmutableGraph.copyOf(mutableGraph);
}
@SuppressWarnings("UnstableApiUsage")
@Override
public Graph resolveForInstallation(Entity entity,
Map parameters,
Map entities) {
ensureV1(entity);
return resolveEntityV1((EntityV1) entity, entities);
}
@SuppressWarnings("UnstableApiUsage")
private Graph resolveEntityV1(EntityV1 entity,
Map entities) {
final ViewEntity viewEntity = objectMapper.convertValue(entity.data(), ViewEntity.class);
return resolveViewEntity(entity, viewEntity, entities);
}
@SuppressWarnings("UnstableApiUsage")
protected Graph resolveViewEntity(EntityV1 entity,
ViewEntity viewEntity,
Map entities) {
final MutableGraph mutableGraph = GraphBuilder.directed().build();
mutableGraph.addNode(entity);
viewEntity.search().usedStreamIds().stream()
.map(s -> EntityDescriptor.create(s, ModelTypes.STREAM_V1))
.map(entities::get)
.filter(Objects::nonNull)
.forEach(stream -> mutableGraph.putEdge(entity, stream));
return ImmutableGraph.copyOf(mutableGraph);
}
}
