Telemetry

go-telemetry-api-extension/extensionApi/client.go

@@ -59,12 +59,16 @@ const (
 
 // Client is a simple client for the Lambda Extensions API
 type Client struct {
-	httpClient  *http.Client
-	baseUrl     string
-	ExtensionID string
+	httpClient   *http.Client
+	baseUrl      string
+	ExtensionID  string
 	functionName string
 }
 
+func (e *Client) GetFunctionName() string {
+	return e.functionName
+}
+
 var l = log.WithFields(log.Fields{"pkg": "extensionApi"})
 
 // Returns a Lambda Extensions API client

go-telemetry-api-extension/main.go

@@ -13,9 +13,9 @@ Notes:
 package main
 
 import (
+	"context"
 	"newrelic-lambda-extension/go-telemetry-api-extension/extensionApi"
 	"newrelic-lambda-extension/go-telemetry-api-extension/telemetryApi"
-	"context"
 	"os"
 	"os/signal"
 	"path"
@@ -65,7 +65,7 @@ func main() {
 		panic(err)
 	}
 	l.Info("[main] Subscription success")
-	dispatcher := telemetryApi.NewDispatcher(extensionApiClient.functionName)
+	dispatcher := telemetryApi.NewDispatcher(extensionApiClient.GetFunctionName())
 
 	// Will block until invoke or shutdown event is received or cancelled via the context.
 	for {

go-telemetry-api-extension/telemetryApi/listener.go

@@ -75,7 +75,7 @@ func (s *TelemetryApiListener) http_handler(w http.ResponseWriter, r *http.Reque
 		return
 	}
 	// Parse and put the log messages into the queue
-	var slice []interface{}
+	var slice []LambdaTelemetryEvent
 	_ = json.Unmarshal(body, &slice)
 
 	for _, el := range slice {
@@ -98,3 +98,9 @@ func (s *TelemetryApiListener) Shutdown() {
 		}
 	}
 }
+
+type LambdaTelemetryEvent struct {
+	Time   string
+	Type   string
+	Record any
+}