gerrit.fd Code Review - hc2vpp.git/blob

   1 /*
   2  * Copyright (c) 2018 Bell Canada, Pantheon Technologies and/or its affiliates.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at:
   7  *
   8  *     http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 package io.fd.hc2vpp.fib.management.read;
  18
  19 import com.google.inject.Inject;
  20 import io.fd.hc2vpp.common.translate.util.JvppReplyConsumer;
  21 import io.fd.hc2vpp.fib.management.FibManagementIIds;
  22 import io.fd.honeycomb.translate.impl.read.GenericInitListReader;
  23 import io.fd.honeycomb.translate.read.ReaderFactory;
  24 import io.fd.honeycomb.translate.read.registry.ModifiableReaderRegistryBuilder;
  25 import io.fd.honeycomb.translate.util.read.cache.DumpCacheManager;
  26 import io.fd.vpp.jvpp.core.dto.Ip6FibDetailsReplyDump;
  27 import io.fd.vpp.jvpp.core.dto.Ip6FibDump;
  28 import io.fd.vpp.jvpp.core.dto.IpFibDetailsReplyDump;
  29 import io.fd.vpp.jvpp.core.dto.IpFibDump;
  30 import io.fd.vpp.jvpp.core.future.FutureJVppCore;
  31 import javax.annotation.Nonnull;
  32 import org.opendaylight.yang.gen.v1.http.fd.io.hc2vpp.yang.vpp.fib.table.management.rev180521.FibTableManagementBuilder;
  33 import org.opendaylight.yang.gen.v1.http.fd.io.hc2vpp.yang.vpp.fib.table.management.rev180521.vpp.fib.table.management.FibTablesBuilder;
  34
  35 /**
  36  * Factory producing readers for FIB management plugin's data.
  37  */
  38 public final class FibManagementReaderFactory implements ReaderFactory, JvppReplyConsumer {
  39
  40     @Inject
  41     private FutureJVppCore vppApi;
  42
  43     @Override
  44     public void init(@Nonnull final ModifiableReaderRegistryBuilder registry) {
  45         final DumpCacheManager<IpFibDetailsReplyDump, Void> ipv4DumpManager = newIpv4RoutesDumpManager(vppApi);
  46         final DumpCacheManager<Ip6FibDetailsReplyDump, Void> ipv6DumpManager = newIpv6RoutesDumpManager(vppApi);
  47
  48         registry.addStructuralReader(FibManagementIIds.FIB_MNGMNT, FibTableManagementBuilder.class);
  49         registry.addStructuralReader(FibManagementIIds.FM_FIB_TABLES, FibTablesBuilder.class);
  50         registry.add(new GenericInitListReader<>(FibManagementIIds.FM_FTBLS_TABLE,
  51                 new FibTableCustomizer(ipv4DumpManager, ipv6DumpManager)));
  52     }
  53
  54     private DumpCacheManager<IpFibDetailsReplyDump, Void> newIpv4RoutesDumpManager(
  55             @Nonnull final FutureJVppCore vppApi) {
  56         return new DumpCacheManager.DumpCacheManagerBuilder<IpFibDetailsReplyDump, Void>()
  57                 .withExecutor(
  58                         (identifier, params) -> getReplyForRead(vppApi.ipFibDump(new IpFibDump()).toCompletableFuture(),
  59                                 identifier))
  60                 .acceptOnly(IpFibDetailsReplyDump.class)
  61                 .build();
  62     }
  63
  64     private DumpCacheManager<Ip6FibDetailsReplyDump, Void> newIpv6RoutesDumpManager(
  65             @Nonnull final FutureJVppCore vppApi) {
  66         return new DumpCacheManager.DumpCacheManagerBuilder<Ip6FibDetailsReplyDump, Void>()
  67                 .withExecutor(
  68                         (identifier, params) -> getReplyForRead(
  69                                 vppApi.ip6FibDump(new Ip6FibDump()).toCompletableFuture(), identifier))
  70                 .acceptOnly(Ip6FibDetailsReplyDump.class)
  71                 .build();
  72     }
  73 }