diff --git a/lighty-modules/lighty-gnmi/lighty-gnmi-sb/src/main/java/io/lighty/gnmi/southbound/schema/certstore/rpc/CertificationStorageServiceRpcImpl.java b/lighty-modules/lighty-gnmi/lighty-gnmi-sb/src/main/java/io/lighty/gnmi/southbound/schema/certstore/rpc/CertificationStorageServiceRpcImpl.java index 586cac087..0fcc6842b 100644 --- a/lighty-modules/lighty-gnmi/lighty-gnmi-sb/src/main/java/io/lighty/gnmi/southbound/schema/certstore/rpc/CertificationStorageServiceRpcImpl.java +++ b/lighty-modules/lighty-gnmi/lighty-gnmi-sb/src/main/java/io/lighty/gnmi/southbound/schema/certstore/rpc/CertificationStorageServiceRpcImpl.java @@ -9,8 +9,6 @@ package io.lighty.gnmi.southbound.schema.certstore.rpc; -import com.google.common.collect.ClassToInstanceMap; -import com.google.common.collect.ImmutableClassToInstanceMap; import com.google.common.util.concurrent.FutureCallback; import com.google.common.util.concurrent.Futures; import com.google.common.util.concurrent.ListenableFuture; @@ -18,6 +16,8 @@ import com.google.common.util.concurrent.SettableFuture; import io.lighty.gnmi.southbound.schema.certstore.service.CertificationStorageService; import java.security.GeneralSecurityException; +import java.util.Collection; +import java.util.List; import org.opendaylight.mdsal.common.api.CommitInfo; import org.opendaylight.yang.gen.v1.urn.lighty.gnmi.certificate.storage.rev210504.AddKeystoreCertificate; import org.opendaylight.yang.gen.v1.urn.lighty.gnmi.certificate.storage.rev210504.AddKeystoreCertificateInput; @@ -90,10 +90,9 @@ public void onFailure(final Throwable throwable) { return rpcResult; } - public ClassToInstanceMap> getRpcClassToInstanceMap() { - return ImmutableClassToInstanceMap.>builder() - .put(AddKeystoreCertificate.class, this::addKeystoreCertificate) - .put(RemoveKeystoreCertificate.class, this::removeKeystoreCertificate) - .build(); + public Collection> getRpcClassToInstanceMap() { + return List.of( + (AddKeystoreCertificate) this::addKeystoreCertificate, + (RemoveKeystoreCertificate) this::removeKeystoreCertificate); } }