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

Example 6 with StandardRevisionUpdate

use of org.apache.nifi.web.revision.StandardRevisionUpdate 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 7 with StandardRevisionUpdate

use of org.apache.nifi.web.revision.StandardRevisionUpdate 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 8 with StandardRevisionUpdate

use of org.apache.nifi.web.revision.StandardRevisionUpdate 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);
}

Example 9 with StandardRevisionUpdate

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

the class StandardNiFiServiceFacade method updateSnippet.

@Override
public SnippetEntity updateSnippet(final Set<Revision> revisions, final SnippetDTO snippetDto) {
    final NiFiUser user = NiFiUserUtils.getNiFiUser();
    final RevisionClaim revisionClaim = new StandardRevisionClaim(revisions);
    final RevisionUpdate<SnippetDTO> snapshot;
    try {
        snapshot = revisionManager.updateRevision(revisionClaim, user, new UpdateRevisionTask<SnippetDTO>() {

            @Override
            public RevisionUpdate<SnippetDTO> update() {
                // get the updated component
                final Snippet snippet = snippetDAO.updateSnippetComponents(snippetDto);
                // drop the snippet
                snippetDAO.dropSnippet(snippet.getId());
                // save updated controller
                controllerFacade.save();
                // increment the revisions
                final Set<Revision> updatedRevisions = revisions.stream().map(revision -> {
                    final Revision currentRevision = revisionManager.getRevision(revision.getComponentId());
                    return currentRevision.incrementRevision(revision.getClientId());
                }).collect(Collectors.toSet());
                final SnippetDTO dto = dtoFactory.createSnippetDto(snippet);
                return new StandardRevisionUpdate<>(dto, null, updatedRevisions);
            }
        });
    } catch (final ExpiredRevisionClaimException e) {
        throw new InvalidRevisionException("Failed to update Snippet", e);
    }
    return entityFactory.createSnippetEntity(snapshot.getComponent());
}