2 * Copyright (c) 2015 Brocade Communications Systems, 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
9 package org.opendaylight.controller.cluster.raft.utils;
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 com.google.common.collect.Lists;
19 import java.util.ArrayList;
20 import java.util.Collections;
21 import java.util.Iterator;
22 import java.util.List;
24 import java.util.concurrent.ConcurrentHashMap;
25 import org.slf4j.Logger;
26 import org.slf4j.LoggerFactory;
27 import scala.concurrent.Future;
30 * An akka SnapshotStore implementation that stores data in memory. This is intended for testing.
32 * @author Thomas Pantelis
34 public class InMemorySnapshotStore extends SnapshotStore {
36 static final Logger LOG = LoggerFactory.getLogger(InMemorySnapshotStore.class);
38 private static Map<String, List<StoredSnapshot>> snapshots = new ConcurrentHashMap<>();
40 public static void addSnapshot(String persistentId, Object snapshot) {
41 List<StoredSnapshot> snapshotList = snapshots.get(persistentId);
43 if(snapshotList == null) {
44 snapshotList = new ArrayList<>();
45 snapshots.put(persistentId, snapshotList);
48 synchronized (snapshotList) {
49 snapshotList.add(new StoredSnapshot(new SnapshotMetadata(persistentId, snapshotList.size(),
50 System.currentTimeMillis()), snapshot));
54 @SuppressWarnings("unchecked")
55 public static <T> List<T> getSnapshots(String persistentId, Class<T> type) {
56 List<StoredSnapshot> stored = snapshots.get(persistentId);
58 return Collections.emptyList();
62 synchronized (stored) {
63 retList = Lists.newArrayListWithCapacity(stored.size());
64 for(StoredSnapshot s: stored) {
65 if(type.isInstance(s.getData())) {
66 retList.add((T) s.getData());
74 public static void clear() {
79 public Future<Option<SelectedSnapshot>> doLoadAsync(String s,
80 SnapshotSelectionCriteria snapshotSelectionCriteria) {
81 List<StoredSnapshot> snapshotList = snapshots.get(s);
82 if(snapshotList == null){
83 return Futures.successful(Option.<SelectedSnapshot>none());
86 StoredSnapshot snapshot = Iterables.getLast(snapshotList);
87 SelectedSnapshot selectedSnapshot =
88 new SelectedSnapshot(snapshot.getMetadata(), snapshot.getData());
89 return Futures.successful(Option.some(selectedSnapshot));
93 public Future<Void> doSaveAsync(SnapshotMetadata snapshotMetadata, Object o) {
94 List<StoredSnapshot> snapshotList = snapshots.get(snapshotMetadata.persistenceId());
96 if(snapshotList == null){
97 snapshotList = new ArrayList<>();
98 snapshots.put(snapshotMetadata.persistenceId(), snapshotList);
100 synchronized (snapshotList) {
101 snapshotList.add(new StoredSnapshot(snapshotMetadata, o));
104 return Futures.successful(null);
108 public void onSaved(SnapshotMetadata snapshotMetadata) throws Exception {
112 public void doDelete(SnapshotMetadata snapshotMetadata) throws Exception {
113 List<StoredSnapshot> snapshotList = snapshots.get(snapshotMetadata.persistenceId());
115 if(snapshotList == null){
119 int deleteIndex = -1;
121 synchronized (snapshotList) {
122 for(int i=0;i<snapshotList.size(); i++){
123 StoredSnapshot snapshot = snapshotList.get(i);
124 if(snapshotMetadata.equals(snapshot.getMetadata())){
130 if(deleteIndex != -1){
131 snapshotList.remove(deleteIndex);
137 public void doDelete(String persistentId, SnapshotSelectionCriteria snapshotSelectionCriteria)
139 List<StoredSnapshot> snapshotList = snapshots.get(persistentId);
141 if(snapshotList == null){
145 synchronized (snapshotList) {
146 Iterator<StoredSnapshot> iter = snapshotList.iterator();
147 while(iter.hasNext()) {
148 StoredSnapshot s = iter.next();
149 LOG.trace("doDelete: sequenceNr: {}, maxSequenceNr: {}", s.getMetadata().sequenceNr(),
150 snapshotSelectionCriteria.maxSequenceNr());
152 if(s.getMetadata().sequenceNr() <= snapshotSelectionCriteria.maxSequenceNr()) {
159 private static class StoredSnapshot {
160 private final SnapshotMetadata metadata;
161 private final Object data;
163 private StoredSnapshot(SnapshotMetadata metadata, Object data) {
164 this.metadata = metadata;
168 public SnapshotMetadata getMetadata() {
172 public Object getData() {