feat: Add functions_log_stackdriver

grant · grant · commit 0d6226a306e7 · 2019-09-11T15:08:30.000-03:00
diff --git a/functions/snippets/src/main/java/LogEntry.java b/functions/snippets/src/main/java/LogEntry.java
@@ -0,0 +1,42 @@
+/*
+ * Copyright 2019 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+// [START functions_log_stackdriver]
+import java.io.IOException;
+import java.util.Base64;
+import java.util.Map;
+import java.util.logging.Logger;
+
+public class LogEntry {
+
+  private static Logger LOGGER = Logger.getLogger(LogEntry.class.getName());
+
+  public void helloPubSub(PubSubMessage message) throws IOException {
+    String data = new String(Base64.getDecoder().decode(message.data.getBytes("UTF-8")));
+    if (data.isEmpty()) {
+      data = "World";
+    }
+    LOGGER.info(String.format("Hello, %s", data));
+  }
+}
+
+class PubSubMessage {
+  String data;
+  Map<String, String> attributes;
+  String messageId;
+  String publishTime;
+}
+// [END functions_log_stackdriver]
diff --git a/functions/snippets/src/test/java/SnippetsTests.java b/functions/snippets/src/test/java/SnippetsTests.java
@@ -179,6 +179,15 @@ public void filesTest() throws IOException {
     assertThat(responseOut.toString(), containsString("Files:"));
   }
 
+  @Test
+  public void logEntry() throws IOException {
+    PubSubMessage message = new PubSubMessage();
+    message.data = "data";
+    message.messageId = "id";
+    new LogEntry().helloPubSub(message);
+    assertThat(responseOut.toString(), containsString(""));
+  }
+
   @Test
   public void envTest() throws IOException {
     environmentVariables.set("FOO", "BAR");
