Трансформаційна Lambda

Потокові дані з AWS Kinesis і Lambda

Maksim Pecherskiy

Data Engineer

У розділі 1...

У розділі 2...

У цьому розділі...

Попередній підхід

Обробка один раз у S3 vs Lambda transform

Обробка один раз у S3

Використовує функцію Lambda, що спрацьовує під час запису об'єкта в S3
Довша затримка до трансформації даних
Сирі дані зберігаються до очищення
Потрібне призначення Firehose до S3

Обробка через Lambda transform

Використовує функцію Lambda, що спрацьовує в середині потоку Firehose
Трансформація відбувається одразу
Зберігаються лише очищені дані
Дозволяє інші призначення у Firehose

Вхідні дані

record_id	timestamp	vin	lon	lat	speed
939ed1d1-1740-420c-8906-445278573c7f	4:25:06.000	4FTEX4944AK844294	106.9447146	-6.3385652	25
f29a5b3d-d0fa-43c0-9e1a-e2a5cdb8be7a	8:10:47.000	3FTEX1G5XAK844393	108.580681	34.79925	37
ff8e7131-408d-463b-8d07-d016419b0656	20:26:44.000	2LAXX1C8XAK844292	114.392392	36.097577	90

Трансформаційна Lambda

Зразок події

{
  "invocationId": "invocationIdExample",
  "deliveryStreamArn": "arn:aws:firehose:us-east-1:458912630:deliverystream/gps-delivery-stream",
  "region": "us-east-1",
  "records": [
    {

      "recordId": "49546986683135544286507457936321625675700192471156785154",

      "approximateArrivalTimestamp": 1495072949453,

      "data": "NjQuMjQyLjg4LjEwIC0gLSBbMDcvTWFyLzIwMDQ6MTY6MTA6MDIgLTA4MDBdICJHRVQgL21haWxtYW4vbGlzdGluZm8vaHNkaXZpc2lvbiBIVFRQLzEuMSIgMjAwIDYyOTE=="

    }
  ]
}

Base64

Обробка даних

import base64
from datetime import datetime as dt

def convert_timestamp(record_time_val):

    # Отримати сьогоднішню дату як рядок
    today = dt.today().strftime("%Y-%m-%d")

    # Поєднати сьогоднішню дату з часом запису та створити об'єкт datetime
    new_ts = dt.strptime(f"{today} {record_time_val}", "%Y-%m-%d %H:%M:%S.%f")

    # Перетворити datetime на зручний рядок
    return new_ts.strftime("%Y-%m-%dT%H:%M:%S")

Обробка даних

def lambda_handler(event, context):
    output = []

    for record in event['records']:
        payload = base64.b64decode(record['data'])

        payload = payload.decode()

        payload = payload.split(" ")

        payload[1] = convert_timestamp(payload[1])

Обробка даних

    for record in event['records']:
        ...
        payload = " ".join(payload)
        payload_enc = base64.b64encode(payload.encode())

        output.append({

          'recordId': record['recordId'],

          'result': 'Ok',

          'data': payload_enc

        })

    return {'records': output}

Підсумок

import base64
def lambda_handler(event, context):

    output = []
    for record in event['records']: # Ітеруйте записи

        payload = base64.b64decode(record['data']).decode() # Декодуйте вміст

        # Змініть його

        payload_enc = base64.b64encode(payload.encode()) # Закодуйте знову

        output.append({ # Додайте до словника
            'recordId': record['recordId'], 'result': 'Ok', 'data': payload_enc,
        })

    return {'records': output}

Створення Lambda в AWS

Скрипт для створення Lambda

https://drive.google.com/file/d/1fTzIp2mgWt04nVdILc7ClyAMOF1yNxkv/view?usp=sharing

Давайте потренуємось!

Потокові дані з AWS Kinesis і Lambda