src/cmd/pprof/testdata/cpu.go

// 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 {
	}
}