Add AbstractDOMSchemaService and FixedDOMSchemaService
[mdsal.git] / dom / mdsal-dom-broker / src / main / java / org / opendaylight / mdsal / dom / broker / schema / ScanningSchemaServiceProvider.java
1 /*
2  * Copyright (c) 2017 Pantheon Technologies s.r.o. 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.mdsal.dom.broker.schema;
9
10 import com.google.common.annotations.VisibleForTesting;
11 import com.google.common.collect.Iterables;
12 import com.google.common.util.concurrent.ListenableFuture;
13 import java.io.IOException;
14 import java.net.URL;
15 import java.util.ArrayList;
16 import java.util.List;
17 import java.util.Optional;
18 import javax.annotation.concurrent.GuardedBy;
19 import org.opendaylight.mdsal.dom.spi.AbstractDOMSchemaService;
20 import org.opendaylight.yangtools.concepts.ListenerRegistration;
21 import org.opendaylight.yangtools.concepts.Registration;
22 import org.opendaylight.yangtools.util.ListenerRegistry;
23 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
24 import org.opendaylight.yangtools.yang.model.api.SchemaContextListener;
25 import org.opendaylight.yangtools.yang.model.parser.api.YangSyntaxErrorException;
26 import org.opendaylight.yangtools.yang.model.repo.api.SchemaSourceException;
27 import org.opendaylight.yangtools.yang.model.repo.api.SourceIdentifier;
28 import org.opendaylight.yangtools.yang.model.repo.api.YangTextSchemaSource;
29 import org.opendaylight.yangtools.yang.parser.repo.YangTextSchemaContextResolver;
30 import org.slf4j.Logger;
31 import org.slf4j.LoggerFactory;
32
33 public class ScanningSchemaServiceProvider extends AbstractDOMSchemaService.WithYangTextSources
34         implements AutoCloseable {
35     private static final Logger LOG = LoggerFactory.getLogger(ScanningSchemaServiceProvider.class);
36
37     private final YangTextSchemaContextResolver contextResolver = YangTextSchemaContextResolver.create("global-bundle");
38     @GuardedBy("lock")
39     private final ListenerRegistry<SchemaContextListener> listeners = ListenerRegistry.create();
40     private final Object lock = new Object();
41
42     public void tryToUpdateSchemaContext() {
43         synchronized (lock) {
44             final Optional<SchemaContext> schema = contextResolver.getSchemaContext();
45             if (schema.isPresent()) {
46                 if (LOG.isDebugEnabled()) {
47                     LOG.debug("Got new SchemaContext: # of modules {}", schema.get().getModules().size());
48                 }
49                 notifyListeners(schema.get());
50             }
51         }
52     }
53
54     @VisibleForTesting
55     @SuppressWarnings("checkstyle:IllegalCatch")
56     public void notifyListeners(final SchemaContext schemaContext) {
57         synchronized (lock) {
58             for (final ListenerRegistration<SchemaContextListener> registration : listeners) {
59                 try {
60                     registration.getInstance().onGlobalContextUpdated(schemaContext);
61                 } catch (final Exception e) {
62                     LOG.error("Exception occured during invoking listener", e);
63                 }
64             }
65         }
66     }
67
68     public List<Registration> registerAvailableYangs(final List<URL> yangs) {
69         final List<Registration> sourceRegistrator = new ArrayList<>();
70         for (final URL url : yangs) {
71             try {
72                 sourceRegistrator.add(contextResolver.registerSource(url));
73             } catch (SchemaSourceException | IOException | YangSyntaxErrorException e) {
74                 LOG.warn("Failed to register {}, ignoring it", url, e);
75             }
76         }
77         return sourceRegistrator;
78     }
79
80     public void removeListener(final SchemaContextListener schemaContextListener) {
81         synchronized (lock) {
82             for (final ListenerRegistration<SchemaContextListener> listenerRegistration : listeners.getListeners()) {
83                 if (listenerRegistration.getInstance().equals(schemaContextListener)) {
84                     listenerRegistration.close();
85                     break;
86                 }
87             }
88         }
89     }
90
91     public boolean hasListeners() {
92         synchronized (lock) {
93             return !Iterables.isEmpty(listeners.getListeners());
94         }
95     }
96
97     @Override
98     public SchemaContext getGlobalContext() {
99         return contextResolver.getSchemaContext().orElse(null);
100     }
101
102     @Override
103     public ListenerRegistration<SchemaContextListener> registerSchemaContextListener(
104             final SchemaContextListener listener) {
105         synchronized (lock) {
106             contextResolver.getSchemaContext().ifPresent(listener::onGlobalContextUpdated);
107             return listeners.register(listener);
108         }
109     }
110
111     @Override
112     public ListenableFuture<? extends YangTextSchemaSource> getSource(final SourceIdentifier sourceIdentifier) {
113         return contextResolver.getSource(sourceIdentifier);
114     }
115
116     @Override
117     public void close() {
118         synchronized (lock) {
119             listeners.forEach(ListenerRegistration::close);
120         }
121     }
122 }