openecomp-be/lib/openecomp-healing-lib/openecomp-sdc-healing-impl/src/main/java/org/openecomp/sdc/healing/healers/OwnerHealer.java

   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.*;
  18
  19 /**
  20  * Created by ayalaben on 8/28/2017
  21  */
  22 public class OwnerHealer implements Healer {
  23   private static final String HEALING_USER_SUFFIX = "_healer";
  24   private static final ItemPermissionsDao permissionsDao =
  25       ItemPermissionsDaoFactory.getInstance().createInterface();
  26   private static final ItemDao itemDao = ItemDaoFactory.getInstance().createInterface();
  27
  28   private static final SubscribersDao subscribersDao = SubscribersDaoFactory.getInstance()
  29       .createInterface();
  30
  31   @Override
  32   public boolean isHealingNeeded(String itemId, Version version) {
  33     return permissionsDao.listItemPermissions(itemId).stream().noneMatch(this::isOwnerPermission) ||
  34         isOwnerMissingOnItem(itemId);
  35   }
  36
  37   public void heal(String itemId, Version version) {
  38     Collection<ItemPermissionsEntity> itemPermissions = permissionsDao.listItemPermissions(itemId);
  39
  40     if (itemPermissions.stream().noneMatch(this::isOwnerPermission)) {
  41       String currentUserId =
  42           SessionContextProviderFactory.getInstance().createInterface().get().getUser().getUserId()
  43               .replace(HEALING_USER_SUFFIX, "");
  44
  45       permissionsDao.updateItemPermissions(itemId, PermissionTypes.Owner.name(),
  46           Collections.singleton(currentUserId), new HashSet<>());
  47
  48       updateItemOwner(itemId, currentUserId);
  49
  50       subscribersDao.subscribe(currentUserId, itemId);
  51
  52     } else if (isOwnerMissingOnItem(itemId)) {
  53       Optional<ItemPermissionsEntity> ownerOpt =
  54           itemPermissions.stream().filter(this::isOwnerPermission).findFirst();
  55       if (ownerOpt.isPresent()) {
  56         updateItemOwner(itemId, ownerOpt.get().getUserId());
  57       } else {
  58         throw new SdcRuntimeException("Unexpected error in Owner Healer. Item id: " + itemId);
  59       }
  60     }
  61   }
  62
  63   private void updateItemOwner(String itemId, String userId) {
  64     Item item = new Item();
  65     item.setId(itemId);
  66     Item retrievedItem = itemDao.get(item);
  67     if (Objects.nonNull(retrievedItem)) {
  68       retrievedItem.setOwner(userId);
  69       itemDao.update(retrievedItem);
  70     }
  71   }
  72
  73   private boolean isOwnerMissingOnItem(String itemId) {
  74     Item item = new Item();
  75     item.setId(itemId);
  76     Item retrievedItem = itemDao.get(item);
  77     return Objects.nonNull(retrievedItem) && Objects.isNull(retrievedItem.getOwner());
  78   }
  79
  80   private boolean isOwnerPermission(ItemPermissionsEntity permissionsEntity) {
  81     return permissionsEntity.getPermission().equals(PermissionTypes.Owner.name());
  82   }
  83 }