vin/aplus
1
Fork 0
mirror of https://codeberg.org/flpolyaplus/aplus.git synced 2024-11-22 01:10:29 -05:00
Code Issues Packages Projects Releases Wiki Activity

add cpu profiling support

Browse Source
This commit is contained in:
Vineet K 2024-01-12 19:15:35 -05:00
parent 3fbd4dc645
commit 4d0051f516
1 changed files with 21 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
main.go
View File

@ -5,6 +5,7 @@ import (
"fmt" "fmt"
"net/http" "net/http"
"os" "os"
"runtime/pprof"
) )
var base_link string var base_link string
@ -22,6 +23,7 @@ func usage() {
fmt.Println(" -F, --listfav\tlists favourited canvas courses with name and ID") fmt.Println(" -F, --listfav\tlists favourited canvas courses with name and ID")
fmt.Println(" -v, --verbose\tprints with verbose output") fmt.Println(" -v, --verbose\tprints with verbose output")
fmt.Println(" -h, --help\tdisplays this help message") fmt.Println(" -h, --help\tdisplays this help message")
fmt.Println(" -p, --profile\twrites cpu profile to a file")
} }
func main() { func main() {
@ -32,6 +34,7 @@ func main() {
var listall bool var listall bool
var listfav bool var listfav bool
var help bool var help bool
var cpuprofile string
flag.StringVar(&code, "code", "", "5 character attendance code") flag.StringVar(&code, "code", "", "5 character attendance code")
flag.StringVar(&code, "c", "", "5 character attendance code") flag.StringVar(&code, "c", "", "5 character attendance code")
@ -45,9 +48,27 @@ func main() {
flag.BoolVar(&verbose, "v", false, "verbose output") flag.BoolVar(&verbose, "v", false, "verbose output")
flag.BoolVar(&help, "help", false, "view usage message") flag.BoolVar(&help, "help", false, "view usage message")
flag.BoolVar(&help, "h", false, "view usage message") flag.BoolVar(&help, "h", false, "view usage message")
flag.StringVar(&cpuprofile, "p", "", "CPU profiling report location")
flag.StringVar(&cpuprofile, "profile", "", "CPU profiling report location")
flag.Parse() flag.Parse()
if cpuprofile != "" {
cpuf, err := os.Create(cpuprofile)
if err != nil {
fmt.Println(err)
os.Exit(1)
}
defer cpuf.Close()
err = pprof.StartCPUProfile(cpuf)
if err != nil {
fmt.Println(err)
os.Exit(1)
}
defer pprof.StopCPUProfile()
}
if code == "" && course == -1 && !listall && !listfav && !help { if code == "" && course == -1 && !listall && !listfav && !help {
usage() usage()
os.Exit(1) os.Exit(1)