Examples with StandardRevisionClaim org.apache.nifi.web.revision.StandardRevisionClaim used on opensource projects

Example 6 with StandardRevisionClaim

use of org.apache.nifi.web.revision.StandardRevisionClaim in project nifi by apache.

the class StandardNiFiServiceFacade method createControllerService.

@Override
public ControllerServiceEntity createControllerService(final Revision revision, final String groupId, final ControllerServiceDTO controllerServiceDTO) {
    controllerServiceDTO.setParentGroupId(groupId);
    final NiFiUser user = NiFiUserUtils.getNiFiUser();
    // request claim for component to be created... revision already verified (version == 0)
    final RevisionClaim claim = new StandardRevisionClaim(revision);
    final RevisionUpdate<ControllerServiceDTO> snapshot;
    if (groupId == null) {
        // update revision through revision manager
        snapshot = revisionManager.updateRevision(claim, user, () -> {
            // Unfortunately, we can not use the createComponent() method here because createComponent() wants to obtain the read lock
            // on the group. The Controller Service may or may not have a Process Group (it won't if it's controller-scoped).
            final ControllerServiceNode controllerService = controllerServiceDAO.createControllerService(controllerServiceDTO);
            final ControllerServiceDTO dto = dtoFactory.createControllerServiceDto(controllerService);
            controllerFacade.save();
            final FlowModification lastMod = new FlowModification(revision.incrementRevision(revision.getClientId()), user.getIdentity());
            return new StandardRevisionUpdate<>(dto, lastMod);
        });
    } else {
        snapshot = revisionManager.updateRevision(claim, user, () -> {
            final ControllerServiceNode controllerService = controllerServiceDAO.createControllerService(controllerServiceDTO);
            final ControllerServiceDTO dto = dtoFactory.createControllerServiceDto(controllerService);
            controllerFacade.save();
            final FlowModification lastMod = new FlowModification(revision.incrementRevision(revision.getClientId()), user.getIdentity());
            return new StandardRevisionUpdate<>(dto, lastMod);
        });
    }
    final ControllerServiceNode controllerService = controllerServiceDAO.getControllerService(controllerServiceDTO.getId());
    final PermissionsDTO permissions = dtoFactory.createPermissionsDto(controllerService);
    final List<BulletinDTO> bulletins = dtoFactory.createBulletinDtos(bulletinRepository.findBulletinsForSource(controllerServiceDTO.getId()));
    final List<BulletinEntity> bulletinEntities = bulletins.stream().map(bulletin -> entityFactory.createBulletinEntity(bulletin, permissions.getCanRead())).collect(Collectors.toList());
    return entityFactory.createControllerServiceEntity(snapshot.getComponent(), dtoFactory.createRevisionDTO(snapshot.getLastModification()), permissions, bulletinEntities);
}

Example 7 with StandardRevisionClaim

use of org.apache.nifi.web.revision.StandardRevisionClaim in project nifi by apache.

the class StandardNiFiServiceFacade method updateControllerServiceReferencingComponents.

@Override
public ControllerServiceReferencingComponentsEntity updateControllerServiceReferencingComponents(final Map<String, Revision> referenceRevisions, final String controllerServiceId, final ScheduledState scheduledState, final ControllerServiceState controllerServiceState) {
    final RevisionClaim claim = new StandardRevisionClaim(referenceRevisions.values());
    final NiFiUser user = NiFiUserUtils.getNiFiUser();
    final RevisionUpdate<ControllerServiceReferencingComponentsEntity> update = revisionManager.updateRevision(claim, user, new UpdateRevisionTask<ControllerServiceReferencingComponentsEntity>() {

        @Override
        public RevisionUpdate<ControllerServiceReferencingComponentsEntity> update() {
            final Set<ConfiguredComponent> updated = controllerServiceDAO.updateControllerServiceReferencingComponents(controllerServiceId, scheduledState, controllerServiceState);
            final ControllerServiceReference updatedReference = controllerServiceDAO.getControllerService(controllerServiceId).getReferences();
            // get the revisions of the updated components
            final Map<String, Revision> updatedRevisions = new HashMap<>();
            for (final ConfiguredComponent component : updated) {
                final Revision currentRevision = revisionManager.getRevision(component.getIdentifier());
                final Revision requestRevision = referenceRevisions.get(component.getIdentifier());
                updatedRevisions.put(component.getIdentifier(), currentRevision.incrementRevision(requestRevision.getClientId()));
            }
            // ensure the revision for all referencing components is included regardless of whether they were updated in this request
            for (final ConfiguredComponent component : findAllReferencingComponents(updatedReference)) {
                updatedRevisions.putIfAbsent(component.getIdentifier(), revisionManager.getRevision(component.getIdentifier()));
            }
            final ControllerServiceReferencingComponentsEntity entity = createControllerServiceReferencingComponentsEntity(updatedReference, updatedRevisions);
            return new StandardRevisionUpdate<>(entity, null, new HashSet<>(updatedRevisions.values()));
        }
    });
    return update.getComponent();
}

