summaryrefslogtreecommitdiff
path: root/meta/classes/buildstats.bbclass
blob: f57d018060934a928c5e6601f0014edde9023e61 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
BUILDSTATS_BASE = ${TMPDIR}/buildstats/
BNFILE = ${BUILDSTATS_BASE}/.buildname

################################################################################
# Build statistics gathering. 
#
# The CPU and Time gathering/tracking functions and bbevent inspiration
# were written by Christopher Larson and can be seen here: 
# http://kergoth.pastey.net/142813
# 
################################################################################

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

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

def set_timedata(var, data):
    import time

    time = time.time()
    cputime = get_cputime()
    proctime = get_process_cputime(os.getpid())
    data.setVar(var, (time, cputime, proctime))

def get_timedata(var, data):
    import time
    timedata = data.getVar(var, False)
    if timedata is None:
        return
    oldtime, oldcpu, oldproc = timedata
    procdiff = get_process_cputime(os.getpid()) - oldproc
    cpudiff = get_cputime() - oldcpu
    timediff = time.time() - oldtime
    if cpudiff > 0:
        cpuperc = float(procdiff) * 100 / cpudiff
    else:
        cpuperc = None
    return timediff, cpuperc
    
##############################################
# We need to set the buildname to a file since
# BUILDNAME changes throughout a build
##############################################

def set_bn(e):
    bn = e.getPkgs()[0] + "-" + bb.data.getVar('MACHINE',e.data, True)
    try:
        os.remove(bb.data.getVar('BNFILE',e.data, True))
    except:
        pass
    file = open(bb.data.getVar('BNFILE',e.data, True), "w")
    file.write(os.path.join(bn, bb.data.getVar('BUILDNAME', e.data, True)))
    file.close()

def get_bn(e):
    file = open(bb.data.getVar('BNFILE',e.data, True))
    bn = file.readline()
    file.close()
    return bn

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

    if isinstance(e, bb.event.BuildStarted):
        ##############################################
        # at first pass make the buildstats heriarchy and then
        # set the buildname
        ##############################################
        try:
            bb.mkdirhier(bb.data.getVar('BUILDSTATS_BASE', e.data, True))
        except:
            pass
        set_bn(e)
        bn = get_bn(e)
        bb.warn(bn)
        bsdir = os.path.join(bb.data.getVar('BUILDSTATS_BASE', e.data, True), bn)
        try:
            bb.mkdirhier(bsdir)
        except:
            pass
        set_timedata("__timedata_build", e.data)
        build_time = os.path.join(bsdir, "build_stats")
        # write start of build into build_time
        file = open(build_time,"a")
        # We do this here because subprocess within BuildStarted is messy
        host_info = subprocess.Popen(["uname", "-a"], stdout=subprocess.PIPE).stdout.read() 
        file.write("Host Info: %s" % host_info)
        file.write("Build Started: %0.2f \n" % time.time())
        file.close()
                
    elif isinstance(e, bb.event.BuildCompleted):
        bn=get_bn(e)
        timedata = get_timedata("__timedata_build", e.data)
        if not timedata:
            return
        time, cpu = timedata
        bsdir = os.path.join(bb.data.getVar('BUILDSTATS_BASE', e.data, True), bn)
        build_time = os.path.join(bsdir, "build_stats")
        # write end of build and cpu used into build_time
        file = open(build_time, "a")
        file.write("Elapsed time: %0.2f seconds \n" % (time))
        if cpu:
            file.write("CPU usage: %0.1f%% \n" % cpu)
        file.close()


    if isinstance(e, bb.build.TaskStarted):
        bn=get_bn(e)
        set_timedata("__timedata_task", e.data)
        
        bsdir = os.path.join(bb.data.getVar('BUILDSTATS_BASE', e.data, True), bn)
        taskdir = os.path.join(bsdir, bb.data.expand("${PF}", e.data))
        try:
            bb.mkdirhier(taskdir)
        except:
            pass
        # write into the task event file the name and start time
        file = open(os.path.join(taskdir, e.task), "a")
        file.write("Event: %s \n" % bb.event.getName(e))
        file.write("Started: %0.2f \n" % time.time())
        file.close()

    elif isinstance(e, bb.build.TaskSucceeded):
        bn=get_bn(e)
        timedata = get_timedata("__timedata_task", e.data)
        if not timedata:
            return
        time, cpu = timedata
        bsdir = os.path.join(bb.data.getVar('BUILDSTATS_BASE', e.data, True), bn)
        taskdir = os.path.join(bsdir, bb.data.expand("${PF}", e.data))
        file = open(os.path.join(taskdir, e.task), "a")
        file.write(bb.data.expand("${PF}: %s: Elapsed time: %0.2f seconds \n" %
                               (e.task, time), e.data))
        if cpu:
            file.write("CPU usage: %0.1f%% \n" % cpu)
    	
    	file.write("Status: PASSED")
        file.close()

        ##############################################
        # Alot of metric gathering occurs here. 
        # Reminder: I stripped out some in process stuff here
        ##############################################

        if e.task == "do_rootfs":
            bs=os.path.join(bsdir, "build_stats")
            file = open(bs,"a") 
            rootfs = bb.data.getVar('IMAGE_ROOTFS', e.data, True)
            rootfs_size = subprocess.Popen(["du", "-sh", rootfs], stdout=subprocess.PIPE).stdout.read() 
            file.write("Uncompressed Rootfs size: %s" % rootfs_size)
            file.close()

    elif isinstance(e, bb.build.TaskFailed):
        bn=get_bn(e)
        timedata = get_timedata("__timedata_task", e.data)
        if not timedata:
            return
        time, cpu = timedata
        bsdir = os.path.join(bb.data.getVar('BUILDSTATS_BASE', e.data, True), bn)
        taskdir = os.path.join(bsdir, bb.data.expand("${PF}", e.data))
        ##############################################
        # If the task fails dump the regular data. 
        # fgrep -R "FAILED" <bsdir> 
        # will grep all the events that failed. 
        ##############################################
        file = open(os.path.join(taskdir, e.task), "a")
        file.write(bb.data.expand("${PF}: %s: Elapsed time: %0.2f seconds \n" %
                               (e.task, time), e.data))
        if cpu:
            file.write("CPU usage: %0.1f%% \n" % cpu)
        file.write("Status: FAILED")
        file.close()
        ##############################################
        # 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")
        file = open(build_status,"a")
        file.write(bb.data.expand("Failed at: ${PF} at task: %s \n", e.task))
        file.close()

}

addhandler run_buildstats