Changeset View
Changeset View
Standalone View
Standalone View
swh/storage/in_memory.py
| Show First 20 Lines • Show All 474 Lines • ▼ Show 20 Lines | class InMemoryCqlRunner: | ||||
| def origin_iter_all(self) -> Iterable[OriginRow]: | def origin_iter_all(self) -> Iterable[OriginRow]: | ||||
| return ( | return ( | ||||
| row | row | ||||
| for (token, partition) in self._origins.data.items() | for (token, partition) in self._origins.data.items() | ||||
| for (clustering_key, row) in partition.items() | for (clustering_key, row) in partition.items() | ||||
| ) | ) | ||||
| def origin_bump_next_visit_id(self, origin_url: str, visit_id: int) -> None: | |||||
| origin = list(self.origin_get_by_url(origin_url))[0] | |||||
| origin.next_visit_id = max(origin.next_visit_id, visit_id + 1) | |||||
| def origin_generate_unique_visit_id(self, origin_url: str) -> int: | def origin_generate_unique_visit_id(self, origin_url: str) -> int: | ||||
| origin = list(self.origin_get_by_url(origin_url))[0] | origin = list(self.origin_get_by_url(origin_url))[0] | ||||
| visit_id = origin.next_visit_id | visit_id = origin.next_visit_id | ||||
| origin.next_visit_id += 1 | origin.next_visit_id += 1 | ||||
| return visit_id | return visit_id | ||||
| ########################## | ########################## | ||||
| # 'origin_visit' table | # 'origin_visit' table | ||||
| ▲ Show 20 Lines • Show All 244 Lines • Show Last 20 Lines | |||||