From f952ffcf74763c5d16677635631441983da2b77d Mon Sep 17 00:00:00 2001 From: hoseinaghaei Date: Thu, 15 Feb 2024 06:28:34 +0330 Subject: [PATCH] change write style --- kafka_server/broker/file/segment.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/kafka_server/broker/file/segment.py b/kafka_server/broker/file/segment.py index 192bb99..6e3fdeb 100644 --- a/kafka_server/broker/file/segment.py +++ b/kafka_server/broker/file/segment.py @@ -34,11 +34,11 @@ def append(self, key: str, value: bytes): data_file_path = os.path.join(segment_path, f'{self.indexer.get_write()}.dat') kb = 1024 - with open(data_file_path, 'wb+') as entry_file: - entry_file.write(f'{key}: '.encode('utf-8')) + with open(data_file_path, 'w') as entry_file: + entry_file.write(f'{key}: ') for i in range(0, len(value), kb): chunk = value[i:i + kb] - entry_file.write(chunk) + entry_file.write(chunk.decode('utf-8')) except Exception as e: print(f"Error appending data to segment: {e}") return False @@ -81,10 +81,10 @@ def read(self): if os.path.exists(data_file_path): # TODO: get lock try: - with open(data_file_path, 'rb') as entry_file: + with open(data_file_path, 'r') as entry_file: data = entry_file.read() - key, value = data.decode('utf-8').split(': ', 1) + key, value = data.split(': ', 1) return key, value except Exception as e: print(f"Error reading file {data_file_path}: {e}")