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
  18 package io.fd.hc2vpp.fib.management.read;
  19
  20 import io.fd.hc2vpp.common.translate.util.ByteDataTranslator;
  21 import io.fd.honeycomb.translate.read.ReadContext;
  22 import io.fd.honeycomb.translate.read.ReadFailedException;
  23 import io.fd.honeycomb.translate.spi.read.ListReaderCustomizer;
  24 import io.fd.honeycomb.translate.util.read.cache.DumpCacheManager;
  25 import io.fd.vpp.jvpp.core.dto.Ip6FibDetailsReplyDump;
  26 import io.fd.vpp.jvpp.core.dto.IpFibDetailsReplyDump;
  27 import java.util.List;
  28 import java.util.stream.Collectors;
  29 import java.util.stream.Stream;
  30 import javax.annotation.Nonnull;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.vpp.fib.table.management.rev180521.AddressFamilyIdentity;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.vpp.fib.table.management.rev180521.Ipv4;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.vpp.fib.table.management.rev180521.Ipv6;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.vpp.fib.table.management.rev180521.VniReference;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.vpp.fib.table.management.rev180521.vpp.fib.table.management.FibTablesBuilder;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.vpp.fib.table.management.rev180521.vpp.fib.table.management.fib.tables.Table;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.vpp.fib.table.management.rev180521.vpp.fib.table.management.fib.tables.TableBuilder;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.vpp.fib.table.management.rev180521.vpp.fib.table.management.fib.tables.TableKey;
  39 import org.opendaylight.yangtools.concepts.Builder;
  40 import org.opendaylight.yangtools.yang.binding.DataObject;
  41 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  42
  43 class FibTableCustomizer implements ListReaderCustomizer<Table, TableKey, TableBuilder>, ByteDataTranslator {
  44     private final DumpCacheManager<IpFibDetailsReplyDump, Void> ipv4DumpManager;
  45     private final DumpCacheManager<Ip6FibDetailsReplyDump, Void> ipv6DumpManager;
  46
  47     FibTableCustomizer(final DumpCacheManager<IpFibDetailsReplyDump, Void> ipv4DumpManager,
  48                        final DumpCacheManager<Ip6FibDetailsReplyDump, Void> ipv6DumpManager) {
  49
  50         this.ipv4DumpManager = ipv4DumpManager;
  51         this.ipv6DumpManager = ipv6DumpManager;
  52     }
  53
  54     @Nonnull
  55     @Override
  56     public List<TableKey> getAllIds(@Nonnull final InstanceIdentifier<Table> instanceIdentifier,
  57                                     @Nonnull final ReadContext readContext) throws ReadFailedException {
  58         return Stream.concat(ipv4DumpManager.getDump(instanceIdentifier, readContext.getModificationCache())
  59                         .or(new IpFibDetailsReplyDump())
  60                         .ipFibDetails.stream()
  61                         .filter(ipFibDetails -> ipFibDetails.tableId >= 0)
  62                         .map(ipFibDetails -> new TableKey(Ipv4.class, new VniReference((long) ipFibDetails.tableId)))
  63                         .distinct(),
  64                 ipv6DumpManager.getDump(instanceIdentifier, readContext.getModificationCache())
  65                         .or(new Ip6FibDetailsReplyDump())
  66                         .ip6FibDetails.stream()
  67                         .filter(ip6FibDetails -> ip6FibDetails.tableId >= 0)
  68                         .map(ipFibDetails -> new TableKey(Ipv6.class, new VniReference((long) ipFibDetails.tableId))))
  69                 .distinct()
  70                 .collect(Collectors.toList());
  71     }
  72
  73     @Override
  74     public void merge(@Nonnull final Builder<? extends DataObject> builder, @Nonnull final List<Table> list) {
  75         ((FibTablesBuilder) builder).setTable(list);
  76     }
  77
  78     @Nonnull
  79     @Override
  80     public TableBuilder getBuilder(@Nonnull final InstanceIdentifier<Table> instanceIdentifier) {
  81         return new TableBuilder();
  82     }
  83
  84     @Override
  85     public void readCurrentAttributes(@Nonnull final InstanceIdentifier<Table> instanceIdentifier,
  86                                       @Nonnull final TableBuilder builder, @Nonnull final ReadContext readContext)
  87             throws ReadFailedException {
  88         TableKey tableKey = instanceIdentifier.firstKeyOf(Table.class);
  89
  90         if (tableKey.getAddressFamily().equals(Ipv4.class)) {
  91             ipv4DumpManager.getDump(instanceIdentifier, readContext.getModificationCache())
  92                     .or(new IpFibDetailsReplyDump())
  93                     .ipFibDetails.stream()
  94                     .filter(ipFibDetails -> ipFibDetails.tableId == tableKey.getTableId().getValue().intValue())
  95                     .findFirst().ifPresent(
  96                     ipFibDetails -> parseFibDetails(ipFibDetails.tableId, ipFibDetails.tableName, Ipv4.class, builder));
  97
  98         } else {
  99             ipv6DumpManager.getDump(instanceIdentifier, readContext.getModificationCache())
 100                     .or(new Ip6FibDetailsReplyDump())
 101                     .ip6FibDetails.stream()
 102                     .filter(ipFibDetails -> ipFibDetails.tableId == tableKey.getTableId().getValue().intValue())
 103                     .findFirst().ifPresent(
 104                     ipFibDetails -> parseFibDetails(ipFibDetails.tableId, ipFibDetails.tableName, Ipv6.class, builder));
 105         }
 106     }
 107
 108     private void parseFibDetails(final Integer tableId, final byte[] tableName,
 109                                  final Class<? extends AddressFamilyIdentity> addressFamily,
 110                                  final TableBuilder builder) {
 111         builder.setAddressFamily(addressFamily)
 112                 .setTableId(new VniReference(Integer.toUnsignedLong(tableId)));
 113
 114         if (tableName != null) {
 115             // table name is optional
 116             builder.setName(toString(tableName));
 117         }
 118     }
 119 }