50 lines
1.1 KiB
Python
50 lines
1.1 KiB
Python
"""
|
|
Duplicate page content database.
|
|
"""
|
|
|
|
class DupesOnDisk(object):
|
|
def __init__(self, filename):
|
|
import lmdb
|
|
# TODO: lmdb needs a sparse file; fail early instead of using 1TB
|
|
# of disk on filesystems with no sparse file support
|
|
for map_size in (1024*1024*1024*1024, 2**31-1):
|
|
try:
|
|
self._env = lmdb.open(
|
|
filename,
|
|
writemap=True,
|
|
sync=False,
|
|
metasync=False,
|
|
# http://lmdb.readthedocs.org/en/release/#lmdb.Environment
|
|
map_size=map_size)
|
|
except OverflowError:
|
|
pass
|
|
else:
|
|
break
|
|
|
|
def get_old_url(self, digest):
|
|
with self._env.begin() as txn:
|
|
maybe_url = txn.get(digest)
|
|
if maybe_url is None:
|
|
return maybe_url
|
|
return maybe_url.decode('utf-8')
|
|
|
|
def set_old_url(self, digest, url):
|
|
with self._env.begin(write=True) as txn:
|
|
return txn.put(digest, url.encode("utf-8"))
|
|
|
|
|
|
class DupesInMemory(object):
|
|
def __init__(self):
|
|
self._digests = {}
|
|
|
|
def get_old_url(self, digest):
|
|
return self._digests.get(digest)
|
|
|
|
def set_old_url(self, digest, url):
|
|
self._digests[digest] = url
|
|
|
|
|
|
__all__ = [
|
|
'DupesOnDisk', 'DupesInMemory'
|
|
]
|