package main

import (
	"fmt"
	"net/http"
	_ "net/http/pprof"
	"os"
	"os/exec"

	"code.byted.org/gopkg/logs"
)

var heap int = 1

func init() {
	go func() {
		http.ListenAndServe("0.0.0.0:8080", nil)
	}()
}

func pprof() {
	go func() {
		http.ListenAndServe("0.0.0.0:8080", nil)
	}()
}

func DownloadHeapAnalysis(fileType string, cate string) {
	cmd := exec.Command("go", "tool", "pprof", "-"+fileType, "http://localhost:8080/debug/pprof/"+cate)
	bytes, err := cmd.CombinedOutput()
	if err != nil {
		logs.Info(err.Error())
		return
	}
	start := 0
	for i := 0; ; i++ {
		if string(bytes[i]) == "g" && string(bytes[i+1]) == "z" {
			start = i + 3
			break
		}
	}
	f, err := os.OpenFile(fmt.Sprintf("heap%d.png", heap), os.O_CREATE|os.O_WRONLY, 0777)
	defer func() {
		f.Close()
	}()
	if err != nil {
		logs.Info(err.Error())
		return
	}
	f.Write(bytes[start:])
}