natservice/impl/src/main/java/org/opendaylight/netvirt/natservice/internal/VipStateTracker.java

   1 /*
   2  * Copyright (c) 2018 Red Hat, Inc. and others. All rights reserved.
   3  *
   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
   7  */
   8 package org.opendaylight.netvirt.natservice.internal;
   9
  10 import static org.opendaylight.genius.infra.Datastore.OPERATIONAL;
  11
  12 import com.google.common.util.concurrent.FluentFuture;
  13 import java.math.BigInteger;
  14 import javax.inject.Inject;
  15 import javax.inject.Singleton;
  16 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  17 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  18 import org.opendaylight.genius.infra.ManagedNewTransactionRunner;
  19 import org.opendaylight.genius.infra.ManagedNewTransactionRunnerImpl;
  20 import org.opendaylight.genius.mdsalutil.cache.DataObjectCache;
  21 import org.opendaylight.infrautils.caches.CacheProvider;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.natservice.rev160111.NeutronVipStates;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.natservice.rev160111.neutron.vip.states.VipState;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.natservice.rev160111.neutron.vip.states.VipStateBuilder;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.natservice.rev160111.neutron.vip.states.VipStateKey;
  26 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  27
  28 @Singleton
  29 public class VipStateTracker extends DataObjectCache<String, VipState> {
  30
  31     private ManagedNewTransactionRunner txRunner;
  32
  33     @Inject
  34     public VipStateTracker(DataBroker dataBroker, CacheProvider cacheProvider) {
  35         super(VipState.class,
  36             dataBroker,
  37             LogicalDatastoreType.OPERATIONAL,
  38             InstanceIdentifier.builder(NeutronVipStates.class).child(VipState.class).build(),
  39             cacheProvider,
  40             (iid, vipState) -> vipState.key().getIp(),
  41             ip -> InstanceIdentifier.builder(NeutronVipStates.class).child(VipState.class,
  42                 new VipStateKey(ip)).build());
  43         this.txRunner = new ManagedNewTransactionRunnerImpl(dataBroker);
  44     }
  45
  46     public VipState buildVipState(String ip, BigInteger dpnId, String ifcName) {
  47         return new VipStateBuilder().setIp(ip).setDpnId(dpnId).setIfcName(ifcName).build();
  48     }
  49
  50     public FluentFuture<Void> writeVipState(VipState vipState) {
  51         return txRunner.callWithNewWriteOnlyTransactionAndSubmit(OPERATIONAL, tx -> {
  52             tx.put(InstanceIdentifier.builder(NeutronVipStates.class)
  53                             .child(VipState.class, vipState.key()).build(),
  54                     vipState, true);
  55         });
  56     }
  57 }