All Downloads are FREE. Search and download functionalities are using the official Maven repository.

com.eurodyn.qlack.fuse.aaa.service.UserGroupService Maven / Gradle / Ivy

The newest version!
package com.eurodyn.qlack.fuse.aaa.service;

import com.eurodyn.qlack.fuse.aaa.criteria.UserGroupSearchCriteria;
import com.eurodyn.qlack.fuse.aaa.dto.UserGroupDTO;
import com.eurodyn.qlack.fuse.aaa.exception.InvalidGroupHierarchyException;
import com.eurodyn.qlack.fuse.aaa.mapper.UserGroupMapper;
import com.eurodyn.qlack.fuse.aaa.model.QUserGroup;
import com.eurodyn.qlack.fuse.aaa.model.User;
import com.eurodyn.qlack.fuse.aaa.model.UserGroup;
import com.eurodyn.qlack.fuse.aaa.repository.UserGroupRepository;
import com.eurodyn.qlack.fuse.aaa.repository.UserRepository;
import com.querydsl.core.BooleanBuilder;
import com.querydsl.core.types.Predicate;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.Pageable;
import org.springframework.data.domain.Sort;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.validation.annotation.Validated;

import java.util.ArrayList;
import java.util.Collection;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import java.util.stream.Collectors;

/**
 * @author European Dynamics SA
 */
@Service
@Validated
@Transactional
public class UserGroupService {

  // QueryDSL helpers.
  private static QUserGroup qUserGroup = QUserGroup.userGroup;
  // Repositories
  private final UserGroupRepository userGroupRepository;
  private final UserRepository userRepository;
  // Mappers
  private final UserGroupMapper userGroupMapper;

  public UserGroupService(UserGroupRepository userGroupRepository,
    UserRepository userRepository, UserGroupMapper userGroupMapper) {
    this.userGroupRepository = userGroupRepository;
    this.userRepository = userRepository;
    this.userGroupMapper = userGroupMapper;
  }

  public String createGroup(UserGroupDTO userGroupDTO) {
    UserGroup userGroup = userGroupMapper.mapToEntity(userGroupDTO);
    if (userGroupDTO.getParentId() != null) {
      userGroup
        .setParent(userGroupRepository.fetchById(userGroupDTO.getParentId()));
    }
    userGroupRepository.save(userGroup);

    return userGroup.getId();
  }

  public void updateGroup(UserGroupDTO userGroupDTO) {
    UserGroup userGroup = userGroupRepository.fetchById(userGroupDTO.getId());
    userGroupMapper.mapToExistingEntity(userGroupDTO, userGroup);
  }

  public void deleteGroup(String groupID) {
    userGroupRepository.delete(userGroupRepository.fetchById(groupID));
  }

  public void moveGroup(String groupID, String newParentId) {
    UserGroup userGroup = userGroupRepository.fetchById(groupID);
    UserGroup newParent = userGroupRepository.fetchById(newParentId);

    // Check the moving the userGroup under the new parent will not
    // create a cyclic dependency.
    UserGroup checkedUserGroup = newParent;
    while (checkedUserGroup != null) {
      if (checkedUserGroup.getId().equals(userGroup.getId())) {
        throw new InvalidGroupHierarchyException(
          "Cannot move userGroup with ID " + groupID
            + " under userGroup with ID " + newParentId
            + " since this will create a cyclic dependency between userGroups.");
      }
      checkedUserGroup = checkedUserGroup.getParent();
    }

    userGroup.setParent(newParent);
  }

  public UserGroupDTO getGroupByID(String groupID, boolean lazyRelatives) {
    return userGroupMapper
      .mapToDTO(userGroupRepository.fetchById(groupID), lazyRelatives);
  }

  public List getGroupsByID(Collection groupIds,
    boolean lazyRelatives) {
    Predicate predicate = qUserGroup.id.in(groupIds);

    return userGroupMapper.mapToDTO(
      userGroupRepository.findAll(predicate, Sort.by("name").ascending()),
      lazyRelatives);

  }

