Changeset View
Changeset View
Standalone View
Standalone View
swh/storage/in_memory.py
| Show All 28 Lines | def now(): | ||||
| return datetime.datetime.now(tz=datetime.timezone.utc) | return datetime.datetime.now(tz=datetime.timezone.utc) | ||||
| class Storage: | class Storage: | ||||
| def __init__(self, journal_writer=None): | def __init__(self, journal_writer=None): | ||||
| self._contents = {} | self._contents = {} | ||||
| self._content_indexes = defaultdict(lambda: defaultdict(set)) | self._content_indexes = defaultdict(lambda: defaultdict(set)) | ||||
| self.reset() | |||||
| if journal_writer: | |||||
| self.journal_writer = get_journal_writer(**journal_writer) | |||||
| else: | |||||
| self.journal_writer = None | |||||
| def reset(self): | |||||
| self._directories = {} | self._directories = {} | ||||
| self._revisions = {} | self._revisions = {} | ||||
| self._releases = {} | self._releases = {} | ||||
| self._snapshots = {} | self._snapshots = {} | ||||
| self._origins = [] | self._origins = [] | ||||
| self._origin_visits = [] | self._origin_visits = [] | ||||
| self._persons = [] | self._persons = [] | ||||
| self._origin_metadata = defaultdict(list) | self._origin_metadata = defaultdict(list) | ||||
| self._tools = {} | self._tools = {} | ||||
| self._metadata_providers = {} | self._metadata_providers = {} | ||||
| self._objects = defaultdict(list) | self._objects = defaultdict(list) | ||||
| # ideally we would want a skip list for both fast inserts and searches | # ideally we would want a skip list for both fast inserts and searches | ||||
| self._sorted_sha1s = [] | self._sorted_sha1s = [] | ||||
| self.objstorage = get_objstorage('memory', {}) | self.objstorage = get_objstorage('memory', {}) | ||||
| if journal_writer: | |||||
| self.journal_writer = get_journal_writer(**journal_writer) | |||||
| else: | |||||
| self.journal_writer = None | |||||
| def check_config(self, *, check_write): | def check_config(self, *, check_write): | ||||
| """Check that the storage is configured and ready to go.""" | """Check that the storage is configured and ready to go.""" | ||||
| return True | return True | ||||
| def _content_add(self, contents, with_data): | def _content_add(self, contents, with_data): | ||||
| if self.journal_writer: | if self.journal_writer: | ||||
| for content in contents: | for content in contents: | ||||
| ▲ Show 20 Lines • Show All 1,547 Lines • Show Last 20 Lines | |||||