def compress_one_file(self, filetype, force_popen):
base_dir = self.get_test_dir()
write_test_file = os.path.join(base_dir, "write_test." + filetype)
assert not os.path.exists(write_test_file)
c = CompressedFile(write_test_file, mode="w", force_popen=force_popen)
# Write data to file.lzma
lines = self.get_test_data()
for line in lines:
c.write(line + "\n")
c.close()
# Read it back
after = []
c = CompressedFile(write_test_file, mode="r", force_popen=force_popen)
for line in c:
after.append(line.strip())
# make sure it looks ok
self.assertEqual(len(lines), len(after))
for i in range(len(lines)):
self.assertEqual(lines[i], after[i])
# all is well, remove the file.
os.remove(write_test_file)
def compress_one_file(self, filetype, force_popen):
base_dir = self.get_test_dir()
write_test_file = os.path.join(base_dir, "write_test." + filetype)
assert not os.path.exists(write_test_file)
c = CompressedFile(write_test_file, mode="w", force_popen=force_popen)
# Write data to file.lzma
lines = self.get_test_data()
for line in lines:
c.write(line + "\n")
c.close()
# Read it back
after = []
c = CompressedFile(write_test_file, mode="r", force_popen=force_popen)
for line in c:
after.append(line.strip())
# make sure it looks ok
self.assertEqual(len(lines), len(after))
for i in range(len(lines)):
self.assertEqual(lines[i], after[i])
# all is well, remove the file.
os.remove(write_test_file)
def test_compress_from_cleanup(self):
base_dir = self.get_test_dir()
comp_test_file = os.path.join(base_dir, "cleanup_test.gz")
raw_test_file = self.get_raw_test_file()
assert os.path.exists(raw_test_file)
c = CompressedFile(comp_test_file, mode="w")
c.compress_from(raw_test_file, remove_original=True)
c.close()
assert not os.path.exists(raw_test_file)
os.remove(comp_test_file)
def compress_from(self, filetype, force_popen):
base_dir = self.get_test_dir()
raw_test_file = self.get_raw_test_file()
comp_test_file = os.path.join(base_dir, "from_test." + filetype)
assert not os.path.exists(comp_test_file)
c = CompressedFile(comp_test_file, mode="w", force_popen=force_popen)
c.compress_from(raw_test_file)
c.close()
self.verify_contents(comp_test_file, force_popen)
os.remove(comp_test_file)
def test_compress_from_cleanup(self):
base_dir = self.get_test_dir()
comp_test_file = os.path.join(base_dir, "cleanup_test.gz")
raw_test_file = self.get_raw_test_file()
assert os.path.exists(raw_test_file)
c = CompressedFile(comp_test_file, mode="w")
c.compress_from(raw_test_file, remove_original=True)
c.close()
assert not os.path.exists(raw_test_file)
os.remove(comp_test_file)
def compress_from(self, filetype, force_popen):
base_dir = self.get_test_dir()
raw_test_file = self.get_raw_test_file()
comp_test_file = os.path.join(base_dir, "from_test." + filetype)
assert not os.path.exists(comp_test_file)
c = CompressedFile(comp_test_file, mode="w", force_popen=force_popen)
c.compress_from(raw_test_file)
c.close()
self.verify_contents(comp_test_file, force_popen)
os.remove(comp_test_file)
def decompress_one_file(self, filetype, force_popen):
base_dir = self.get_test_dir()
c = CompressedFile(os.path.join(base_dir, "test.txt." + filetype), force_popen=force_popen)
lines = []
for line in c:
lines.append(line.strip())
expected = self.get_test_data()
self.assertEqual(len(expected), c.line_num)
c.close()
for i in range(c.line_num):
self.assertEqual(expected[i], lines[i])
def light_versus_heavy(self, file_type, force_popen):
base_dir = self.get_test_dir()
test_lines = []
for i in range(2000):
test_lines.append("Hello there {0}!".format(i))
test_contents = "\n".join(test_lines)
# Write it with a little compression.
write_light = os.path.join(
base_dir, "clevel_test.1.{}.{}".format(force_popen, file_type))
assert not os.path.exists(write_light)
c = CompressedFile(write_light,
mode="w",
force_popen=force_popen,
compression_level=1)
c.write(test_contents)
c.close()
# Check the resulting size
light_compression_size = os.stat(write_light).st_size
os.remove(write_light)
# Write it with max compression.
write_heavy = os.path.join(
base_dir, "clevel_test.9.{}.{}".format(force_popen, file_type))
assert not os.path.exists(write_heavy)
c = CompressedFile(write_heavy,
mode="w",
force_popen=force_popen,
compression_level=9)
c.write(test_contents)
c.close()
# Check the size again.
heavy_compression_size = os.stat(write_heavy).st_size
os.remove(write_heavy)
common_msg = "size should be less than raw size for type {0} " \
"(popen={1})".format(file_type, force_popen)
self.assertTrue(light_compression_size < len(test_contents),
msg="Lightly Compressed " + common_msg)
self.assertTrue(heavy_compression_size < len(test_contents),
msg="Heavily Compressed " + common_msg)
#print "{0}, popen={1} - raw: {2}, light: {3}, heavy: {4}".format(
# t, popen, len(test_contents), light_compression_size,
# heavy_compression_size)
self.assertTrue(light_compression_size > heavy_compression_size,
msg="Light compression ({0}) should be larger " \
"than heavy compression ({1}) for type {2} " \
"(popen={3})".format(light_compression_size,
heavy_compression_size, file_type, force_popen))
def decompress_one_file(self, filetype, force_popen):
base_dir = self.get_test_dir()
c = CompressedFile(os.path.join(base_dir, "test.txt." + filetype),
force_popen=force_popen)
lines = []
for line in c:
lines.append(line.strip())
expected = self.get_test_data()
self.assertEqual(len(expected), c.line_num)
c.close()
for i in range(c.line_num):
self.assertEqual(expected[i], lines[i])
def handle(self, record):
filename = record
base_ends = filename.find(".log") + 4
if base_ends < 4:
self.log("Bad filename encountered, skipping: " + filename)
self.stats.increment(records_read=1,
bad_records=1,
bad_record_type="bad_filename")
return
basename = filename[0:base_ends]
# Get a unique name for the compressed file:
comp_name = basename + "." + uuid.uuid4(
).hex + StorageLayout.COMPRESSED_SUFFIX
comp_file = CompressedFile(comp_name,
mode="w",
open_now=True,
compression_level=1)
# Rename uncompressed file to a temp name
tmp_name = comp_name + ".compressing"
os.rename(filename, tmp_name)
start = now()
try:
comp_file.compress_from(tmp_name, remove_original=False)
comp_file.close()
except Exception as e:
self.stats.increment(records_read=1,
bad_records=1,
bad_record_type="compression_error")
self.log("Error compressing file {0}: {1}".format(filename, e))
return
raw_bytes = os.stat(tmp_name).st_size
comp_bytes = os.stat(comp_name).st_size
raw_mb = float(raw_bytes) / 1024.0 / 1024.0
comp_mb = float(comp_bytes) / 1024.0 / 1024.0
self.stats.increment(records_read=1,
records_written=1,
bytes_read=raw_bytes,
bytes_written=comp_bytes)
# Remove raw file
os.remove(tmp_name)
sec = timer.delta_sec(start, now())
self.log("Compressed %s as %s in %.2fs. Size before: %.2fMB, after:" \
" %.2fMB (r: %.2fMB/s, w: %.2fMB/s)" % (filename, comp_name,
sec, raw_mb, comp_mb, (raw_mb/sec), (comp_mb/sec)))
def light_versus_heavy(self, file_type, force_popen):
base_dir = self.get_test_dir()
test_lines = []
for i in range(2000):
test_lines.append("Hello there {0}!".format(i))
test_contents = "\n".join(test_lines)
# Write it with a little compression.
write_light = os.path.join(base_dir, "clevel_test.1.{}.{}".format(force_popen, file_type))
assert not os.path.exists(write_light)
c = CompressedFile(write_light, mode="w", force_popen=force_popen, compression_level=1)
c.write(test_contents)
c.close()
# Check the resulting size
light_compression_size = os.stat(write_light).st_size
os.remove(write_light)
# Write it with max compression.
write_heavy = os.path.join(base_dir, "clevel_test.9.{}.{}".format(force_popen, file_type))
assert not os.path.exists(write_heavy)
c = CompressedFile(write_heavy, mode="w", force_popen=force_popen, compression_level=9)
c.write(test_contents)
c.close()
# Check the size again.
heavy_compression_size = os.stat(write_heavy).st_size
os.remove(write_heavy)
common_msg = "size should be less than raw size for type {0} " "(popen={1})".format(file_type, force_popen)
self.assertTrue(light_compression_size < len(test_contents), msg="Lightly Compressed " + common_msg)
self.assertTrue(heavy_compression_size < len(test_contents), msg="Heavily Compressed " + common_msg)
# print "{0}, popen={1} - raw: {2}, light: {3}, heavy: {4}".format(
# t, popen, len(test_contents), light_compression_size,
# heavy_compression_size)
self.assertTrue(
light_compression_size > heavy_compression_size,
msg="Light compression ({0}) should be larger "
"than heavy compression ({1}) for type {2} "
"(popen={3})".format(light_compression_size, heavy_compression_size, file_type, force_popen),
)
def handle(self, record):
filename = record
base_ends = filename.find(".log") + 4
if base_ends < 4:
self.log("Bad filename encountered, skipping: " + filename)
self.stats.increment(records_read=1, bad_records=1,
bad_record_type="bad_filename")
return
basename = filename[0:base_ends]
# Get a unique name for the compressed file:
comp_name = basename + "." + uuid.uuid4().hex + StorageLayout.COMPRESSED_SUFFIX
comp_file = CompressedFile(comp_name, mode="w", open_now=True, compression_level=1)
# Rename uncompressed file to a temp name
tmp_name = comp_name + ".compressing"
os.rename(filename, tmp_name)
start = now()
try:
comp_file.compress_from(tmp_name, remove_original=False)
comp_file.close()
except Exception as e:
self.stats.increment(records_read=1, bad_records=1,
bad_record_type="compression_error")
self.log("Error compressing file {0}: {1}".format(filename, e))
return
raw_bytes = os.stat(tmp_name).st_size
comp_bytes = os.stat(comp_name).st_size
raw_mb = float(raw_bytes) / 1024.0 / 1024.0
comp_mb = float(comp_bytes) / 1024.0 / 1024.0
self.stats.increment(records_read=1, records_written=1,
bytes_read=raw_bytes, bytes_written=comp_bytes)
# Remove raw file
os.remove(tmp_name)
sec = timer.delta_sec(start, now())
self.log("Compressed %s as %s in %.2fs. Size before: %.2fMB, after:" \
" %.2fMB (r: %.2fMB/s, w: %.2fMB/s)" % (filename, comp_name,
sec, raw_mb, comp_mb, (raw_mb/sec), (comp_mb/sec)))
