neutron-mapper/src/main/java/org/opendaylight/groupbasedpolicy/neutron/mapper/mapping/group/SecGroupDao.java

   1 package org.opendaylight.groupbasedpolicy.neutron.mapper.mapping.group;
   2
   3 import static com.google.common.base.Preconditions.checkNotNull;
   4
   5 import java.util.HashMap;
   6 import java.util.Map;
   7
   8 import org.opendaylight.groupbasedpolicy.neutron.mapper.util.Utils;
   9 import org.opendaylight.neutron.spi.NeutronSecurityGroup;
  10 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.EndpointGroupId;
  11
  12 import com.google.common.base.Strings;
  13
  14 public class SecGroupDao {
  15
  16     private final Map<EndpointGroupId, NeutronSecurityGroup> secGroupById = new HashMap<>();
  17
  18     public void addSecGroup(NeutronSecurityGroup secGrp) {
  19         checkNotNull(secGrp);
  20         secGroupById.put(new EndpointGroupId(Utils.normalizeUuid(secGrp.getSecurityGroupUUID())), secGrp);
  21     }
  22
  23     public NeutronSecurityGroup getSecGroupById(EndpointGroupId id) {
  24         return secGroupById.get(id);
  25     }
  26
  27     /**
  28      * @param id
  29      * @return {@code empty string} if security group with given ID does not exist; returns
  30      *         {@code name of security group} if has some; otherwise security group id
  31      */
  32     public String getNameOrIdOfSecGroup(EndpointGroupId id) {
  33         NeutronSecurityGroup secGrp = secGroupById.get(checkNotNull(id));
  34         if (secGrp == null) {
  35             return "";
  36         }
  37         if (!Strings.isNullOrEmpty(secGrp.getSecurityGroupName())) {
  38             return secGrp.getSecurityGroupName();
  39         }
  40         return id.getValue();
  41     }
  42
  43     public void removeSecGroup(EndpointGroupId id) {
  44         secGroupById.remove(checkNotNull(id));
  45     }
  46
  47 }