要求
写一个kv store,需要实现按版本查找的method,简单的每次take新snapshot,就把map复制一次,放到list里面,这样按版本找可以实现O(1)。如果想省空间可以次snapshot只存diff。查的时候要把每个snapshot都查一次,就变成O(num_snapshots)。
Python 实现
class kvstore:
def __init__(self):
self.store = {}
self.snapshots = []
self.changed_keys = set()
def set(self, k, v):
self.store[k] = v
self.changed_keys.add(k)
def get(self, k):
return self.store.get(k, None)
def snapshot(self):
d = {}
for k in self.changed_keys:
d[k] = self.store[k]
self.snapshots.append(d)
self.changed_keys = set()
return len(self.snapshots) - 1
def get_from_snap(self, snap_version, key):
for version in range(snap_version, -1, -1):
if key in self.snapshots[version]:
return self.snapshots[version][key]
return None
store = kvstore()
store.set("k1", "v1")
store.set("k2", "v2")
assert store.get("k1") == "v1"
last_snapshot = store.snapshot()
assert last_snapshot == 0
assert store.get("k1") == "v1"
assert store.get_from_snap(last_snapshot, "k1") == "v1"
assert store.get_from_snap(last_snapshot, "k2") == "v2"
assert store.get_from_snap(last_snapshot, "k3") == None
last_snapshot = store.snapshot()
assert last_snapshot == 1
assert store.get("k1") == "v1"
assert store.get_from_snap(last_snapshot, "k1") == "v1"
assert store.get_from_snap(last_snapshot, "k2") == "v2"
assert store.get_from_snap(last_snapshot, "k3") == None
store.set("k3", "v3")
store.set("k2", "k2-1")
last_snapshot = store.snapshot()
assert last_snapshot == 2
assert store.get("k1") == "v1"
assert store.get_from_snap(last_snapshot, "k1") == "v1"
assert store.get_from_snap(last_snapshot, "k2") == "k2-1"
assert store.get_from_snap(last_snapshot, "k3") == "v3"
