diff options
Diffstat (limited to 'kafka/codec.py')
-rw-r--r-- | kafka/codec.py | 45 |
1 files changed, 21 insertions, 24 deletions
diff --git a/kafka/codec.py b/kafka/codec.py index cb3858b..35102ca 100644 --- a/kafka/codec.py +++ b/kafka/codec.py @@ -1,51 +1,48 @@ from cStringIO import StringIO import gzip -import logging - -log = logging.getLogger("kafka.codec") try: import snappy - hasSnappy = True + _has_snappy = True except ImportError: - log.warn("Snappy codec not available") - hasSnappy = False + _has_snappy = False + def has_gzip(): return True -def has_snappy(): - return hasSnappy +def _has_snappy(): + return _has_snappy def gzip_encode(payload): - buf = StringIO() - f = gzip.GzipFile(fileobj=buf, mode='w', compresslevel=6) - f.write(payload) - f.close() - buf.seek(0) - out = buf.read() - buf.close() - return out + buffer = StringIO() + handle = gzip.GzipFile(fileobj=buffer, mode="w") + handle.write(payload) + handle.close() + buffer.seek(0) + result = buffer.read() + buffer.close() + return result def gzip_decode(payload): - buf = StringIO(payload) - f = gzip.GzipFile(fileobj=buf, mode='r') - out = f.read() - f.close() - buf.close() - return out + buffer = StringIO(payload) + handle = gzip.GzipFile(fileobj=buffer, mode='r') + result = handle.read() + handle.close() + buffer.close() + return result def snappy_encode(payload): - if not hasSnappy: + if not _has_snappy: raise NotImplementedError("Snappy codec not available") return snappy.compress(payload) def snappy_decode(payload): - if not hasSnappy: + if not _has_snappy: raise NotImplementedError("Snappy codec not available") return snappy.decompress(payload) |