  public UserGroupDTO getGroupByName(String groupName, boolean lazyRelatives) {
    return userGroupMapper.mapToDTO(
      userGroupRepository.findByName(groupName), lazyRelatives);
  }

  public List getGroupByNames(List groupNames,
    boolean lazyRelatives) {
    Predicate predicate = qUserGroup.name.in(groupNames);

    return userGroupMapper
      .mapToDTO(userGroupRepository.findAll(predicate), lazyRelatives);
  }

  public UserGroupDTO getGroupByObjectId(String objectId,
    boolean lazyRelatives) {
    return userGroupMapper.mapToDTO(
      userGroupRepository.findByObjectId(objectId), lazyRelatives);
  }

  public List listGroups() {
    return userGroupMapper.mapToDTO(
      userGroupRepository.findAll(Sort.by("name").ascending()), false);
  }

  public List listGroupsAsTree() {
    Predicate predicate = qUserGroup.parent.isNull();

    return userGroupMapper.mapToDTO(userGroupRepository.findAll(
      predicate, Sort.by("name").ascending()), false);
  }

  public UserGroupDTO getGroupParent(String groupID) {
    UserGroup userGroup = userGroupRepository.fetchById(groupID);
    return userGroupMapper.mapToDTO(userGroup.getParent(), false);
  }

  public List getGroupChildren(String groupID) {
    Predicate predicate;
    if (groupID == null) {
      predicate = qUserGroup.parent.isNull();
    } else {
      predicate = qUserGroup.parent.id.eq(groupID);
    }
    return userGroupMapper.mapToDTO(userGroupRepository.findAll(
      predicate, Sort.by("name").ascending()), false);

  }

  /**
   * Returns the users belonging to a given userGroup and (optionally) its
   * hierarchy
   *
   * @param userGroup The userGroup the users of which to retrieve
   * @param includeAncestors true if users belonging to ancestors of this
   * userGroup (the userGroup's parent and its parent's parent, etc.) should
   * be retrieved
   * @param includeDescendants true if users belonging to descendants of this
   * userGroup (the userGroup's children and its children's children, etc.)
   * should be retrieved
   * @return The IDs of the users belonging to the specified userGroup
   * hierarchy.
   */
  private Set getGroupHierarchyUsersIds(UserGroup userGroup,
    boolean includeAncestors,
    boolean includeDescendants) {
    Set retVal = new HashSet<>(userGroup.getUsers().size());
    for (User user : userGroup.getUsers()) {
      retVal.add(user.getId());
    }

    // If children userGroup users should be included iterate over them
    // (and their children recursively) and add their users to
    // the return value. Same for the userGroup parents.
    if (includeDescendants) {
      for (UserGroup child : userGroup.getChildren()) {
        retVal.addAll(getGroupHierarchyUsersIds(child, false, true));
      }
    }
    if ((includeAncestors) && (userGroup.getParent() != null)) {
      retVal
        .addAll(getGroupHierarchyUsersIds(userGroup.getParent(), true, false));
    }

    return retVal;
  }

  private void addUsers(Collection userIDs, UserGroup userGroup) {
    for (String userID : userIDs) {
      User user = userRepository.fetchById(userID);

      if (userGroup.getUsers() == null) {
        userGroup.setUsers(new ArrayList());
      }
      if (!userGroup.getUsers().contains(user)) {
        userGroup.getUsers().add(user);
      }

      if (user.getUserGroups() == null) {
        user.setUserGroups(new ArrayList());
      }

      if (!user.getUserGroups().contains(userGroup)) {
        user.getUserGroups().add(userGroup);
      }
    }
  }

  public void addUser(String userID, String groupId) {
    List userIds = new ArrayList<>(1);
    userIds.add(userID);
    addUsers(userIds, groupId);
  }