Example 8 with StandardRevisionClaim

use of org.apache.nifi.web.revision.StandardRevisionClaim in project nifi by apache.

the class StandardNiFiServiceFacade method createRegistryClient.

@Override
public RegistryClientEntity createRegistryClient(Revision revision, RegistryDTO registryDTO) {
    final NiFiUser user = NiFiUserUtils.getNiFiUser();
    // request claim for component to be created... revision already verified (version == 0)
    final RevisionClaim claim = new StandardRevisionClaim(revision);
    // update revision through revision manager
    final RevisionUpdate<FlowRegistry> revisionUpdate = revisionManager.updateRevision(claim, user, () -> {
        // add the component
        final FlowRegistry registry = registryDAO.createFlowRegistry(registryDTO);
        // save the flow
        controllerFacade.save();
        final FlowModification lastMod = new FlowModification(revision.incrementRevision(revision.getClientId()), user.getIdentity());
        return new StandardRevisionUpdate<>(registry, lastMod);
    });
    final FlowRegistry registry = revisionUpdate.getComponent();
    return createRegistryClientEntity(registry);
}

Example 9 with StandardRevisionClaim

use of org.apache.nifi.web.revision.StandardRevisionClaim in project nifi by apache.

the class StandardNiFiServiceFacade method createComponent.

/**
 * Creates a component using the optimistic locking manager.
 *
 * @param componentDto the DTO that will be used to create the component
 * @param daoCreation  A Supplier that will create the NiFi Component to use
 * @param dtoCreation  a Function that will convert the NiFi Component into a corresponding DTO
 * @param <D>          the DTO Type
 * @param <C>          the NiFi Component Type
 * @return a RevisionUpdate that represents the updated configuration
 */
private <D, C> RevisionUpdate<D> createComponent(final Revision revision, final ComponentDTO componentDto, final Supplier<C> daoCreation, final Function<C, D> dtoCreation) {
    final NiFiUser user = NiFiUserUtils.getNiFiUser();
    // read lock on the containing group
    // request claim for component to be created... revision already verified (version == 0)
    final RevisionClaim claim = new StandardRevisionClaim(revision);
    // update revision through revision manager
    return revisionManager.updateRevision(claim, user, () -> {
        // add the component
        final C component = daoCreation.get();
        // save the flow
        controllerFacade.save();
        final D dto = dtoCreation.apply(component);
        final FlowModification lastMod = new FlowModification(revision.incrementRevision(revision.getClientId()), user.getIdentity());
        return new StandardRevisionUpdate<>(dto, lastMod);
    });
}

Example 10 with StandardRevisionClaim

use of org.apache.nifi.web.revision.StandardRevisionClaim in project nifi by apache.

the class StandardNiFiServiceFacade method updateRegistryClient.

@Override
public RegistryClientEntity updateRegistryClient(Revision revision, RegistryDTO registryDTO) {
    final RevisionClaim revisionClaim = new StandardRevisionClaim(revision);
    final NiFiUser user = NiFiUserUtils.getNiFiUser();
    final FlowRegistry registry = registryDAO.getFlowRegistry(registryDTO.getId());
    final RevisionUpdate<FlowRegistry> revisionUpdate = revisionManager.updateRevision(revisionClaim, user, () -> {
        final boolean duplicateName = registryDAO.getFlowRegistries().stream().anyMatch(reg -> reg.getName().equals(registryDTO.getName()) && !reg.getIdentifier().equals(registryDTO.getId()));
        if (duplicateName) {
            throw new IllegalStateException("Cannot update Flow Registry because a Flow Registry already exists with the name " + registryDTO.getName());
        }
        registry.setDescription(registryDTO.getDescription());
        registry.setName(registryDTO.getName());
        registry.setURL(registryDTO.getUri());
        controllerFacade.save();
        final Revision updatedRevision = revisionManager.getRevision(revision.getComponentId()).incrementRevision(revision.getClientId());
        final FlowModification lastModification = new FlowModification(updatedRevision, user.getIdentity());
        return new StandardRevisionUpdate<>(registry, lastModification);
    });
    final FlowRegistry updatedReg = revisionUpdate.getComponent();
    return createRegistryClientEntity(updatedReg);
}