Memphis.dev is more than a broker. It's a new streaming stack.
Memphis.dev is a highly scalable event streaming and processing engine.
Before Memphis came along, handling ingestion and processing of events on a large scale took months to adopt and was a capability reserved for the top 20% of mega-companies. Now, Memphis opens the door for the other 80% to unleash their event and data streaming superpowers quickly, easily, and with great cost-effectiveness.
This repository is responsible for the Memphis Functions Python SDK
$ pip3 install memphis-functions
from memphis import create_function
Memphis provides a create_function utility for more easily creatin Memphis Functions.
The user created event_handler
will be called for every message in the given batch of events. The user's event_handler
will take in a msg_payload
as bytes, msg_headers
as a dict and inputs
as a dict, and should return a modified version of the payload and headers in the same data types.
The user function should raise an exception if the message processing has failed. If any exception is raised (deliberately or by a failed operation) the message will be sent to the dead letter station.
If the returned modified version of the msg_payload
or msg_headers
are returned as None
, then the message will be skipped and will not be sent to the station or dead letter station.
Make sure to encode the modified
msg_payload
bytes object with utf-8 encoding!
This example function takes the bytes object msg_payload
and encodes it into a string so that it may be parsed as JSON.
import json
import base64
from memphis import create_function
def handler(event, context): # The name of this file and this function should match the handler field in the memphis.yaml file in the following format <file name>.<function name>
return create_function(event, event_handler = event_handler)
def event_handler(msg_payload, msg_headers, inputs):
payload = str(msg_payload, 'utf-8')
as_json = json.loads(payload)
as_json['modified'] = True
return bytes(json.dumps(as_json), encoding='utf-8'), msg_headers
Instead of taking msg_payload
as a bytes object, the as_dict flag can be used to have the JSON parsed to a dictionary.
import json
import base64
from memphis import create_function
def handler(event, context): # The name of this file and this function should match the handler field in the memphis.yaml file in the following format <file name>.<function name>
return create_function(event, event_handler = event_handler, as_dict=True)
def event_handler(msg_payload, msg_headers, inputs):
msg_payload['modified'] = True
return msg_payload, msg_headers
Memphis Functions support using Async functions through asyncio. When functions are async, set the use_async parameter to true.
import json
import base64
import asyncio
from memphis import create_function
def handler(event, context): # The name of this file and this function should match the handler field in the memphis.yaml file in the following format <file name>.<function name>
return create_function(event, event_handler = event_handler, use_async = True)
async def event_handler(msg_payload, msg_headers, inputs):
payload = str(msg_payload, 'utf-8')
as_json = json.loads(payload)
as_json['modified'] = True
asyncio.sleep(1)
return bytes(json.dumps(as_json), encoding='utf-8'), msg_headers
If the user would want to have a message that they would want to validate and send to the dead letter station if the validation fails then the user can raise an exception. In the following example, the field check
is simply a boolean. The following function will send any messages which fail the check
to the dead letter station.
import json
import base64
from memphis import create_function
def handler(event, context): # The name of this file and this function should match the handler field in the memphis.yaml file in the following format <file name>.<function name>
return create_function(event, event_handler = event_handler)
def event_handler(msg_payload, msg_headers, inputs):
payload = str(msg_payload, 'utf-8')
as_json = json.loads(payload)
if as_json['check'] == False:
raise Exception("Validation Failed!")
return bytes(json.dumps(as_json), encoding='utf-8'), msg_headers
If a user would rather just skip the message and not have it be sent to the station or dead letter station, the cuser could instead return None
, None
:
import json
import base64
from memphis import create_function
def handler(event, context): # The name of this file and this function should match the handler field in the memphis.yaml file in the following format <file name>.<function name>
return create_function(event, event_handler = event_handler)
def event_handler(msg_payload, msg_headers, inputs):
payload = str(msg_payload, 'utf-8')
as_json = json.loads(payload)
if as_json['check'] == False:
return None, None
return bytes(json.dumps(as_json), encoding='utf-8'), msg_headers
Lastly, if the user is using another data format like Protocol Buffers, the user may simply decode the msg_payload
into that format instead of JSON. Assuming we have a .proto definition like this:
syntax = "proto3";
package protobuf_example;
message Message{
string data_field = 1;
}
We can decode this and get the data_field out like this:
import json
import base64
from memphis import create_function
import message_pb2
def handler(event, context): # The name of this file and this function should match the handler field in the memphis.yaml file in the following format <file name>.<function name>
return create_function(event, event_handler = event_handler)
def event_handler(msg_payload, msg_headers, inputs):
message = message_pb2.Message()
message.ParseFromString(base64.b64decode(encoded_str))
# Arbitrarily changing the data_field
message.data_field = "my new data"
# SerializeToString returns bytes, which is the type we want
return message.SerializeToString(), msg_headers