mirror of
https://github.com/go-micro/go-micro.git
synced 2025-07-12 22:41:07 +02:00
Add Debug.Cache method
This commit is contained in:
@ -5,6 +5,7 @@ import (
|
||||
"context"
|
||||
"time"
|
||||
|
||||
"github.com/micro/go-micro/v2/client"
|
||||
"github.com/micro/go-micro/v2/debug/log"
|
||||
proto "github.com/micro/go-micro/v2/debug/service/proto"
|
||||
"github.com/micro/go-micro/v2/debug/stats"
|
||||
@ -13,11 +14,12 @@ import (
|
||||
)
|
||||
|
||||
// NewHandler returns an instance of the Debug Handler
|
||||
func NewHandler() *Debug {
|
||||
func NewHandler(c client.Client) *Debug {
|
||||
return &Debug{
|
||||
log: log.DefaultLog,
|
||||
stats: stats.DefaultStats,
|
||||
trace: trace.DefaultTracer,
|
||||
cache: c.Options().Cache,
|
||||
}
|
||||
}
|
||||
|
||||
@ -30,6 +32,8 @@ type Debug struct {
|
||||
stats stats.Stats
|
||||
// the tracer
|
||||
trace trace.Tracer
|
||||
// the cache
|
||||
cache *client.Cache
|
||||
}
|
||||
|
||||
func (d *Debug) Health(ctx context.Context, req *proto.HealthRequest, rsp *proto.HealthResponse) error {
|
||||
@ -164,3 +168,9 @@ func (d *Debug) Log(ctx context.Context, stream server.Stream) error {
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// Cache returns all the key value pairs in the client cache
|
||||
func (d *Debug) Cache(ctx context.Context, req *proto.CacheRequest, rsp *proto.CacheResponse) error {
|
||||
rsp.Values = d.cache.List()
|
||||
return nil
|
||||
}
|
||||
|
Reference in New Issue
Block a user