2 * Copyright (c) 2018 Red Hat, Inc. and others. All rights reserved.
4 * This program and the accompanying materials are made available under the
5 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6 * and is available at http://www.eclipse.org/legal/epl-v10.html
8 package org.opendaylight.netvirt.natservice.internal;
10 import static org.opendaylight.genius.infra.Datastore.OPERATIONAL;
12 import com.google.common.util.concurrent.FluentFuture;
13 import java.math.BigInteger;
14 import javax.inject.Inject;
15 import javax.inject.Singleton;
17 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
18 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
19 import org.opendaylight.genius.infra.ManagedNewTransactionRunner;
20 import org.opendaylight.genius.infra.ManagedNewTransactionRunnerImpl;
21 import org.opendaylight.genius.mdsalutil.cache.DataObjectCache;
22 import org.opendaylight.infrautils.caches.CacheProvider;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.natservice.rev160111.NeutronVipStates;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.natservice.rev160111.neutron.vip.states.VipState;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.natservice.rev160111.neutron.vip.states.VipStateBuilder;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.natservice.rev160111.neutron.vip.states.VipStateKey;
27 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
30 public class VipStateTracker extends DataObjectCache<String, VipState> {
32 private ManagedNewTransactionRunner txRunner;
35 public VipStateTracker(DataBroker dataBroker, CacheProvider cacheProvider) {
38 LogicalDatastoreType.OPERATIONAL,
39 InstanceIdentifier.builder(NeutronVipStates.class).child(VipState.class).build(),
41 (iid, vipState) -> vipState.key().getIp(),
42 ip -> InstanceIdentifier.builder(NeutronVipStates.class).child(VipState.class,
43 new VipStateKey(ip)).build());
44 this.txRunner = new ManagedNewTransactionRunnerImpl(dataBroker);
47 public VipState buildVipState(String ip, String mac, BigInteger dpnId, String ifcName) {
48 return new VipStateBuilder().setIp(ip).setDpnId(dpnId).setIfcName(ifcName).build();
51 public FluentFuture<Void> writeVipState(VipState vipState) {
52 return txRunner.callWithNewWriteOnlyTransactionAndSubmit(OPERATIONAL, tx -> {
53 tx.put(InstanceIdentifier.builder(NeutronVipStates.class)
54 .child(VipState.class, vipState.key()).build(),