def test_base(self):
ldb = LooseObjectDB(fixture_path("../../../.git/objects"))
for sha1 in ldb.sha_iter():
oinfo = ldb.info(sha1)
ostream = ldb.stream(sha1)
assert oinfo[:3] == ostream[:3]
assert len(ostream.read()) == ostream.size
assert ldb.has_object(oinfo.binsha)
# END for each sha in database
# assure we close all files
try:
del(ostream)
del(oinfo)
except UnboundLocalError:
pass
# END ignore exception if there are no loose objects
data = "my data"
istream = IStream("blob", len(data), StringIO(data))
# the object does not yet have a sha
assert istream.binsha is None
ldb.store(istream)
# now the sha is set
assert len(istream.binsha) == 20
assert ldb.has_object(istream.binsha)
# async operation
# Create a reader from an iterator
reader = IteratorReader(ldb.sha_iter())
# get reader for object streams
info_reader = ldb.stream_async(reader)
# read one
info = info_reader.read(1)[0]
# read all the rest until depletion
ostreams = info_reader.read()
# set the pool to use two threads
pool.set_size(2)
# synchronize the mode of operation
pool.set_size(0)
def test_large_data_streaming(self, path):
ldb = LooseObjectDB(path)
string_ios = list() # list of streams we previously created
# serial mode
for randomize in range(2):
desc = (randomize and 'random ') or ''
print >> sys.stderr, "Creating %s data ..." % desc
st = time()
size, stream = make_memory_file(self.large_data_size_bytes, randomize)
elapsed = time() - st
print >> sys.stderr, "Done (in %f s)" % elapsed
string_ios.append(stream)
# writing - due to the compression it will seem faster than it is
st = time()
sha = ldb.store(IStream('blob', size, stream)).binsha
elapsed_add = time() - st
assert ldb.has_object(sha)
db_file = ldb.readable_db_object_path(bin_to_hex(sha))
fsize_kib = os.path.getsize(db_file) / 1000
size_kib = size / 1000
print >> sys.stderr, "Added %i KiB (filesize = %i KiB) of %s data to loose odb in %f s ( %f Write KiB / s)" % (size_kib, fsize_kib, desc, elapsed_add, size_kib / elapsed_add)
# reading all at once
st = time()
ostream = ldb.stream(sha)
shadata = ostream.read()
elapsed_readall = time() - st
stream.seek(0)
assert shadata == stream.getvalue()
print >> sys.stderr, "Read %i KiB of %s data at once from loose odb in %f s ( %f Read KiB / s)" % (size_kib, desc, elapsed_readall, size_kib / elapsed_readall)
# reading in chunks of 1 MiB
cs = 512*1000
chunks = list()
st = time()
ostream = ldb.stream(sha)
while True:
data = ostream.read(cs)
chunks.append(data)
if len(data) < cs:
break
# END read in chunks
elapsed_readchunks = time() - st
stream.seek(0)
assert ''.join(chunks) == stream.getvalue()
cs_kib = cs / 1000
print >> sys.stderr, "Read %i KiB of %s data in %i KiB chunks from loose odb in %f s ( %f Read KiB / s)" % (size_kib, desc, cs_kib, elapsed_readchunks, size_kib / elapsed_readchunks)
# del db file so we keep something to do
os.remove(db_file)
# END for each randomization factor
# multi-threaded mode
# want two, should be supported by most of todays cpus
pool.set_size(2)
total_kib = 0
nsios = len(string_ios)
for stream in string_ios:
stream.seek(0)
total_kib += len(stream.getvalue()) / 1000
# END rewind
def istream_iter():
for stream in string_ios:
stream.seek(0)
yield IStream(str_blob_type, len(stream.getvalue()), stream)
# END for each stream
# END util
# write multiple objects at once, involving concurrent compression
reader = IteratorReader(istream_iter())
istream_reader = ldb.store_async(reader)
istream_reader.task().max_chunksize = 1
st = time()
istreams = istream_reader.read(nsios)
assert len(istreams) == nsios
elapsed = time() - st
print >> sys.stderr, "Threads(%i): Compressed %i KiB of data in loose odb in %f s ( %f Write KiB / s)" % (pool.size(), total_kib, elapsed, total_kib / elapsed)
# decompress multiple at once, by reading them
# chunk size is not important as the stream will not really be decompressed
# until its read
istream_reader = IteratorReader(iter([ i.binsha for i in istreams ]))
ostream_reader = ldb.stream_async(istream_reader)
chunk_task = TestStreamReader(ostream_reader, "chunker", None)
output_reader = pool.add_task(chunk_task)
output_reader.task().max_chunksize = 1
st = time()
assert len(output_reader.read(nsios)) == nsios
elapsed = time() - st
print >> sys.stderr, "Threads(%i): Decompressed %i KiB of data in loose odb in %f s ( %f Read KiB / s)" % (pool.size(), total_kib, elapsed, total_kib / elapsed)
# store the files, and read them back. For the reading, we use a task
# as well which is chunked into one item per task. Reading all will
# very quickly result in two threads handling two bytestreams of
# chained compression/decompression streams
reader = IteratorReader(istream_iter())
istream_reader = ldb.store_async(reader)
istream_reader.task().max_chunksize = 1
istream_to_sha = lambda items: [ i.binsha for i in items ]
istream_reader.set_post_cb(istream_to_sha)
ostream_reader = ldb.stream_async(istream_reader)
chunk_task = TestStreamReader(ostream_reader, "chunker", None)
output_reader = pool.add_task(chunk_task)
output_reader.max_chunksize = 1
st = time()
assert len(output_reader.read(nsios)) == nsios
elapsed = time() - st
print >> sys.stderr, "Threads(%i): Compressed and decompressed and read %i KiB of data in loose odb in %f s ( %f Combined KiB / s)" % (pool.size(), total_kib, elapsed, total_kib / elapsed)
def test_large_data_streaming(self, path):
ldb = LooseObjectDB(path)
string_ios = list() # list of streams we previously created
# serial mode
for randomize in range(2):
desc = (randomize and 'random ') or ''
print >> sys.stderr, "Creating %s data ..." % desc
st = time()
size, stream = make_memory_file(self.large_data_size_bytes,
randomize)
elapsed = time() - st
print >> sys.stderr, "Done (in %f s)" % elapsed
string_ios.append(stream)
# writing - due to the compression it will seem faster than it is
st = time()
sha = ldb.store(IStream('blob', size, stream)).binsha
elapsed_add = time() - st
assert ldb.has_object(sha)
db_file = ldb.readable_db_object_path(bin_to_hex(sha))
fsize_kib = os.path.getsize(db_file) / 1000
size_kib = size / 1000
print >> sys.stderr, "Added %i KiB (filesize = %i KiB) of %s data to loose odb in %f s ( %f Write KiB / s)" % (
size_kib, fsize_kib, desc, elapsed_add, size_kib / elapsed_add)
# reading all at once
st = time()
ostream = ldb.stream(sha)
shadata = ostream.read()
elapsed_readall = time() - st
stream.seek(0)
assert shadata == stream.getvalue()
print >> sys.stderr, "Read %i KiB of %s data at once from loose odb in %f s ( %f Read KiB / s)" % (
size_kib, desc, elapsed_readall, size_kib / elapsed_readall)
# reading in chunks of 1 MiB
cs = 512 * 1000
chunks = list()
st = time()
ostream = ldb.stream(sha)
while True:
data = ostream.read(cs)
chunks.append(data)
if len(data) < cs:
break
# END read in chunks
elapsed_readchunks = time() - st
stream.seek(0)
assert ''.join(chunks) == stream.getvalue()
cs_kib = cs / 1000
print >> sys.stderr, "Read %i KiB of %s data in %i KiB chunks from loose odb in %f s ( %f Read KiB / s)" % (
size_kib, desc, cs_kib, elapsed_readchunks,
size_kib / elapsed_readchunks)
# del db file so we keep something to do
os.remove(db_file)
# END for each randomization factor
# multi-threaded mode
# want two, should be supported by most of todays cpus
pool.set_size(2)
total_kib = 0
nsios = len(string_ios)
for stream in string_ios:
stream.seek(0)
total_kib += len(stream.getvalue()) / 1000
# END rewind
def istream_iter():
for stream in string_ios:
stream.seek(0)
yield IStream(str_blob_type, len(stream.getvalue()), stream)
# END for each stream
# END util
# write multiple objects at once, involving concurrent compression
reader = IteratorReader(istream_iter())
istream_reader = ldb.store_async(reader)
istream_reader.task().max_chunksize = 1
st = time()
istreams = istream_reader.read(nsios)
assert len(istreams) == nsios
elapsed = time() - st
print >> sys.stderr, "Threads(%i): Compressed %i KiB of data in loose odb in %f s ( %f Write KiB / s)" % (
pool.size(), total_kib, elapsed, total_kib / elapsed)
# decompress multiple at once, by reading them
# chunk size is not important as the stream will not really be decompressed
# until its read
istream_reader = IteratorReader(iter([i.binsha for i in istreams]))
ostream_reader = ldb.stream_async(istream_reader)
chunk_task = TestStreamReader(ostream_reader, "chunker", None)
output_reader = pool.add_task(chunk_task)
output_reader.task().max_chunksize = 1
st = time()
assert len(output_reader.read(nsios)) == nsios
elapsed = time() - st
print >> sys.stderr, "Threads(%i): Decompressed %i KiB of data in loose odb in %f s ( %f Read KiB / s)" % (
pool.size(), total_kib, elapsed, total_kib / elapsed)
# store the files, and read them back. For the reading, we use a task
# as well which is chunked into one item per task. Reading all will
# very quickly result in two threads handling two bytestreams of
# chained compression/decompression streams
reader = IteratorReader(istream_iter())
istream_reader = ldb.store_async(reader)
istream_reader.task().max_chunksize = 1
istream_to_sha = lambda items: [i.binsha for i in items]
istream_reader.set_post_cb(istream_to_sha)
ostream_reader = ldb.stream_async(istream_reader)
chunk_task = TestStreamReader(ostream_reader, "chunker", None)
output_reader = pool.add_task(chunk_task)
output_reader.max_chunksize = 1
st = time()
assert len(output_reader.read(nsios)) == nsios
elapsed = time() - st
print >> sys.stderr, "Threads(%i): Compressed and decompressed and read %i KiB of data in loose odb in %f s ( %f Combined KiB / s)" % (
pool.size(), total_kib, elapsed, total_kib / elapsed)
