opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/datastore/utils/InMemorySnapshotStore.java

   1 /*
   2  * Copyright (c) 2014 Cisco Systems, 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
   9 package org.opendaylight.controller.cluster.datastore.utils;
  10
  11 import akka.dispatch.Futures;
  12 import akka.japi.Option;
  13 import akka.persistence.SelectedSnapshot;
  14 import akka.persistence.SnapshotMetadata;
  15 import akka.persistence.SnapshotSelectionCriteria;
  16 import akka.persistence.snapshot.japi.SnapshotStore;
  17 import com.google.common.collect.Iterables;
  18 import scala.concurrent.Future;
  19 import java.util.ArrayList;
  20 import java.util.List;
  21 import java.util.Map;
  22 import java.util.concurrent.ConcurrentHashMap;
  23 import org.opendaylight.controller.cluster.raft.Snapshot;
  24
  25 public class InMemorySnapshotStore extends SnapshotStore {
  26
  27     private static Map<String, List<StoredSnapshot>> snapshots = new ConcurrentHashMap<>();
  28
  29     public static void addSnapshot(String persistentId, Snapshot snapshot) {
  30         List<StoredSnapshot> snapshotList = snapshots.get(persistentId);
  31
  32         if(snapshotList == null) {
  33             snapshotList = new ArrayList<>();
  34             snapshots.put(persistentId, snapshotList);
  35         }
  36
  37         snapshotList.add(new StoredSnapshot(new SnapshotMetadata(persistentId, snapshotList.size(),
  38                 System.currentTimeMillis()), snapshot));
  39     }
  40
  41     public static void clear() {
  42         snapshots.clear();
  43     }
  44
  45     @Override
  46     public Future<Option<SelectedSnapshot>> doLoadAsync(String s,
  47         SnapshotSelectionCriteria snapshotSelectionCriteria) {
  48         List<StoredSnapshot> snapshotList = snapshots.get(s);
  49         if(snapshotList == null){
  50             return Futures.successful(Option.<SelectedSnapshot>none());
  51         }
  52
  53         StoredSnapshot snapshot = Iterables.getLast(snapshotList);
  54         SelectedSnapshot selectedSnapshot =
  55             new SelectedSnapshot(snapshot.getMetadata(), snapshot.getData());
  56         return Futures.successful(Option.some(selectedSnapshot));
  57     }
  58
  59     @Override
  60     public Future<Void> doSaveAsync(SnapshotMetadata snapshotMetadata, Object o) {
  61         List<StoredSnapshot> snapshotList = snapshots.get(snapshotMetadata.persistenceId());
  62
  63         if(snapshotList == null){
  64             snapshotList = new ArrayList<>();
  65             snapshots.put(snapshotMetadata.persistenceId(), snapshotList);
  66         }
  67         snapshotList.add(new StoredSnapshot(snapshotMetadata, o));
  68
  69         return Futures.successful(null);
  70     }
  71
  72     @Override
  73     public void onSaved(SnapshotMetadata snapshotMetadata) throws Exception {
  74     }
  75
  76     @Override
  77     public void doDelete(SnapshotMetadata snapshotMetadata) throws Exception {
  78         List<StoredSnapshot> snapshotList = snapshots.get(snapshotMetadata.persistenceId());
  79
  80         if(snapshotList == null){
  81             return;
  82         }
  83
  84         int deleteIndex = -1;
  85
  86         for(int i=0;i<snapshotList.size(); i++){
  87             StoredSnapshot snapshot = snapshotList.get(i);
  88             if(snapshotMetadata.equals(snapshot.getMetadata())){
  89                 deleteIndex = i;
  90                 break;
  91             }
  92         }
  93
  94         if(deleteIndex != -1){
  95             snapshotList.remove(deleteIndex);
  96         }
  97
  98     }
  99
 100     @Override
 101     public void doDelete(String s, SnapshotSelectionCriteria snapshotSelectionCriteria)
 102         throws Exception {
 103         List<StoredSnapshot> snapshotList = snapshots.get(s);
 104
 105         if(snapshotList == null){
 106             return;
 107         }
 108
 109         // TODO : This is a quick and dirty implementation. Do actual match later.
 110         snapshotList.clear();
 111         snapshots.remove(s);
 112     }
 113
 114     private static class StoredSnapshot {
 115         private final SnapshotMetadata metadata;
 116         private final Object data;
 117
 118         private StoredSnapshot(SnapshotMetadata metadata, Object data) {
 119             this.metadata = metadata;
 120             this.data = data;
 121         }
 122
 123         public SnapshotMetadata getMetadata() {
 124             return metadata;
 125         }
 126
 127         public Object getData() {
 128             return data;
 129         }
 130     }
 131 }