@Override public ResolvedReference resolve(final CommandOption option, final Integer position) { final VersionedFlow versionedFlow = backRefs.get(position); if (versionedFlow != null) { if (option != null && option == CommandOption.BUCKET_ID) { return new ResolvedReference(option, position, versionedFlow.getBucketName(), versionedFlow.getBucketIdentifier()); } else { return new ResolvedReference(option, position, versionedFlow.getName(), versionedFlow.getIdentifier()); } } else { return null; } }
@Override public OkResult doExecute(final NiFiRegistryClient client, final Properties properties) throws IOException, NiFiRegistryException, ParseException { final String flowId = getRequiredArg(properties, CommandOption.FLOW_ID); final boolean forceDelete = properties.containsKey(CommandOption.FORCE.getLongName()); final FlowClient flowClient = client.getFlowClient(); final VersionedFlow versionedFlow = flowClient.get(flowId); final FlowSnapshotClient flowSnapshotClient = client.getFlowSnapshotClient(); final List<VersionedFlowSnapshotMetadata> snapshotMetadata = flowSnapshotClient.getSnapshotMetadata(flowId); if (snapshotMetadata != null && snapshotMetadata.size() > 0 && !forceDelete) { throw new NiFiRegistryException("Flow has versions, use --" + CommandOption.FORCE.getLongName() + " to delete"); } else { flowClient.delete(versionedFlow.getBucketIdentifier(), versionedFlow.getIdentifier()); return new OkResult(getContext().isInteractive()); } } }
metadata.setBucketIdentifier(versionedFlow.getBucketIdentifier()); metadata.setFlowIdentifier(flowId); metadata.setVersion(version);
@Override public VersionedFlowSnapshot registerVersionedFlowSnapshot(final VersionedFlow flow, final VersionedProcessGroup snapshot, final String comments, final int expectedVersion, final NiFiUser user) throws IOException, NiFiRegistryException { final FlowSnapshotClient snapshotClient = getFlowSnapshotClient(user); final VersionedFlowSnapshot versionedFlowSnapshot = new VersionedFlowSnapshot(); versionedFlowSnapshot.setFlowContents(snapshot); final VersionedFlowSnapshotMetadata metadata = new VersionedFlowSnapshotMetadata(); metadata.setBucketIdentifier(flow.getBucketIdentifier()); metadata.setFlowIdentifier(flow.getIdentifier()); metadata.setAuthor(getIdentity(user)); metadata.setTimestamp(System.currentTimeMillis()); metadata.setVersion(expectedVersion); metadata.setComments(comments); versionedFlowSnapshot.setSnapshotMetadata(metadata); return snapshotClient.create(versionedFlowSnapshot); }
private VersionedFlowEntity createVersionedFlowEntity(final String registryId, final VersionedFlow versionedFlow) { if (versionedFlow == null) { return null; } final VersionedFlowDTO dto = new VersionedFlowDTO(); dto.setRegistryId(registryId); dto.setBucketId(versionedFlow.getBucketIdentifier()); dto.setFlowId(versionedFlow.getIdentifier()); dto.setFlowName(versionedFlow.getName()); dto.setDescription(versionedFlow.getDescription()); final VersionedFlowEntity entity = new VersionedFlowEntity(); entity.setVersionedFlow(dto); return entity; }
destMetadata.setBucketIdentifier(destFlow.getBucketIdentifier()); destMetadata.setFlowIdentifier(destFlowId); destMetadata.setVersion(srcVersion);
destMetadata.setBucketIdentifier(destFlow.getBucketIdentifier()); destMetadata.setFlowIdentifier(destFlowId); destMetadata.setVersion(destFlowVersion);
if (versionedFlow.getBucketIdentifier() != null && !bucketIdentifier.equals(versionedFlow.getBucketIdentifier())) { throw new IllegalArgumentException("Bucket identifiers must match"); if (versionedFlow.getBucketIdentifier() == null) { versionedFlow.setBucketIdentifier(bucketIdentifier);
if (StringUtils.isBlank(versionedFlow.getBucketIdentifier())) { throw new IllegalArgumentException("Versioned flow bucket identifier cannot be null or blank"); try { final BucketEntity existingBucket = metadataService.getBucketById(versionedFlow.getBucketIdentifier()); if (existingBucket == null) { LOGGER.warn("The specified bucket id [{}] does not exist.", versionedFlow.getBucketIdentifier()); throw new ResourceNotFoundException("The specified bucket ID does not exist in this registry.");
public static Event flowUpdated(final VersionedFlow versionedFlow) { return new StandardEvent.Builder() .eventType(EventType.UPDATE_FLOW) .addField(EventFieldName.BUCKET_ID, versionedFlow.getBucketIdentifier()) .addField(EventFieldName.FLOW_ID, versionedFlow.getIdentifier()) .addField(EventFieldName.USER, NiFiUserUtils.getNiFiUserIdentity()) .build(); }
public static Event flowDeleted(final VersionedFlow versionedFlow) { return new StandardEvent.Builder() .eventType(EventType.DELETE_FLOW) .addField(EventFieldName.BUCKET_ID, versionedFlow.getBucketIdentifier()) .addField(EventFieldName.FLOW_ID, versionedFlow.getIdentifier()) .addField(EventFieldName.USER, NiFiUserUtils.getNiFiUserIdentity()) .build(); }
public static Event flowCreated(final VersionedFlow versionedFlow) { return new StandardEvent.Builder() .eventType(EventType.CREATE_FLOW) .addField(EventFieldName.BUCKET_ID, versionedFlow.getBucketIdentifier()) .addField(EventFieldName.FLOW_ID, versionedFlow.getIdentifier()) .addField(EventFieldName.USER, NiFiUserUtils.getNiFiUserIdentity()) .build(); }