import stew/ranges, tables, sets, eth/trie/db type CachingDB* = ref object of RootObj backing: TrieDatabaseRef changed: Table[seq[byte], seq[byte]] deleted: HashSet[seq[byte]] proc newCachingDB*(backing: TrieDatabaseRef): CachingDB = result.new() result.backing = backing result.changed = initTable[seq[byte], seq[byte]]() result.deleted = initSet[seq[byte]]() proc get*(db: CachingDB, key: openarray[byte]): seq[byte] = let key = @key result = db.changed.getOrDefault(key) if result.len == 0 and key notin db.deleted: result = db.backing.get(key) proc put*(db: CachingDB, key, value: openarray[byte]) = let key = @key db.deleted.excl(key) db.changed[key] = @value proc contains*(db: CachingDB, key: openarray[byte]): bool = let key = @key result = key in db.changed if not result and key notin db.deleted: result = db.backing.contains(key) proc del*(db: CachingDB, key: openarray[byte]) = let key = @key db.changed.del(key) db.deleted.incl(key) proc commit*(db: CachingDB) = for k in db.deleted: db.backing.del(k) for k, v in db.changed: db.backing.put(k, v)