bundle2: store changeset count when creating file bundles The bundle2 changegroup part has an advisory param saying how many changesets are in the part. Before this patch, we were setting this part when generating bundle2 parts via the wire protocol but not when generating local bundle2 files. A side effect of not setting the changeset count part is that progress bars don't work when applying changesets. As the tests show, this impacted clone bundles, shelve, backup bundles, `hg unbundle`, and anything touching bundle2 files. This patch adds a backdoor to allow us to pass state from changegroup generation into the unbundler. We store the number of changesets in the changegroup in this state and use it to populate the aforementioned advisory part parameter when generating the bundle2 bundle. I concede that I'm not thrilled by how state is being passed in changegroup.py (it feels a bit hacky). I would love to overhaul the rather confusing set of functions in changegroup.py with something that passes rich objects around instead of e.g. low-level generators. However, given the code freeze for 3.9 is imminent, I'd rather not undertake this endeavor right now. This feels like the easiest way to get the parameter added to the changegroup part.

     1

#require serve

     2

     3

  $ hg init server

     4

  $ cd server

     5

  $ cat >> .hg/hgrc << EOF

     6

  > [extensions]

     7

  > strip=

     8

  > EOF

     9

    10

  $ echo 1 > foo

    11

  $ hg commit -A -m 'first'

    12

  adding foo

    13

  $ echo 2 > bar

    14

  $ hg commit -A -m 'second'

    15

  adding bar

    16

    17

Produce a bundle to use

    18

    19

  $ hg strip -r 1

    20

  0 files updated, 0 files merged, 1 files removed, 0 files unresolved

    21

  saved backup bundle to $TESTTMP/server/.hg/strip-backup/ed602e697e0f-cc9fff6a-backup.hg (glob)

    22

    23

Serve from a bundle file

    24

    25

  $ hg serve -R .hg/strip-backup/ed602e697e0f-cc9fff6a-backup.hg -d -p $HGPORT --pid-file=hg.pid

    26

  $ cat hg.pid >> $DAEMON_PIDS

    27

    28

Ensure we're serving from the bundle

    29

    30

  $ (get-with-headers.py localhost:$HGPORT 'file/tip/?style=raw')

    31

  200 Script output follows

    32

    33

    34

  -rw-r--r-- 2 bar

    35

  -rw-r--r-- 2 foo

    36

    37