nat/nat2vpp/src/main/java/io/fd/honeycomb/nat/read/NatReaderFactory.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.nat.read;
  18
  19 import com.google.common.collect.Sets;
  20 import com.google.inject.Inject;
  21 import io.fd.honeycomb.nat.util.MappingEntryContext;
  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.snat.dto.SnatAddressDetailsReplyDump;
  27 import io.fd.vpp.jvpp.snat.dto.SnatStaticMappingDetailsReplyDump;
  28 import io.fd.vpp.jvpp.snat.future.FutureJVppSnatFacade;
  29 import javax.annotation.Nonnull;
  30 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.nat.rev150908.NatState;
  31 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.nat.rev150908.NatStateBuilder;
  32 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.nat.rev150908.mapping.entry.ExternalSrcPort;
  33 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.nat.rev150908.mapping.entry.InternalSrcPort;
  34 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.nat.rev150908.nat.parameters.ExternalIpAddressPool;
  35 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.nat.rev150908.nat.state.NatInstances;
  36 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.nat.rev150908.nat.state.NatInstancesBuilder;
  37 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.nat.rev150908.nat.state.nat.instances.NatInstance;
  38 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.nat.rev150908.nat.state.nat.instances.nat.instance.MappingTable;
  39 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.nat.rev150908.nat.state.nat.instances.nat.instance.MappingTableBuilder;
  40 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.nat.rev150908.nat.state.nat.instances.nat.instance.NatCurrentConfig;
  41 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.nat.rev150908.nat.state.nat.instances.nat.instance.NatCurrentConfigBuilder;
  42 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.nat.rev150908.nat.state.nat.instances.nat.instance.mapping.table.MappingEntry;
  43 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  44
  45 public class NatReaderFactory implements ReaderFactory {
  46
  47     private static final InstanceIdentifier<NatState> NAT_OPER_ID = InstanceIdentifier.create(NatState.class);
  48     private static final InstanceIdentifier<NatInstances> NAT_INSTANCES_ID = NAT_OPER_ID.child(NatInstances.class);
  49     private static final InstanceIdentifier<NatInstance> NAT_INSTANCE_ID = NAT_INSTANCES_ID.child(NatInstance.class);
  50     private static final InstanceIdentifier<NatCurrentConfig> CURRENT_CONFIG =
  51             NAT_INSTANCE_ID.child(NatCurrentConfig.class);
  52     private static final InstanceIdentifier<MappingTable> MAP_TABLE_ID = NAT_INSTANCE_ID.child(MappingTable.class);
  53     private static final InstanceIdentifier<MappingEntry> MAP_ENTRY_ID = MAP_TABLE_ID.child(MappingEntry.class);
  54
  55     private final MappingEntryContext mappingEntryContext;
  56     private final DumpCacheManager<SnatStaticMappingDetailsReplyDump, Void> mapEntryDumpMgr;
  57     private final DumpCacheManager<SnatAddressDetailsReplyDump, Void> addressRangeDumpMgr;
  58
  59
  60     @Inject
  61     public NatReaderFactory(final FutureJVppSnatFacade jvppSnat,
  62                             final MappingEntryContext mappingEntryContext) {
  63         this.mappingEntryContext = mappingEntryContext;
  64         this.mapEntryDumpMgr =
  65                 new DumpCacheManager.DumpCacheManagerBuilder<SnatStaticMappingDetailsReplyDump, Void>()
  66                         .withExecutor(new MappingEntryCustomizer.MappingEntryDumpExecutor(jvppSnat))
  67                         .build();
  68
  69         this.addressRangeDumpMgr =
  70                 new DumpCacheManager.DumpCacheManagerBuilder<SnatAddressDetailsReplyDump, Void>()
  71                         .withExecutor(new ExternalIpPoolCustomizer.AddressRangeDumpExecutor(jvppSnat))
  72                         .build();
  73     }
  74
  75     @Override
  76     public void init(@Nonnull final ModifiableReaderRegistryBuilder registry) {
  77         registry.addStructuralReader(NAT_OPER_ID, NatStateBuilder.class);
  78         registry.addStructuralReader(NAT_INSTANCES_ID, NatInstancesBuilder.class);
  79         registry.add(new GenericInitListReader<>(NAT_INSTANCE_ID, new NatInstanceCustomizer(mapEntryDumpMgr)));
  80         registry.addStructuralReader(MAP_TABLE_ID, MappingTableBuilder.class);
  81         registry.subtreeAdd(Sets.newHashSet(InstanceIdentifier.create(MappingEntry.class).child(ExternalSrcPort.class),
  82                 InstanceIdentifier.create(MappingEntry.class).child(InternalSrcPort.class)),
  83                 new GenericInitListReader<>(MAP_ENTRY_ID,
  84                         new MappingEntryCustomizer(mapEntryDumpMgr, mappingEntryContext)));
  85
  86         registry.addStructuralReader(CURRENT_CONFIG, NatCurrentConfigBuilder.class);
  87         registry.add(new GenericInitListReader<>(CURRENT_CONFIG.child(ExternalIpAddressPool.class),
  88                 new ExternalIpPoolCustomizer(addressRangeDumpMgr)));
  89     }
  90 }