org.opendaylight.jsonrpc.binding.MultiModelProxy Maven / Gradle / Ivy
/*
* Copyright (c) 2018 Lumina Networks, Inc. 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.jsonrpc.binding;
import com.google.common.collect.ClassToInstanceMap;
import com.google.common.collect.ImmutableClassToInstanceMap;
import java.util.Objects;
import java.util.Set;
import java.util.stream.Collectors;
import org.opendaylight.yangtools.yang.binding.RpcService;
/**
* Proxy context for remote RPC service which implements multiple yang models.
*
* @author Richard Kosegi
* @since Oct 16, 2018
*/
public class MultiModelProxy implements AutoCloseable {
private final ClassToInstanceMap proxyMap;
private final Set> proxies;
public MultiModelProxy(Set> proxies) {
this.proxies = proxies;
proxyMap = ImmutableClassToInstanceMap
.copyOf(proxies.stream().collect(Collectors.toMap(ProxyContext::getType, ProxyContext::getProxy)));
}
@Override
public void close() {
proxies.stream().forEach(ProxyContext::close);
}
/**
* Get RPC proxy for particular {@link RpcService}.
*
* @param type subtype of {@link RpcService} to get proxy for
* @return proxy for given RpcService subtype
*/
@SuppressWarnings("unchecked")
public T getRpcService(Class type) {
return (T) Objects.requireNonNull(proxyMap.get(type),
"Service is not supported by this requester instance : " + type);
}
}