2 * Copyright (c) 2014 Cisco 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.datastore.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 scala.concurrent.Future;
19 import java.util.ArrayList;
20 import java.util.List;
22 import java.util.concurrent.ConcurrentHashMap;
23 import org.opendaylight.controller.cluster.raft.Snapshot;
25 public class InMemorySnapshotStore extends SnapshotStore {
27 private static Map<String, List<StoredSnapshot>> snapshots = new ConcurrentHashMap<>();
29 public static void addSnapshot(String persistentId, Snapshot snapshot) {
30 List<StoredSnapshot> snapshotList = snapshots.get(persistentId);
32 if(snapshotList == null) {
33 snapshotList = new ArrayList<>();
34 snapshots.put(persistentId, snapshotList);
37 snapshotList.add(new StoredSnapshot(new SnapshotMetadata(persistentId, snapshotList.size(),
38 System.currentTimeMillis()), snapshot));
41 public static void clear() {
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());
53 StoredSnapshot snapshot = Iterables.getLast(snapshotList);
54 SelectedSnapshot selectedSnapshot =
55 new SelectedSnapshot(snapshot.getMetadata(), snapshot.getData());
56 return Futures.successful(Option.some(selectedSnapshot));
60 public Future<Void> doSaveAsync(SnapshotMetadata snapshotMetadata, Object o) {
61 List<StoredSnapshot> snapshotList = snapshots.get(snapshotMetadata.persistenceId());
63 if(snapshotList == null){
64 snapshotList = new ArrayList<>();
65 snapshots.put(snapshotMetadata.persistenceId(), snapshotList);
67 snapshotList.add(new StoredSnapshot(snapshotMetadata, o));
69 return Futures.successful(null);
73 public void onSaved(SnapshotMetadata snapshotMetadata) throws Exception {
77 public void doDelete(SnapshotMetadata snapshotMetadata) throws Exception {
78 List<StoredSnapshot> snapshotList = snapshots.get(snapshotMetadata.persistenceId());
80 if(snapshotList == null){
86 for(int i=0;i<snapshotList.size(); i++){
87 StoredSnapshot snapshot = snapshotList.get(i);
88 if(snapshotMetadata.equals(snapshot.getMetadata())){
94 if(deleteIndex != -1){
95 snapshotList.remove(deleteIndex);
101 public void doDelete(String s, SnapshotSelectionCriteria snapshotSelectionCriteria)
103 List<StoredSnapshot> snapshotList = snapshots.get(s);
105 if(snapshotList == null){
109 // TODO : This is a quick and dirty implementation. Do actual match later.
110 snapshotList.clear();
114 private static class StoredSnapshot {
115 private final SnapshotMetadata metadata;
116 private final Object data;
118 private StoredSnapshot(SnapshotMetadata metadata, Object data) {
119 this.metadata = metadata;
123 public SnapshotMetadata getMetadata() {
127 public Object getData() {