opendaylight/md-sal/sal-dom-broker/src/main/java/org/opendaylight/controller/md/sal/dom/broker/impl/mount/DOMMountPointServiceImpl.java

   1 /*
   2  * Copyright (c) 2014 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8
   9 package org.opendaylight.controller.md.sal.dom.broker.impl.mount;
  10
  11 import com.google.common.annotations.VisibleForTesting;
  12 import com.google.common.base.Optional;
  13 import com.google.common.base.Preconditions;
  14 import com.google.common.collect.ClassToInstanceMap;
  15 import com.google.common.collect.MutableClassToInstanceMap;
  16 import org.opendaylight.controller.md.sal.dom.api.DOMMountPoint;
  17 import org.opendaylight.controller.md.sal.dom.api.DOMMountPointService;
  18 import org.opendaylight.controller.md.sal.dom.api.DOMService;
  19 import org.opendaylight.controller.md.sal.dom.broker.spi.mount.SimpleDOMMountPoint;
  20 import org.opendaylight.mdsal.dom.api.DOMMountPointListener;
  21 import org.opendaylight.yangtools.concepts.ListenerRegistration;
  22 import org.opendaylight.yangtools.concepts.ObjectRegistration;
  23 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  24 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
  25
  26 public class DOMMountPointServiceImpl implements DOMMountPointService {
  27
  28     private final org.opendaylight.mdsal.dom.api.DOMMountPointService delegate;
  29
  30     @VisibleForTesting
  31     public DOMMountPointServiceImpl() {
  32         this(new org.opendaylight.mdsal.dom.broker.DOMMountPointServiceImpl());
  33     }
  34
  35     public DOMMountPointServiceImpl(final org.opendaylight.mdsal.dom.api.DOMMountPointService delegate) {
  36         this.delegate = delegate;
  37     }
  38
  39     @Override
  40     public Optional<DOMMountPoint> getMountPoint(final YangInstanceIdentifier path) {
  41         return Optional.fromJavaUtil(delegate.getMountPoint(path).map(DOMMountPointServiceImpl::convert));
  42     }
  43
  44     private static DOMMountPoint convert(final org.opendaylight.mdsal.dom.api.DOMMountPoint from) {
  45         return new DOMMountPoint() {
  46             @Override
  47             public YangInstanceIdentifier getIdentifier() {
  48                 return from.getIdentifier();
  49             }
  50
  51             @Override
  52             public <T extends DOMService> Optional<T> getService(final Class<T> cls) {
  53                 return Optional.fromJavaUtil(from.getService(cls));
  54             }
  55
  56             @Override
  57             public SchemaContext getSchemaContext() {
  58                 return from.getSchemaContext();
  59             }
  60
  61             @Override
  62             public int hashCode() {
  63                 return from.getIdentifier().hashCode();
  64             }
  65
  66             @Override
  67             public boolean equals(final Object obj) {
  68                 if (this == obj) {
  69                     return true;
  70                 }
  71
  72                 if (!(obj instanceof DOMMountPoint)) {
  73                     return false;
  74                 }
  75
  76                 DOMMountPoint other = (DOMMountPoint) obj;
  77                 return from.getIdentifier().equals(other.getIdentifier());
  78             }
  79         };
  80     }
  81
  82     @Override
  83     public DOMMountPointBuilder createMountPoint(final YangInstanceIdentifier path) {
  84         return new DOMMountPointBuilderImpl(path);
  85     }
  86
  87     @Override
  88     public ListenerRegistration<DOMMountPointListener> registerProvisionListener(final DOMMountPointListener listener) {
  89         return delegate.registerProvisionListener(listener);
  90     }
  91
  92     @SuppressWarnings("unchecked")
  93     ObjectRegistration<DOMMountPoint> registerMountPoint(final SimpleDOMMountPoint mountPoint) {
  94         final org.opendaylight.mdsal.dom.api.DOMMountPointService.DOMMountPointBuilder delegateBuilder =
  95                 delegate.createMountPoint(mountPoint.getIdentifier());
  96
  97         if (mountPoint.getSchemaContext() != null) {
  98             delegateBuilder.addInitialSchemaContext(mountPoint.getSchemaContext());
  99         }
 100
 101         mountPoint.getAllServices().forEach(
 102             entry -> delegateBuilder.addService((Class<DOMService>)entry.getKey(), entry.getValue()));
 103
 104         final ObjectRegistration<org.opendaylight.mdsal.dom.api.DOMMountPoint> delegateReg = delegateBuilder.register();
 105         return new ObjectRegistration<DOMMountPoint>() {
 106             @Override
 107             public void close() {
 108                 delegateReg.close();
 109             }
 110
 111             @Override
 112             public DOMMountPoint getInstance() {
 113                 return mountPoint;
 114             }
 115         };
 116     }
 117
 118     public class DOMMountPointBuilderImpl implements DOMMountPointBuilder {
 119
 120         ClassToInstanceMap<DOMService> services = MutableClassToInstanceMap.create();
 121         private SimpleDOMMountPoint mountPoint;
 122         private final YangInstanceIdentifier path;
 123         private SchemaContext schemaContext;
 124
 125         public DOMMountPointBuilderImpl(final YangInstanceIdentifier path) {
 126             this.path = path;
 127         }
 128
 129         @Override
 130         public <T extends DOMService> DOMMountPointBuilder addService(final Class<T> type, final T impl) {
 131             services.putInstance(type, impl);
 132             return this;
 133         }
 134
 135         @Override
 136         public DOMMountPointBuilder addInitialSchemaContext(final SchemaContext ctx) {
 137             schemaContext = ctx;
 138             return this;
 139         }
 140
 141         @Override
 142         public ObjectRegistration<DOMMountPoint> register() {
 143             Preconditions.checkState(mountPoint == null, "Mount point is already built.");
 144             mountPoint = SimpleDOMMountPoint.create(path, services, schemaContext);
 145             return registerMountPoint(mountPoint);
 146         }
 147     }
 148 }