D405.id4548.diff
No OneTemporary
Actions

Size

837 B

Subscribers

None

D405.id4548.diff
View Options

	diff --git a/swh/vault/cookers/base.py b/swh/vault/cookers/base.py
	--- a/swh/vault/cookers/base.py
	+++ b/swh/vault/cookers/base.py
	@@ -6,6 +6,7 @@
	import abc
	import io
	import logging
	+from psycopg2.extensions import QueryCanceledError

	from swh.model import hashutil

	@@ -111,7 +112,11 @@

	self.fileobj = BytesIOBundleSizeLimit(size_limit=self.max_bundle_size)
	try:
	- self.prepare_bundle()
	+ try:
	+ self.prepare_bundle()
	+ except QueryCanceledError:
	+ raise PolicyError(
	+ "Timeout reached while assembling the requested bundle")
	bundle = self.fileobj.getvalue()
	# TODO: use proper content streaming instead of put_bundle()
	self.backend.put_bundle(self.CACHE_TYPE_KEY, self.obj_id, bundle)