Exemplo n.º 1
0
 private void createKey(ClusterConfiguration clusterConfig, User user)
     throws JSchException, FileNotFoundException, IOException {
   String dir = "sshKey";
   Utils.checkDirAvailable(dir);
   JSch jsch = new JSch();
   String name = "cloudgene_" + System.currentTimeMillis() + "";
   KeyPair key;
   key = KeyPair.genKeyPair(jsch, com.jcraft.jsch.KeyPair.RSA);
   String publicKeyLoc = dir + File.separator + name + ".pub";
   String privateKeyLoc = dir + File.separatorChar + name + ".key";
   key.writePrivateKey(privateKeyLoc);
   key.writePublicKey(publicKeyLoc, "");
   // set for launch cluster
   clusterConfig.setSshPublic(publicKeyLoc);
   clusterConfig.setSshPrivate(privateKeyLoc);
   user.setSshKey(privateKeyLoc);
   user.setSshPub(publicKeyLoc);
   createZip(publicKeyLoc, privateKeyLoc);
   key.dispose();
 }
Exemplo n.º 2
0
  @Post
  public Representation acceptRepresentation(Representation entity) {
    Representation representation = null;
    try {
      JsonRepresentation represent = new JsonRepresentation(entity);

      UserSessions sessions = UserSessions.getInstance();
      User user = sessions.getUserByRequest(getRequest());

      JSONObject obj = represent.getJsonObject();
      if (user != null) {
        ClusterConfiguration clusterConfig = new ClusterConfiguration();
        String usr = obj.get("loginUsername").toString();
        String pwd = obj.get("loginPassword").toString();
        String program = obj.get("program").toString();
        String ssh = obj.get("key").toString();
        String name = obj.get("name").toString();
        String provider = obj.get("cluster").toString();
        String amount = obj.get("amount").toString();
        String bucket = obj.get("bucketName").toString();
        String type = obj.get("type").toString();
        String dirLog = "logs";

        /** check credentials */
        ConnectionUtil.getInputStream(usr, pwd);
        UserDao dao = new UserDao();
        if (!obj.has("saveCre")) {
          user.setCloudKey("");
          user.setCloudSecure("");
          dao.updateCredential(user);
        } else {
          user.setCloudKey(usr);
          user.setCloudSecure(pwd);
          dao.updateCredential(user);
        }
        user.setCloudKey(usr);
        user.setCloudSecure(pwd);

        /** save ssh data */
        if (ssh.equals("1")) {
          createKey(clusterConfig, user);
        } else if (ssh.equals("3")) {
          clusterConfig.setSshPrivate(user.getSshKey());
          clusterConfig.setSshPublic(user.getSshPub());
        }
        if (obj.has("saveSsh")) {
          dao.updateSSH(user);
        }
        if (obj.has("s3Export")) {
          clusterConfig.setS3Bucket(bucket);
        }

        /** CONFIGURE CLUSTER */
        Program prog = Programs.getProgramByName(program);
        clusterConfig.setProgram(prog.getCluster());
        Utils.checkDirAvailable(dirLog);
        clusterConfig.setLog(dirLog + File.separatorChar + System.currentTimeMillis() + ".txt");
        clusterConfig.setCloudID(String.valueOf(System.currentTimeMillis()));
        clusterConfig.setName(name);
        clusterConfig.getProgram().setProvider(provider);
        clusterConfig.setCloudUsername(usr);
        clusterConfig.setCloudPassword(pwd);
        clusterConfig.setAmount(Integer.valueOf(amount));
        clusterConfig.setCloudgeneUser(user);
        clusterConfig.setSSHAvailable(true);
        clusterConfig.setInstanceType(type);
        clusterConfig.setStartTime(System.currentTimeMillis());
        clusterConfig.setActionType(ClusterConfiguration.CREATE_CLUSTER);
        clusterConfig.setState(ClusterConfiguration.QUEUE);

        /** add to queue */
        ClusterQueue.getInstance().submit(clusterConfig);

        /** add to threadpool */
        ClusterTask task = new ClusterTask(clusterConfig);
        ClusterThreadPoolCreate.getInstance().runTask(task);
        getResponse().setStatus(Status.SUCCESS_OK);

      } else {
        representation = new StringRepresentation("No user");
        getResponse().setStatus(Status.CLIENT_ERROR_BAD_REQUEST);
        getResponse().setEntity(representation);
        return representation;
      }
    } catch (S3ServiceException e) {
      representation = new StringRepresentation("Please check your security credentials");
      getResponse().setEntity(representation);
      getResponse().setStatus(Status.CLIENT_ERROR_BAD_REQUEST);
      e.printStackTrace();
      return representation;
    } catch (Exception e) {
      // TODO Auto-generated catch block
      representation = new StringRepresentation("Error occured");
      getResponse().setStatus(Status.CLIENT_ERROR_BAD_REQUEST);
      getResponse().setEntity(representation);
      e.printStackTrace();
      return representation;
    }
    return representation;
  }