Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Make some document properties extensible #1163

Merged
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
30 changes: 27 additions & 3 deletions esrally/track/track.py
Original file line number Diff line number Diff line change
Expand Up @@ -213,9 +213,9 @@ def __init__(self, source_format, document_file=None, document_archive=None, bas
self.document_archive = document_archive
self.base_url = base_url
self.includes_action_and_meta_data = includes_action_and_meta_data
self.number_of_documents = number_of_documents
self.compressed_size_in_bytes = compressed_size_in_bytes
self.uncompressed_size_in_bytes = uncompressed_size_in_bytes
self._number_of_documents = number_of_documents
self._compressed_size_in_bytes = compressed_size_in_bytes
self._uncompressed_size_in_bytes = uncompressed_size_in_bytes
self.target_index = target_index
self.target_data_stream = target_data_stream
self.target_type = target_type
Expand All @@ -227,6 +227,30 @@ def has_compressed_corpus(self):
def has_uncompressed_corpus(self):
return self.document_file is not None

@property
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

TIL

def number_of_documents(self):
return self._number_of_documents

@number_of_documents.setter
def number_of_documents(self, value):
self._number_of_documents = value

@property
def uncompressed_size_in_bytes(self):
return self._uncompressed_size_in_bytes

@uncompressed_size_in_bytes.setter
def uncompressed_size_in_bytes(self, value):
self._uncompressed_size_in_bytes = value

@property
def compressed_size_in_bytes(self):
return self._compressed_size_in_bytes

@compressed_size_in_bytes.setter
def compressed_size_in_bytes(self, value):
self._compressed_size_in_bytes = value

@property
def number_of_lines(self):
if self.includes_action_and_meta_data:
Expand Down