poky/meta/classes/buildstats.bbclass - mdmillerii/openbmc - Gitiles

 BUILDSTATS_BASE = "${TMPDIR}/buildstats/"

 ################################################################################
 # Build statistics gathering.
 #
 # The CPU and Time gathering/tracking functions and bbevent inspiration
 # were written by Christopher Larson.
 #
 ################################################################################

 def get_buildprocess_cputime(pid):
     with open("/proc/%d/stat" % pid, "r") as f:
         fields = f.readline().rstrip().split()
     # 13: utime, 14: stime, 15: cutime, 16: cstime
     return sum(int(field) for field in fields[13:16])

 def get_process_cputime(pid):
     import resource
     with open("/proc/%d/stat" % pid, "r") as f:
         fields = f.readline().rstrip().split()
     stats = {
         'utime'  : fields[13],
         'stime'  : fields[14],
         'cutime' : fields[15],
         'cstime' : fields[16],
     }
     iostats = {}
     if os.path.isfile("/proc/%d/io" % pid):
         with open("/proc/%d/io" % pid, "r") as f:
             while True:
                 i = f.readline().strip()
                 if not i:
                     break
                 if not ":" in i:
                     # one more extra line is appended (empty or containing "0")
                     # most probably due to race condition in kernel while
                     # updating IO stats
                     break
                 i = i.split(": ")
                 iostats[i[0]] = i[1]
     resources = resource.getrusage(resource.RUSAGE_SELF)
     childres = resource.getrusage(resource.RUSAGE_CHILDREN)
     return stats, iostats, resources, childres

 def get_cputime():
     with open("/proc/stat", "r") as f:
         fields = f.readline().rstrip().split()[1:]
     return sum(int(field) for field in fields)

 def set_timedata(var, d, server_time):
     d.setVar(var, server_time)

 def get_timedata(var, d, end_time):
     oldtime = d.getVar(var, False)
     if oldtime is None:
         return
     return end_time - oldtime

 def set_buildtimedata(var, d):
     import time
     time = time.time()
     cputime = get_cputime()
     proctime = get_buildprocess_cputime(os.getpid())
     d.setVar(var, (time, cputime, proctime))

 def get_buildtimedata(var, d):
     import time
     timedata = d.getVar(var, False)
     if timedata is None:
         return
     oldtime, oldcpu, oldproc = timedata
     procdiff = get_buildprocess_cputime(os.getpid()) - oldproc
     cpudiff = get_cputime() - oldcpu
     end_time = time.time()
     timediff = end_time - oldtime
     if cpudiff > 0:
         cpuperc = float(procdiff) * 100 / cpudiff
     else:
         cpuperc = None
     return timediff, cpuperc

 def write_task_data(status, logfile, e, d):
     bn = d.getVar('BUILDNAME')
     bsdir = os.path.join(d.getVar('BUILDSTATS_BASE'), bn)
     with open(os.path.join(logfile), "a") as f:
         elapsedtime = get_timedata("__timedata_task", d, e.time)
         if elapsedtime:
             f.write(d.expand("${PF}: %s\n" % e.task))
             f.write(d.expand("Elapsed time: %0.2f seconds\n" % elapsedtime))
             cpu, iostats, resources, childres = get_process_cputime(os.getpid())
             if cpu:
                 f.write("utime: %s\n" % cpu['utime'])
                 f.write("stime: %s\n" % cpu['stime'])
                 f.write("cutime: %s\n" % cpu['cutime'])
                 f.write("cstime: %s\n" % cpu['cstime'])
             for i in iostats:
                 f.write("IO %s: %s\n" % (i, iostats[i]))
             rusages = ["ru_utime", "ru_stime", "ru_maxrss", "ru_minflt", "ru_majflt", "ru_inblock", "ru_oublock", "ru_nvcsw", "ru_nivcsw"]
             for i in rusages:
                 f.write("rusage %s: %s\n" % (i, getattr(resources, i)))
             for i in rusages:
                 f.write("Child rusage %s: %s\n" % (i, getattr(childres, i)))
         if status is "passed":
             f.write("Status: PASSED \n")
         else:
             f.write("Status: FAILED \n")
         f.write("Ended: %0.2f \n" % e.time)

 python run_buildstats () {
     import bb.build
     import bb.event
     import time, subprocess, platform

     bn = d.getVar('BUILDNAME')
     bsdir = os.path.join(d.getVar('BUILDSTATS_BASE'), bn)
     taskdir = os.path.join(bsdir, d.getVar('PF'))

     if isinstance(e, bb.event.BuildStarted):
         ########################################################################
         # If the kernel was not configured to provide I/O statistics, issue
         # a one time warning.
         ########################################################################
         if not os.path.isfile("/proc/%d/io" % os.getpid()):
             bb.warn("The Linux kernel on your build host was not configured to provide process I/O statistics. (CONFIG_TASK_IO_ACCOUNTING is not set)")

         ########################################################################
         # at first pass make the buildstats hierarchy and then
         # set the buildname
         ########################################################################
         bb.utils.mkdirhier(bsdir)
         set_buildtimedata("__timedata_build", d)
         build_time = os.path.join(bsdir, "build_stats")
         # write start of build into build_time
         with open(build_time, "a") as f:
             host_info = platform.uname()
             f.write("Host Info: ")
             for x in host_info:
                 if x:
                     f.write(x + " ")
             f.write("\n")
             f.write("Build Started: %0.2f \n" % time.time())

     elif isinstance(e, bb.event.BuildCompleted):
         build_time = os.path.join(bsdir, "build_stats")
         with open(build_time, "a") as f:
             ########################################################################
             # Write build statistics for the build
             ########################################################################
             timedata = get_buildtimedata("__timedata_build", d)
             if timedata:
                 time, cpu = timedata
                 # write end of build and cpu used into build_time
                 f.write("Elapsed time: %0.2f seconds \n" % (time))
                 if cpu:
                     f.write("CPU usage: %0.1f%% \n" % cpu)

     if isinstance(e, bb.build.TaskStarted):
         set_timedata("__timedata_task", d, e.time)
         bb.utils.mkdirhier(taskdir)
         # write into the task event file the name and start time
         with open(os.path.join(taskdir, e.task), "a") as f:
             f.write("Event: %s \n" % bb.event.getName(e))
             f.write("Started: %0.2f \n" % e.time)

     elif isinstance(e, bb.build.TaskSucceeded):
         write_task_data("passed", os.path.join(taskdir, e.task), e, d)
         if e.task == "do_rootfs":
             bs = os.path.join(bsdir, "build_stats")
             with open(bs, "a") as f:
                 rootfs = d.getVar('IMAGE_ROOTFS')
                 if os.path.isdir(rootfs):
                     try:
                         rootfs_size = subprocess.check_output(["du", "-sh", rootfs],
                                 stderr=subprocess.STDOUT).decode('utf-8')
                         f.write("Uncompressed Rootfs size: %s" % rootfs_size)
                     except subprocess.CalledProcessError as err:
                         bb.warn("Failed to get rootfs size: %s" % err.output.decode('utf-8'))

     elif isinstance(e, bb.build.TaskFailed):
         # Can have a failure before TaskStarted so need to mkdir here too
         bb.utils.mkdirhier(taskdir)
         write_task_data("failed", os.path.join(taskdir, e.task), e, d)
         ########################################################################
         # Lets make things easier and tell people where the build failed in
         # build_status. We do this here because BuildCompleted triggers no
         # matter what the status of the build actually is
         ########################################################################
         build_status = os.path.join(bsdir, "build_stats")
         with open(build_status, "a") as f:
             f.write(d.expand("Failed at: ${PF} at task: %s \n" % e.task))
 }

 addhandler run_buildstats
 run_buildstats[eventmask] = "bb.event.BuildStarted bb.event.BuildCompleted bb.build.TaskStarted bb.build.TaskSucceeded bb.build.TaskFailed"

 python runqueue_stats () {
     import buildstats
     from bb import event, runqueue
     # We should not record any samples before the first task has started,
     # because that's the first activity shown in the process chart.
     # Besides, at that point we are sure that the build variables
     # are available that we need to find the output directory.
     # The persistent SystemStats is stored in the datastore and
     # closed when the build is done.
     system_stats = d.getVar('_buildstats_system_stats', False)
     if not system_stats and isinstance(e, (bb.runqueue.sceneQueueTaskStarted, bb.runqueue.runQueueTaskStarted)):
         system_stats = buildstats.SystemStats(d)
         d.setVar('_buildstats_system_stats', system_stats)
     if system_stats:
         # Ensure that we sample at important events.
         done = isinstance(e, bb.event.BuildCompleted)
         system_stats.sample(e, force=done)
         if done:
             system_stats.close()
             d.delVar('_buildstats_system_stats')
 }

 addhandler runqueue_stats
 runqueue_stats[eventmask] = "bb.runqueue.sceneQueueTaskStarted bb.runqueue.runQueueTaskStarted bb.event.HeartbeatEvent bb.event.BuildCompleted bb.event.MonitorDiskEvent"
	BUILDSTATS_BASE = "${TMPDIR}/buildstats/"

	################################################################################
	# Build statistics gathering.
	#
	# The CPU and Time gathering/tracking functions and bbevent inspiration
	# were written by Christopher Larson.
	#
	################################################################################

	def get_buildprocess_cputime(pid):
	with open("/proc/%d/stat" % pid, "r") as f:
	fields = f.readline().rstrip().split()
	# 13: utime, 14: stime, 15: cutime, 16: cstime
	return sum(int(field) for field in fields[13:16])

	def get_process_cputime(pid):
	import resource
	with open("/proc/%d/stat" % pid, "r") as f:
	fields = f.readline().rstrip().split()
	stats = {
	'utime' : fields[13],
	'stime' : fields[14],
	'cutime' : fields[15],
	'cstime' : fields[16],
	}
	iostats = {}
	if os.path.isfile("/proc/%d/io" % pid):
	with open("/proc/%d/io" % pid, "r") as f:
	while True:
	i = f.readline().strip()
	if not i:
	break
	if not ":" in i:
	# one more extra line is appended (empty or containing "0")
	# most probably due to race condition in kernel while
	# updating IO stats
	break
	i = i.split(": ")
	iostats[i[0]] = i[1]
	resources = resource.getrusage(resource.RUSAGE_SELF)
	childres = resource.getrusage(resource.RUSAGE_CHILDREN)
	return stats, iostats, resources, childres

	def get_cputime():
	with open("/proc/stat", "r") as f:
	fields = f.readline().rstrip().split()[1:]
	return sum(int(field) for field in fields)

	def set_timedata(var, d, server_time):
	d.setVar(var, server_time)

	def get_timedata(var, d, end_time):
	oldtime = d.getVar(var, False)
	if oldtime is None:
	return
	return end_time - oldtime

	def set_buildtimedata(var, d):
	import time
	time = time.time()
	cputime = get_cputime()
	proctime = get_buildprocess_cputime(os.getpid())
	d.setVar(var, (time, cputime, proctime))

	def get_buildtimedata(var, d):
	import time
	timedata = d.getVar(var, False)
	if timedata is None:
	return
	oldtime, oldcpu, oldproc = timedata
	procdiff = get_buildprocess_cputime(os.getpid()) - oldproc
	cpudiff = get_cputime() - oldcpu
	end_time = time.time()
	timediff = end_time - oldtime
	if cpudiff > 0:
	cpuperc = float(procdiff) * 100 / cpudiff
	else:
	cpuperc = None
	return timediff, cpuperc

	def write_task_data(status, logfile, e, d):
	bn = d.getVar('BUILDNAME')
	bsdir = os.path.join(d.getVar('BUILDSTATS_BASE'), bn)
	with open(os.path.join(logfile), "a") as f:
	elapsedtime = get_timedata("__timedata_task", d, e.time)
	if elapsedtime:
	f.write(d.expand("${PF}: %s\n" % e.task))
	f.write(d.expand("Elapsed time: %0.2f seconds\n" % elapsedtime))
	cpu, iostats, resources, childres = get_process_cputime(os.getpid())
	if cpu:
	f.write("utime: %s\n" % cpu['utime'])
	f.write("stime: %s\n" % cpu['stime'])
	f.write("cutime: %s\n" % cpu['cutime'])
	f.write("cstime: %s\n" % cpu['cstime'])
	for i in iostats:
	f.write("IO %s: %s\n" % (i, iostats[i]))
	rusages = ["ru_utime", "ru_stime", "ru_maxrss", "ru_minflt", "ru_majflt", "ru_inblock", "ru_oublock", "ru_nvcsw", "ru_nivcsw"]
	for i in rusages:
	f.write("rusage %s: %s\n" % (i, getattr(resources, i)))
	for i in rusages:
	f.write("Child rusage %s: %s\n" % (i, getattr(childres, i)))
	if status is "passed":
	f.write("Status: PASSED \n")
	else:
	f.write("Status: FAILED \n")
	f.write("Ended: %0.2f \n" % e.time)

	python run_buildstats () {
	import bb.build
	import bb.event
	import time, subprocess, platform

	bn = d.getVar('BUILDNAME')
	bsdir = os.path.join(d.getVar('BUILDSTATS_BASE'), bn)
	taskdir = os.path.join(bsdir, d.getVar('PF'))

	if isinstance(e, bb.event.BuildStarted):
	########################################################################
	# If the kernel was not configured to provide I/O statistics, issue
	# a one time warning.
	########################################################################
	if not os.path.isfile("/proc/%d/io" % os.getpid()):
	bb.warn("The Linux kernel on your build host was not configured to provide process I/O statistics. (CONFIG_TASK_IO_ACCOUNTING is not set)")

	########################################################################
	# at first pass make the buildstats hierarchy and then
	# set the buildname
	########################################################################
	bb.utils.mkdirhier(bsdir)
	set_buildtimedata("__timedata_build", d)
	build_time = os.path.join(bsdir, "build_stats")
	# write start of build into build_time
	with open(build_time, "a") as f:
	host_info = platform.uname()
	f.write("Host Info: ")
	for x in host_info:
	if x:
	f.write(x + " ")
	f.write("\n")
	f.write("Build Started: %0.2f \n" % time.time())

	elif isinstance(e, bb.event.BuildCompleted):
	build_time = os.path.join(bsdir, "build_stats")
	with open(build_time, "a") as f:
	########################################################################
	# Write build statistics for the build
	########################################################################
	timedata = get_buildtimedata("__timedata_build", d)
	if timedata:
	time, cpu = timedata
	# write end of build and cpu used into build_time
	f.write("Elapsed time: %0.2f seconds \n" % (time))
	if cpu:
	f.write("CPU usage: %0.1f%% \n" % cpu)

	if isinstance(e, bb.build.TaskStarted):
	set_timedata("__timedata_task", d, e.time)
	bb.utils.mkdirhier(taskdir)
	# write into the task event file the name and start time
	with open(os.path.join(taskdir, e.task), "a") as f:
	f.write("Event: %s \n" % bb.event.getName(e))
	f.write("Started: %0.2f \n" % e.time)

	elif isinstance(e, bb.build.TaskSucceeded):
	write_task_data("passed", os.path.join(taskdir, e.task), e, d)
	if e.task == "do_rootfs":
	bs = os.path.join(bsdir, "build_stats")
	with open(bs, "a") as f:
	rootfs = d.getVar('IMAGE_ROOTFS')
	if os.path.isdir(rootfs):
	try:
	rootfs_size = subprocess.check_output(["du", "-sh", rootfs],
	stderr=subprocess.STDOUT).decode('utf-8')
	f.write("Uncompressed Rootfs size: %s" % rootfs_size)
	except subprocess.CalledProcessError as err:
	bb.warn("Failed to get rootfs size: %s" % err.output.decode('utf-8'))

	elif isinstance(e, bb.build.TaskFailed):
	# Can have a failure before TaskStarted so need to mkdir here too
	bb.utils.mkdirhier(taskdir)
	write_task_data("failed", os.path.join(taskdir, e.task), e, d)
	########################################################################
	# Lets make things easier and tell people where the build failed in
	# build_status. We do this here because BuildCompleted triggers no
	# matter what the status of the build actually is
	########################################################################
	build_status = os.path.join(bsdir, "build_stats")
	with open(build_status, "a") as f:
	f.write(d.expand("Failed at: ${PF} at task: %s \n" % e.task))
	}

	addhandler run_buildstats
	run_buildstats[eventmask] = "bb.event.BuildStarted bb.event.BuildCompleted bb.build.TaskStarted bb.build.TaskSucceeded bb.build.TaskFailed"

	python runqueue_stats () {
	import buildstats
	from bb import event, runqueue
	# We should not record any samples before the first task has started,
	# because that's the first activity shown in the process chart.
	# Besides, at that point we are sure that the build variables
	# are available that we need to find the output directory.
	# The persistent SystemStats is stored in the datastore and
	# closed when the build is done.
	system_stats = d.getVar('_buildstats_system_stats', False)
	if not system_stats and isinstance(e, (bb.runqueue.sceneQueueTaskStarted, bb.runqueue.runQueueTaskStarted)):
	system_stats = buildstats.SystemStats(d)
	d.setVar('_buildstats_system_stats', system_stats)
	if system_stats:
	# Ensure that we sample at important events.
	done = isinstance(e, bb.event.BuildCompleted)
	system_stats.sample(e, force=done)
	if done:
	system_stats.close()
	d.delVar('_buildstats_system_stats')
	}

	addhandler runqueue_stats
	runqueue_stats[eventmask] = "bb.runqueue.sceneQueueTaskStarted bb.runqueue.runQueueTaskStarted bb.event.HeartbeatEvent bb.event.BuildCompleted bb.event.MonitorDiskEvent"