| pulkit |
| hg-reviewers |
This prevents the worker subprocesses from contacting the server individually,
which is either inefficient, or leads to problems if the connection is shared
among them.
| No Linters Available |
| No Unit Test Coverage |
| Path | Packages | |||
|---|---|---|---|---|
| M | hgext/fix.py (28 lines) |
| Commit | Parents | Author | Summary | Date |
|---|---|---|---|---|
| bec4def8b8a8 | bf468cf746eb | Rodrigo Damazio Bovendorp | Jul 9 2020, 11:46 PM |
| revstofix = getrevstofix(ui, repo, opts) | revstofix = getrevstofix(ui, repo, opts) | ||||
| basectxs = getbasectxs(repo, opts, revstofix) | basectxs = getbasectxs(repo, opts, revstofix) | ||||
| workqueue, numitems = getworkqueue( | workqueue, numitems = getworkqueue( | ||||
| ui, repo, pats, opts, revstofix, basectxs | ui, repo, pats, opts, revstofix, basectxs | ||||
| ) | ) | ||||
| basepaths = getbasepaths(repo, opts, workqueue, basectxs) | basepaths = getbasepaths(repo, opts, workqueue, basectxs) | ||||
| fixers = getfixers(ui) | fixers = getfixers(ui) | ||||
| # Rather than letting each worker independently fetch the files | |||||
| # (which also would add complications for shared/keepalive | |||||
| # connections), prefetch them all first. | |||||
| _prefetchfiles(repo, workqueue, basepaths) | |||||
| # There are no data dependencies between the workers fixing each file | # There are no data dependencies between the workers fixing each file | ||||
| # revision, so we can use all available parallelism. | # revision, so we can use all available parallelism. | ||||
| def getfixes(items): | def getfixes(items): | ||||
| for rev, path in items: | for rev, path in items: | ||||
| ctx = repo[rev] | ctx = repo[rev] | ||||
| olddata = ctx[path].data() | olddata = ctx[path].data() | ||||
| metadata, newdata = fixfile( | metadata, newdata = fixfile( | ||||
| ui, repo, opts, fixers, ctx, path, basepaths, basectxs[rev] | ui, repo, opts, fixers, ctx, path, basepaths, basectxs[rev] | ||||
| for pctx in ctx.parents(): | for pctx in ctx.parents(): | ||||
| if pctx.rev() in basectxs: | if pctx.rev() in basectxs: | ||||
| basectxs[rev].update(basectxs[pctx.rev()]) | basectxs[rev].update(basectxs[pctx.rev()]) | ||||
| else: | else: | ||||
| basectxs[rev].add(pctx) | basectxs[rev].add(pctx) | ||||
| return basectxs | return basectxs | ||||
| def _prefetchfiles(repo, workqueue, basepaths): | |||||
| toprefetch = set() | |||||
| # Prefetch the files that will be fixed. | |||||
| for rev, path in workqueue: | |||||
| if rev == wdirrev: | |||||
| continue | |||||
| toprefetch.add((rev, path)) | |||||
| # Prefetch the base contents for lineranges(). | |||||
| for (baserev, fixrev, path), basepath in basepaths.items(): | |||||
| toprefetch.add((baserev, basepath)) | |||||
| if toprefetch: | |||||
| scmutil.prefetchfiles( | |||||
| repo, | |||||
| [ | |||||
| (rev, scmutil.matchfiles(repo, [path])) | |||||
| for rev, path in toprefetch | |||||
| ], | |||||
| ) | |||||
| def fixfile(ui, repo, opts, fixers, fixctx, path, basepaths, basectxs): | def fixfile(ui, repo, opts, fixers, fixctx, path, basepaths, basectxs): | ||||
| """Run any configured fixers that should affect the file in this context | """Run any configured fixers that should affect the file in this context | ||||
| Returns the file content that results from applying the fixers in some order | Returns the file content that results from applying the fixers in some order | ||||
| starting with the file's content in the fixctx. Fixers that support line | starting with the file's content in the fixctx. Fixers that support line | ||||
| ranges will affect lines that have changed relative to any of the basectxs | ranges will affect lines that have changed relative to any of the basectxs | ||||
| (i.e. they will only avoid lines that are common to all basectxs). | (i.e. they will only avoid lines that are common to all basectxs). | ||||