def on_post(self, req, resp, project_id, queue_name): uuid = req.get_header('Client-ID', required=True) # Pull out just the fields we care about messages = wsgi_helpers.filter_stream( req.stream, req.content_length, MESSAGE_POST_SPEC, doctype=wsgi_helpers.JSONArray) # Verify that at least one message was provided. try: first_message = next(messages) except StopIteration: description = _('No messages were provided.') raise wsgi_exceptions.HTTPBadRequestBody(description) # Hack to make message_controller oblivious to the # fact that we just popped the first message. messages = itertools.chain((first_message,), messages) # Enqueue the messages partial = False try: message_ids = self.message_controller.post( queue_name, messages=messages, project=project_id, client_uuid=uuid) except storage_exceptions.DoesNotExist: raise falcon.HTTPNotFound() except storage_exceptions.MessageConflict as ex: LOG.exception(ex) partial = True message_ids = ex.succeeded_ids if not message_ids: # TODO(kgriffs): Include error code that is different # from the code used in the generic case, below. description = _('No messages could be enqueued.') raise wsgi_exceptions.HTTPServiceUnavailable(description) except Exception as ex: LOG.exception(ex) description = _('Messages could not be enqueued.') raise wsgi_exceptions.HTTPServiceUnavailable(description) # Prepare the response resp.status = falcon.HTTP_201 ids_value = ','.join(message_ids) resp.location = req.path + '?ids=' + ids_value hrefs = [req.path + '/' + id for id in message_ids] body = {'resources': hrefs, 'partial': partial} resp.body = helpers.to_json(body)
def test_filter_stream_expect_obj(self): obj = {u'body': {'event': 'start_backup'}, 'id': 'DEADBEEF'} document = json.dumps(obj, ensure_ascii=False) stream = io.StringIO(document) spec = [('body', dict), ('id', basestring)] filtered_object, = helpers.filter_stream(stream, len(document), spec) self.assertEqual(filtered_object, obj) stream.seek(0) self.assertRaises(falcon.HTTPBadRequest, helpers.filter_stream, stream, len(document), spec, doctype=helpers.JSONArray)
def test_filter_stream_expect_array(self): array = [{u'body': {u'x': 1}}, {u'body': {u'x': 2}}] document = json.dumps(array, ensure_ascii=False) stream = io.StringIO(document) spec = [('body', dict)] filtered_objects = list(helpers.filter_stream( stream, len(document), spec, doctype=helpers.JSONArray)) self.assertEqual(filtered_objects, array) stream.seek(0) self.assertRaises(falcon.HTTPBadRequest, helpers.filter_stream, stream, len(document), spec, doctype=helpers.JSONObject)
def on_post(self, req, resp, project_id, queue_name): # Check for an explicit limit on the # of messages to claim limit = req.get_param_as_int('limit') claim_options = {} if limit is None else {'limit': limit} # Read claim metadata (e.g., TTL) and raise appropriate # HTTP errors as needed. metadata, = wsgi_helpers.filter_stream(req.stream, req.content_length, CLAIM_METADATA_SPEC) # Claim some messages try: cid, msgs = self.claim_controller.create( queue_name, metadata=metadata, project=project_id, **claim_options) # Buffer claimed messages # TODO(kgriffs): optimize, along with serialization (below) resp_msgs = list(msgs) except storage_exceptions.DoesNotExist: raise falcon.HTTPNotFound() except Exception as ex: LOG.exception(ex) description = _('Claim could not be created.') raise wsgi_exceptions.HTTPServiceUnavailable(description) # Serialize claimed messages, if any. This logic assumes # the storage driver returned well-formed messages. if len(resp_msgs) != 0: for msg in resp_msgs: msg['href'] = _msg_uri_from_claim( req.path.rpartition('/')[0], msg['id'], cid) del msg['id'] resp.location = req.path + '/' + cid resp.body = helpers.to_json(resp_msgs) resp.status = falcon.HTTP_200 else: resp.status = falcon.HTTP_204
def on_patch(self, req, resp, project_id, queue_name, claim_id): # Read claim metadata (e.g., TTL) and raise appropriate # HTTP errors as needed. metadata, = wsgi_helpers.filter_stream(req.stream, req.content_length, CLAIM_METADATA_SPEC) try: self.claim_controller.update(queue_name, claim_id=claim_id, metadata=metadata, project=project_id) resp.status = falcon.HTTP_204 except storage_exceptions.DoesNotExist: raise falcon.HTTPNotFound except Exception as ex: LOG.exception(ex) description = _('Claim could not be updated.') raise wsgi_exceptions.HTTPServiceUnavailable(description)