/*
* Copyright (c) 2015 Cisco Systems, Inc. and others. All rights reserved.
*
* This program and the accompanying materials are made available under the
* terms of the Eclipse Public License v1.0 which accompanies this distribution,
* and is available at http://www.eclipse.org/legal/epl-v10.html
*/
package org.opendaylight.controller.md.sal.binding.impl;
import com.google.common.base.Preconditions;
import com.google.common.cache.CacheBuilder;
import com.google.common.cache.CacheLoader;
import com.google.common.cache.LoadingCache;
import com.google.common.collect.ClassToInstanceMap;
import com.google.common.collect.ImmutableSet;
import java.util.Set;
import org.opendaylight.controller.md.sal.binding.impl.BindingDOMAdapterBuilder.Factory;
import org.opendaylight.controller.md.sal.dom.api.DOMRpcService;
import org.opendaylight.controller.md.sal.dom.api.DOMService;
import org.opendaylight.controller.sal.binding.api.RpcConsumerRegistry;
import org.opendaylight.yangtools.yang.binding.RpcService;
import org.opendaylight.yangtools.yang.binding.util.BindingReflections;
public class BindingDOMRpcServiceAdapter implements RpcConsumerRegistry {
protected static final Factory<RpcConsumerRegistry> BUILDER_FACTORY = new Factory<RpcConsumerRegistry>() {
@Override
public BindingDOMAdapterBuilder<RpcConsumerRegistry> newBuilder() {
return new Builder();
}
};
private final LoadingCache<Class<? extends RpcService>, RpcServiceAdapter> proxies = CacheBuilder.newBuilder()
.weakKeys()
.build(new CacheLoader<Class<? extends RpcService>, RpcServiceAdapter>() {
@Override
public RpcServiceAdapter load(final Class<? extends RpcService> key) throws Exception {
return createProxy(key);
}
});
private final DOMRpcService domService;
private final BindingToNormalizedNodeCodec codec;
public BindingDOMRpcServiceAdapter(final DOMRpcService domService, final BindingToNormalizedNodeCodec codec) {
super();
this.domService = domService;
this.codec = codec;
}
@SuppressWarnings("unchecked")
@Override
public <T extends RpcService> T getRpcService(final Class<T> rpcService) {
Preconditions.checkArgument(rpcService != null, "Rpc Service needs to be specied.");
return (T) proxies.getUnchecked(rpcService).getProxy();
}
private RpcServiceAdapter createProxy(final Class<? extends RpcService> key) {
Preconditions.checkArgument(BindingReflections.isBindingClass(key));
Preconditions.checkArgument(key.isInterface(), "Supplied RPC service type must be interface.");
return new RpcServiceAdapter(key, codec, domService);
}
private static final class Builder extends BindingDOMAdapterBuilder<RpcConsumerRegistry> {
@Override
protected RpcConsumerRegistry createInstance(final BindingToNormalizedNodeCodec codec,
final ClassToInstanceMap<DOMService> delegates) {
final DOMRpcService domRpc = delegates.getInstance(DOMRpcService.class);
return new BindingDOMRpcServiceAdapter(domRpc , codec);
}
@Override
public Set<? extends Class<? extends DOMService>> getRequiredDelegates() {
return ImmutableSet.of(DOMRpcService.class);
}
}
}