/**
  * Creates a new group in HDFS with the name <code>projectName</code> if it does not exist, then
  * creates the owner in HDFS with the name <code>projectName</code>__<code>username</code> , also
  * if it does not exist, and gets added to the group <code>projectName</code>.
  *
  * <p>
  *
  * @param project
  * @throws java.io.IOException
  */
 public void addProjectFolderOwner(Project project) throws IOException {
   String owner = getHdfsUserName(project, project.getOwner());
   String projectPath = File.separator + settings.DIR_ROOT + File.separator + project.getName();
   Path location = new Path(projectPath);
   // FsPermission(FsAction u, FsAction g, FsAction o) 775
   // Gives owner and group all access and read, execute for others
   // This means group is for data_owners and others for data_scientist
   // This means every body can see the content of a project.
   FsPermission fsPermission =
       new FsPermission(FsAction.ALL, FsAction.ALL, FsAction.READ_EXECUTE); // 775
   fsOps.setOwner(location, owner, project.getName());
   fsOps.setPermission(location, fsPermission);
 }
 /**
  * Removes the user project__username from the group projectName. This means the user is no longer
  * a data_owner in this project. (will be a data_scientist with r, x privileges on datasets inside
  * the project)
  *
  * <p>
  *
  * @param user
  * @param project
  */
 public void modifyProjectMembership(Users user, Project project) {
   if (user == null || project == null || project.getProjectTeamCollection() == null) {
     throw new IllegalArgumentException("One or more arguments are null.");
   }
   String userName = getHdfsUserName(project, user);
   byte[] userId = UsersGroups.getUserID(userName);
   byte[] groupId = UsersGroups.getGroupID(project.getName());
   HdfsGroups hdfsGroup = hdfsGroupsFacade.findHdfsGroup(groupId);
   HdfsUsers hdfsUser = hdfsUsersFacade.findHdfsUser(userId);
   if (hdfsUser == null || hdfsGroup == null) {
     throw new IllegalArgumentException("Hdfs user not found or not in project group.");
   }
   hdfsUser.getHdfsGroupsCollection().remove(hdfsGroup);
   hdfsUsersFacade.merge(hdfsUser);
 }
