Shinichiro Hamaji | b69bf8a | 2015-06-10 14:52:06 +0900 | [diff] [blame] | 1 | // Copyright 2015 Google Inc. All rights reserved |
| 2 | // |
| 3 | // Licensed under the Apache License, Version 2.0 (the "License"); |
| 4 | // you may not use this file except in compliance with the License. |
| 5 | // You may obtain a copy of the License at |
| 6 | // |
| 7 | // http://www.apache.org/licenses/LICENSE-2.0 |
| 8 | // |
| 9 | // Unless required by applicable law or agreed to in writing, software |
| 10 | // distributed under the License is distributed on an "AS IS" BASIS, |
| 11 | // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 12 | // See the License for the specific language governing permissions and |
| 13 | // limitations under the License. |
| 14 | |
Fumitoshi Ukai | 6ac7f69 | 2015-04-15 17:13:51 +0900 | [diff] [blame] | 15 | package main |
| 16 | |
| 17 | import ( |
| 18 | "fmt" |
Fumitoshi Ukai | 432a242 | 2015-06-11 15:16:29 +0900 | [diff] [blame] | 19 | "io" |
| 20 | "os" |
Fumitoshi Ukai | 6ac7f69 | 2015-04-15 17:13:51 +0900 | [diff] [blame] | 21 | "sort" |
Fumitoshi Ukai | f543f4d | 2015-06-15 15:21:47 +0900 | [diff] [blame] | 22 | "sync" |
Fumitoshi Ukai | 6ac7f69 | 2015-04-15 17:13:51 +0900 | [diff] [blame] | 23 | "time" |
| 24 | ) |
| 25 | |
Fumitoshi Ukai | 432a242 | 2015-06-11 15:16:29 +0900 | [diff] [blame] | 26 | type traceEventT struct { |
Fumitoshi Ukai | f543f4d | 2015-06-15 15:21:47 +0900 | [diff] [blame] | 27 | mu sync.Mutex |
Fumitoshi Ukai | 432a242 | 2015-06-11 15:16:29 +0900 | [diff] [blame] | 28 | f io.WriteCloser |
| 29 | t0 time.Time |
| 30 | pid int |
| 31 | } |
| 32 | |
Fumitoshi Ukai | f543f4d | 2015-06-15 15:21:47 +0900 | [diff] [blame] | 33 | const ( |
| 34 | traceEventMain = iota + 1 |
| 35 | traceEventFindCache |
Fumitoshi Ukai | 8341013 | 2015-06-15 14:50:07 +0900 | [diff] [blame] | 36 | traceEventFindCacheLeaves |
| 37 | traceEventFindCacheFiles |
Fumitoshi Ukai | f543f4d | 2015-06-15 15:21:47 +0900 | [diff] [blame] | 38 | ) |
| 39 | |
Fumitoshi Ukai | 432a242 | 2015-06-11 15:16:29 +0900 | [diff] [blame] | 40 | var traceEvent traceEventT |
| 41 | |
| 42 | func (t *traceEventT) start(f io.WriteCloser) { |
| 43 | t.f = f |
| 44 | t.t0 = time.Now() |
| 45 | fmt.Fprint(t.f, "[ ") |
| 46 | } |
| 47 | |
| 48 | func (t *traceEventT) enabled() bool { |
| 49 | return t.f != nil |
| 50 | } |
| 51 | |
| 52 | func (t *traceEventT) stop() { |
| 53 | fmt.Fprint(t.f, "\n]\n") |
| 54 | t.f.Close() |
| 55 | } |
| 56 | |
| 57 | type event struct { |
| 58 | name, v string |
Fumitoshi Ukai | f543f4d | 2015-06-15 15:21:47 +0900 | [diff] [blame] | 59 | tid int |
Fumitoshi Ukai | 432a242 | 2015-06-11 15:16:29 +0900 | [diff] [blame] | 60 | t time.Time |
Fumitoshi Ukai | cb393c5 | 2015-06-11 15:34:10 +0900 | [diff] [blame] | 61 | emit bool |
Fumitoshi Ukai | 432a242 | 2015-06-11 15:16:29 +0900 | [diff] [blame] | 62 | } |
| 63 | |
Fumitoshi Ukai | 8341013 | 2015-06-15 14:50:07 +0900 | [diff] [blame] | 64 | func (t *traceEventT) begin(name string, v Value, tid int) event { |
Fumitoshi Ukai | 432a242 | 2015-06-11 15:16:29 +0900 | [diff] [blame] | 65 | var e event |
Fumitoshi Ukai | f543f4d | 2015-06-15 15:21:47 +0900 | [diff] [blame] | 66 | e.tid = tid |
Fumitoshi Ukai | 432a242 | 2015-06-11 15:16:29 +0900 | [diff] [blame] | 67 | e.t = time.Now() |
Fumitoshi Ukai | 6ab72dd | 2015-06-11 15:45:52 +0900 | [diff] [blame] | 68 | if t.f != nil || katiEvalStatsFlag { |
| 69 | e.name = name |
Fumitoshi Ukai | 8341013 | 2015-06-15 14:50:07 +0900 | [diff] [blame] | 70 | e.v = v.String() |
Fumitoshi Ukai | 6ab72dd | 2015-06-11 15:45:52 +0900 | [diff] [blame] | 71 | } |
Fumitoshi Ukai | 432a242 | 2015-06-11 15:16:29 +0900 | [diff] [blame] | 72 | if t.f != nil { |
Fumitoshi Ukai | f543f4d | 2015-06-15 15:21:47 +0900 | [diff] [blame] | 73 | e.emit = name == "include" || name == "shell" || name == "findcache" |
Fumitoshi Ukai | cb393c5 | 2015-06-11 15:34:10 +0900 | [diff] [blame] | 74 | if e.emit { |
Fumitoshi Ukai | 8341013 | 2015-06-15 14:50:07 +0900 | [diff] [blame] | 75 | t.emit("B", e, e.t.Sub(t.t0)) |
Fumitoshi Ukai | 432a242 | 2015-06-11 15:16:29 +0900 | [diff] [blame] | 76 | } |
| 77 | } |
| 78 | return e |
| 79 | } |
| 80 | |
Fumitoshi Ukai | 8341013 | 2015-06-15 14:50:07 +0900 | [diff] [blame] | 81 | func (t *traceEventT) emit(ph string, e event, ts time.Duration) { |
Fumitoshi Ukai | f543f4d | 2015-06-15 15:21:47 +0900 | [diff] [blame] | 82 | t.mu.Lock() |
| 83 | defer t.mu.Unlock() |
Fumitoshi Ukai | 8341013 | 2015-06-15 14:50:07 +0900 | [diff] [blame] | 84 | |
| 85 | if t.pid == 0 { |
| 86 | t.pid = os.Getpid() |
| 87 | } else { |
| 88 | fmt.Fprintf(t.f, ",\n") |
| 89 | } |
| 90 | fmt.Fprintf(t.f, `{"pid":%d,"tid":%d,"ts":%d,"ph":%q,"cat":%q,"name":%q,"args":{}}`, |
| 91 | t.pid, |
| 92 | e.tid, |
| 93 | ts.Nanoseconds()/1e3, |
| 94 | ph, |
| 95 | e.name, |
| 96 | e.v, |
| 97 | ) |
| 98 | } |
| 99 | |
| 100 | func (t *traceEventT) end(e event) { |
Fumitoshi Ukai | 432a242 | 2015-06-11 15:16:29 +0900 | [diff] [blame] | 101 | if t.f != nil { |
Fumitoshi Ukai | cb393c5 | 2015-06-11 15:34:10 +0900 | [diff] [blame] | 102 | if e.emit { |
Fumitoshi Ukai | 8341013 | 2015-06-15 14:50:07 +0900 | [diff] [blame] | 103 | t.emit("E", e, time.Since(t.t0)) |
Fumitoshi Ukai | 432a242 | 2015-06-11 15:16:29 +0900 | [diff] [blame] | 104 | } |
| 105 | } |
| 106 | addStats(e.name, e.v, e.t) |
| 107 | } |
| 108 | |
Fumitoshi Ukai | 6ac7f69 | 2015-04-15 17:13:51 +0900 | [diff] [blame] | 109 | type statsData struct { |
| 110 | Name string |
| 111 | Count int |
| 112 | Longest time.Duration |
| 113 | Total time.Duration |
| 114 | } |
| 115 | |
| 116 | var stats = map[string]statsData{} |
| 117 | |
Fumitoshi Ukai | 432a242 | 2015-06-11 15:16:29 +0900 | [diff] [blame] | 118 | func addStats(name, v string, t time.Time) { |
Fumitoshi Ukai | 586b02a | 2015-05-08 00:23:10 +0900 | [diff] [blame] | 119 | if !katiEvalStatsFlag { |
Fumitoshi Ukai | 6ac7f69 | 2015-04-15 17:13:51 +0900 | [diff] [blame] | 120 | return |
| 121 | } |
Fumitoshi Ukai | 358c68a | 2015-06-08 13:12:55 +0900 | [diff] [blame] | 122 | d := time.Since(t) |
Fumitoshi Ukai | 432a242 | 2015-06-11 15:16:29 +0900 | [diff] [blame] | 123 | key := fmt.Sprintf("%s:%s", name, v) |
Fumitoshi Ukai | 6ac7f69 | 2015-04-15 17:13:51 +0900 | [diff] [blame] | 124 | s := stats[key] |
| 125 | if d > s.Longest { |
| 126 | s.Longest = d |
| 127 | } |
| 128 | s.Total += d |
| 129 | s.Count++ |
| 130 | stats[key] = s |
| 131 | } |
| 132 | |
| 133 | func dumpStats() { |
Fumitoshi Ukai | 586b02a | 2015-05-08 00:23:10 +0900 | [diff] [blame] | 134 | if !katiEvalStatsFlag { |
Fumitoshi Ukai | 6ac7f69 | 2015-04-15 17:13:51 +0900 | [diff] [blame] | 135 | return |
| 136 | } |
| 137 | var sv byTotalTime |
| 138 | for k, v := range stats { |
| 139 | v.Name = k |
| 140 | sv = append(sv, v) |
| 141 | } |
| 142 | sort.Sort(sv) |
Fumitoshi Ukai | 47f401f | 2015-04-30 17:29:50 +0900 | [diff] [blame] | 143 | fmt.Println("count,longest(ns),total(ns),longest,total,name") |
Fumitoshi Ukai | 6ac7f69 | 2015-04-15 17:13:51 +0900 | [diff] [blame] | 144 | for _, s := range sv { |
Fumitoshi Ukai | 47f401f | 2015-04-30 17:29:50 +0900 | [diff] [blame] | 145 | fmt.Printf("%d,%d,%d,%v,%v,%s\n", s.Count, s.Longest, s.Total, s.Longest, s.Total, s.Name) |
Fumitoshi Ukai | 6ac7f69 | 2015-04-15 17:13:51 +0900 | [diff] [blame] | 146 | } |
| 147 | } |
| 148 | |
| 149 | type byTotalTime []statsData |
| 150 | |
| 151 | func (b byTotalTime) Len() int { return len(b) } |
| 152 | func (b byTotalTime) Swap(i, j int) { b[i], b[j] = b[j], b[i] } |
| 153 | func (b byTotalTime) Less(i, j int) bool { |
| 154 | return b[i].Total > b[j].Total |
| 155 | } |