1 package org.opendaylight.controller.sal.binding.test.util;
3 import java.io.InputStream;
4 import java.util.ArrayList;
7 import java.util.concurrent.Future;
9 import javassist.ClassPool;
11 import org.opendaylight.controller.sal.binding.api.RpcProviderRegistry;
12 import org.opendaylight.controller.sal.binding.api.data.DataProviderService;
13 import org.opendaylight.controller.sal.binding.api.mount.MountProviderService;
14 import org.opendaylight.controller.sal.binding.impl.DataBrokerImpl;
15 import org.opendaylight.controller.sal.binding.impl.NotificationBrokerImpl;
16 import org.opendaylight.controller.sal.binding.impl.RpcProviderRegistryImpl;
17 import org.opendaylight.controller.sal.binding.impl.connect.dom.BindingDomConnectorDeployer;
18 import org.opendaylight.controller.sal.binding.impl.connect.dom.BindingIndependentConnector;
19 import org.opendaylight.controller.sal.binding.impl.forward.DomForwardedBindingBrokerImpl;
20 import org.opendaylight.controller.sal.core.api.Broker.ProviderSession;
21 import org.opendaylight.controller.sal.core.api.Broker.RoutedRpcRegistration;
22 import org.opendaylight.controller.sal.core.api.Broker.RpcRegistration;
23 import org.opendaylight.controller.sal.core.api.BrokerService;
24 import org.opendaylight.controller.sal.core.api.RpcImplementation;
25 import org.opendaylight.controller.sal.core.api.RpcProvisionRegistry;
26 import org.opendaylight.controller.sal.core.api.RpcRegistrationListener;
27 import org.opendaylight.controller.sal.core.api.data.DataStore;
28 import org.opendaylight.controller.sal.core.api.mount.MountProvisionService;
29 import org.opendaylight.controller.sal.dom.broker.BrokerImpl;
30 import org.opendaylight.controller.sal.dom.broker.MountPointManagerImpl;
31 import org.opendaylight.controller.sal.dom.broker.impl.DataStoreStatsWrapper;
32 import org.opendaylight.controller.sal.dom.broker.impl.HashMapDataStore;
33 import org.opendaylight.controller.sal.dom.broker.impl.SchemaAwareDataStoreAdapter;
34 import org.opendaylight.controller.sal.dom.broker.impl.SchemaAwareRpcBroker;
35 import org.opendaylight.controller.sal.dom.broker.impl.SchemaContextProvider;
36 import org.opendaylight.yangtools.concepts.ListenerRegistration;
37 import org.opendaylight.yangtools.sal.binding.generator.impl.RuntimeGeneratedMappingServiceImpl;
38 import org.opendaylight.yangtools.yang.common.QName;
39 import org.opendaylight.yangtools.yang.common.RpcResult;
40 import org.opendaylight.yangtools.yang.data.api.CompositeNode;
41 import org.opendaylight.yangtools.yang.data.impl.codec.BindingIndependentMappingService;
42 import org.opendaylight.yangtools.yang.model.api.Module;
43 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
44 import org.opendaylight.yangtools.yang.parser.impl.YangParserImpl;
45 import org.reflections.Reflections;
46 import org.reflections.scanners.ResourcesScanner;
47 import org.slf4j.Logger;
48 import org.slf4j.LoggerFactory;
50 import com.google.common.base.Predicate;
51 import com.google.common.collect.ClassToInstanceMap;
52 import com.google.common.collect.ImmutableClassToInstanceMap;
53 import com.google.common.util.concurrent.ListeningExecutorService;
55 import static com.google.common.base.Preconditions.*;
57 public class BindingTestContext implements AutoCloseable, SchemaContextProvider {
59 public static final org.opendaylight.yangtools.yang.data.api.InstanceIdentifier TREE_ROOT = org.opendaylight.yangtools.yang.data.api.InstanceIdentifier
60 .builder().toInstance();
62 private static final Logger LOG = LoggerFactory.getLogger(BindingTestContext.class);
64 private RuntimeGeneratedMappingServiceImpl mappingServiceImpl;
66 private DomForwardedBindingBrokerImpl baBrokerImpl;
67 private DataBrokerImpl baDataImpl;
68 private NotificationBrokerImpl baNotifyImpl;
69 private BindingIndependentConnector baConnectImpl;
71 private org.opendaylight.controller.sal.dom.broker.DataBrokerImpl biDataImpl;
72 private BrokerImpl biBrokerImpl;
73 private HashMapDataStore rawDataStore;
74 private SchemaAwareDataStoreAdapter schemaAwareDataStore;
75 private DataStoreStatsWrapper dataStoreStats;
76 private DataStore dataStore;
78 private boolean dataStoreStatisticsEnabled = false;
80 private final ListeningExecutorService executor;
81 private final ClassPool classPool;
83 private final boolean startWithSchema;
85 private MountPointManagerImpl biMountImpl;
87 private SchemaContext schemaContext;
89 public SchemaContext getSchemaContext() {
93 protected BindingTestContext(ListeningExecutorService executor, ClassPool classPool, boolean startWithSchema) {
94 this.executor = executor;
95 this.classPool = classPool;
96 this.startWithSchema = startWithSchema;
99 public void startDomDataStore() {
100 checkState(dataStore == null, "DataStore already started.");
101 checkState(biDataImpl != null, "Dom Data Broker not present");
102 rawDataStore = new HashMapDataStore();
103 schemaAwareDataStore = new SchemaAwareDataStoreAdapter();
104 schemaAwareDataStore.changeDelegate(rawDataStore);
105 if (dataStoreStatisticsEnabled) {
106 dataStoreStats = new DataStoreStatsWrapper(schemaAwareDataStore);
107 dataStore = dataStoreStats;
109 dataStore = schemaAwareDataStore;
112 biDataImpl.registerConfigurationReader(TREE_ROOT, dataStore);
113 biDataImpl.registerOperationalReader(TREE_ROOT, dataStore);
114 biDataImpl.registerCommitHandler(TREE_ROOT, dataStore);
117 public void startDomDataBroker() {
118 checkState(executor != null, "Executor needs to be set");
119 biDataImpl = new org.opendaylight.controller.sal.dom.broker.DataBrokerImpl();
120 biDataImpl.setExecutor(executor);
123 public void startBindingDataBroker() {
124 checkState(executor != null, "Executor needs to be set");
125 baDataImpl = new DataBrokerImpl();
126 baDataImpl.setExecutor(executor);
129 public void startBindingBroker() {
130 checkState(executor != null, "Executor needs to be set");
131 checkState(baDataImpl != null, "Binding Data Broker must be started");
132 checkState(baNotifyImpl != null, "Notification Service must be started");
133 baBrokerImpl = new DomForwardedBindingBrokerImpl("test");
135 baBrokerImpl.getMountManager().setDataCommitExecutor(executor);
136 baBrokerImpl.getMountManager().setNotificationExecutor(executor);
137 baBrokerImpl.setRpcBroker(new RpcProviderRegistryImpl("test"));
138 baBrokerImpl.setDataBroker(baDataImpl);
139 baBrokerImpl.setNotificationBroker(baNotifyImpl);
140 baBrokerImpl.start();
143 public void startForwarding() {
144 checkState(baDataImpl != null, "Binding Data Broker needs to be started");
145 checkState(biDataImpl != null, "DOM Data Broker needs to be started.");
146 checkState(mappingServiceImpl != null, "DOM Mapping Service needs to be started.");
148 baConnectImpl = BindingDomConnectorDeployer.createConnector(getBindingToDomMappingService());
149 baConnectImpl.setDomRpcRegistry(getDomRpcRegistry());
150 baBrokerImpl.setConnector(baConnectImpl);
151 baBrokerImpl.setDomProviderContext(createMockContext());
152 baBrokerImpl.startForwarding();
155 private ProviderSession createMockContext() {
156 // TODO Auto-generated method stub
157 final ClassToInstanceMap<BrokerService> domBrokerServices = ImmutableClassToInstanceMap
158 .<BrokerService> builder()
160 .put(org.opendaylight.controller.sal.core.api.data.DataProviderService.class, biDataImpl) //
161 .put(RpcProvisionRegistry.class, biBrokerImpl.getRouter()) //
162 .put(MountProvisionService.class, biMountImpl) //
165 return new ProviderSession() {
168 public Future<RpcResult<CompositeNode>> rpc(QName rpc, CompositeNode input) {
169 throw new UnsupportedOperationException();
173 public <T extends BrokerService> T getService(Class<T> service) {
174 return domBrokerServices.getInstance(service);
178 public boolean isClosed() {
183 public Set<QName> getSupportedRpcs() {
188 public void close() {
192 public ListenerRegistration<RpcRegistrationListener> addRpcRegistrationListener(
193 RpcRegistrationListener listener) {
198 public RpcRegistration addRpcImplementation(QName rpcType, RpcImplementation implementation)
199 throws IllegalArgumentException {
204 public RoutedRpcRegistration addRoutedRpcImplementation(QName rpcType, RpcImplementation implementation) {
209 public RoutedRpcRegistration addMountedRpcImplementation(QName rpcType, RpcImplementation implementation) {
215 public void startBindingToDomMappingService() {
216 checkState(classPool != null, "ClassPool needs to be present");
217 mappingServiceImpl = new RuntimeGeneratedMappingServiceImpl();
218 mappingServiceImpl.setPool(classPool);
219 mappingServiceImpl.init();
222 public void updateYangSchema(String[] files) {
223 schemaContext = getContext(files);
224 if (schemaAwareDataStore != null) {
225 schemaAwareDataStore.onGlobalContextUpdated(schemaContext);
227 if (mappingServiceImpl != null) {
228 mappingServiceImpl.onGlobalContextUpdated(schemaContext);
232 public static String[] getAllYangFilesOnClasspath() {
233 Predicate<String> predicate = new Predicate<String>() {
235 public boolean apply(String input) {
236 return input.endsWith(".yang");
239 Reflections reflection = new Reflections("META-INF.yang", new ResourcesScanner());
240 Set<String> result = reflection.getResources(predicate);
241 return (String[]) result.toArray(new String[result.size()]);
244 private static SchemaContext getContext(String[] yangFiles) {
245 ClassLoader loader = BindingTestContext.class.getClassLoader();
246 List<InputStream> streams = new ArrayList<>();
247 for (String string : yangFiles) {
248 InputStream stream = loader.getResourceAsStream(string);
251 YangParserImpl parser = new YangParserImpl();
252 Set<Module> modules = parser.parseYangModelsFromStreams(streams);
253 return parser.resolveSchemaContext(modules);
256 public void start() {
257 startBindingDataBroker();
258 startBindingNotificationBroker();
259 startBindingBroker();
260 startDomDataBroker();
263 startDomMountPoint();
264 startBindingToDomMappingService();
266 if (startWithSchema) {
267 loadYangSchemaFromClasspath();
271 private void startDomMountPoint() {
272 biMountImpl = new MountPointManagerImpl();
273 biMountImpl.setDataBroker(getDomDataBroker());
276 private void startDomBroker() {
277 checkState(executor != null);
278 biBrokerImpl = new BrokerImpl();
279 biBrokerImpl.setExecutor(executor);
280 biBrokerImpl.setRouter(new SchemaAwareRpcBroker("/", this));
283 public void startBindingNotificationBroker() {
284 checkState(executor != null);
285 baNotifyImpl = new NotificationBrokerImpl(executor);
289 public void loadYangSchemaFromClasspath() {
290 String[] files = getAllYangFilesOnClasspath();
291 updateYangSchema(files);
294 public DataProviderService getBindingDataBroker() {
298 public org.opendaylight.controller.sal.core.api.data.DataProviderService getDomDataBroker() {
302 public DataStore getDomDataStore() {
306 public BindingIndependentMappingService getBindingToDomMappingService() {
307 return mappingServiceImpl;
310 public void logDataStoreStatistics() {
311 if (dataStoreStats == null) {
315 LOG.info("BIDataStore Statistics: Configuration Read Count: {} TotalTime: {} ms AverageTime (ns): {} ms",
316 dataStoreStats.getConfigurationReadCount(), dataStoreStats.getConfigurationReadTotalTime(),
317 dataStoreStats.getConfigurationReadAverageTime());
319 LOG.info("BIDataStore Statistics: Operational Read Count: {} TotalTime: {} ms AverageTime (ns): {} ms",
320 dataStoreStats.getOperationalReadCount(), dataStoreStats.getOperationalReadTotalTime(),
321 dataStoreStats.getOperationalReadAverageTime());
323 LOG.info("BIDataStore Statistics: Request Commit Count: {} TotalTime: {} ms AverageTime (ns): {} ms",
324 dataStoreStats.getRequestCommitCount(), dataStoreStats.getRequestCommitTotalTime(),
325 dataStoreStats.getRequestCommitAverageTime());
328 public RpcProviderRegistry getBindingRpcRegistry() {
329 return baBrokerImpl.getRoot();
332 public RpcProvisionRegistry getDomRpcRegistry() {
333 if (biBrokerImpl == null) {
336 return biBrokerImpl.getRouter();
339 public RpcImplementation getDomRpcInvoker() {
340 return biBrokerImpl.getRouter();
344 public void close() throws Exception {
348 public MountProviderService getBindingMountProviderService() {
349 return baBrokerImpl.getMountManager();
352 public MountProvisionService getDomMountProviderService() {