def CommitManifest(self, request, context=None):
retrieval_token = self.retrieval_token(request.staging_session_token)
with filesystems.FileSystems.create(
self._manifest_path(retrieval_token)) as fout:
fout.write(request.manifest.SerializeToString())
return beam_artifact_api_pb2.CommitManifestResponse(
retrieval_token=retrieval_token)
def CommitManifest(self, request, context=None):
retrieval_token = self.retrieval_token(request.staging_session_token)
proxy_manifest = beam_artifact_api_pb2.ProxyManifest(
manifest=request.manifest,
location=[
beam_artifact_api_pb2.ProxyManifest.Location(
name=metadata.name,
uri=self._artifact_path(retrieval_token, metadata.name))
for metadata in request.manifest.artifact])
with self._open(self._manifest_path(retrieval_token), 'w') as fout:
fout.write(json_format.MessageToJson(proxy_manifest).encode('utf-8'))
return beam_artifact_api_pb2.CommitManifestResponse(
retrieval_token=retrieval_token)
def CommitManifest(self, request, context):
self.manifest = request.manifest
return beam_artifact_api_pb2.CommitManifestResponse(
staging_token='token')
def CommitManifest(self, request, context):
self.manifest = request.manifest
self.retrieval_tokens.add(request.staging_session_token)
return beam_artifact_api_pb2.CommitManifestResponse(
retrieval_token='token')