Skip to content

Commit 8a0b1c3

Browse files
committed
Move method
1 parent 73b51e6 commit 8a0b1c3

File tree

2 files changed

+17
-16
lines changed

2 files changed

+17
-16
lines changed

src/azul/azulclient.py

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -66,6 +66,7 @@
6666
IndexService,
6767
)
6868
from azul.plugins import (
69+
File,
6970
MetadataPlugin,
7071
RepositoryPlugin,
7172
)
@@ -130,6 +131,21 @@ def mirror_source_message(self,
130131
group_id=source.id
131132
)
132133

134+
def mirror_file_message(self,
135+
catalog: CatalogName,
136+
source: SourceRef,
137+
file: File,
138+
) -> SQSFifoMessage:
139+
return SQSFifoMessage(
140+
body={
141+
'action': MirrorAction.mirror_file.to_json(),
142+
'catalog': catalog,
143+
'source': source.to_json(),
144+
'file': file.to_json()
145+
},
146+
group_id=file.digest.value
147+
)
148+
133149
def local_reindex(self, catalog: CatalogName, prefix: str) -> int:
134150
service = self.index_repository_service
135151
plugin = self.repository_plugin(catalog)

src/azul/indexer/mirror_controller.py

Lines changed: 1 addition & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -182,7 +182,7 @@ def messages() -> Iterable[SQSMessage]:
182182
log.info('Not mirroring file to save cost: %r', file)
183183
else:
184184
log.debug('Queueing file %r', file)
185-
yield self.mirror_file_message(catalog, source, file)
185+
yield self.client.mirror_file_message(catalog, source, file)
186186

187187
self.client.queue_mirror_messages(messages())
188188
log.info('Queued %d files in partition %r of source %r in catalog %r',
@@ -294,21 +294,6 @@ def mirror_partition_message(self,
294294
group_id=f'{source.id}:{prefix}'
295295
)
296296

297-
def mirror_file_message(self,
298-
catalog: CatalogName,
299-
source: SourceRef,
300-
file: File,
301-
) -> SQSFifoMessage:
302-
return SQSFifoMessage(
303-
body={
304-
'action': MirrorAction.mirror_file.to_json(),
305-
'catalog': catalog,
306-
'source': source.to_json(),
307-
'file': file.to_json()
308-
},
309-
group_id=file.digest.value
310-
)
311-
312297
def mirror_part_message(self,
313298
catalog: CatalogName,
314299
file: File,

0 commit comments

Comments
 (0)