summaryrefslogtreecommitdiffstats
path: root/src/cmd/pprof/testdata/cpu.go
blob: 5b682870db8f69ad5ee1ef539c8dcb505c596e58 (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
// 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 {
	}
}