mirror of https://github.com/rocboss/paopao-ce
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
49 lines
1.1 KiB
49 lines
1.1 KiB
2 years ago
|
// Copyright 2023 ROC. All rights reserved.
|
||
|
|
// Use of this source code is governed by a MIT style
|
||
|
|
// license that can be found in the LICENSE file.
|
||
|
|
|
||
|
|
//go:build pyroscope
|
||
|
|
// +build pyroscope
|
||
|
|
|
||
|
|
package debug
|
||
|
|
|
||
|
|
import (
|
||
|
|
"os"
|
||
|
|
|
||
|
|
"github.com/alimy/cfg"
|
||
|
|
"github.com/pyroscope-io/client/pyroscope"
|
||
|
|
"github.com/rocboss/paopao-ce/internal/conf"
|
||
|
|
"github.com/sirupsen/logrus"
|
||
|
|
)
|
||
|
|
|
||
|
|
func StartPyroscope() {
|
||
|
|
if !cfg.If("Pyroscope") {
|
||
|
|
logrus.Infoln("skip Pyroscope because not add Pyroscope feature in config.yaml")
|
||
|
|
return
|
||
|
|
}
|
||
|
|
s := conf.PyroscopeSetting
|
||
|
|
c := pyroscope.Config{
|
||
|
|
ApplicationName: s.AppName,
|
||
|
|
ServerAddress: s.Endpoint,
|
||
|
|
AuthToken: os.Getenv("PYROSCOPE_AUTH_TOKEN"),
|
||
|
|
Logger: s.GetLogger(),
|
||
|
|
ProfileTypes: []pyroscope.ProfileType{
|
||
|
|
pyroscope.ProfileCPU,
|
||
|
|
pyroscope.ProfileGoroutines,
|
||
|
|
pyroscope.ProfileAllocObjects,
|
||
|
|
pyroscope.ProfileAllocSpace,
|
||
|
|
pyroscope.ProfileInuseObjects,
|
||
|
|
pyroscope.ProfileInuseSpace,
|
||
|
|
pyroscope.ProfileBlockCount,
|
||
|
|
pyroscope.ProfileBlockDuration,
|
||
|
|
pyroscope.ProfileMutexCount,
|
||
|
|
pyroscope.ProfileMutexDuration,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
if len(c.AuthToken) == 0 {
|
||
|
|
c.AuthToken = s.AuthToken
|
||
|
|
}
|
||
|
|
pyroscope.Start(c)
|
||
|
|
|
||
|
|
}
|