mirror of
https://github.com/golang/go
synced 2024-11-26 07:47:57 -07:00
666abfb64c
reorganize benchmark computation so that it is incremental. if it times out, it doesn't lose the pieces it already computed, so that next time it has a fighting chance to finish. R=agl1, agl CC=golang-dev https://golang.org/cl/216046
665 lines
22 KiB
Python
665 lines
22 KiB
Python
# Copyright 2009 The Go Authors. All rights reserved.
|
|
# Use of this source code is governed by a BSD-style
|
|
# license that can be found in the LICENSE file.
|
|
|
|
# This is the server part of the continuous build system for Go. It must be run
|
|
# by AppEngine.
|
|
|
|
from google.appengine.api import memcache
|
|
from google.appengine.runtime import DeadlineExceededError
|
|
from google.appengine.ext import db
|
|
from google.appengine.ext import webapp
|
|
from google.appengine.ext.webapp import template
|
|
from google.appengine.ext.webapp.util import run_wsgi_app
|
|
import binascii
|
|
import datetime
|
|
import hashlib
|
|
import hmac
|
|
import logging
|
|
import os
|
|
import re
|
|
import struct
|
|
|
|
# local imports
|
|
import key
|
|
|
|
# The majority of our state are commit objects. One of these exists for each of
|
|
# the commits known to the build system. Their key names are of the form
|
|
# <commit number (%08x)> "-" <hg hash>. This means that a sorting by the key
|
|
# name is sufficient to order the commits.
|
|
#
|
|
# The commit numbers are purely local. They need not match up to the commit
|
|
# numbers in an hg repo. When inserting a new commit, the parent commit must be
|
|
# given and this is used to generate the new commit number. In order to create
|
|
# the first Commit object, a special command (/init) is used.
|
|
class Commit(db.Model):
|
|
num = db.IntegerProperty() # internal, monotonic counter.
|
|
node = db.StringProperty() # Hg hash
|
|
parentnode = db.StringProperty() # Hg hash
|
|
user = db.StringProperty()
|
|
date = db.DateTimeProperty()
|
|
desc = db.BlobProperty()
|
|
|
|
# This is the list of builds. Each element is a string of the form <builder
|
|
# name> "`" <log hash>. If the log hash is empty, then the build was
|
|
# successful.
|
|
builds = db.StringListProperty()
|
|
|
|
class Benchmark(db.Model):
|
|
name = db.StringProperty()
|
|
version = db.IntegerProperty()
|
|
|
|
class BenchmarkResult(db.Model):
|
|
num = db.IntegerProperty()
|
|
builder = db.StringProperty()
|
|
iterations = db.IntegerProperty()
|
|
nsperop = db.IntegerProperty()
|
|
|
|
# A Log contains the textual build log of a failed build. The key name is the
|
|
# hex digest of the SHA256 hash of the contents.
|
|
class Log(db.Model):
|
|
log = db.BlobProperty()
|
|
|
|
# For each builder, we store the last revision that it built. So, if it
|
|
# crashes, it knows where to start up from. The key names for these objects are
|
|
# "hw-" <builder name>
|
|
class Highwater(db.Model):
|
|
commit = db.StringProperty()
|
|
|
|
N = 30
|
|
|
|
def builderInfo(b):
|
|
f = b.split('-', 3)
|
|
goos = f[0]
|
|
goarch = f[1]
|
|
note = ""
|
|
if len(f) > 2:
|
|
note = f[2]
|
|
return {'name': b, 'goos': goos, 'goarch': goarch, 'note': note}
|
|
|
|
def builderset():
|
|
q = Commit.all()
|
|
q.order('-__key__')
|
|
results = q.fetch(N)
|
|
builders = set()
|
|
for c in results:
|
|
builders.update(set(parseBuild(build)['builder'] for build in c.builds))
|
|
return builders
|
|
|
|
class MainPage(webapp.RequestHandler):
|
|
def get(self):
|
|
self.response.headers['Content-Type'] = 'text/html; charset=utf-8'
|
|
|
|
q = Commit.all()
|
|
q.order('-__key__')
|
|
results = q.fetch(N)
|
|
|
|
revs = [toRev(r) for r in results]
|
|
builders = {}
|
|
|
|
for r in revs:
|
|
for b in r['builds']:
|
|
builders[b['builder']] = builderInfo(b['builder'])
|
|
|
|
for r in revs:
|
|
have = set(x['builder'] for x in r['builds'])
|
|
need = set(builders.keys()).difference(have)
|
|
for n in need:
|
|
r['builds'].append({'builder': n, 'log':'', 'ok': False})
|
|
r['builds'].sort(cmp = byBuilder)
|
|
|
|
builders = list(builders.items())
|
|
builders.sort()
|
|
values = {"revs": revs, "builders": [v for k,v in builders]}
|
|
|
|
path = os.path.join(os.path.dirname(__file__), 'main.html')
|
|
self.response.out.write(template.render(path, values))
|
|
|
|
class GetHighwater(webapp.RequestHandler):
|
|
def get(self):
|
|
builder = self.request.get('builder')
|
|
|
|
key = 'hw-%s' % builder
|
|
node = memcache.get(key)
|
|
if node is None:
|
|
hw = Highwater.get_by_key_name('hw-%s' % builder)
|
|
if hw is None:
|
|
# If no highwater has been recorded for this builder,
|
|
# we go back N+1 commits and return that.
|
|
q = Commit.all()
|
|
q.order('-__key__')
|
|
c = q.fetch(N+1)[-1]
|
|
node = c.node
|
|
else:
|
|
# if the proposed hw is too old, bump it forward
|
|
node = hw.commit
|
|
found = False
|
|
q = Commit.all()
|
|
q.order('-__key__')
|
|
recent = q.fetch(N+1)
|
|
for c in recent:
|
|
if c.node == node:
|
|
found = True
|
|
break
|
|
if not found:
|
|
node = recent[-1].node
|
|
memcache.set(key, node, 3600)
|
|
self.response.set_status(200)
|
|
self.response.out.write(node)
|
|
|
|
def auth(req):
|
|
k = req.get('key')
|
|
return k == hmac.new(key.accessKey, req.get('builder')).hexdigest() or k == key.accessKey
|
|
|
|
class SetHighwater(webapp.RequestHandler):
|
|
def post(self):
|
|
if not auth(self.request):
|
|
self.response.set_status(403)
|
|
return
|
|
|
|
builder = self.request.get('builder')
|
|
newhw = self.request.get('hw')
|
|
q = Commit.all()
|
|
q.filter('node =', newhw)
|
|
c = q.get()
|
|
if c is None:
|
|
self.response.set_status(404)
|
|
return
|
|
|
|
# if the proposed hw is too old, bump it forward
|
|
found = False
|
|
q = Commit.all()
|
|
q.order('-__key__')
|
|
recent = q.fetch(N+1)
|
|
for c in head:
|
|
if c.node == newhw:
|
|
found = True
|
|
break
|
|
if not found:
|
|
c = recent[-1]
|
|
|
|
key = 'hw-%s' % builder
|
|
memcache.delete(key)
|
|
hw = Highwater(key_name = key)
|
|
hw.commit = c.node
|
|
hw.put()
|
|
|
|
class LogHandler(webapp.RequestHandler):
|
|
def get(self):
|
|
self.response.headers['Content-Type'] = 'text/plain; charset=utf-8'
|
|
hash = self.request.path[5:]
|
|
l = Log.get_by_key_name(hash)
|
|
if l is None:
|
|
self.response.set_status(404)
|
|
return
|
|
self.response.set_status(200)
|
|
self.response.out.write(l.log)
|
|
|
|
# Init creates the commit with id 0. Since this commit doesn't have a parent,
|
|
# it cannot be created by Build.
|
|
class Init(webapp.RequestHandler):
|
|
def post(self):
|
|
if not auth(self.request):
|
|
self.response.set_status(403)
|
|
return
|
|
|
|
date = parseDate(self.request.get('date'))
|
|
node = self.request.get('node')
|
|
if not validNode(node) or date is None:
|
|
logging.error("Not valid node ('%s') or bad date (%s %s)", node, date, self.request.get('date'))
|
|
self.response.set_status(500)
|
|
return
|
|
|
|
commit = Commit(key_name = '00000000-%s' % node)
|
|
commit.num = 0
|
|
commit.node = node
|
|
commit.parentnode = ''
|
|
commit.user = self.request.get('user')
|
|
commit.date = date
|
|
commit.desc = self.request.get('desc').encode('utf8')
|
|
|
|
commit.put()
|
|
|
|
self.response.set_status(200)
|
|
|
|
# Build is the main command: it records the result of a new build.
|
|
class Build(webapp.RequestHandler):
|
|
def post(self):
|
|
if not auth(self.request):
|
|
self.response.set_status(403)
|
|
return
|
|
|
|
builder = self.request.get('builder')
|
|
log = self.request.get('log').encode('utf-8')
|
|
|
|
loghash = ''
|
|
if len(log) > 0:
|
|
loghash = hashlib.sha256(log).hexdigest()
|
|
l = Log(key_name = loghash)
|
|
l.log = log
|
|
l.put()
|
|
|
|
date = parseDate(self.request.get('date'))
|
|
node = self.request.get('node')
|
|
parent = self.request.get('parent')
|
|
if not validNode(node) or not validNode(parent) or date is None:
|
|
logging.error("Not valid node ('%s') or bad date (%s %s)", node, date, self.request.get('date'))
|
|
self.response.set_status(500)
|
|
return
|
|
|
|
q = Commit.all()
|
|
q.filter('node =', parent)
|
|
p = q.get()
|
|
if p is None:
|
|
logging.error('Cannot find parent %s of node %s' % (parent, node))
|
|
self.response.set_status(404)
|
|
return
|
|
parentnum, _ = p.key().name().split('-', 1)
|
|
nodenum = int(parentnum, 16) + 1
|
|
|
|
def add_build():
|
|
key_name = '%08x-%s' % (nodenum, node)
|
|
n = Commit.get_by_key_name(key_name)
|
|
if n is None:
|
|
n = Commit(key_name = key_name)
|
|
n.num = nodenum
|
|
n.node = node
|
|
n.parentnode = parent
|
|
n.user = self.request.get('user')
|
|
n.date = date
|
|
n.desc = self.request.get('desc').encode('utf8')
|
|
s = '%s`%s' % (builder, loghash)
|
|
for i, b in enumerate(n.builds):
|
|
if b.split('`', 1)[0] == builder:
|
|
n.builds[i] = s
|
|
break
|
|
else:
|
|
n.builds.append(s)
|
|
n.put()
|
|
|
|
db.run_in_transaction(add_build)
|
|
|
|
hw = Highwater.get_by_key_name('hw-%s' % builder)
|
|
if hw is None:
|
|
hw = Highwater(key_name = 'hw-%s' % builder)
|
|
hw.commit = node
|
|
hw.put()
|
|
memcache.delete('hw')
|
|
memcache.delete('bench')
|
|
|
|
self.response.set_status(200)
|
|
|
|
class Benchmarks(webapp.RequestHandler):
|
|
def json(self):
|
|
q = Benchmark.all()
|
|
q.filter('__key__ >', Benchmark.get_or_insert('v002.').key())
|
|
bs = q.fetch(10000)
|
|
|
|
self.response.set_status(200)
|
|
self.response.headers['Content-Type'] = 'text/plain; charset=utf-8'
|
|
self.response.out.write('{"benchmarks": [')
|
|
|
|
first = True
|
|
sep = "\n\t"
|
|
for b in bs:
|
|
self.response.out.write('%s"%s"' % (sep, b.name))
|
|
sep = ",\n\t"
|
|
self.response.out.write('\n]}\n')
|
|
|
|
def get(self):
|
|
if self.request.get('fmt') == 'json':
|
|
return self.json()
|
|
self.response.set_status(200)
|
|
self.response.headers['Content-Type'] = 'text/html; charset=utf-8'
|
|
page = memcache.get('bench')
|
|
if not page:
|
|
num = memcache.get('hw')
|
|
if num is None:
|
|
q = Commit.all()
|
|
q.order('-__key__')
|
|
n = q.fetch(1)[0]
|
|
memcache.set('hw', num)
|
|
page, full = self.compute(n.num)
|
|
if full:
|
|
memcache.set('bench', page, 3600)
|
|
self.response.out.write(page)
|
|
|
|
def compute(self, num):
|
|
q = Benchmark.all()
|
|
q.filter('__key__ >', Benchmark.get_or_insert('v002.').key())
|
|
benchmarks = q.fetch(10000)
|
|
|
|
# Which builders have sent benchmarks recently?
|
|
builders = set()
|
|
q = BenchmarkResult.all()
|
|
q.ancestor(benchmarks[0])
|
|
q.order('-__key__')
|
|
for r in q.fetch(50):
|
|
builders.add(r.builder)
|
|
builders = list(builders)
|
|
builders.sort()
|
|
|
|
NB = 80
|
|
last = num
|
|
first = num+1 - NB
|
|
|
|
# Build list of rows, one per benchmark
|
|
rows = [{"name": bm.name, "builds": [{"url": ""} for b in builders]} for bm in benchmarks]
|
|
|
|
full = True
|
|
try:
|
|
for i in range(len(rows)):
|
|
data = None
|
|
bm = benchmarks[i]
|
|
builds = rows[i]["builds"]
|
|
all = None
|
|
for j in range(len(builders)):
|
|
cell = builds[j]
|
|
b = builders[j]
|
|
# Build cell: a URL for the chart server or an empty string.
|
|
# Cache individual graphs because they're so damn expensive.
|
|
key = "bench(%s,%s,%d)" % (bm.name, b, num)
|
|
url = memcache.get(key)
|
|
if url is not None:
|
|
cell["url"] = url
|
|
continue
|
|
|
|
# Page in all data for benchmark for all builders,
|
|
# on demand. It might be faster to ask for just the
|
|
# builder that we need, but q.filter('builder = ', b) is
|
|
# broken right now (the index is corrupt).
|
|
if all is None:
|
|
q = BenchmarkResult.all()
|
|
q.ancestor(bm)
|
|
q.order('-__key__')
|
|
all = q.fetch(1000)
|
|
|
|
data = [-1 for x in range(first, last+1)]
|
|
for r in all:
|
|
if r.builder == b and first <= r.num and r.num <= last:
|
|
data[r.num - first] = r.nsperop
|
|
present = [x for x in data if x >= 0]
|
|
if len(present) == 0:
|
|
memcache.set(key, "", 3600)
|
|
continue
|
|
avg = sum(present) / len(present)
|
|
maxval = max(2*avg, max(present))
|
|
# Encoding is 0-61, which is fine enough granularity for our tiny graphs. _ means missing.
|
|
encoding = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789"
|
|
s = ''.join([x < 0 and "_" or encoding[int((len(encoding)-1)*x/maxval)] for x in data])
|
|
url = "http://chart.apis.google.com/chart?cht=ls&chd=s:"+s
|
|
memcache.set(key, url, 3600)
|
|
cell["url"] = url
|
|
except DeadlineExceededError:
|
|
# forge ahead with partial benchmark results
|
|
# the url caches above should make the next page quicker to compute
|
|
full = False
|
|
|
|
names = [bm.name for bm in benchmarks]
|
|
|
|
bs = []
|
|
for b in builders:
|
|
f = b.split('-', 3)
|
|
goos = f[0]
|
|
goarch = f[1]
|
|
note = ""
|
|
if len(f) > 2:
|
|
note = f[2]
|
|
bs.append({'goos': goos, 'goarch': goarch, 'note': note})
|
|
|
|
values = {"benchmarks": rows, "builders": bs}
|
|
|
|
path = os.path.join(os.path.dirname(__file__), 'benchmarks.html')
|
|
return template.render(path, values), full
|
|
|
|
def post(self):
|
|
if not auth(self.request):
|
|
self.response.set_status(403)
|
|
return
|
|
|
|
builder = self.request.get('builder')
|
|
node = self.request.get('node')
|
|
if not validNode(node):
|
|
logging.error("Not valid node ('%s')", node)
|
|
self.response.set_status(500)
|
|
return
|
|
|
|
benchmarkdata = self.request.get('benchmarkdata')
|
|
benchmarkdata = binascii.a2b_base64(benchmarkdata)
|
|
|
|
def get_string(i):
|
|
l, = struct.unpack('>H', i[:2])
|
|
s = i[2:2+l]
|
|
if len(s) != l:
|
|
return None, None
|
|
return s, i[2+l:]
|
|
|
|
benchmarks = {}
|
|
while len(benchmarkdata) > 0:
|
|
name, benchmarkdata = get_string(benchmarkdata)
|
|
iterations_str, benchmarkdata = get_string(benchmarkdata)
|
|
time_str, benchmarkdata = get_string(benchmarkdata)
|
|
iterations = int(iterations_str)
|
|
time = int(time_str)
|
|
|
|
benchmarks[name] = (iterations, time)
|
|
|
|
q = Commit.all()
|
|
q.filter('node =', node)
|
|
n = q.get()
|
|
if n is None:
|
|
logging.error('Client asked for unknown commit while uploading benchmarks')
|
|
self.response.set_status(404)
|
|
return
|
|
|
|
for (benchmark, (iterations, time)) in benchmarks.items():
|
|
b = Benchmark.get_or_insert('v002.' + benchmark.encode('base64'), name = benchmark, version = 2)
|
|
r = BenchmarkResult(key_name = '%08x/%s' % (n.num, builder), parent = b, num = n.num, iterations = iterations, nsperop = time, builder = builder)
|
|
r.put()
|
|
key = "bench(%s,%s,%d)" % (benchmark, builder, n.num)
|
|
memcache.delete(key)
|
|
|
|
memcache.delete('bench')
|
|
self.response.set_status(200)
|
|
|
|
def node(num):
|
|
q = Commit.all()
|
|
q.filter('num =', num)
|
|
n = q.get()
|
|
return n
|
|
|
|
class GetBenchmarks(webapp.RequestHandler):
|
|
def get(self):
|
|
benchmark = self.request.path[12:]
|
|
bm = Benchmark.get_by_key_name('v002.' + benchmark.encode('base64'))
|
|
if bm is None:
|
|
self.response.set_status(404)
|
|
return
|
|
|
|
q = BenchmarkResult.all()
|
|
q.ancestor(bm)
|
|
q.order('-__key__')
|
|
results = q.fetch(10000)
|
|
|
|
if len(results) == 0:
|
|
self.response.set_status(404)
|
|
return
|
|
|
|
maxv = -1
|
|
minv = 2000000000
|
|
builders = set()
|
|
for r in results:
|
|
if maxv < r.num:
|
|
maxv = r.num
|
|
if minv > r.num:
|
|
minv = r.num
|
|
builders.add(r.builder)
|
|
|
|
builders = list(builders)
|
|
builders.sort()
|
|
|
|
res = {}
|
|
for b in builders:
|
|
res[b] = [[-1] * ((maxv - minv) + 1), [-1] * ((maxv - minv) + 1)]
|
|
|
|
for r in results:
|
|
res[r.builder][0][r.num - minv] = r.iterations
|
|
res[r.builder][1][r.num - minv] = r.nsperop
|
|
|
|
minhash = node(minv).node
|
|
maxhash = node(maxv).node
|
|
if self.request.get('fmt') == 'json':
|
|
self.response.headers['Content-Type'] = 'text/plain; charset=utf-8'
|
|
self.response.out.write('{"min": "%s", "max": "%s", "data": {' % (minhash, maxhash))
|
|
sep = "\n\t"
|
|
for b in builders:
|
|
self.response.out.write('%s"%s": {"iterations": %s, "nsperop": %s}' % (sep, b, str(res[b][0]).replace("L", ""), str(res[b][1]).replace("L", "")))
|
|
sep = ",\n\t"
|
|
self.response.out.write("\n}}\n")
|
|
return
|
|
|
|
def bgraph(builder):
|
|
data = res[builder][1]
|
|
encoding = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789.-"
|
|
m = max(data) # max ns timing
|
|
if m == -1:
|
|
return ""
|
|
tot = 0
|
|
ntot = 0
|
|
for d in data:
|
|
if d < 0:
|
|
continue
|
|
tot += d
|
|
ntot += 1
|
|
avg = tot / ntot
|
|
if 2*avg > m:
|
|
m = 2*avg
|
|
s = ""
|
|
for d in data:
|
|
if d < 0:
|
|
s += "__"
|
|
continue
|
|
val = int(d*4095.0/m)
|
|
s += encoding[val/64] + encoding[val%64]
|
|
return "http://chart.apis.google.com/chart?cht=lc&chxt=x,y&chxl=0:|%s|%s|1:|0|%g ns|%g ns&chd=e:%s" % (minhash[0:12], maxhash[0:12], m/2, m, s)
|
|
|
|
graphs = []
|
|
for b in builders:
|
|
graphs.append({"builder": b, "url": bgraph(b)})
|
|
|
|
revs = []
|
|
for i in range(minv, maxv+1):
|
|
r = nodeInfo(node(i))
|
|
ns = []
|
|
for b in builders:
|
|
t = res[b][1][i - minv]
|
|
if t < 0:
|
|
t = None
|
|
ns.append(t)
|
|
r["ns_by_builder"] = ns
|
|
revs.append(r)
|
|
|
|
path = os.path.join(os.path.dirname(__file__), 'benchmark1.html')
|
|
data = {
|
|
"benchmark": bm.name,
|
|
"builders": [builderInfo(b) for b in builders],
|
|
"graphs": graphs,
|
|
"revs": revs
|
|
}
|
|
self.response.out.write(template.render(path, data))
|
|
|
|
class FixedOffset(datetime.tzinfo):
|
|
"""Fixed offset in minutes east from UTC."""
|
|
|
|
def __init__(self, offset):
|
|
self.__offset = datetime.timedelta(seconds = offset)
|
|
|
|
def utcoffset(self, dt):
|
|
return self.__offset
|
|
|
|
def tzname(self, dt):
|
|
return None
|
|
|
|
def dst(self, dt):
|
|
return datetime.timedelta(0)
|
|
|
|
def validNode(node):
|
|
if len(node) != 40:
|
|
return False
|
|
for x in node:
|
|
o = ord(x)
|
|
if (o < ord('0') or o > ord('9')) and (o < ord('a') or o > ord('f')):
|
|
return False
|
|
return True
|
|
|
|
def parseDate(date):
|
|
if '-' in date:
|
|
(a, offset) = date.split('-', 1)
|
|
try:
|
|
return datetime.datetime.fromtimestamp(float(a), FixedOffset(0-int(offset)))
|
|
except ValueError:
|
|
return None
|
|
if '+' in date:
|
|
(a, offset) = date.split('+', 1)
|
|
try:
|
|
return datetime.datetime.fromtimestamp(float(a), FixedOffset(int(offset)))
|
|
except ValueError:
|
|
return None
|
|
try:
|
|
return datetime.datetime.utcfromtimestamp(float(date))
|
|
except ValueError:
|
|
return None
|
|
|
|
email_re = re.compile('^[^<]+<([^>]*)>$')
|
|
|
|
def toUsername(user):
|
|
r = email_re.match(user)
|
|
if r is None:
|
|
return user
|
|
email = r.groups()[0]
|
|
return email.replace('@golang.org', '')
|
|
|
|
def dateToShortStr(d):
|
|
return d.strftime('%a %b %d %H:%M')
|
|
|
|
def parseBuild(build):
|
|
[builder, logblob] = build.split('`')
|
|
return {'builder': builder, 'log': logblob, 'ok': len(logblob) == 0}
|
|
|
|
def nodeInfo(c):
|
|
return {
|
|
"node": c.node,
|
|
"user": toUsername(c.user),
|
|
"date": dateToShortStr(c.date),
|
|
"desc": c.desc,
|
|
"shortdesc": c.desc.split('\n', 2)[0]
|
|
}
|
|
|
|
def toRev(c):
|
|
b = nodeInfo(c)
|
|
b['builds'] = [parseBuild(build) for build in c.builds]
|
|
return b
|
|
|
|
def byBuilder(x, y):
|
|
return cmp(x['builder'], y['builder'])
|
|
|
|
# This is the URL map for the server. The first three entries are public, the
|
|
# rest are only used by the builders.
|
|
application = webapp.WSGIApplication(
|
|
[('/', MainPage),
|
|
('/log/.*', LogHandler),
|
|
('/hw-get', GetHighwater),
|
|
('/hw-set', SetHighwater),
|
|
|
|
('/init', Init),
|
|
('/build', Build),
|
|
('/benchmarks', Benchmarks),
|
|
('/benchmarks/.*', GetBenchmarks),
|
|
], debug=True)
|
|
|
|
def main():
|
|
run_wsgi_app(application)
|
|
|
|
if __name__ == "__main__":
|
|
main()
|