Exemplo n.º 1
0
 private Map<Agent, String> runAgents(
     List<Agent> endpoints, ParameterizedRunnable<BoxAgentHelper> runnable)
     throws ServiceException {
   try {
     Map<Agent, ByteArrayOutputStream> streams = createAgentStreams();
     BoxAgentHelper helper = createAgentHelper(endpoints, streams);
     runnable.run(helper);
     helper.join();
     return getAgentOutput(streams);
   } catch (Exception ex) {
     throw new ServiceException("Failed calling agents", ex);
   }
 }
Exemplo n.º 2
0
  private BoxAgentHelper createAgentHelper(
      final List<Agent> agents, final Map<Agent, ByteArrayOutputStream> streams) {
    BoxAgentHelper agentHelper = new BoxAgentHelper();
    agentHelper.setStreamFactory(
        new PrintStreamFactory() {
          public PrintStream create(Endpoint endpoint) {
            ByteArrayOutputStream bstream = new ByteArrayOutputStream();
            streams.put(getAgent(endpoint), bstream);
            return new PrintStream(bstream);
          }
        });
    CompositeEndpointConnectionFactory<BoxInterface> connectionFactory =
        new CompositeEndpointConnectionFactory<BoxInterface>();
    connectionFactory.add(new BoxDirectConnectionFactory());
    connectionFactory.add(new BoxTunnelConnectionFactory());
    agentHelper.setEndpointConnectionFactory(connectionFactory);

    agentHelper.setEndpoints(getEndpoints(agents));

    return agentHelper;
  }