Exemple #3
0
 public DataSetDTO(Dataset ds, Project project, List<String> sharedWith) {
   this.inodeId = ds.getInode().getId();
   this.name = ds.getInode().getInodePK().getName();
   this.description = ds.getDescription();
   this.projectName = project.getName();
   this.sharedWith = sharedWith;
   this.projectTeam = new ArrayList<>();
   // this have to be done because project team contains too much info.
   for (ProjectTeam member : project.getProjectTeamCollection()) {
     projectTeam.add(
         new UserCardDTO(
             member.getUser().getFname(),
             member.getUser().getLname(),
             member.getUser().getEmail()));
   }
 }
 /**
  * Adds a user to project group if the member have a Data owner role in the project.
  *
  * <p>throws IllegalArgumentException if the project group is not found.
  *
  * <p>
  *
  * @param project
  * @param member
  */
 public void addUserToProjectGroup(Project project, ProjectTeam member) {
   byte[] groupId = UsersGroups.getGroupID(project.getName());
   HdfsGroups hdfsGroup = hdfsGroupsFacade.findHdfsGroup(groupId);
   if (hdfsGroup == null) {
     throw new IllegalArgumentException("No group found for project in HDFS.");
   }
   String hdfsUsername;
   HdfsUsers memberHdfsUser;
   byte[] memberUserId;
   Users newMember = userFacade.findByEmail(member.getProjectTeamPK().getTeamMember());
   hdfsUsername = getHdfsUserName(project, newMember);
   memberUserId = UsersGroups.getUserID(hdfsUsername);
   memberHdfsUser = hdfsUsersFacade.findHdfsUser(memberUserId);
   if (memberHdfsUser == null) {
     memberHdfsUser = new HdfsUsers(memberUserId, hdfsUsername);
     hdfsUsersFacade.persist(memberHdfsUser);
   }
   if (memberHdfsUser.getHdfsGroupsCollection() == null) {
     memberHdfsUser.setHdfsGroupsCollection(new ArrayList<HdfsGroups>());
   }
   // add only data_owners to project group
   if (member.getTeamRole().equals(AllowedRoles.DATA_OWNER)) {
     if (!memberHdfsUser.getHdfsGroupsCollection().contains(hdfsGroup)) {
       memberHdfsUser.getHdfsGroupsCollection().add(hdfsGroup);
     }
   }
 }
 /**
  * Deletes the project group from HDFS
  *
  * <p>
  *
  * @param project
  */
 public void deleteProjectGroup(Project project) {
   if (project == null) {
     throw new IllegalArgumentException("One or more arguments are null.");
   }
   byte[] groupId = UsersGroups.getGroupID(project.getName());
   HdfsGroups hdfsGroup = hdfsGroupsFacade.findHdfsGroup(groupId);
   if (hdfsGroup != null) {
     hdfsGroupsFacade.remove(hdfsGroup);
   }
 }
 /**
  * Adds a new member to the project. This will create a new user in HDFS with the name <code>
  * projectName</code>__<code>username</code> and adds it to the group <code>projectName</code>.
  * throws IllegalArgumentException if the project group is not found.
  *
  * <p>
  *
  * @param project
  * @param member
  * @throws java.io.IOException
  */
 public void addNewProjectMember(Project project, ProjectTeam member) throws IOException {
   byte[] groupId = UsersGroups.getGroupID(project.getName());
   HdfsGroups hdfsGroup = hdfsGroupsFacade.findHdfsGroup(groupId);
   if (hdfsGroup == null) {
     throw new IllegalArgumentException("No group found for project in HDFS.");
   }
   String hdfsUsername;
   HdfsUsers memberHdfsUser;
   byte[] memberUserId;
   Users newMember = userFacade.findByEmail(member.getProjectTeamPK().getTeamMember());
   hdfsUsername = getHdfsUserName(project, newMember);
   memberUserId = UsersGroups.getUserID(hdfsUsername);
   memberHdfsUser = hdfsUsersFacade.findHdfsUser(memberUserId);
   if (memberHdfsUser == null) {
     memberHdfsUser = new HdfsUsers(memberUserId, hdfsUsername);
     hdfsUsersFacade.persist(memberHdfsUser);
   }
   if (memberHdfsUser.getHdfsGroupsCollection() == null) {
     memberHdfsUser.setHdfsGroupsCollection(new ArrayList<HdfsGroups>());
   }
   // add only data_owners to project group
   if (member.getTeamRole().equals(AllowedRoles.DATA_OWNER)) {
     if (!memberHdfsUser.getHdfsGroupsCollection().contains(hdfsGroup)) {
       memberHdfsUser.getHdfsGroupsCollection().add(hdfsGroup);
     }
   }
   byte[] dsGroupId;
   String dsGroups;
   HdfsGroups hdfsDsGroup;
   // add the member to all dataset groups in the project.
   List<Dataset> dsInProject = datasetFacade.findByProject(project);
   for (Dataset ds : dsInProject) {
     dsGroups = getHdfsGroupName(ds);
     dsGroupId = UsersGroups.getGroupID(dsGroups);
     hdfsDsGroup = hdfsGroupsFacade.findHdfsGroup(dsGroupId);
     if (hdfsDsGroup != null) {
       if (!memberHdfsUser.getHdfsGroupsCollection().contains(hdfsDsGroup)) {
         memberHdfsUser.getHdfsGroupsCollection().add(hdfsDsGroup);
       }
     }
   }
   hdfsUsersFacade.merge(memberHdfsUser);
 }
 /**
  * Deletes the project group and all associated groups from HDFS
  *
  * <p>
  *
  * @param project
  * @param dsInProject
  */
 public void deleteProjectGroupsRecursive(Project project, List<Dataset> dsInProject) {
   if (project == null) {
     throw new IllegalArgumentException("One or more arguments are null.");
   }
   byte[] groupId = UsersGroups.getGroupID(project.getName());
   HdfsGroups hdfsGroup = hdfsGroupsFacade.findHdfsGroup(groupId);
   if (hdfsGroup != null) {
     hdfsGroupsFacade.remove(hdfsGroup);
   }
   byte[] dsGroupId;
   String dsGroups;
   HdfsGroups hdfsDsGroup;
   for (Dataset ds : dsInProject) {
     dsGroups = getHdfsGroupName(project, ds);
     dsGroupId = UsersGroups.getGroupID(dsGroups);
     hdfsDsGroup = hdfsGroupsFacade.findHdfsGroup(dsGroupId);
     if (hdfsDsGroup != null) {
       hdfsGroupsFacade.remove(hdfsDsGroup);
     }
   }
 }
 /**
  * If the dataset is shared with this project we will get a group name that does not exist.
  *
  * <p>
  *
  * @param project
  * @param ds
  * @return
  */
 public String getHdfsGroupName(Project project, Dataset ds) {
   if (project == null || ds == null) {
     return null;
   }
   return project.getName() + USER_NAME_DELIMITER + ds.getInode().getInodePK().getName();
 }
 /**
  * Returns the hdfs username for the user in this project
  *
  * <p>
  *
  * @param project
  * @param user
  * @return
  */
 public String getHdfsUserName(Project project, Users user) {
   if (project == null || user == null) {
     return null;
   }
   return project.getName() + USER_NAME_DELIMITER + user.getUsername();
 }
  /**
   * Create a new group in HDFS with the name project.name__datasetName if it does not exist, then
   * adds all members of the project to this group. This is done when a new dataset is created in a
   * project. If stickyBit is set true: all members of the project will be given r, w, x privileges.
   * If stickyBit is set false: user will get all privileges, and all other members will have r and
   * x privileges.
   *
   * <p>
   *
   * @param owner
   * @param project
   * @param dataset
   * @param stickyBit
   * @throws java.io.IOException
   */
  public void addDatasetUsersGroups(
      Users owner, Project project, Dataset dataset, boolean stickyBit) throws IOException {
    if (owner == null
        || project == null
        || project.getProjectTeamCollection() == null
        || dataset == null) {
      throw new IllegalArgumentException("One or more arguments are null.");
    }
    String datasetGroup = getHdfsGroupName(project, dataset);
    String dsOwner = getHdfsUserName(project, owner);
    String dsPath =
        File.separator
            + settings.DIR_ROOT
            + File.separator
            + project.getName()
            + File.separator
            + dataset.getInode().getInodePK().getName();
    Path location = new Path(dsPath);
    // FsPermission(FsAction u, FsAction g, FsAction o, boolean sb)
    FsPermission fsPermission =
        new FsPermission(
            FsAction.ALL, FsAction.READ_EXECUTE, FsAction.NONE); // Permission hdfs dfs -chmod 750
    if (stickyBit) {
      fsPermission =
          new FsPermission(
              FsAction.ALL,
              FsAction.ALL,
              FsAction.NONE,
              stickyBit); // Permission hdfs dfs -chmod 1770
    }
    fsOps.setOwner(location, dsOwner, datasetGroup);
    fsOps.setPermission(location, fsPermission);

    String hdfsUsername;
    HdfsUsers hdfsUser;
    byte[] userId;
    byte[] groupId = UsersGroups.getGroupID(datasetGroup);
    HdfsGroups hdfsGroup = hdfsGroupsFacade.findHdfsGroup(groupId);
    if (hdfsGroup == null) {
      throw new IllegalArgumentException("Could not create dataset group in HDFS.");
    }
    if (hdfsGroup.getHdfsUsersCollection() == null) {
      hdfsGroup.setHdfsUsersCollection(new ArrayList<HdfsUsers>());
    }
    // add every member to the new ds group
    for (ProjectTeam member : project.getProjectTeamCollection()) {
      hdfsUsername = getHdfsUserName(project, member.getUser());
      userId = UsersGroups.getUserID(hdfsUsername);
      hdfsUser = hdfsUsersFacade.findHdfsUser(userId);
      // the owner does not need to be added to the group.
      if (hdfsUsername.equals(dsOwner)) {
        continue;
      }
      if (hdfsUser == null) {
        hdfsUser = new HdfsUsers(userId, hdfsUsername);
        hdfsUsersFacade.persist(hdfsUser);
      }
      if (!hdfsGroup.getHdfsUsersCollection().contains(hdfsUser)) {
        hdfsGroup.getHdfsUsersCollection().add(hdfsUser);
      }
    }
    hdfsGroupsFacade.merge(hdfsGroup);
  }