Wrapper for grpc ServerStream providing context override

sscontext.go

package streamutil

import (
    "golang.org/x/net/context"
    "google.golang.org/grpc"
)

struct serverStreamWithContext {
    grpc.ServerStream
    ctx context.Context
}

func (ss *serverStreamWithContext) Context() context.Context {
    return ss.ctx
}

func ServerStreamWithContext(ss ServerStream, ctx context.Context) {
    return &serverStreamWithContext{ss, ctx}
}