You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
110 lines
3.6 KiB
110 lines
3.6 KiB
6 years ago
|
# test_performance.py
|
||
|
|
# Copyright (C) 2008, 2009 Michael Trier (mtrier@gmail.com) and contributors
|
||
|
|
#
|
||
|
|
# This module is part of GitPython and is released under
|
||
|
|
# the BSD License: http://www.opensource.org/licenses/bsd-license.php
|
||
|
|
from __future__ import print_function
|
||
|
|
from io import BytesIO
|
||
|
|
from time import time
|
||
|
|
import sys
|
||
|
|
|
||
|
|
from .lib import TestBigRepoRW
|
||
|
|
from git import Commit
|
||
|
|
from gitdb import IStream
|
||
|
|
from git.compat import xrange
|
||
|
|
from git.test.test_commit import assert_commit_serialization
|
||
|
|
|
||
|
|
|
||
|
|
class TestPerformance(TestBigRepoRW):
|
||
|
|
|
||
|
|
def tearDown(self):
|
||
|
|
import gc
|
||
|
|
gc.collect()
|
||
|
|
|
||
|
|
# ref with about 100 commits in its history
|
||
|
|
ref_100 = '0.1.6'
|
||
|
|
|
||
|
|
def _query_commit_info(self, c):
|
||
|
|
c.author
|
||
|
|
c.authored_date
|
||
|
|
c.author_tz_offset
|
||
|
|
c.committer
|
||
|
|
c.committed_date
|
||
|
|
c.committer_tz_offset
|
||
|
|
c.message
|
||
|
|
c.parents
|
||
|
|
|
||
|
|
def test_iteration(self):
|
||
|
|
no = 0
|
||
|
|
nc = 0
|
||
|
|
|
||
|
|
# find the first commit containing the given path - always do a full
|
||
|
|
# iteration ( restricted to the path in question ), but in fact it should
|
||
|
|
# return quite a lot of commits, we just take one and hence abort the operation
|
||
|
|
|
||
|
|
st = time()
|
||
|
|
for c in self.rorepo.iter_commits(self.ref_100):
|
||
|
|
nc += 1
|
||
|
|
self._query_commit_info(c)
|
||
|
|
for obj in c.tree.traverse():
|
||
|
|
obj.size
|
||
|
|
no += 1
|
||
|
|
# END for each object
|
||
|
|
# END for each commit
|
||
|
|
elapsed_time = time() - st
|
||
|
|
print("Traversed %i Trees and a total of %i uncached objects in %s [s] ( %f objs/s )"
|
||
|
|
% (nc, no, elapsed_time, no / elapsed_time), file=sys.stderr)
|
||
|
|
|
||
|
|
def test_commit_traversal(self):
|
||
|
|
# bound to cat-file parsing performance
|
||
|
|
nc = 0
|
||
|
|
st = time()
|
||
|
|
for c in self.gitrorepo.commit().traverse(branch_first=False):
|
||
|
|
nc += 1
|
||
|
|
self._query_commit_info(c)
|
||
|
|
# END for each traversed commit
|
||
|
|
elapsed_time = time() - st
|
||
|
|
print("Traversed %i Commits in %s [s] ( %f commits/s )"
|
||
|
|
% (nc, elapsed_time, nc / elapsed_time), file=sys.stderr)
|
||
|
|
|
||
|
|
def test_commit_iteration(self):
|
||
|
|
# bound to stream parsing performance
|
||
|
|
nc = 0
|
||
|
|
st = time()
|
||
|
|
for c in Commit.iter_items(self.gitrorepo, self.gitrorepo.head):
|
||
|
|
nc += 1
|
||
|
|
self._query_commit_info(c)
|
||
|
|
# END for each traversed commit
|
||
|
|
elapsed_time = time() - st
|
||
|
|
print("Iterated %i Commits in %s [s] ( %f commits/s )"
|
||
|
|
% (nc, elapsed_time, nc / elapsed_time), file=sys.stderr)
|
||
|
|
|
||
|
|
def test_commit_serialization(self):
|
||
|
|
assert_commit_serialization(self.gitrwrepo, '58c78e6', True)
|
||
|
|
|
||
|
|
rwrepo = self.gitrwrepo
|
||
|
|
make_object = rwrepo.odb.store
|
||
|
|
# direct serialization - deserialization can be tested afterwards
|
||
|
|
# serialization is probably limited on IO
|
||
|
|
hc = rwrepo.commit(rwrepo.head)
|
||
|
|
|
||
|
|
nc = 5000
|
||
|
|
st = time()
|
||
|
|
for i in xrange(nc):
|
||
|
|
cm = Commit(rwrepo, Commit.NULL_BIN_SHA, hc.tree,
|
||
|
|
hc.author, hc.authored_date, hc.author_tz_offset,
|
||
|
|
hc.committer, hc.committed_date, hc.committer_tz_offset,
|
||
|
|
str(i), parents=hc.parents, encoding=hc.encoding)
|
||
|
|
|
||
|
|
stream = BytesIO()
|
||
|
|
cm._serialize(stream)
|
||
|
|
slen = stream.tell()
|
||
|
|
stream.seek(0)
|
||
|
|
|
||
|
|
cm.binsha = make_object(IStream(Commit.type, slen, stream)).binsha
|
||
|
|
# END commit creation
|
||
|
|
elapsed = time() - st
|
||
|
|
|
||
|
|
print("Serialized %i commits to loose objects in %f s ( %f commits / s )"
|
||
|
|
% (nc, elapsed, nc / elapsed), file=sys.stderr)
|