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 package org.opendaylight.controller.md.sal.dom.broker.impl.mount;
   9
  10 import com.google.common.annotations.VisibleForTesting;
  11 import com.google.common.base.Optional;
  12 import com.google.common.base.Preconditions;
  13 import com.google.common.collect.ClassToInstanceMap;
  14 import com.google.common.collect.MutableClassToInstanceMap;
  15 import org.opendaylight.controller.md.sal.dom.api.DOMMountPoint;
  16 import org.opendaylight.controller.md.sal.dom.api.DOMMountPointService;
  17 import org.opendaylight.controller.md.sal.dom.api.DOMService;
  18 import org.opendaylight.controller.md.sal.dom.broker.spi.mount.SimpleDOMMountPoint;
  19 import org.opendaylight.controller.sal.core.compat.DOMMountPointAdapter;
  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     private final org.opendaylight.mdsal.dom.api.DOMMountPointService delegate;
  28
  29     @VisibleForTesting
  30     public DOMMountPointServiceImpl() {
  31         this(new org.opendaylight.mdsal.dom.broker.DOMMountPointServiceImpl());
  32     }
  33
  34     public DOMMountPointServiceImpl(final org.opendaylight.mdsal.dom.api.DOMMountPointService delegate) {
  35         this.delegate = delegate;
  36     }
  37
  38     @Override
  39     public Optional<DOMMountPoint> getMountPoint(final YangInstanceIdentifier path) {
  40         return delegate.getMountPoint(path).transform(DOMMountPointAdapter::new);
  41     }
  42
  43     @Override
  44     public DOMMountPointBuilder createMountPoint(final YangInstanceIdentifier path) {
  45         return new DOMMountPointBuilderImpl(path);
  46     }
  47
  48     @Override
  49     public ListenerRegistration<DOMMountPointListener> registerProvisionListener(final DOMMountPointListener listener) {
  50         return delegate.registerProvisionListener(listener);
  51     }
  52
  53     /**
  54      * Deprecated.
  55
  56      * @deprecated this method should never have been exposed publicly - registration should be done via the
  57      *         public {@link #createMountPoint} interface. As such, this method expects the {@code mountPoint} param
  58      *         to be of type {@link SimpleDOMMountPoint}.
  59      */
  60     @Deprecated
  61     public ObjectRegistration<DOMMountPoint> registerMountPoint(final DOMMountPoint mountPoint) {
  62         Preconditions.checkArgument(mountPoint instanceof SimpleDOMMountPoint, "Expected SimpleDOMMountPoint");
  63         return doRegisterMountPoint((SimpleDOMMountPoint) mountPoint);
  64     }
  65
  66     @SuppressWarnings("unchecked")
  67     ObjectRegistration<DOMMountPoint> doRegisterMountPoint(final SimpleDOMMountPoint mountPoint) {
  68         final org.opendaylight.mdsal.dom.api.DOMMountPointService.DOMMountPointBuilder delegateBuilder =
  69                 delegate.createMountPoint(mountPoint.getIdentifier());
  70
  71         if (mountPoint.getSchemaContext() != null) {
  72             delegateBuilder.addInitialSchemaContext(mountPoint.getSchemaContext());
  73         }
  74
  75         mountPoint.getAllServices().forEach(
  76             entry -> delegateBuilder.addService((Class<DOMService>)entry.getKey(), entry.getValue()));
  77
  78         final ObjectRegistration<org.opendaylight.mdsal.dom.api.DOMMountPoint> delegateReg = delegateBuilder.register();
  79         return new ObjectRegistration<DOMMountPoint>() {
  80             @Override
  81             public void close() {
  82                 delegateReg.close();
  83             }
  84
  85             @Override
  86             public DOMMountPoint getInstance() {
  87                 return mountPoint;
  88             }
  89         };
  90     }
  91
  92     public class DOMMountPointBuilderImpl implements DOMMountPointBuilder {
  93
  94         ClassToInstanceMap<DOMService> services = MutableClassToInstanceMap.create();
  95         private SimpleDOMMountPoint mountPoint;
  96         private final YangInstanceIdentifier path;
  97         private SchemaContext schemaContext;
  98
  99         public DOMMountPointBuilderImpl(final YangInstanceIdentifier path) {
 100             this.path = path;
 101         }
 102
 103         @Override
 104         public <T extends DOMService> DOMMountPointBuilder addService(final Class<T> type, final T impl) {
 105             services.putInstance(type, impl);
 106             return this;
 107         }
 108
 109         @Override
 110         public DOMMountPointBuilder addInitialSchemaContext(final SchemaContext ctx) {
 111             schemaContext = ctx;
 112             return this;
 113         }
 114
 115         @Override
 116         public ObjectRegistration<DOMMountPoint> register() {
 117             Preconditions.checkState(mountPoint == null, "Mount point is already built.");
 118             mountPoint = SimpleDOMMountPoint.create(path, services, schemaContext);
 119             return doRegisterMountPoint(mountPoint);
 120         }
 121     }
 122 }