Introduce the context type to the datastore package
Minor changes to pubsub/storage due to moving some common functions to the internal package