From 724da2b2a384661042995b80c350cb692c45d5a2 Mon Sep 17 00:00:00 2001
From: Daniel Bimschas <daniel@bimschas.com>
Date: Sat, 7 Dec 2024 17:14:17 +0100
Subject: [PATCH] HOLI-10718: try functionsframework
---
src/main.py | 23 ++++-------------------
src/requirements.txt | 1 +
src/test_main.py | 1 -
3 files changed, 5 insertions(+), 20 deletions(-)
diff --git a/src/main.py b/src/main.py
index 04ea4da..f8105ee 100644
--- a/src/main.py
+++ b/src/main.py
@@ -1,21 +1,6 @@
import base64
+import functions_framework
-def process_message(event, context):
- """
- Cloud Function triggered by a Pub/Sub message.
-
- Args:
- event (dict): The event payload, which includes the message data.
- context (google.cloud.functions.Context): Metadata about the event.
-
- """
- # Decode the Pub/Sub message
- if 'data' in event:
- message = base64.b64decode(event['data']).decode('utf-8')
- print(f"Received message: {message}")
- else:
- print("No data found in event.")
-
- # Log context information
- print(f"Event ID: {context.event_id}")
- print(f"Event type: {context.event_type}")
+@functions_framework.cloud_event
+def process_message(cloud_event):
+ print(f"Received message: {cloud_event}")
diff --git a/src/requirements.txt b/src/requirements.txt
index d197ada..6c9d154 100644
--- a/src/requirements.txt
+++ b/src/requirements.txt
@@ -1 +1,2 @@
+functions_framework==3.8.2
pytest==8.3.4
diff --git a/src/test_main.py b/src/test_main.py
index 6bab790..c862d60 100644
--- a/src/test_main.py
+++ b/src/test_main.py
@@ -1,4 +1,3 @@
-from src.main import process_message
def test_process_message():
pass
\ No newline at end of file
--
GitLab