2 * Copyright (c) 2015 Cisco Systems, Inc. and others. All rights reserved.
4 * This program and the accompanying materials are made available under the terms of the Eclipse
5 * Public License v1.0 which accompanies this distribution, and is available at
6 * http://www.eclipse.org/legal/epl-v10.html
8 package org.opendaylight.controller.md.sal.binding.impl;
10 import com.google.common.base.Predicate;
11 import com.google.common.base.Throwables;
12 import com.google.common.util.concurrent.SettableFuture;
14 import java.util.Collection;
15 import java.util.Date;
16 import java.util.List;
17 import java.util.concurrent.CopyOnWriteArrayList;
18 import java.util.concurrent.ExecutionException;
19 import java.util.concurrent.TimeUnit;
20 import java.util.concurrent.TimeoutException;
21 import org.opendaylight.yangtools.sal.binding.generator.util.BindingRuntimeContext;
22 import org.opendaylight.yangtools.yang.binding.Augmentation;
24 class FutureSchema implements AutoCloseable {
26 private final List<FutureSchemaPredicate> postponedOperations = new CopyOnWriteArrayList<>();
27 private final long duration;
28 private final TimeUnit unit;
30 protected FutureSchema(final long time, final TimeUnit unit) {
35 void onRuntimeContextUpdated(final BindingRuntimeContext context) {
36 for (final FutureSchemaPredicate op : postponedOperations) {
37 op.unlockIfPossible(context);
51 for (final FutureSchemaPredicate op : postponedOperations) {
56 private static boolean isSchemaAvailable(final Class<?> clz, final BindingRuntimeContext context) {
58 if (Augmentation.class.isAssignableFrom(clz)) {
59 schema = context.getAugmentationDefinition(clz);
61 schema = context.getSchemaDefinition(clz);
63 return schema != null;
66 boolean waitForSchema(final URI namespace, final Date revision) {
67 final FutureSchemaPredicate postponedOp = new FutureSchemaPredicate() {
70 public boolean apply(final BindingRuntimeContext input) {
71 return input.getSchemaContext().findModuleByNamespaceAndRevision(namespace, revision) != null;
74 return postponedOp.waitForSchema();
77 boolean waitForSchema(final Collection<Class<?>> bindingClasses) {
78 final FutureSchemaPredicate postponedOp = new FutureSchemaPredicate() {
81 public boolean apply(final BindingRuntimeContext context) {
82 for (final Class<?> clz : bindingClasses) {
83 if (!isSchemaAvailable(clz, context)) {
90 return postponedOp.waitForSchema();
93 private abstract class FutureSchemaPredicate implements Predicate<BindingRuntimeContext> {
95 final boolean waitForSchema() {
97 schemaPromise.get(duration, unit);
99 } catch (final InterruptedException | ExecutionException e) {
100 throw Throwables.propagate(e);
101 } catch (final TimeoutException e) {
104 postponedOperations.remove(this);
108 final void unlockIfPossible(final BindingRuntimeContext context) {
109 if (!schemaPromise.isDone() && apply(context)) {
110 schemaPromise.set(null);
114 final void cancel() {
115 schemaPromise.cancel(true);
118 private final SettableFuture<?> schemaPromise = SettableFuture.create();