mirror/borg
1
0
Fork 0
mirror of https://github.com/borgbackup/borg.git synced 2025-02-21 21:57:36 +00:00
Code Issues Releases Wiki Activity

Implement --debug-profile

Browse source
This commit is contained in:
Marian Beermann 2017-05-15 20:06:13 +02:00
parent a596ecc59f
commit 4c441c75b1
2 changed files with 28 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
src/borg/archiver.py
View file

@ -2049,6 +2049,8 @@ def define_common_options(add_common_option):
add_common_option('--consider-part-files', dest='consider_part_files',
action='store_true', default=False,
help='treat part files like normal files (e.g. to list/extract them)')
add_common_option('--debug-profile', dest='debug_profile', default=None, metavar='FILE',
help='Store a Python profile at FILE')
parser = argparse.ArgumentParser(prog=self.prog, description='Borg - Deduplicated Backups',
add_help=False)
@ -3542,7 +3544,22 @@ def run(self, args):
self.prerun_checks(logger)
if is_slow_msgpack():
logger.warning("Using a pure-python msgpack! This will result in lower performance.")
return set_ec(func(args))
if args.debug_profile:
# Import these only when needed - avoids a further increase in startup time
import cProfile
import marshal
logger.debug('Writing execution profile to %s', args.debug_profile)
# Open the file early, before running the main program, to avoid
# a very late crash in case the specified path is invalid
with open(args.debug_profile, 'wb') as fd:
profiler = cProfile.Profile()
variables = dict(locals())
profiler.runctx('rc = set_ec(func(args))', globals(), variables)
profiler.snapshot_stats()
marshal.dump(profiler.stats, fd)
return variables['rc']
else:
return set_ec(func(args))
def sig_info_handler(sig_no, stack): # pragma: no cover

10
src/borg/testsuite/archiver.py
View file

@ -3,6 +3,7 @@
import json
import logging
import os
import pstats
import random
import shutil
import socket
@ -1681,6 +1682,15 @@ def test_log_json(self):
assert log_message['name'].startswith('borg.')
assert isinstance(log_message['message'], str)
def test_debug_profile(self):
self.create_test_files()
self.cmd('init', '--encryption=repokey', self.repository_location)
self.cmd('create', self.repository_location + '::test', 'input', '--debug-profile=create.prof')
stats = pstats.Stats('create.prof')
stats.strip_dirs()
stats.sort_stats('cumtime')
# Ok, stats can be loaded, good enough.
def test_common_options(self):
self.create_test_files()
self.cmd('init', '--encryption=repokey', self.repository_location)