Bug 5403 - Support yang-library schema resolution
[netconf.git] / restconf / sal-rest-connector / src / main / java / org / opendaylight / restconf / rest / RestConnectorProvider.java
1 /*
2  * Copyright (c) 2016 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.restconf.rest;
9
10 import com.google.common.base.Preconditions;
11 import java.util.Collection;
12 import java.util.Collections;
13 import org.opendaylight.controller.md.sal.dom.api.DOMMountPointService;
14 import org.opendaylight.controller.sal.core.api.Broker.ProviderSession;
15 import org.opendaylight.controller.sal.core.api.Provider;
16 import org.opendaylight.controller.sal.core.api.model.SchemaService;
17 import org.opendaylight.netconf.sal.rest.api.RestConnector;
18 import org.opendaylight.restconf.rest.api.schema.context.SchemaContextHandler;
19 import org.opendaylight.restconf.rest.handlers.api.DOMMountPointServiceHandler;
20 import org.opendaylight.restconf.rest.handlers.impl.DOMMountPointServiceHandlerImpl;
21 import org.opendaylight.restconf.rest.impl.schema.context.SchemaContextHandlerImpl;
22 import org.opendaylight.restconf.rest.impl.services.Draft11ServicesWrapperImpl;
23 import org.opendaylight.yangtools.concepts.ListenerRegistration;
24 import org.opendaylight.yangtools.yang.model.api.SchemaContextListener;
25
26 /**
27  * Provider for restconf draft11.
28  *
29  */
30 public class RestConnectorProvider implements Provider, RestConnector, AutoCloseable {
31
32     private ListenerRegistration<SchemaContextListener> listenerRegistration;
33
34     @Override
35     public void onSessionInitiated(final ProviderSession session) {
36         final SchemaService schemaService = Preconditions.checkNotNull(session.getService(SchemaService.class));
37         final DOMMountPointServiceHandler domMountPointServiceHandler = new DOMMountPointServiceHandlerImpl();
38         final SchemaContextHandler schemaCtxHandler = new SchemaContextHandlerImpl();
39         domMountPointServiceHandler.setDOMMountPointService(session.getService(DOMMountPointService.class));
40         final Draft11ServicesWrapperImpl wrapperServices = Draft11ServicesWrapperImpl.getInstance();
41         this.listenerRegistration = schemaService.registerSchemaContextListener(schemaCtxHandler);
42
43         wrapperServices.setHandlers(schemaCtxHandler, domMountPointServiceHandler);
44     }
45
46     @Override
47     public Collection<ProviderFunctionality> getProviderFunctionality() {
48         return Collections.emptySet();
49     }
50
51     @Override
52     public void close() throws Exception {
53         if (this.listenerRegistration != null) {
54             this.listenerRegistration.close();
55         }
56     }
57 }