public ScheduleState generate(List<WorkItem> expandworkSets) { // topologyId -> SpoutSpec Map<String, SpoutSpec> topoSpoutSpecsMap = generateSpoutMonitorMetadata(); // grp-by meta spec(sort & grp) Map<String, RouterSpec> groupSpecsMap = generateGroupbyMonitorMetadata(); // alert bolt spec Map<String, AlertBoltSpec> alertSpecsMap = generateAlertMonitorMetadata(); Map<String, PublishSpec> publishSpecsMap = generatePublishMetadata(); String uniqueVersion = generateVersion(); ScheduleState status = new ScheduleState(uniqueVersion, topoSpoutSpecsMap, groupSpecsMap, alertSpecsMap, publishSpecsMap, context.getPolicyAssignments().values(), context.getMonitoredStreams().values(), context.getPolicies().values(), context.getStreamSchemas().values()); return status; }
private static ScheduleState createScheduleState() { ScheduleState ss = new ScheduleState(); ss.setVersion(V1); ss.getMonitoredStreams().add(createMonitoredStream()); ss.getAssignments().add(createAssignment()); return ss; }
@Ignore @Test public void addScheduleState() throws Exception { ConfigFactory.invalidateCaches(); System.setProperty("config.resource", "/test-application.conf"); Config config = ConfigFactory.load("test-application.conf").getConfig("coordinator"); MetadataServiceClientImpl client = new MetadataServiceClientImpl(config); ScheduleState ss = new ScheduleState(); ss.setVersion("spec_version_1463764252582"); client.addScheduleState(ss); client.close(); ss.setVersion("spec_version_1464764252582"); ZKConfig zkConfig = ZKConfigBuilder.getZKConfig(config); ConfigBusProducer producer = new ConfigBusProducer(zkConfig); Coordinator.postSchedule(client, ss, producer); } }