| // Copyright 2021 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. |
| |
| package main |
| |
| import ( |
| "flag" |
| "fmt" |
| "os" |
| "runtime/pprof" |
| "time" |
| ) |
| |
| var output = flag.String("output", "", "pprof profile output file") |
| |
| func main() { |
| flag.Parse() |
| if *output == "" { |
| fmt.Fprintf(os.Stderr, "usage: %s -output file.pprof\n", os.Args[0]) |
| os.Exit(2) |
| } |
| |
| f, err := os.Create(*output) |
| if err != nil { |
| fmt.Fprintln(os.Stderr, err) |
| os.Exit(2) |
| } |
| defer f.Close() |
| |
| if err := pprof.StartCPUProfile(f); err != nil { |
| fmt.Fprintln(os.Stderr, err) |
| os.Exit(2) |
| } |
| defer pprof.StopCPUProfile() |
| |
| // Spin for long enough to collect some samples. |
| start := time.Now() |
| for time.Since(start) < time.Second { |
| } |
| } |
