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 |