Page MenuHomeDevCentral
Files F4043888

D1004.id2563.diff
No OneTemporary
Actions

D1004.id2563.diff
View Options

diff --git a/app.py b/app.py
--- a/app.py
+++ b/app.py
@@ -16,6 +16,7 @@
"""
from flask import Flask, abort, jsonify, request
+import base64
import uuid
import pika
import os
@@ -99,6 +100,32 @@
# -------------------------------------------------------------
+# Message format helper methods
+# - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
+
+
+def get_valid_formats():
+ return [
+ "as-is",
+ "base64"
+ ]
+
+
+def is_valid_format(message_format):
+ return message_format in get_valid_formats()
+
+
+def format_message(message_format, message):
+ if message_format == "as-is":
+ return message.decode("utf-8")
+
+ if message_format == "base64":
+ return base64.b64encode(message).decode("ascii")
+
+ raise ValueError('Unknown format: ' + message_format)
+
+
+# -------------------------------------------------------------
# API helper methods
# - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
@@ -183,6 +210,21 @@
abort(400)
+def get_single_message(queue, message_format, no_ack):
+ connection = get_broker_connection(service['config'])
+ channel = connection.channel()
+
+ method_frame, header_frame, body = channel.basic_get(
+ queue=queue, no_ack=no_ack)
+ if method_frame:
+ return {
+ "key": method_frame.delivery_tag,
+ "format": message_format,
+ "content": format_message(message_format, body)
+ }
+
+ return {}
+
# -------------------------------------------------------------
# API methods
# - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
@@ -235,6 +277,30 @@
return error_handler(err)
+@app.route(endpoint + "/get", methods=['POST'])
+def get():
+ """Consume a message from the queue."""
+
+ # Reads request
+ if not request.json or "key" not in request.json:
+ abort(400)
+
+ queue = get_queue_name(request.json["key"])
+ message_format = get_variable_from_request("format", "base64")
+ no_ack = get_variable_from_request("no-ack", False)
+
+ if not is_valid_format(message_format):
+ print("Invalid format: {0}".format(message_format))
+ abort(400)
+
+ # Handles request
+ try:
+ message = get_single_message(queue, message_format, no_ack)
+ return jsonify(message)
+ except pika.exceptions.ChannelClosed as err:
+ return error_handler(err)
+
+
# -------------------------------------------------------------
# Start application
# - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -

File Metadata

Mime Type
text/plain
Expires
Thu, Jan 23, 11:08 (18 h, 4 m)
Storage Engine
blob
Storage Format
Raw Data
Storage Handle
2371926
Default Alt Text
D1004.id2563.diff (2 KB)

Event Timeline

Nasqueron DevCentral · If it had been much bigger the moon would have had a core of ice. · Powered by Phabricator