Sua primeira lambda ao vivo!

Streaming de dados com AWS Kinesis e Lambda

Maksim Pecherskiy

Data Engineer

Aula anterior

Sem título

Esta aula

Sem título

Formato comum de log

bc75da5f-1bf6-444c-80ad-49c180e1b8de 23:16:06.000 3FTEX1G5XAK844393 -76.6990172 2.481207 40
ff8e7131-408d-463b-8d07-d016419b0656 20:26:44.000 2LAXX1C8XAK844292 114.39239199999999 36.097577 90
f29a5b3d-d0fa-43c0-9e1a-e2a5cdb8be7a 8:10:47.000 3FTEX1G5XAK844393 108.58068100000001 34.79925 37

Lendo arquivos do S3

import pandas as pd

results in:

Response:
{
  "errorMessage": "Unable to import module 'lambda_function': No module named 'pandas'",
  "errorType": "Runtime.ImportModuleError"
}

Atualizando o handler

#lambda_function.py
import json, boto3, pandas as pd
# Inicializa clientes
...

SPEED_ALERT_THRESHOLD = 45
ALERT_PHONE_NUMBER = "+1234567890"

# Função auxiliar para ler dataframes dos registros gravados
def get_new_data(event):
    pd.read_csv()...
    ...
    return data

# Handler da Lambda
def record_created_handler(event, context):
    data = get_new_data(event)
    ...
    sns.publish()
    ...

get_new_data()

def get_new_data(event):
    # Cria uma lista para guardar as chaves dos novos objetos.
    written_objects = []

    # Itera por cada registro do evento do S3.
    for record in event['Records']:

        # Variáveis a verificar
        event_name = record['eventName']
        bucket_name = record['s3']['bucket']['name']
        obj_key = record['s3']['object']['key']

get_new_data()

def get_new_data(event):
        ...
        # Confirma que o evento veio do bucket sd-vehicle-data.
        if event_name == 'ObjectCreated:Put' and bucket_name == 'sd-vehicle-data':

            obj = s3.get_object(Bucket=bucket_name, Key = obj_key)
            df = pd.read_csv(obj['Body'], delimiter = " ", 
                   names=["record_id", "timestamp", "vin", "lon", "lat", "speed"])

            written_objects.append(df)

       # Concatena os novos registros em um único dataframe.
       return pd.concat(written_objects)

record_created_handler()

SPEED_ALERT_THRESHOLD = 45
ALERT_PHONE_NUMBER = "+1234567890"
...
def record_created_handler(event, context):
    # Chama a função auxiliar
    data = get_new_data(event)

    ## Velocidades máximas
    top_speeds = data.groupby(['vin'])['speed'].max().reset_index()

    ## Velocidades acima do limite 45
    too_fast = top_speeds.loc[top_speeds.speed > SPEED_ALERT_THRESHOLD, :]

record_created_handler()

SPEED_ALERT_THRESHOLD = 45
ALERT_PHONE_NUMBER = "+1234567890"
...

def record_created_handler(event, context):
    ...
    ## Envia SMS
    sns.publish(PhoneNumber=ALERT_PHONE_NUMBER,
        Message="Speeding Alert \n" + too_fast.to_string())

    ## Ainda não vai pra lugar nenhum, mas precisamos retornar algo.
    totals = data.groupby(['vin'])['speed'].max().reset_index()
    return totals.to_csv(sep=" ", index=False)

Testar a função Lambda

Sem título

Adicionando variáveis de ambiente

import os
os.environ.get("ENV_VARIABLE_NAME", "DEFAULT_VALUE")

import os
SPEED_ALERT_THRESHOLD = os.environ.get("SPEED_ALERT_THRESHOLD", 45)
ALERT_PHONE_NUMBER = os.environ.get("ALERT_PHONE_NUMBER", None)
...

def record_created_handler(event, context):
    ...