HONEYCOMB-58 - Routing Api
[honeycomb.git] / lisp / lisp2vpp / src / main / java / io / fd / honeycomb / lisp / translate / read / MapResolverCustomizer.java
1 /*
2  * Copyright (c) 2016 Cisco 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.honeycomb.lisp.translate.read;
18
19 import static io.fd.honeycomb.translate.util.read.cache.EntityDumpExecutor.NO_PARAMS;
20
21 import com.google.common.base.Optional;
22 import io.fd.honeycomb.translate.read.ReadContext;
23 import io.fd.honeycomb.translate.read.ReadFailedException;
24 import io.fd.honeycomb.translate.spi.read.ListReaderCustomizer;
25 import io.fd.honeycomb.translate.util.RWUtils;
26 import io.fd.honeycomb.translate.util.read.cache.DumpCacheManager;
27 import io.fd.honeycomb.translate.vpp.util.AddressTranslator;
28 import io.fd.honeycomb.translate.vpp.util.FutureJVppCustomizer;
29 import io.fd.honeycomb.translate.vpp.util.JvppReplyConsumer;
30 import io.fd.vpp.jvpp.core.dto.LispMapResolverDetails;
31 import io.fd.vpp.jvpp.core.dto.LispMapResolverDetailsReplyDump;
32 import io.fd.vpp.jvpp.core.dto.LispMapResolverDump;
33 import io.fd.vpp.jvpp.core.future.FutureJVppCore;
34 import java.util.Arrays;
35 import java.util.Collections;
36 import java.util.List;
37 import java.util.stream.Collectors;
38 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddress;
39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.lisp.rev161214.map.resolvers.grouping.MapResolversBuilder;
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.lisp.rev161214.map.resolvers.grouping.map.resolvers.MapResolver;
41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.lisp.rev161214.map.resolvers.grouping.map.resolvers.MapResolverBuilder;
42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.lisp.rev161214.map.resolvers.grouping.map.resolvers.MapResolverKey;
43 import org.opendaylight.yangtools.concepts.Builder;
44 import org.opendaylight.yangtools.yang.binding.DataObject;
45 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
46 import org.slf4j.Logger;
47 import org.slf4j.LoggerFactory;
48
49 public class MapResolverCustomizer extends FutureJVppCustomizer
50         implements ListReaderCustomizer<MapResolver, MapResolverKey, MapResolverBuilder>, AddressTranslator,
51         JvppReplyConsumer {
52
53     private static final Logger LOG = LoggerFactory.getLogger(MapResolverCustomizer.class);
54
55     private final DumpCacheManager<LispMapResolverDetailsReplyDump, Void> dumpManager;
56
57     public MapResolverCustomizer(FutureJVppCore futureJvpp) {
58         super(futureJvpp);
59         this.dumpManager =
60                 new DumpCacheManager.DumpCacheManagerBuilder<LispMapResolverDetailsReplyDump, Void>()
61                         .withExecutor((identifier, params) -> getReplyForRead(
62                                 futureJvpp.lispMapResolverDump(new LispMapResolverDump()).toCompletableFuture(),
63                                 identifier))
64                         .build();
65     }
66
67     @Override
68     public MapResolverBuilder getBuilder(InstanceIdentifier<MapResolver> id) {
69         return new MapResolverBuilder();
70     }
71
72     @Override
73     public void readCurrentAttributes(InstanceIdentifier<MapResolver> id, MapResolverBuilder builder, ReadContext ctx)
74             throws ReadFailedException {
75         LOG.debug("Reading attributes...");
76
77         final Optional<LispMapResolverDetailsReplyDump> dumpOptional =
78                 dumpManager.getDump(id, ctx.getModificationCache(), NO_PARAMS);
79
80         if (!dumpOptional.isPresent() || dumpOptional.get().lispMapResolverDetails.isEmpty()) {
81             LOG.warn("No data dumped");
82             return;
83         }
84
85         final MapResolverKey key = id.firstKeyOf(MapResolver.class);
86         final LispMapResolverDetails mapResolverDetails =
87                 dumpOptional.get().lispMapResolverDetails.stream()
88                         .filter(a -> addressesEqual(key.getIpAddress(),
89                                 arrayToIpAddressReversed(byteToBoolean(a.isIpv6), a.ipAddress)))
90                         .collect(RWUtils.singleItemCollector());
91
92         builder.setKey(key);
93         builder.setIpAddress(
94                 arrayToIpAddress(byteToBoolean(mapResolverDetails.isIpv6), mapResolverDetails.ipAddress));
95     }
96
97     // safest way to compare addresses - prevents returning false while using different types from hierarchy
98     // Ex. Key for MapResolver contains Ipv4Address as value but we translate addresses from binary data to Ipv4AddressNoZone
99     private boolean addressesEqual(final IpAddress left, final IpAddress right) {
100         return Arrays.equals(left.getValue(), right.getValue());
101     }
102
103     @Override
104     public List<MapResolverKey> getAllIds(InstanceIdentifier<MapResolver> id, ReadContext context)
105             throws ReadFailedException {
106         LOG.debug("Dumping MapResolver...");
107
108         final Optional<LispMapResolverDetailsReplyDump> dumpOptional =
109                 dumpManager.getDump(id, context.getModificationCache(), NO_PARAMS);
110
111         if (!dumpOptional.isPresent() || dumpOptional.get().lispMapResolverDetails.isEmpty()) {
112             return Collections.emptyList();
113         }
114
115         return dumpOptional.get().lispMapResolverDetails.stream()
116                 .map(resolver -> new MapResolverKey(
117                         arrayToIpAddressReversed(byteToBoolean(resolver.isIpv6), resolver.ipAddress)))
118                 .collect(Collectors.toList());
119     }
120
121     @Override
122     public void merge(Builder<? extends DataObject> builder, List<MapResolver> readData) {
123         ((MapResolversBuilder) builder).setMapResolver(readData);
124     }
125 }