Using https://golang.org/x/net/context is probably a good direction to move in. It should be moving into the standard library too (golang/go#14660).