78 lines
1.5 KiB
Go
78 lines
1.5 KiB
Go
package handler
|
|
|
|
import (
|
|
"context"
|
|
"runtime"
|
|
"time"
|
|
|
|
"github.com/micro/go-micro/debug/log"
|
|
|
|
proto "github.com/micro/go-micro/debug/proto"
|
|
)
|
|
|
|
var (
|
|
// DefaultHandler is default debug handler
|
|
DefaultHandler = newDebug()
|
|
)
|
|
|
|
type Debug struct {
|
|
started int64
|
|
log.Log
|
|
}
|
|
|
|
func newDebug() *Debug {
|
|
return &Debug{
|
|
started: time.Now().Unix(),
|
|
}
|
|
}
|
|
|
|
func (d *Debug) Health(ctx context.Context, req *proto.HealthRequest, rsp *proto.HealthResponse) error {
|
|
rsp.Status = "ok"
|
|
return nil
|
|
}
|
|
|
|
func (d *Debug) Stats(ctx context.Context, req *proto.StatsRequest, rsp *proto.StatsResponse) error {
|
|
var mstat runtime.MemStats
|
|
runtime.ReadMemStats(&mstat)
|
|
|
|
rsp.Started = uint64(d.started)
|
|
rsp.Uptime = uint64(time.Now().Unix() - d.started)
|
|
rsp.Memory = mstat.Alloc
|
|
rsp.Gc = mstat.PauseTotalNs
|
|
rsp.Threads = uint64(runtime.NumGoroutine())
|
|
return nil
|
|
}
|
|
|
|
func (d *Debug) Logs(ctx context.Context, req *proto.LogRequest, stream proto.Debug_LogsStream) error {
|
|
var records []log.Record
|
|
since := time.Unix(0, req.Since)
|
|
if !since.IsZero() {
|
|
records = d.Log.Read(log.Since(since))
|
|
} else {
|
|
records = d.Log.Read(log.Count(int(req.Count)))
|
|
}
|
|
|
|
defer stream.Close()
|
|
|
|
// TODO: figure out the stream later on
|
|
// stream the logs
|
|
for _, record := range records {
|
|
metadata := make(map[string]string)
|
|
for k, v := range record.Metadata {
|
|
metadata[k] = v
|
|
}
|
|
|
|
recLog := &proto.Log{
|
|
Timestamp: record.Timestamp.UnixNano(),
|
|
Value: record.Value.(string),
|
|
Metadata: metadata,
|
|
}
|
|
|
|
if err := stream.Send(recLog); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|