gerrit.onap Code Review - sdc.git/blob

   1 package org.openecomp.sdc.healing.healers;
   2
   3 import org.openecomp.sdc.common.errors.SdcRuntimeException;
   4 import org.openecomp.sdc.common.session.SessionContextProviderFactory;
   5 import org.openecomp.sdc.healing.interfaces.Healer;
   6 import org.openecomp.sdc.itempermissions.dao.ItemPermissionsDao;
   7 import org.openecomp.sdc.itempermissions.dao.ItemPermissionsDaoFactory;
   8 import org.openecomp.sdc.itempermissions.impl.types.PermissionTypes;
   9 import org.openecomp.sdc.itempermissions.type.ItemPermissionsEntity;
  10 import org.openecomp.sdc.notification.dao.SubscribersDao;
  11 import org.openecomp.sdc.notification.factories.SubscribersDaoFactory;
  12 import org.openecomp.sdc.versioning.dao.ItemDao;
  13 import org.openecomp.sdc.versioning.dao.ItemDaoFactory;
  14 import org.openecomp.sdc.versioning.dao.types.Version;
  15 import org.openecomp.sdc.versioning.types.Item;
  16
  17 import java.util.Collection;
  18 import java.util.Collections;
  19 import java.util.HashSet;
  20 import java.util.Objects;
  21 import java.util.Optional;
  22
  23 /**
  24  * Created by ayalaben on 8/28/2017
  25  */
  26 public class OwnerHealer implements Healer {
  27   private static final String HEALING_USER_SUFFIX = "_healer";
  28   private static final ItemPermissionsDao permissionsDao =
  29       ItemPermissionsDaoFactory.getInstance().createInterface();
  30   private static final ItemDao itemDao = ItemDaoFactory.getInstance().createInterface();
  31
  32   private static final SubscribersDao subscribersDao = SubscribersDaoFactory.getInstance()
  33       .createInterface();
  34
  35   @Override
  36   public boolean isHealingNeeded(String itemId, Version version) {
  37     return permissionsDao.listItemPermissions(itemId).stream().noneMatch(this::isOwnerPermission) ||
  38         isOwnerMissingOnItem(itemId);
  39   }
  40
  41   public void heal(String itemId, Version version) {
  42     Collection<ItemPermissionsEntity> itemPermissions = permissionsDao.listItemPermissions(itemId);
  43
  44     if (itemPermissions.stream().noneMatch(this::isOwnerPermission)) {
  45       String currentUserId =
  46           SessionContextProviderFactory.getInstance().createInterface().get().getUser().getUserId()
  47               .replace(HEALING_USER_SUFFIX, "");
  48
  49       permissionsDao.updateItemPermissions(itemId, PermissionTypes.Owner.name(),
  50           Collections.singleton(currentUserId), new HashSet<>());
  51
  52       updateItemOwner(itemId, currentUserId);
  53
  54       subscribersDao.subscribe(currentUserId, itemId);
  55
  56     } else if (isOwnerMissingOnItem(itemId)) {
  57       Optional<ItemPermissionsEntity> ownerOpt =
  58           itemPermissions.stream().filter(this::isOwnerPermission).findFirst();
  59       if (ownerOpt.isPresent()) {
  60         updateItemOwner(itemId, ownerOpt.get().getUserId());
  61       } else {
  62         throw new SdcRuntimeException("Unexpected error in Owner Healer. Item id: " + itemId);
  63       }
  64     }
  65   }
  66
  67   private void updateItemOwner(String itemId, String userId) {
  68     Item item = new Item();
  69     item.setId(itemId);
  70     Item retrievedItem = itemDao.get(item);
  71     if (Objects.nonNull(retrievedItem)) {
  72       retrievedItem.setOwner(userId);
  73       itemDao.update(retrievedItem);
  74     }
  75   }
  76
  77   private boolean isOwnerMissingOnItem(String itemId) {
  78     Item item = new Item();
  79     item.setId(itemId);
  80     Item retrievedItem = itemDao.get(item);
  81     return Objects.nonNull(retrievedItem) && Objects.isNull(retrievedItem.getOwner());
  82   }
  83
  84   private boolean isOwnerPermission(ItemPermissionsEntity permissionsEntity) {
  85     return permissionsEntity.getPermission().equals(PermissionTypes.Owner.name());
  86   }
  87 }