opendaylight/md-sal/sal-rest-docgen/src/main/java/org/opendaylight/controller/sal/rest/doc/DocProvider.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.sal.rest.doc;
   9
  10 import java.util.Collection;
  11 import java.util.Collections;
  12
  13 import org.opendaylight.controller.sal.core.api.Broker;
  14 import org.opendaylight.controller.sal.core.api.Provider;
  15 import org.opendaylight.controller.sal.core.api.model.SchemaService;
  16 import org.opendaylight.controller.sal.rest.doc.impl.ApiDocGenerator;
  17 import org.osgi.framework.BundleActivator;
  18 import org.osgi.framework.BundleContext;
  19 import org.osgi.framework.ServiceReference;
  20 import org.osgi.util.tracker.ServiceTracker;
  21 import org.osgi.util.tracker.ServiceTrackerCustomizer;
  22 import org.slf4j.Logger;
  23 import org.slf4j.LoggerFactory;
  24
  25 public class DocProvider implements BundleActivator, ServiceTrackerCustomizer<Broker, Broker>, Provider, AutoCloseable {
  26
  27     private static final Logger _logger = LoggerFactory.getLogger(DocProvider.class);
  28
  29     private ServiceTracker<Broker, Broker> brokerServiceTracker;
  30     private BundleContext bundleContext;
  31     private Broker.ProviderSession session;
  32
  33     @Override
  34     public void close() throws Exception {
  35         stop(bundleContext);
  36     }
  37
  38     @Override
  39     public void onSessionInitiated(final Broker.ProviderSession providerSession) {
  40         SchemaService schemaService = providerSession.getService(SchemaService.class);
  41         ApiDocGenerator.getInstance().setSchemaService(schemaService);
  42
  43         _logger.debug("Restconf API Explorer started");
  44     }
  45
  46     @Override
  47     public Collection<ProviderFunctionality> getProviderFunctionality() {
  48         return Collections.emptySet();
  49     }
  50
  51     @Override
  52     public void start(final BundleContext context) throws Exception {
  53         bundleContext = context;
  54         brokerServiceTracker = new ServiceTracker<>(context, Broker.class, this);
  55         brokerServiceTracker.open();
  56     }
  57
  58     @Override
  59     public void stop(final BundleContext context) throws Exception {
  60         if (brokerServiceTracker != null) {
  61             brokerServiceTracker.close();
  62         }
  63
  64         if (session != null) {
  65             session.close();
  66         }
  67     }
  68
  69     @Override
  70     public Broker addingService(final ServiceReference<Broker> reference) {
  71         Broker broker = bundleContext.getService(reference);
  72         session = broker.registerProvider(this, bundleContext);
  73         return broker;
  74     }
  75
  76     @Override
  77     public void modifiedService(final ServiceReference<Broker> reference, final Broker service) {
  78         if (session != null) {
  79             session.close();
  80         }
  81
  82         Broker broker = bundleContext.getService(reference);
  83         session = broker.registerProvider(this, bundleContext);
  84     }
  85
  86     @Override
  87     public void removedService(final ServiceReference<Broker> reference, final Broker service) {
  88         bundleContext.ungetService(reference);
  89     }
  90 }