@Override
  public void execute(DelegateExecution execution) throws Exception {
    Pool pool = (Pool) execution.getVariable(CoreProcessVariables.POOL);
    checkNotNull(
        pool,
        "Please add the pool description as a process " + "variable with the name '%s'.",
        CoreProcessVariables.POOL);

    Machine machine = (Machine) execution.getVariable("machine");
    LOG.info(">> Connecting to machine {} to install packages", machine);

    SSHClient client = Ssh.newClient(machine, pool.getAdminAccess());
    try {
      String puppetScript =
          Mustache.toString(
              InstallPackages.class,
              "/com/axemblr/provisionr/core/puppet/packages.pp.mustache",
              ImmutableMap.of("packages", packagesAsListOfMaps(pool.getSoftware())));

      Ssh.createFile(client, puppetScript, 0600, "/tmp/packages.pp");
      Session session = client.startSession();
      try {
        session.allocateDefaultPTY();
        Session.Command command = session.exec("sudo puppet apply --verbose /tmp/packages.pp");

        Ssh.logCommandOutput(LOG, machine.getExternalId(), command);
        command.join();

        if (command.getExitStatus() != 0) {
          throw new RuntimeException(
              String.format(
                  "Failed to execute puppet. Exit code: %d. Exit message: %s",
                  command.getExitStatus(), command.getExitErrorMessage()));

        } else {
          LOG.info("<< Command completed successfully with exit code 0");
        }

      } finally {
        session.close();
      }
    } finally {
      client.close();
    }
  }
Esempio n. 2
0
File: SSH.java Progetto: ning/atlas
  public String exec(String command, int time, TimeUnit unit) throws IOException {
    Session s = ssh.startSession();
    try {
      Session.Command cmd = s.exec(command);

      StringBuilder all = new StringBuilder();
      BufferedReader out = new BufferedReader(new InputStreamReader(cmd.getInputStream()));
      String buf;
      while (null != (buf = out.readLine())) {
        logger.debug(buf);
        all.append(buf).append("\n");
      }

      cmd.join(time, unit);
      cmd.close();
      return all.toString();
    } finally {
      s.close();
    }
  }
Esempio n. 3
0
  public static void main(String... args) throws IOException {
    final SSHClient ssh = new SSHClient();
    ssh.loadKnownHosts();

    ssh.connect("localhost");
    try {
      ssh.authPublickey(System.getProperty("user.name"));
      final Session session = ssh.startSession();
      try {
        final Command cmd = session.exec("ping -c 1 google.com");
        System.out.println(IOUtils.readFully(cmd.getInputStream()).toString());
        cmd.join(5, TimeUnit.SECONDS);
        System.out.println("\n** exit status: " + cmd.getExitStatus());
      } finally {
        session.close();
      }
    } finally {
      ssh.disconnect();
    }
  }
Esempio n. 4
0
 @Override
 public ExecResponse create() throws Exception {
   try {
     session = acquire(execConnection());
     Command output = session.exec(checkNotNull(command, "command"));
     String outputString = IOUtils.readFully(output.getInputStream()).toString();
     output.join(timeoutMillis, TimeUnit.SECONDS);
     int errorStatus = output.getExitStatus();
     String errorString = IOUtils.readFully(output.getErrorStream()).toString();
     return new ExecResponse(outputString, errorString, errorStatus);
   } finally {
     clear();
   }
 }
Esempio n. 5
0
 @Override
 public void clear() throws TransportException, ConnectionException {
   if (session != null) session.close();
 }