GCF: Add Stackdriver triggers sample (GoogleCloudPlatform#1707)

Ace Nassri · web-flow · commit a34f99108a6e · 2018-09-28T18:08:23.000-07:00
* GCF: Add Stackdriver triggers sample

Change-Id: I55fa7bdc307a23fcb12c8f63c091e787b522626a

* Make sample product-agnostic

Change-Id: Ie0499ac5662c7bf0eaeccceb6a789a0e81f64e0a
diff --git a/functions/log/main.py b/functions/log/main.py
@@ -12,6 +12,12 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+# [START functions_log_bigquery_stackdriver]
+import base64
+import json
+
+# [END functions_log_bigquery_stackdriver]
+
 # [START functions_log_helloworld]
 import logging
 
@@ -66,3 +72,14 @@ def get_log_entries(request):
 
     return 'Done!'
 # [END functions_log_retrieve]
+
+
+# [START functions_log_stackdriver]
+def process_log_entry(data, context):
+    data_buffer = base64.b64decode(data['data'])
+    log_entry = json.loads(data_buffer)['protoPayload']
+
+    print(f"Method: {log_entry['methodName']}")
+    print(f"Resource: {log_entry['resourceName']}")
+    print(f"Initiator: {log_entry['authenticationInfo']['principalEmail']}")
+# [END functions_log_stackdriver]
diff --git a/functions/log/main_test.py b/functions/log/main_test.py
@@ -12,6 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import base64
+import json
 
 import main
 
@@ -21,3 +23,26 @@ def test_hello_world(capsys):
 
     out, _ = capsys.readouterr()
     assert "Hello, stdout!" in out
+
+
+def test_process_log_entry(capsys):
+    inner_json = {
+      'protoPayload': {
+        'methodName': 'method',
+        'resourceName': 'resource',
+        'authenticationInfo': {
+          'principalEmail': 'me@example.com'
+        }
+      }
+    }
+
+    data = {
+      'data': base64.b64encode(json.dumps(inner_json).encode())
+    }
+
+    main.process_log_entry(data, None)
+
+    out, _ = capsys.readouterr()
+    assert 'Method: method' in out
+    assert 'Resource: resource' in out
+    assert 'Initiator: me@example.com' in out
