add stream to meilisearch

streams/events/index_docs.py

@@ -6,30 +6,34 @@ from aws_lambda_powertools.utilities.data_classes import (
 )
 from aws_lambda_powertools.utilities.typing import LambdaContext
 
+from meili import Op
 from settings import MEILISEARCH_API_KEY, MEILISEARCH_HOST
 
 meili_client = meilisearch.Client(MEILISEARCH_HOST, MEILISEARCH_API_KEY)
 
 
+INDEXES = {
+    'saladeaula_courses': 'courses',
+    'betaeducacao-prod-enrollments': 'enrollments',
+    'betaeducacao-prod-orders': 'orders',
+    'betaeducacao-prod-users_d2o3r5gmm4it7j': 'users',
+}
+
+
 @event_source(data_class=DynamoDBStreamEvent)
 def lambda_handler(event: DynamoDBStreamEvent, context: LambdaContext):
-    for record in event.records:
-        pk = record.dynamodb.keys['id']  # type: ignore
-        new_image = record.dynamodb.new_image  # type: ignore
-        index_name = table_from_arn(record.event_source_arn)  # type: ignore
+    with Op(meili_client) as op:
+        for record in event.records:
+            pk = record.dynamodb.keys['id']  # type: ignore
+            new_image = record.dynamodb.new_image  # type: ignore
+            table_name = table_from_arn(record.event_source_arn)  # type: ignore
+            index = INDEXES[table_name]  # type: ignore
 
-        print(index_name, pk, record.event_name, new_image)
-
-    #     match record.event_name:
-    #         case DynamoDBRecordEventName.MODIFY:
-    #             update_docs.append(new_image)
-    #         case DynamoDBRecordEventName.REMOVE:
-    #             remove_docs.append(new_image)
-    #         case _:
-    #             add_docs.append(new_image)
-
-    # print(add_docs)
-    # meili_client.index('users').add_documents(add_docs, serializer=JSONEncoder)
+            op.append(
+                index,
+                op=record.event_name,  # type: ignore
+                data=new_image or pk,
+            )
 
 
 def table_from_arn(arn: str) -> str: