@Override public List<Subscriber> getSubscribers(int n) { if (context == QUERY_SUBMISSION) { return getPubscribers(n).stream().map(p -> (Subscriber) p).collect(Collectors.toList()); } return IntStream.range(0, n).mapToObj(i -> new DRPCQuerySubscriber(config, maxUncommittedMessages)) .collect(Collectors.toList()); }
@Override public List<Subscriber> getSubscribers(int n) { if (context == QUERY_SUBMISSION) { return getPubscribers(n).stream().map(p -> (Subscriber) p).collect(Collectors.toList()); } return IntStream.range(0, n).mapToObj(i -> new DRPCQuerySubscriber(config, maxUncommittedMessages)) .collect(Collectors.toList()); }
@Override public Subscriber getSubscriber() { return context == QUERY_SUBMISSION ? getPubscriber() : new DRPCQuerySubscriber(config, maxUncommittedMessages); }
@Override public Subscriber getSubscriber() { return context == QUERY_SUBMISSION ? getPubscriber() : new DRPCQuerySubscriber(config, maxUncommittedMessages); }
@BeforeMethod public void setup() { DRPCConfig config = new DRPCConfig("src/test/resources/test_drpc_config.yaml"); // 1 task for the component named "foo" with task index 0 CustomTopologyContext context = new CustomTopologyContext(Collections.singletonList(1), "foo", 0); config.set(DRPCConfig.STORM_CONTEXT, context); Map stormConfig = new Config("src/test/resources/test_storm_config.yaml").getAll(Optional.empty()); config.set(DRPCConfig.STORM_CONFIG, stormConfig); subscriber = new DRPCQuerySubscriber(config, 5); DRPCOutputCollector collector = subscriber.getCollector(); // Override the DRPCSpout with our own that emits using our collector. injectedMockSpout = new MockDRPCSpout("foo", collector); subscriber.setSpout(injectedMockSpout); }