Tidy docs

Author: TheTeaCat

await_shutdown.go

@@ -22,8 +22,6 @@ func awaitShutdown(extensionClient *extensionsapi.Client, ctx context.Context) (
 			if err != nil {
 				return "", errors.WithMessage(err, "failed to get next event")
 			}
-
-			// Exit if we receive a SHUTDOWN event
 			if res.EventType == extensionsapi.Shutdown {
 				return "received shutdown event", nil
 			}

extensionsapi/client.go

@@ -1,6 +1,3 @@
-// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
-// SPDX-License-Identifier: MIT-0
-
 package extensionsapi
 
 import (
@@ -16,7 +13,6 @@ type Client struct {
 	ExtensionID      string
 }
 
-// NewClient returns a Lambda Extensions API client
 func NewClient() *Client {
 	return &Client{
 		extensionsApiUrl: fmt.Sprintf("http://%s/2020-01-01/extension", os.Getenv("AWS_LAMBDA_RUNTIME_API")),

extensionsapi/register.go

@@ -14,11 +14,8 @@ import (
 type EventType string
 
 const (
-	// Invoke is a lambda invoke
-	Invoke EventType = "INVOKE"
-
-	// Shutdown is a shutdown event for the environment
-	Shutdown EventType = "SHUTDOWN"
+	Invoke   EventType = "INVOKE"   // a lambda invoke
+	Shutdown EventType = "SHUTDOWN" // shutdown event for the environment
 
 	extensionNameHeader      = "Lambda-Extension-Name"
 	extensionIdentiferHeader = "Lambda-Extension-Identifier"

main.go

@@ -1,6 +1,3 @@
-// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
-// SPDX-License-Identifier: MIT-0
-
 package main
 
 import (
@@ -35,6 +32,7 @@ func main() {
 		panic(err)
 	}
 
+	// Create a logsApiClient & remember to shut it down when we're done
 	logsApiClient, err := initLogsApiClient(logsapi.Options{
 		ExtensionID:      extensionClient.ExtensionID,
 		LogServerAddress: "sandbox:1234",