use of com.hortonworks.streamline.streams.catalog.TopologyWindow in project streamline by hortonworks.
the class TopologyExportVisitor method createRuleHandlerMap.
private Map<String, RulesHandler> createRuleHandlerMap() {
ImmutableMap.Builder<String, RulesHandler> rulesHandlerBuilder = ImmutableMap.builder();
RulesHandler rulesHandler = new RulesHandler() {
@Override
public void handle(Rule rule) throws Exception {
TopologyRule topologyRule = streamCatalogService.getRule(topologyId, rule.getId());
topologyData.addRule(topologyRule);
}
};
rulesHandlerBuilder.put(ComponentTypes.RULE, rulesHandler);
rulesHandlerBuilder.put(ComponentTypes.PROJECTION, rulesHandler);
rulesHandlerBuilder.put(ComponentTypes.BRANCH, new RulesHandler() {
@Override
public void handle(Rule rule) throws Exception {
TopologyBranchRule topologyBranchRule = streamCatalogService.getBranchRule(topologyId, rule.getId());
topologyData.addBranch(topologyBranchRule);
}
});
rulesHandlerBuilder.put(ComponentTypes.WINDOW, new RulesHandler() {
@Override
public void handle(Rule rule) throws Exception {
TopologyWindow topologyWindow = streamCatalogService.getWindow(topologyId, rule.getId());
topologyData.addWindow(topologyWindow);
}
});
return rulesHandlerBuilder.build();
}
use of com.hortonworks.streamline.streams.catalog.TopologyWindow in project streamline by hortonworks.
the class StreamCatalogService method getWindow.
public TopologyWindow getWindow(Long topologyId, Long windowId, Long versionId) {
TopologyWindow topologyTopologyWindow = new TopologyWindow();
topologyTopologyWindow.setId(windowId);
topologyTopologyWindow.setVersionId(versionId);
TopologyWindow windowInfo = dao.get(new StorableKey(TOPOLOGY_WINDOWINFO_NAMESPACE, topologyTopologyWindow.getPrimaryKey()));
if (windowInfo == null || !windowInfo.getTopologyId().equals(topologyId)) {
return null;
}
windowInfo.setVersionTimestamp(getVersionTimestamp(versionId));
return windowInfo;
}
use of com.hortonworks.streamline.streams.catalog.TopologyWindow in project streamline by hortonworks.
the class StreamCatalogService method removeWindow.
public TopologyWindow removeWindow(Long topologyId, Long windowId, Long versionId) throws Exception {
TopologyWindow topologyWindow = getWindow(topologyId, windowId, versionId);
if (topologyWindow != null) {
topologyWindow = dao.remove(new StorableKey(TOPOLOGY_WINDOWINFO_NAMESPACE, topologyWindow.getPrimaryKey()));
topologyWindow.setVersionTimestamp(updateVersionTimestamp(versionId).getTimestamp());
}
return topologyWindow;
}
use of com.hortonworks.streamline.streams.catalog.TopologyWindow in project streamline by hortonworks.
the class StreamCatalogService method doImportTopology.
private Topology doImportTopology(Topology newTopology, TopologyData topologyData) throws Exception {
List<TopologySource> topologySources = topologyData.getSources();
Map<Long, Long> oldToNewComponentIds = new HashMap<>();
Map<Long, Long> oldToNewRuleIds = new HashMap<>();
Map<Long, Long> oldToNewWindowIds = new HashMap<>();
Map<Long, Long> oldToNewBranchRuleIds = new HashMap<>();
Map<Long, Long> oldToNewStreamIds = new HashMap<>();
// import source streams
for (TopologySource topologySource : topologySources) {
topologySource.setOutputStreamIds(importOutputStreams(newTopology.getId(), oldToNewStreamIds, topologySource.getOutputStreams()));
topologySource.setOutputStreams(null);
}
// import processor streams
for (TopologyProcessor topologyProcessor : topologyData.getProcessors()) {
topologyProcessor.setOutputStreamIds(importOutputStreams(newTopology.getId(), oldToNewStreamIds, topologyProcessor.getOutputStreams()));
topologyProcessor.setOutputStreams(null);
}
// import rules
for (TopologyRule rule : topologyData.getRules()) {
Long currentId = rule.getId();
rule.setId(null);
TopologyRule addedRule = addRule(newTopology.getId(), rule);
oldToNewRuleIds.put(currentId, addedRule.getId());
}
// import windowed rules
for (TopologyWindow window : topologyData.getWindows()) {
Long currentId = window.getId();
window.setId(null);
TopologyWindow addedWindow = addWindow(newTopology.getId(), window);
oldToNewWindowIds.put(currentId, addedWindow.getId());
}
// import branch rules
for (TopologyBranchRule branchRule : topologyData.getBranchRules()) {
Long currentId = branchRule.getId();
branchRule.setId(null);
TopologyBranchRule addedBranchRule = addBranchRule(newTopology.getId(), branchRule);
oldToNewBranchRuleIds.put(currentId, addedBranchRule.getId());
}
// import sources
for (TopologySource topologySource : topologySources) {
Long oldComponentId = topologySource.getId();
topologySource.setId(null);
topologySource.setTopologyId(newTopology.getId());
TopologyComponentBundle bundle = getCurrentTopologyComponentBundle(TopologyComponentBundle.TopologyComponentType.SOURCE, topologyData.getBundleIdToType().get(topologySource.getTopologyComponentBundleId().toString()));
topologySource.setTopologyComponentBundleId(bundle.getId());
addTopologySource(newTopology.getId(), topologySource);
oldToNewComponentIds.put(oldComponentId, topologySource.getId());
}
// import processors
for (TopologyProcessor topologyProcessor : topologyData.getProcessors()) {
Long oldComponentId = topologyProcessor.getId();
topologyProcessor.setId(null);
topologyProcessor.setTopologyId(newTopology.getId());
TopologyComponentBundle bundle;
String subType = topologyData.getBundleIdToType().get(topologyProcessor.getTopologyComponentBundleId().toString());
if (TopologyLayoutConstants.JSON_KEY_CUSTOM_PROCESSOR_SUB_TYPE.equals(subType)) {
QueryParam queryParam = new QueryParam(CustomProcessorInfo.NAME, topologyProcessor.getConfig().get(CustomProcessorInfo.NAME));
Collection<TopologyComponentBundle> result = listCustomProcessorBundlesWithFilter(Collections.singletonList(queryParam));
if (result.size() != 1) {
throw new IllegalStateException("Not able to find topology component bundle for custom processor :" + topologyProcessor.getConfig().get(CustomProcessorInfo.NAME));
}
bundle = result.iterator().next();
} else {
bundle = getCurrentTopologyComponentBundle(TopologyComponentBundle.TopologyComponentType.PROCESSOR, subType);
}
topologyProcessor.setTopologyComponentBundleId(bundle.getId());
Optional<Object> ruleListObj = topologyProcessor.getConfig().getAnyOptional(RulesProcessor.CONFIG_KEY_RULES);
ruleListObj.ifPresent(ruleList -> {
List<Long> ruleIds = new ObjectMapper().convertValue(ruleList, new TypeReference<List<Long>>() {
});
List<Long> updatedRuleIds = new ArrayList<>();
if (ComponentTypes.RULE.equals(bundle.getSubType()) || ComponentTypes.PROJECTION.equals(bundle.getSubType())) {
ruleIds.forEach(ruleId -> updatedRuleIds.add(oldToNewRuleIds.get(ruleId)));
} else if (bundle.getSubType().equals(ComponentTypes.BRANCH)) {
ruleIds.forEach(ruleId -> updatedRuleIds.add(oldToNewBranchRuleIds.get(ruleId)));
} else if (bundle.getSubType().equals(ComponentTypes.WINDOW)) {
ruleIds.forEach(ruleId -> updatedRuleIds.add(oldToNewWindowIds.get(ruleId)));
}
topologyProcessor.getConfig().setAny(RulesProcessor.CONFIG_KEY_RULES, updatedRuleIds);
});
addTopologyProcessor(newTopology.getId(), topologyProcessor);
oldToNewComponentIds.put(oldComponentId, topologyProcessor.getId());
}
// import sinks
for (TopologySink topologySink : topologyData.getSinks()) {
topologySink.setTopologyId(newTopology.getId());
Long currentId = topologySink.getId();
topologySink.setId(null);
TopologyComponentBundle bundle = getCurrentTopologyComponentBundle(TopologyComponentBundle.TopologyComponentType.SINK, topologyData.getBundleIdToType().get(topologySink.getTopologyComponentBundleId().toString()));
topologySink.setTopologyComponentBundleId(bundle.getId());
if (bundle.getSubType().equals(NOTIFICATION)) {
updateNotifierJarFileName(topologySink);
}
addTopologySink(newTopology.getId(), topologySink);
oldToNewComponentIds.put(currentId, topologySink.getId());
}
// import edges
for (TopologyEdge topologyEdge : topologyData.getEdges()) {
List<StreamGrouping> streamGroupings = topologyEdge.getStreamGroupings();
for (StreamGrouping streamGrouping : streamGroupings) {
Long newStreamId = oldToNewStreamIds.get(streamGrouping.getStreamId());
streamGrouping.setStreamId(newStreamId);
}
topologyEdge.setId(null);
topologyEdge.setTopologyId(newTopology.getId());
topologyEdge.setFromId(oldToNewComponentIds.get(topologyEdge.getFromId()));
topologyEdge.setToId(oldToNewComponentIds.get(topologyEdge.getToId()));
addTopologyEdge(newTopology.getId(), topologyEdge);
}
// import topology editor metadata
TopologyEditorMetadata topologyEditorMetadata = topologyData.getTopologyEditorMetadata();
topologyEditorMetadata.setTopologyId(newTopology.getId());
if (topologyEditorMetadata.getData() != null) {
TopologyUIData topologyUIData = new ObjectMapper().readValue(topologyEditorMetadata.getData(), TopologyUIData.class);
topologyUIData.getSources().forEach(c -> c.setId(oldToNewComponentIds.get(c.getId())));
topologyUIData.getProcessors().forEach(c -> c.setId(oldToNewComponentIds.get(c.getId())));
topologyUIData.getSinks().forEach(c -> c.setId(oldToNewComponentIds.get(c.getId())));
topologyEditorMetadata.setData(new ObjectMapper().writeValueAsString(topologyUIData));
} else {
topologyEditorMetadata.setData(StringUtils.EMPTY);
}
addTopologyEditorMetadata(newTopology.getId(), topologyData.getTopologyEditorMetadata());
return newTopology;
}
use of com.hortonworks.streamline.streams.catalog.TopologyWindow in project streamline by hortonworks.
the class WindowCatalogResource method getTopologyWindowById.
@GET
@Path("/topologies/{topologyId}/windows/{id}")
@Timed
@Produces(MediaType.APPLICATION_JSON)
public Response getTopologyWindowById(@PathParam("topologyId") Long topologyId, @PathParam("id") Long windowId, @Context SecurityContext securityContext) throws Exception {
SecurityUtil.checkRoleOrPermissions(authorizer, securityContext, Roles.ROLE_TOPOLOGY_USER, Topology.NAMESPACE, topologyId, READ);
TopologyWindow topologyWindow = catalogService.getWindow(topologyId, windowId);
if (topologyWindow != null) {
return WSUtils.respondEntity(topologyWindow, OK);
}
throw EntityNotFoundException.byId(buildMessageForCompositeId(topologyId, windowId));
}
Aggregations