private List<Long> getOutputStreamIds(TopologySource topologySource) { List<Long> streamIds = new ArrayList<>(); if (topologySource != null) { QueryParam qp1 = new QueryParam(TopologySourceStreamMap.FIELD_SOURCE_ID, String.valueOf(topologySource.getId())); QueryParam qp2 = new QueryParam(TopologySourceStreamMap.FIELD_VERSION_ID, String.valueOf(topologySource.getVersionId())); for (TopologySourceStreamMap mapping : listTopologySourceStreamMapping(ImmutableList.of(qp1, qp2))) { streamIds.add(mapping.getStreamId()); } } return streamIds; }
private void removeAllTopologyTestSources(TopologySource topologySource) { QueryParam sourceIdQuery = new QueryParam("sourceId", String.valueOf(topologySource.getId())); Collection<TopologyTestRunCaseSource> sources = listTopologyTestRunCaseSource(Collections.singletonList(sourceIdQuery)); if (sources != null) { sources.forEach(s -> removeTopologyTestRunCaseSource(s.getId())); } }
private void addSourceStreamMapping(TopologySource topologySource, List<Long> streamIds) { for (Long outputStreamId : streamIds) { dao.<TopologySourceStreamMap>add(new TopologySourceStreamMap(topologySource.getId(), topologySource.getVersionId(), outputStreamId)); } }
private List<TopologyStream> getOutputStreams(TopologySource topologySource) { List<TopologyStream> streams = new ArrayList<>(); if (topologySource != null) { QueryParam qp1 = new QueryParam(TopologySourceStreamMap.FIELD_SOURCE_ID, String.valueOf(topologySource.getId())); QueryParam qp2 = new QueryParam(TopologySourceStreamMap.FIELD_VERSION_ID, String.valueOf(topologySource.getVersionId())); for (TopologySourceStreamMap mapping : listTopologySourceStreamMapping(ImmutableList.of(qp1, qp2))) { TopologyStream topologyStream = getStreamInfo(topologySource.getTopologyId(), mapping.getStreamId(), topologySource.getVersionId()); if (topologyStream != null) { streams.add(topologyStream); } } } return streams; }
private void removeSourceStreamMapping(TopologySource topologySource, List<Long> streamIds) { if (topologySource != null) { for (Long outputStreamId : streamIds) { TopologySourceStreamMap mapping = new TopologySourceStreamMap( topologySource.getId(), topologySource.getVersionId(), outputStreamId); dao.<TopologySourceStreamMap>remove(mapping.getStorableKey()); } } }
Long oldComponentId = topologySource.getId(); topologySource.setId(null); topologySource.setTopologyId(newTopology.getId()); topologySource.setTopologyComponentBundleId(bundle.getId()); addTopologySource(newTopology.getId(), topologySource); oldToNewComponentIds.put(oldComponentId, topologySource.getId());
public TopologySource addTopologySource(Long topologyId, Long versionId, TopologySource topologySource) { if (topologySource.getId() == null) { topologySource.setId(getNextTopologyComponentId()); } topologySource.setVersionId(versionId); topologySource.setTopologyId(topologyId); validateTopologySource(topologySource); List<TopologyStream> topologyStreams = addTopologyOutputComponent(topologySource); addSourceStreamMapping(topologySource, topologySource.getOutputStreamIds()); topologySource.setOutputStreams(topologyStreams); topologySource.setVersionTimestamp(updateVersionTimestamp(versionId).getTimestamp()); return topologySource; }
removeTopologySource(topologyId, source.getId(), versionId, false);