File tree Expand file tree Collapse file tree 2 files changed +17
-16
lines changed Expand file tree Collapse file tree 2 files changed +17
-16
lines changed Original file line number Diff line number Diff line change 6666 IndexService ,
6767)
6868from 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 )
Original file line number Diff line number Diff 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 ,
You can’t perform that action at this time.
0 commit comments