/**
* 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);
}
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);
}
