From 787d11949a2b1f210032ee7d8ea3c3ba1c13d307 Mon Sep 17 00:00:00 2001
From: Tianzi Cai <tianzih@outlook.com>
Date: Mon, 13 Sep 2021 17:46:47 -0400
Subject: [PATCH] docs: update pubsub example

---
 README.md | 18 +++++++++---------
 1 file changed, 9 insertions(+), 9 deletions(-)

diff --git a/README.md b/README.md
index 43dbac5b..12eedae7 100644
--- a/README.md
+++ b/README.md
@@ -112,26 +112,26 @@ contents:
 ```java
 package com.example;
 
-import com.example.PubSubBackground.PubSubMessage;
 import com.google.cloud.functions.BackgroundFunction;
 import com.google.cloud.functions.Context;
 import java.util.Map;
 import java.util.logging.Logger;
 
+// This is the Pub/Sub message format from the Pub/Sub emulator. 
+class PubSubMessage {
+  String data;
+  Map<String, String> attributes;
+  String messageId;
+  String publishTime;
+}
+
 public class PubSubBackground implements BackgroundFunction<PubSubMessage> {
   private static final Logger logger =
       Logger.getLogger(PubSubBackground.class.getName());
 
   @Override
   public void accept(PubSubMessage pubSubMessage, Context context) {
-    logger.info("Received message with id " + pubSubMessage.messageId);
-  }
-
-  public static class PubSubMessage {
-    public String data;
-    public Map<String, String> attributes;
-    public String messageId;
-    public String publishTime;
+    logger.info("Received message with id " + context.eventId());
   }
 }
 ```