ref: Make vdl generation use Call instead of Context.
MultiPart: 1/6
Change-Id: Ie739b6bc4540474c8ec11c92aa147ad9793293f4
diff --git a/services/mgmt/vtrace/impl/vtrace.go b/services/mgmt/vtrace/impl/vtrace.go
index 3f33148..c344825 100644
--- a/services/mgmt/vtrace/impl/vtrace.go
+++ b/services/mgmt/vtrace/impl/vtrace.go
@@ -19,13 +19,13 @@
return *tr, nil
}
-func (v *vtraceService) AllTraces(ctx svtrace.StoreAllTracesContext) error {
+func (v *vtraceService) AllTraces(call svtrace.StoreAllTracesServerCall) error {
// TODO(mattr): Consider changing the store to allow us to iterate through traces
// when there are many.
- store := vtrace.GetStore(ctx.Context())
+ store := vtrace.GetStore(call.Context())
traces := store.TraceRecords()
for i := range traces {
- if err := ctx.SendStream().Send(traces[i]); err != nil {
+ if err := call.SendStream().Send(traces[i]); err != nil {
return err
}
}
