Safely record gRPC request cancellation

The way we are checking for client cancellation is not correct. The context.is_active method does not seem to be updated when the callbacks from the servicer context are triggered. Using this wrapper, we can track the updates to cancellation more easily.