You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
59 lines
979 B
Go
59 lines
979 B
Go
package prof
|
|
|
|
import "github.com/tal-tech/go-zero/core/utils"
|
|
|
|
type (
|
|
ProfilePoint struct {
|
|
*utils.ElapsedTimer
|
|
}
|
|
|
|
Profiler interface {
|
|
Start() ProfilePoint
|
|
Report(name string, point ProfilePoint)
|
|
}
|
|
|
|
RealProfiler struct{}
|
|
|
|
NullProfiler struct{}
|
|
)
|
|
|
|
var profiler = newNullProfiler()
|
|
|
|
func EnableProfiling() {
|
|
profiler = newRealProfiler()
|
|
}
|
|
|
|
func Start() ProfilePoint {
|
|
return profiler.Start()
|
|
}
|
|
|
|
func Report(name string, point ProfilePoint) {
|
|
profiler.Report(name, point)
|
|
}
|
|
|
|
func newRealProfiler() Profiler {
|
|
return &RealProfiler{}
|
|
}
|
|
|
|
func (rp *RealProfiler) Start() ProfilePoint {
|
|
return ProfilePoint{
|
|
ElapsedTimer: utils.NewElapsedTimer(),
|
|
}
|
|
}
|
|
|
|
func (rp *RealProfiler) Report(name string, point ProfilePoint) {
|
|
duration := point.Duration()
|
|
report(name, duration)
|
|
}
|
|
|
|
func newNullProfiler() Profiler {
|
|
return &NullProfiler{}
|
|
}
|
|
|
|
func (np *NullProfiler) Start() ProfilePoint {
|
|
return ProfilePoint{}
|
|
}
|
|
|
|
func (np *NullProfiler) Report(string, ProfilePoint) {
|
|
}
|