| package impl |
| |
| import ( |
| "v.io/core/veyron2/ipc" |
| svtrace "v.io/core/veyron2/services/mgmt/vtrace" |
| "v.io/core/veyron2/uniqueid" |
| "v.io/core/veyron2/verror2" |
| "v.io/core/veyron2/vtrace" |
| ) |
| |
| type vtraceService struct{} |
| |
| func (v *vtraceService) Trace(ctx ipc.ServerContext, id uniqueid.ID) (vtrace.TraceRecord, error) { |
| store := vtrace.GetStore(ctx.Context()) |
| tr := store.TraceRecord(id) |
| if tr == nil { |
| return vtrace.TraceRecord{}, verror2.Make(verror2.NoExist, ctx.Context(), "No trace with id %x", id) |
| } |
| return *tr, nil |
| } |
| |
| func (v *vtraceService) AllTraces(ctx svtrace.StoreAllTracesContext) error { |
| // TODO(mattr): Consider changing the store to allow us to iterate through traces |
| // when there are many. |
| store := vtrace.GetStore(ctx.Context()) |
| traces := store.TraceRecords() |
| for i := range traces { |
| if err := ctx.SendStream().Send(traces[i]); err != nil { |
| return err |
| } |
| } |
| return nil |
| } |
| |
| func NewVtraceService() interface{} { |
| return svtrace.StoreServer(&vtraceService{}) |
| } |