  public void addUsers(Collection userIDs, String groupID) {
    addUsers(userIDs, userGroupRepository.fetchById(groupID));
  }

  public void addUserByGroupName(String userId, String groupName) {
    List userIds = new ArrayList<>(1);
    userIds.add(userId);
    addUsersByGroupName(userIds, groupName);
  }

  public void addUsersByGroupName(Collection userIDs,
    String groupName) {
    addUsers(userIDs, userGroupRepository.findByName(groupName));
  }

  public void removeUser(String userID, String groupID) {
    List userIds = new ArrayList<>(1);
    userIds.add(userID);
    removeUsers(userIds, groupID);
  }

  public void removeUsers(Collection userIDs, String groupID) {
    UserGroup userGroup = userGroupRepository.fetchById(groupID);
    for (String userID : userIDs) {
      User user = userRepository.fetchById(userID);
      userGroup.getUsers().remove(user);
    }
  }

  public Set getGroupUsersIds(String groupID, boolean includeChildren) {
    UserGroup userGroup = userGroupRepository.fetchById(groupID);
    return getGroupHierarchyUsersIds(userGroup, false, includeChildren);
  }

  public Set getUserGroupsIds(String userID) {
    User user = userRepository.fetchById(userID);
    Set retVal = new HashSet<>();
    if (user.getUserGroups() != null) {
      for (UserGroup userGroup : user.getUserGroups()) {
        retVal.add(userGroup.getId());
      }
    }
    return retVal;
  }

  /**
   * Retrieves the names of the users who are members of specific groups
   *
   * @param groupIDs The ids of the groups whose members to retrieve
   * @return The names of the retrieved users
   */
  public Set getGroupUsersNames(Collection groupIDs) {
    List groups = userGroupRepository.findByIdIn(groupIDs);
    Set users = groups.stream().flatMap(g -> g.getUsers().stream())
      .collect(Collectors.toSet());
    return users.stream().map(User::getUsername).collect(Collectors.toSet());
  }

  /**
   * Retrieves Groups
   *
   * @param criteria the criteria that is specified to search for a group
   * @return a list of Groups
   */
  public Iterable findGroups(UserGroupSearchCriteria criteria) {
    Predicate predicate = buildPredicate(criteria);
    if (criteria.getPageable() != null) {

      return findAll(predicate, criteria.getPageable());
    } else {

      return listGroups(predicate);
    }
  }

  /**
   * Finds all the groups based on predicate and pagination parameters.
   *
   * @param predicate the Boolean typed expressions to search for.
   * @param pageable the pagination information.
   * @return the responded DTO in page form, useful information for pagination and sorting.
   */
  public Page findAll(Predicate predicate, Pageable pageable) {
    return userGroupMapper.map(userGroupRepository.findAll(predicate, pageable));
  }

  private Predicate buildPredicate(UserGroupSearchCriteria criteria) {
    Predicate predicate = new BooleanBuilder();
    if (criteria.getNameLike() != null){
      predicate = ((BooleanBuilder) predicate)
          .and(qUserGroup.name.like(criteria.getNameLike()));
    }
    if (criteria.getName() != null) {
      predicate = ((BooleanBuilder) predicate)
          .and(qUserGroup.name.eq(criteria.getName()));
    }
    if (criteria.getIncludeIds() != null) {
      predicate = ((BooleanBuilder) predicate)
          .and(qUserGroup.id.in(criteria.getIncludeIds()));
    }
    if (criteria.getExcludeIds() != null) {
      predicate = ((BooleanBuilder) predicate)
          .and(qUserGroup.id.notIn(criteria.getExcludeIds()));
    }

    return predicate;
  }

  private List listGroups(Predicate predicate) {

    return userGroupRepository.findAll(predicate).stream()
        .map(o -> userGroupMapper.mapToDTO(o, false))
        .collect(Collectors.toList());
  }

}




© 2015 - 2024 Weber Informatics LLC | Privacy Policy