origin/sysservice/pprofservice.go

package sysservice

import (
	"encoding/json"
	"fmt"
	"runtime/pprof"
	"time"

	"github.com/duanhf2012/origin/service"
	"github.com/duanhf2012/origin/sysservice/originhttp"
)

type PProfService struct {
	service.BaseService
	fisttime int
}

type ProfileData struct {
	Name  string
	Count int
} 

type Profilestruct struct {
	Fisttime    int
	ProfileList []ProfileData
}

func (slf *PProfService) OnInit() error {
	slf.fisttime = (int)(time.Now().UnixNano())
	return nil
}  

func (slf *PProfService) GetPprof() ([]byte, error) {
	var pfiles Profilestruct
	pfiles.Fisttime = slf.fisttime

	for _, p := range pprof.Profiles() {
		pfiles.ProfileList = append(pfiles.ProfileList, ProfileData{
			Name:  p.Name(),
			Count: p.Count(),
		})
	}

	return json.Marshal(pfiles)
}

func (slf *PProfService) HTTP_DebugPProf(request *originhttp.HttpRequest, resp *originhttp.HttpRespone) error {
	var err error
	resp.Respone, err = slf.GetPprof()
	if err != nil {
		resp.Respone = []byte(fmt.Sprint(err))
	}
	return nil
}

func (slf *PProfService) RPC_DebugPProf(arg *string, ret *Profilestruct) error {

	ret.Fisttime = slf.fisttime
	for _, p := range pprof.Profiles() {
		ret.ProfileList = append(ret.ProfileList, ProfileData{
			Name:  p.Name(),
			Count: p.Count(),
		})
	}

	return nil
}

func (slf *PProfService) HTTP_Test(request *originhttp.HttpRequest, resp *originhttp.HttpRespone) error {

	resp.Respone = []byte(request.Body)
	return nil
}