cmd/midx-cmd.py

   1 #!/usr/bin/env python
   2 import sys, math, struct, glob, resource
   3 import tempfile
   4 from bup import options, git, _helpers
   5 from bup.helpers import *
   6
   7 PAGE_SIZE=4096
   8 SHA_PER_PAGE=PAGE_SIZE/20.
   9
  10 optspec = """
  11 bup midx [options...] <idxnames...>
  12 --
  13 o,output=  output midx filename (default: auto-generated)
  14 a,auto     automatically use all existing .midx/.idx files as input
  15 f,force    merge produce exactly one .midx containing all objects
  16 p,print    print names of generated midx files
  17 max-files= maximum number of idx files to open at once [-1]
  18 dir=       directory containing idx/midx files
  19 """
  20
  21 def _group(l, count):
  22     for i in xrange(0, len(l), count):
  23         yield l[i:i+count]
  24
  25
  26 def max_files():
  27     mf = min(resource.getrlimit(resource.RLIMIT_NOFILE))
  28     if mf > 32:
  29         mf -= 20  # just a safety margin
  30     else:
  31         mf -= 6   # minimum safety margin
  32     return mf
  33
  34 merge_into = _helpers.merge_into
  35
  36 _first = None
  37 def _do_midx(outdir, outfilename, infilenames, prefixstr):
  38     global _first
  39     if not outfilename:
  40         assert(outdir)
  41         sum = Sha1('\0'.join(infilenames)).hexdigest()
  42         outfilename = '%s/midx-%s.midx' % (outdir, sum)
  43
  44     inp = []
  45     total = 0
  46     allfilenames = []
  47     for name in infilenames:
  48         ix = git.open_idx(name)
  49         inp.append((
  50             ix.map,
  51             len(ix),
  52             ix.sha_ofs,
  53             isinstance(ix, git.PackMidx) and ix.which_ofs or 0,
  54             len(allfilenames),
  55         ))
  56         for n in ix.idxnames:
  57             allfilenames.append(os.path.basename(n))
  58         total += len(ix)
  59     inp.sort(lambda x,y: cmp(str(y[0][y[2]:y[2]+20]),str(x[0][x[2]:x[2]+20])))
  60
  61     if not _first: _first = outdir
  62     dirprefix = (_first != outdir) and git.repo_rel(outdir)+': ' or ''
  63     log('midx: %s%screating from %d files (%d objects).\n'
  64         % (dirprefix, prefixstr, len(infilenames), total))
  65     if (not opt.force and (total < 1024 and len(infilenames) < 3)) \
  66        or len(infilenames) < 2 \
  67        or (opt.force and not total):
  68         debug1('midx: nothing to do.\n')
  69         return
  70
  71     pages = int(total/SHA_PER_PAGE) or 1
  72     bits = int(math.ceil(math.log(pages, 2)))
  73     entries = 2**bits
  74     debug1('midx: table size: %d (%d bits)\n' % (entries*4, bits))
  75
  76     try:
  77         os.unlink(outfilename)
  78     except OSError:
  79         pass
  80     f = open(outfilename + '.tmp', 'w+b')
  81     f.write('MIDX')
  82     f.write(struct.pack('!II', git.MIDX_VERSION, bits))
  83     assert(f.tell() == 12)
  84
  85     f.truncate(12 + 4*entries + 20*total + 4*total)
  86
  87     fmap = mmap_readwrite(f, close=False)
  88
  89     count = merge_into(fmap, bits, total, inp)
  90     fmap.flush()
  91     fmap.close()
  92
  93     f.seek(0, git.SEEK_END)
  94     f.write('\0'.join(allfilenames))
  95     f.close()
  96     os.rename(outfilename + '.tmp', outfilename)
  97
  98     # this is just for testing
  99     if 0:
 100         p = git.PackMidx(outfilename)
 101         assert(len(p.idxnames) == len(infilenames))
 102         print p.idxnames
 103         assert(len(p) == total)
 104         for pe, e in p, git.idxmerge(inp, final_progress=False):
 105             assert(i == pi.next())
 106             assert(p.exists(i))
 107
 108     return total, outfilename
 109
 110
 111 def do_midx(outdir, outfilename, infilenames, prefixstr):
 112     rv = _do_midx(outdir, outfilename, infilenames, prefixstr)
 113     if rv and opt['print']:
 114         print rv[1]
 115
 116
 117 def do_midx_dir(path):
 118     already = {}
 119     sizes = {}
 120     if opt.force and not opt.auto:
 121         midxs = []   # don't use existing midx files
 122     else:
 123         midxs = glob.glob('%s/*.midx' % path)
 124         contents = {}
 125         for mname in midxs:
 126             m = git.open_idx(mname)
 127             contents[mname] = [('%s/%s' % (path,i)) for i in m.idxnames]
 128             sizes[mname] = len(m)
 129
 130         # sort the biggest midxes first, so that we can eliminate smaller
 131         # redundant ones that come later in the list
 132         midxs.sort(lambda x,y: -cmp(sizes[x], sizes[y]))
 133
 134         for mname in midxs:
 135             any = 0
 136             for iname in contents[mname]:
 137                 if not already.get(iname):
 138                     already[iname] = 1
 139                     any = 1
 140             if not any:
 141                 debug1('%r is redundant\n' % mname)
 142                 unlink(mname)
 143                 already[mname] = 1
 144
 145     midxs = [k for k in midxs if not already.get(k)]
 146     idxs = [k for k in glob.glob('%s/*.idx' % path) if not already.get(k)]
 147
 148     for iname in idxs:
 149         i = git.open_idx(iname)
 150         sizes[iname] = len(i)
 151
 152     all = [(sizes[n],n) for n in (midxs + idxs)]
 153
 154     # FIXME: what are the optimal values?  Does this make sense?
 155     DESIRED_HWM = opt.force and 1 or 5
 156     DESIRED_LWM = opt.force and 1 or 2
 157     existed = dict((name,1) for sz,name in all)
 158     debug1('midx: %d indexes; want no more than %d.\n'
 159            % (len(all), DESIRED_HWM))
 160     if len(all) <= DESIRED_HWM:
 161         debug1('midx: nothing to do.\n')
 162     while len(all) > DESIRED_HWM:
 163         all.sort()
 164         part1 = [name for sz,name in all[:len(all)-DESIRED_LWM+1]]
 165         part2 = all[len(all)-DESIRED_LWM+1:]
 166         all = list(do_midx_group(path, part1)) + part2
 167         if len(all) > DESIRED_HWM:
 168             debug1('\nStill too many indexes (%d > %d).  Merging again.\n'
 169                    % (len(all), DESIRED_HWM))
 170
 171     if opt['print']:
 172         for sz,name in all:
 173             if not existed.get(name):
 174                 print name
 175
 176
 177 def do_midx_group(outdir, infiles):
 178     groups = list(_group(infiles, opt.max_files))
 179     gprefix = ''
 180     for n,sublist in enumerate(groups):
 181         if len(groups) != 1:
 182             gprefix = 'Group %d: ' % (n+1)
 183         rv = _do_midx(path, None, sublist, gprefix)
 184         if rv:
 185             yield rv
 186
 187
 188 handle_ctrl_c()
 189
 190 o = options.Options(optspec)
 191 (opt, flags, extra) = o.parse(sys.argv[1:])
 192
 193 if extra and (opt.auto or opt.force):
 194     o.fatal("you can't use -f/-a and also provide filenames")
 195
 196 git.check_repo_or_die()
 197
 198 if opt.max_files < 0:
 199     opt.max_files = max_files()
 200 assert(opt.max_files >= 5)
 201
 202 if extra:
 203     do_midx(git.repo('objects/pack'), opt.output, extra, '')
 204 elif opt.auto or opt.force:
 205     paths = opt.dir and [opt.dir] or git.all_packdirs()
 206     for path in paths:
 207         debug1('midx: scanning %s\n' % path)
 208         do_midx_dir(path)
 209 else:
 210     o.fatal("you must use -f or -a or provide input filenames")