github.com/eun/go@v0.0.0-20170811110501-92cfd07a6cfd/src/runtime/pprof/pprof.go (about) 1 // Copyright 2010 The Go Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style 3 // license that can be found in the LICENSE file. 4 5 // Package pprof writes runtime profiling data in the format expected 6 // by the pprof visualization tool. 7 // 8 // Profiling a Go program 9 // 10 // The first step to profiling a Go program is to enable profiling. 11 // Support for profiling benchmarks built with the standard testing 12 // package is built into go test. For example, the following command 13 // runs benchmarks in the current directory and writes the CPU and 14 // memory profiles to cpu.prof and mem.prof: 15 // 16 // go test -cpuprofile cpu.prof -memprofile mem.prof -bench . 17 // 18 // To add equivalent profiling support to a standalone program, add 19 // code like the following to your main function: 20 // 21 // var cpuprofile = flag.String("cpuprofile", "", "write cpu profile `file`") 22 // var memprofile = flag.String("memprofile", "", "write memory profile to `file`") 23 // 24 // func main() { 25 // flag.Parse() 26 // if *cpuprofile != "" { 27 // f, err := os.Create(*cpuprofile) 28 // if err != nil { 29 // log.Fatal("could not create CPU profile: ", err) 30 // } 31 // if err := pprof.StartCPUProfile(f); err != nil { 32 // log.Fatal("could not start CPU profile: ", err) 33 // } 34 // defer pprof.StopCPUProfile() 35 // } 36 // 37 // // ... rest of the program ... 38 // 39 // if *memprofile != "" { 40 // f, err := os.Create(*memprofile) 41 // if err != nil { 42 // log.Fatal("could not create memory profile: ", err) 43 // } 44 // runtime.GC() // get up-to-date statistics 45 // if err := pprof.WriteHeapProfile(f); err != nil { 46 // log.Fatal("could not write memory profile: ", err) 47 // } 48 // f.Close() 49 // } 50 // } 51 // 52 // There is also a standard HTTP interface to profiling data. Adding 53 // the following line will install handlers under the /debug/pprof/ 54 // URL to download live profiles: 55 // 56 // import _ "net/http/pprof" 57 // 58 // See the net/http/pprof package for more details. 59 // 60 // Profiles can then be visualized with the pprof tool: 61 // 62 // go tool pprof cpu.prof 63 // 64 // There are many commands available from the pprof command line. 65 // Commonly used commands include "top", which prints a summary of the 66 // top program hot-spots, and "web", which opens an interactive graph 67 // of hot-spots and their call graphs. Use "help" for information on 68 // all pprof commands. 69 // 70 // For more information about pprof, see 71 // https://github.com/google/pprof/blob/master/doc/pprof.md. 72 package pprof 73 74 import ( 75 "bufio" 76 "bytes" 77 "fmt" 78 "io" 79 "runtime" 80 "sort" 81 "strings" 82 "sync" 83 "text/tabwriter" 84 "time" 85 "unsafe" 86 ) 87 88 // BUG(rsc): Profiles are only as good as the kernel support used to generate them. 89 // See https://golang.org/issue/13841 for details about known problems. 90 91 // A Profile is a collection of stack traces showing the call sequences 92 // that led to instances of a particular event, such as allocation. 93 // Packages can create and maintain their own profiles; the most common 94 // use is for tracking resources that must be explicitly closed, such as files 95 // or network connections. 96 // 97 // A Profile's methods can be called from multiple goroutines simultaneously. 98 // 99 // Each Profile has a unique name. A few profiles are predefined: 100 // 101 // goroutine - stack traces of all current goroutines 102 // heap - a sampling of all heap allocations 103 // threadcreate - stack traces that led to the creation of new OS threads 104 // block - stack traces that led to blocking on synchronization primitives 105 // mutex - stack traces of holders of contended mutexes 106 // 107 // These predefined profiles maintain themselves and panic on an explicit 108 // Add or Remove method call. 109 // 110 // The heap profile reports statistics as of the most recently completed 111 // garbage collection; it elides more recent allocation to avoid skewing 112 // the profile away from live data and toward garbage. 113 // If there has been no garbage collection at all, the heap profile reports 114 // all known allocations. This exception helps mainly in programs running 115 // without garbage collection enabled, usually for debugging purposes. 116 // 117 // The CPU profile is not available as a Profile. It has a special API, 118 // the StartCPUProfile and StopCPUProfile functions, because it streams 119 // output to a writer during profiling. 120 // 121 type Profile struct { 122 name string 123 mu sync.Mutex 124 m map[interface{}][]uintptr 125 count func() int 126 write func(io.Writer, int) error 127 } 128 129 // profiles records all registered profiles. 130 var profiles struct { 131 mu sync.Mutex 132 m map[string]*Profile 133 } 134 135 var goroutineProfile = &Profile{ 136 name: "goroutine", 137 count: countGoroutine, 138 write: writeGoroutine, 139 } 140 141 var threadcreateProfile = &Profile{ 142 name: "threadcreate", 143 count: countThreadCreate, 144 write: writeThreadCreate, 145 } 146 147 var heapProfile = &Profile{ 148 name: "heap", 149 count: countHeap, 150 write: writeHeap, 151 } 152 153 var blockProfile = &Profile{ 154 name: "block", 155 count: countBlock, 156 write: writeBlock, 157 } 158 159 var mutexProfile = &Profile{ 160 name: "mutex", 161 count: countMutex, 162 write: writeMutex, 163 } 164 165 func lockProfiles() { 166 profiles.mu.Lock() 167 if profiles.m == nil { 168 // Initial built-in profiles. 169 profiles.m = map[string]*Profile{ 170 "goroutine": goroutineProfile, 171 "threadcreate": threadcreateProfile, 172 "heap": heapProfile, 173 "block": blockProfile, 174 "mutex": mutexProfile, 175 } 176 } 177 } 178 179 func unlockProfiles() { 180 profiles.mu.Unlock() 181 } 182 183 // NewProfile creates a new profile with the given name. 184 // If a profile with that name already exists, NewProfile panics. 185 // The convention is to use a 'import/path.' prefix to create 186 // separate name spaces for each package. 187 // For compatibility with various tools that read pprof data, 188 // profile names should not contain spaces. 189 func NewProfile(name string) *Profile { 190 lockProfiles() 191 defer unlockProfiles() 192 if name == "" { 193 panic("pprof: NewProfile with empty name") 194 } 195 if profiles.m[name] != nil { 196 panic("pprof: NewProfile name already in use: " + name) 197 } 198 p := &Profile{ 199 name: name, 200 m: map[interface{}][]uintptr{}, 201 } 202 profiles.m[name] = p 203 return p 204 } 205 206 // Lookup returns the profile with the given name, or nil if no such profile exists. 207 func Lookup(name string) *Profile { 208 lockProfiles() 209 defer unlockProfiles() 210 return profiles.m[name] 211 } 212 213 // Profiles returns a slice of all the known profiles, sorted by name. 214 func Profiles() []*Profile { 215 lockProfiles() 216 defer unlockProfiles() 217 218 all := make([]*Profile, 0, len(profiles.m)) 219 for _, p := range profiles.m { 220 all = append(all, p) 221 } 222 223 sort.Slice(all, func(i, j int) bool { return all[i].name < all[j].name }) 224 return all 225 } 226 227 // Name returns this profile's name, which can be passed to Lookup to reobtain the profile. 228 func (p *Profile) Name() string { 229 return p.name 230 } 231 232 // Count returns the number of execution stacks currently in the profile. 233 func (p *Profile) Count() int { 234 p.mu.Lock() 235 defer p.mu.Unlock() 236 if p.count != nil { 237 return p.count() 238 } 239 return len(p.m) 240 } 241 242 // Add adds the current execution stack to the profile, associated with value. 243 // Add stores value in an internal map, so value must be suitable for use as 244 // a map key and will not be garbage collected until the corresponding 245 // call to Remove. Add panics if the profile already contains a stack for value. 246 // 247 // The skip parameter has the same meaning as runtime.Caller's skip 248 // and controls where the stack trace begins. Passing skip=0 begins the 249 // trace in the function calling Add. For example, given this 250 // execution stack: 251 // 252 // Add 253 // called from rpc.NewClient 254 // called from mypkg.Run 255 // called from main.main 256 // 257 // Passing skip=0 begins the stack trace at the call to Add inside rpc.NewClient. 258 // Passing skip=1 begins the stack trace at the call to NewClient inside mypkg.Run. 259 // 260 func (p *Profile) Add(value interface{}, skip int) { 261 if p.name == "" { 262 panic("pprof: use of uninitialized Profile") 263 } 264 if p.write != nil { 265 panic("pprof: Add called on built-in Profile " + p.name) 266 } 267 268 stk := make([]uintptr, 32) 269 n := runtime.Callers(skip+1, stk[:]) 270 stk = stk[:n] 271 if len(stk) == 0 { 272 // The value for skip is too large, and there's no stack trace to record. 273 stk = []uintptr{funcPC(lostProfileEvent)} 274 } 275 276 p.mu.Lock() 277 defer p.mu.Unlock() 278 if p.m[value] != nil { 279 panic("pprof: Profile.Add of duplicate value") 280 } 281 p.m[value] = stk 282 } 283 284 // Remove removes the execution stack associated with value from the profile. 285 // It is a no-op if the value is not in the profile. 286 func (p *Profile) Remove(value interface{}) { 287 p.mu.Lock() 288 defer p.mu.Unlock() 289 delete(p.m, value) 290 } 291 292 // WriteTo writes a pprof-formatted snapshot of the profile to w. 293 // If a write to w returns an error, WriteTo returns that error. 294 // Otherwise, WriteTo returns nil. 295 // 296 // The debug parameter enables additional output. 297 // Passing debug=0 prints only the hexadecimal addresses that pprof needs. 298 // Passing debug=1 adds comments translating addresses to function names 299 // and line numbers, so that a programmer can read the profile without tools. 300 // 301 // The predefined profiles may assign meaning to other debug values; 302 // for example, when printing the "goroutine" profile, debug=2 means to 303 // print the goroutine stacks in the same form that a Go program uses 304 // when dying due to an unrecovered panic. 305 func (p *Profile) WriteTo(w io.Writer, debug int) error { 306 if p.name == "" { 307 panic("pprof: use of zero Profile") 308 } 309 if p.write != nil { 310 return p.write(w, debug) 311 } 312 313 // Obtain consistent snapshot under lock; then process without lock. 314 p.mu.Lock() 315 all := make([][]uintptr, 0, len(p.m)) 316 for _, stk := range p.m { 317 all = append(all, stk) 318 } 319 p.mu.Unlock() 320 321 // Map order is non-deterministic; make output deterministic. 322 sort.Sort(stackProfile(all)) 323 324 return printCountProfile(w, debug, p.name, stackProfile(all)) 325 } 326 327 type stackProfile [][]uintptr 328 329 func (x stackProfile) Len() int { return len(x) } 330 func (x stackProfile) Stack(i int) []uintptr { return x[i] } 331 func (x stackProfile) Swap(i, j int) { x[i], x[j] = x[j], x[i] } 332 func (x stackProfile) Less(i, j int) bool { 333 t, u := x[i], x[j] 334 for k := 0; k < len(t) && k < len(u); k++ { 335 if t[k] != u[k] { 336 return t[k] < u[k] 337 } 338 } 339 return len(t) < len(u) 340 } 341 342 // A countProfile is a set of stack traces to be printed as counts 343 // grouped by stack trace. There are multiple implementations: 344 // all that matters is that we can find out how many traces there are 345 // and obtain each trace in turn. 346 type countProfile interface { 347 Len() int 348 Stack(i int) []uintptr 349 } 350 351 // printCountProfile prints a countProfile at the specified debug level. 352 // The profile will be in compressed proto format unless debug is nonzero. 353 func printCountProfile(w io.Writer, debug int, name string, p countProfile) error { 354 // Build count of each stack. 355 var buf bytes.Buffer 356 key := func(stk []uintptr) string { 357 buf.Reset() 358 fmt.Fprintf(&buf, "@") 359 for _, pc := range stk { 360 fmt.Fprintf(&buf, " %#x", pc) 361 } 362 return buf.String() 363 } 364 count := map[string]int{} 365 index := map[string]int{} 366 var keys []string 367 n := p.Len() 368 for i := 0; i < n; i++ { 369 k := key(p.Stack(i)) 370 if count[k] == 0 { 371 index[k] = i 372 keys = append(keys, k) 373 } 374 count[k]++ 375 } 376 377 sort.Sort(&keysByCount{keys, count}) 378 379 if debug > 0 { 380 // Print debug profile in legacy format 381 tw := tabwriter.NewWriter(w, 1, 8, 1, '\t', 0) 382 fmt.Fprintf(tw, "%s profile: total %d\n", name, p.Len()) 383 for _, k := range keys { 384 fmt.Fprintf(tw, "%d %s\n", count[k], k) 385 printStackRecord(tw, p.Stack(index[k]), false) 386 } 387 return tw.Flush() 388 } 389 390 // Output profile in protobuf form. 391 b := newProfileBuilder(w) 392 b.pbValueType(tagProfile_PeriodType, name, "count") 393 b.pb.int64Opt(tagProfile_Period, 1) 394 b.pbValueType(tagProfile_SampleType, name, "count") 395 396 values := []int64{0} 397 var locs []uint64 398 for _, k := range keys { 399 values[0] = int64(count[k]) 400 locs = locs[:0] 401 for _, addr := range p.Stack(index[k]) { 402 // For count profiles, all stack addresses are 403 // return PCs, which is what locForPC expects. 404 l := b.locForPC(addr) 405 if l == 0 { // runtime.goexit 406 continue 407 } 408 locs = append(locs, l) 409 } 410 b.pbSample(values, locs, nil) 411 } 412 b.build() 413 return nil 414 } 415 416 // keysByCount sorts keys with higher counts first, breaking ties by key string order. 417 type keysByCount struct { 418 keys []string 419 count map[string]int 420 } 421 422 func (x *keysByCount) Len() int { return len(x.keys) } 423 func (x *keysByCount) Swap(i, j int) { x.keys[i], x.keys[j] = x.keys[j], x.keys[i] } 424 func (x *keysByCount) Less(i, j int) bool { 425 ki, kj := x.keys[i], x.keys[j] 426 ci, cj := x.count[ki], x.count[kj] 427 if ci != cj { 428 return ci > cj 429 } 430 return ki < kj 431 } 432 433 // printStackRecord prints the function + source line information 434 // for a single stack trace. 435 func printStackRecord(w io.Writer, stk []uintptr, allFrames bool) { 436 show := allFrames 437 frames := runtime.CallersFrames(stk) 438 for { 439 frame, more := frames.Next() 440 name := frame.Function 441 if name == "" { 442 show = true 443 fmt.Fprintf(w, "#\t%#x\n", frame.PC) 444 } else if name != "runtime.goexit" && (show || !strings.HasPrefix(name, "runtime.")) { 445 // Hide runtime.goexit and any runtime functions at the beginning. 446 // This is useful mainly for allocation traces. 447 show = true 448 fmt.Fprintf(w, "#\t%#x\t%s+%#x\t%s:%d\n", frame.PC, name, frame.PC-frame.Entry, frame.File, frame.Line) 449 } 450 if !more { 451 break 452 } 453 } 454 if !show { 455 // We didn't print anything; do it again, 456 // and this time include runtime functions. 457 printStackRecord(w, stk, true) 458 return 459 } 460 fmt.Fprintf(w, "\n") 461 } 462 463 // Interface to system profiles. 464 465 // WriteHeapProfile is shorthand for Lookup("heap").WriteTo(w, 0). 466 // It is preserved for backwards compatibility. 467 func WriteHeapProfile(w io.Writer) error { 468 return writeHeap(w, 0) 469 } 470 471 // countHeap returns the number of records in the heap profile. 472 func countHeap() int { 473 n, _ := runtime.MemProfile(nil, true) 474 return n 475 } 476 477 // writeHeap writes the current runtime heap profile to w. 478 func writeHeap(w io.Writer, debug int) error { 479 // Find out how many records there are (MemProfile(nil, true)), 480 // allocate that many records, and get the data. 481 // There's a race—more records might be added between 482 // the two calls—so allocate a few extra records for safety 483 // and also try again if we're very unlucky. 484 // The loop should only execute one iteration in the common case. 485 var p []runtime.MemProfileRecord 486 n, ok := runtime.MemProfile(nil, true) 487 for { 488 // Allocate room for a slightly bigger profile, 489 // in case a few more entries have been added 490 // since the call to MemProfile. 491 p = make([]runtime.MemProfileRecord, n+50) 492 n, ok = runtime.MemProfile(p, true) 493 if ok { 494 p = p[0:n] 495 break 496 } 497 // Profile grew; try again. 498 } 499 500 if debug == 0 { 501 return writeHeapProto(w, p, int64(runtime.MemProfileRate)) 502 } 503 504 sort.Slice(p, func(i, j int) bool { return p[i].InUseBytes() > p[j].InUseBytes() }) 505 506 b := bufio.NewWriter(w) 507 tw := tabwriter.NewWriter(b, 1, 8, 1, '\t', 0) 508 w = tw 509 510 var total runtime.MemProfileRecord 511 for i := range p { 512 r := &p[i] 513 total.AllocBytes += r.AllocBytes 514 total.AllocObjects += r.AllocObjects 515 total.FreeBytes += r.FreeBytes 516 total.FreeObjects += r.FreeObjects 517 } 518 519 // Technically the rate is MemProfileRate not 2*MemProfileRate, 520 // but early versions of the C++ heap profiler reported 2*MemProfileRate, 521 // so that's what pprof has come to expect. 522 fmt.Fprintf(w, "heap profile: %d: %d [%d: %d] @ heap/%d\n", 523 total.InUseObjects(), total.InUseBytes(), 524 total.AllocObjects, total.AllocBytes, 525 2*runtime.MemProfileRate) 526 527 for i := range p { 528 r := &p[i] 529 fmt.Fprintf(w, "%d: %d [%d: %d] @", 530 r.InUseObjects(), r.InUseBytes(), 531 r.AllocObjects, r.AllocBytes) 532 for _, pc := range r.Stack() { 533 fmt.Fprintf(w, " %#x", pc) 534 } 535 fmt.Fprintf(w, "\n") 536 printStackRecord(w, r.Stack(), false) 537 } 538 539 // Print memstats information too. 540 // Pprof will ignore, but useful for people 541 s := new(runtime.MemStats) 542 runtime.ReadMemStats(s) 543 fmt.Fprintf(w, "\n# runtime.MemStats\n") 544 fmt.Fprintf(w, "# Alloc = %d\n", s.Alloc) 545 fmt.Fprintf(w, "# TotalAlloc = %d\n", s.TotalAlloc) 546 fmt.Fprintf(w, "# Sys = %d\n", s.Sys) 547 fmt.Fprintf(w, "# Lookups = %d\n", s.Lookups) 548 fmt.Fprintf(w, "# Mallocs = %d\n", s.Mallocs) 549 fmt.Fprintf(w, "# Frees = %d\n", s.Frees) 550 551 fmt.Fprintf(w, "# HeapAlloc = %d\n", s.HeapAlloc) 552 fmt.Fprintf(w, "# HeapSys = %d\n", s.HeapSys) 553 fmt.Fprintf(w, "# HeapIdle = %d\n", s.HeapIdle) 554 fmt.Fprintf(w, "# HeapInuse = %d\n", s.HeapInuse) 555 fmt.Fprintf(w, "# HeapReleased = %d\n", s.HeapReleased) 556 fmt.Fprintf(w, "# HeapObjects = %d\n", s.HeapObjects) 557 558 fmt.Fprintf(w, "# Stack = %d / %d\n", s.StackInuse, s.StackSys) 559 fmt.Fprintf(w, "# MSpan = %d / %d\n", s.MSpanInuse, s.MSpanSys) 560 fmt.Fprintf(w, "# MCache = %d / %d\n", s.MCacheInuse, s.MCacheSys) 561 fmt.Fprintf(w, "# BuckHashSys = %d\n", s.BuckHashSys) 562 fmt.Fprintf(w, "# GCSys = %d\n", s.GCSys) 563 fmt.Fprintf(w, "# OtherSys = %d\n", s.OtherSys) 564 565 fmt.Fprintf(w, "# NextGC = %d\n", s.NextGC) 566 fmt.Fprintf(w, "# LastGC = %d\n", s.LastGC) 567 fmt.Fprintf(w, "# PauseNs = %d\n", s.PauseNs) 568 fmt.Fprintf(w, "# PauseEnd = %d\n", s.PauseEnd) 569 fmt.Fprintf(w, "# NumGC = %d\n", s.NumGC) 570 fmt.Fprintf(w, "# NumForcedGC = %d\n", s.NumForcedGC) 571 fmt.Fprintf(w, "# GCCPUFraction = %v\n", s.GCCPUFraction) 572 fmt.Fprintf(w, "# DebugGC = %v\n", s.DebugGC) 573 574 tw.Flush() 575 return b.Flush() 576 } 577 578 // countThreadCreate returns the size of the current ThreadCreateProfile. 579 func countThreadCreate() int { 580 n, _ := runtime.ThreadCreateProfile(nil) 581 return n 582 } 583 584 // writeThreadCreate writes the current runtime ThreadCreateProfile to w. 585 func writeThreadCreate(w io.Writer, debug int) error { 586 return writeRuntimeProfile(w, debug, "threadcreate", runtime.ThreadCreateProfile) 587 } 588 589 // countGoroutine returns the number of goroutines. 590 func countGoroutine() int { 591 return runtime.NumGoroutine() 592 } 593 594 // writeGoroutine writes the current runtime GoroutineProfile to w. 595 func writeGoroutine(w io.Writer, debug int) error { 596 if debug >= 2 { 597 return writeGoroutineStacks(w) 598 } 599 return writeRuntimeProfile(w, debug, "goroutine", runtime.GoroutineProfile) 600 } 601 602 func writeGoroutineStacks(w io.Writer) error { 603 // We don't know how big the buffer needs to be to collect 604 // all the goroutines. Start with 1 MB and try a few times, doubling each time. 605 // Give up and use a truncated trace if 64 MB is not enough. 606 buf := make([]byte, 1<<20) 607 for i := 0; ; i++ { 608 n := runtime.Stack(buf, true) 609 if n < len(buf) { 610 buf = buf[:n] 611 break 612 } 613 if len(buf) >= 64<<20 { 614 // Filled 64 MB - stop there. 615 break 616 } 617 buf = make([]byte, 2*len(buf)) 618 } 619 _, err := w.Write(buf) 620 return err 621 } 622 623 func writeRuntimeProfile(w io.Writer, debug int, name string, fetch func([]runtime.StackRecord) (int, bool)) error { 624 // Find out how many records there are (fetch(nil)), 625 // allocate that many records, and get the data. 626 // There's a race—more records might be added between 627 // the two calls—so allocate a few extra records for safety 628 // and also try again if we're very unlucky. 629 // The loop should only execute one iteration in the common case. 630 var p []runtime.StackRecord 631 n, ok := fetch(nil) 632 for { 633 // Allocate room for a slightly bigger profile, 634 // in case a few more entries have been added 635 // since the call to ThreadProfile. 636 p = make([]runtime.StackRecord, n+10) 637 n, ok = fetch(p) 638 if ok { 639 p = p[0:n] 640 break 641 } 642 // Profile grew; try again. 643 } 644 645 return printCountProfile(w, debug, name, runtimeProfile(p)) 646 } 647 648 type runtimeProfile []runtime.StackRecord 649 650 func (p runtimeProfile) Len() int { return len(p) } 651 func (p runtimeProfile) Stack(i int) []uintptr { return p[i].Stack() } 652 653 var cpu struct { 654 sync.Mutex 655 profiling bool 656 done chan bool 657 } 658 659 // StartCPUProfile enables CPU profiling for the current process. 660 // While profiling, the profile will be buffered and written to w. 661 // StartCPUProfile returns an error if profiling is already enabled. 662 // 663 // On Unix-like systems, StartCPUProfile does not work by default for 664 // Go code built with -buildmode=c-archive or -buildmode=c-shared. 665 // StartCPUProfile relies on the SIGPROF signal, but that signal will 666 // be delivered to the main program's SIGPROF signal handler (if any) 667 // not to the one used by Go. To make it work, call os/signal.Notify 668 // for syscall.SIGPROF, but note that doing so may break any profiling 669 // being done by the main program. 670 func StartCPUProfile(w io.Writer) error { 671 // The runtime routines allow a variable profiling rate, 672 // but in practice operating systems cannot trigger signals 673 // at more than about 500 Hz, and our processing of the 674 // signal is not cheap (mostly getting the stack trace). 675 // 100 Hz is a reasonable choice: it is frequent enough to 676 // produce useful data, rare enough not to bog down the 677 // system, and a nice round number to make it easy to 678 // convert sample counts to seconds. Instead of requiring 679 // each client to specify the frequency, we hard code it. 680 const hz = 100 681 682 cpu.Lock() 683 defer cpu.Unlock() 684 if cpu.done == nil { 685 cpu.done = make(chan bool) 686 } 687 // Double-check. 688 if cpu.profiling { 689 return fmt.Errorf("cpu profiling already in use") 690 } 691 cpu.profiling = true 692 runtime.SetCPUProfileRate(hz) 693 go profileWriter(w) 694 return nil 695 } 696 697 // readProfile, provided by the runtime, returns the next chunk of 698 // binary CPU profiling stack trace data, blocking until data is available. 699 // If profiling is turned off and all the profile data accumulated while it was 700 // on has been returned, readProfile returns eof=true. 701 // The caller must save the returned data and tags before calling readProfile again. 702 func readProfile() (data []uint64, tags []unsafe.Pointer, eof bool) 703 704 func profileWriter(w io.Writer) { 705 b := newProfileBuilder(w) 706 var err error 707 for { 708 time.Sleep(100 * time.Millisecond) 709 data, tags, eof := readProfile() 710 if e := b.addCPUData(data, tags); e != nil && err == nil { 711 err = e 712 } 713 if eof { 714 break 715 } 716 } 717 if err != nil { 718 // The runtime should never produce an invalid or truncated profile. 719 // It drops records that can't fit into its log buffers. 720 panic("runtime/pprof: converting profile: " + err.Error()) 721 } 722 b.build() 723 cpu.done <- true 724 } 725 726 // StopCPUProfile stops the current CPU profile, if any. 727 // StopCPUProfile only returns after all the writes for the 728 // profile have completed. 729 func StopCPUProfile() { 730 cpu.Lock() 731 defer cpu.Unlock() 732 733 if !cpu.profiling { 734 return 735 } 736 cpu.profiling = false 737 runtime.SetCPUProfileRate(0) 738 <-cpu.done 739 } 740 741 // countBlock returns the number of records in the blocking profile. 742 func countBlock() int { 743 n, _ := runtime.BlockProfile(nil) 744 return n 745 } 746 747 // countMutex returns the number of records in the mutex profile. 748 func countMutex() int { 749 n, _ := runtime.MutexProfile(nil) 750 return n 751 } 752 753 // writeBlock writes the current blocking profile to w. 754 func writeBlock(w io.Writer, debug int) error { 755 var p []runtime.BlockProfileRecord 756 n, ok := runtime.BlockProfile(nil) 757 for { 758 p = make([]runtime.BlockProfileRecord, n+50) 759 n, ok = runtime.BlockProfile(p) 760 if ok { 761 p = p[:n] 762 break 763 } 764 } 765 766 sort.Slice(p, func(i, j int) bool { return p[i].Cycles > p[j].Cycles }) 767 768 b := bufio.NewWriter(w) 769 var tw *tabwriter.Writer 770 w = b 771 if debug > 0 { 772 tw = tabwriter.NewWriter(w, 1, 8, 1, '\t', 0) 773 w = tw 774 } 775 776 fmt.Fprintf(w, "--- contention:\n") 777 fmt.Fprintf(w, "cycles/second=%v\n", runtime_cyclesPerSecond()) 778 for i := range p { 779 r := &p[i] 780 fmt.Fprintf(w, "%v %v @", r.Cycles, r.Count) 781 for _, pc := range r.Stack() { 782 fmt.Fprintf(w, " %#x", pc) 783 } 784 fmt.Fprint(w, "\n") 785 if debug > 0 { 786 printStackRecord(w, r.Stack(), true) 787 } 788 } 789 790 if tw != nil { 791 tw.Flush() 792 } 793 return b.Flush() 794 } 795 796 // writeMutex writes the current mutex profile to w. 797 func writeMutex(w io.Writer, debug int) error { 798 // TODO(pjw): too much common code with writeBlock. FIX! 799 var p []runtime.BlockProfileRecord 800 n, ok := runtime.MutexProfile(nil) 801 for { 802 p = make([]runtime.BlockProfileRecord, n+50) 803 n, ok = runtime.MutexProfile(p) 804 if ok { 805 p = p[:n] 806 break 807 } 808 } 809 810 sort.Slice(p, func(i, j int) bool { return p[i].Cycles > p[j].Cycles }) 811 812 b := bufio.NewWriter(w) 813 var tw *tabwriter.Writer 814 w = b 815 if debug > 0 { 816 tw = tabwriter.NewWriter(w, 1, 8, 1, '\t', 0) 817 w = tw 818 } 819 820 fmt.Fprintf(w, "--- mutex:\n") 821 fmt.Fprintf(w, "cycles/second=%v\n", runtime_cyclesPerSecond()) 822 fmt.Fprintf(w, "sampling period=%d\n", runtime.SetMutexProfileFraction(-1)) 823 for i := range p { 824 r := &p[i] 825 fmt.Fprintf(w, "%v %v @", r.Cycles, r.Count) 826 for _, pc := range r.Stack() { 827 fmt.Fprintf(w, " %#x", pc) 828 } 829 fmt.Fprint(w, "\n") 830 if debug > 0 { 831 printStackRecord(w, r.Stack(), true) 832 } 833 } 834 835 if tw != nil { 836 tw.Flush() 837 } 838 return b.Flush() 839 } 840 841 func runtime_cyclesPerSecond() int64