Add custom LocalStack LogsEgressAPI using logCollector

cmd/localstack/logs_egress_api.go

@@ -0,0 +1,31 @@
+package main
+
+import (
+	"io"
+	"os"
+)
+
+// This LocalStack LogsEgressAPI builder allows to customize log capturing, in our case using the logCollector.
+
+type LocalStackLogsEgressAPI struct {
+	logCollector *LogCollector
+}
+
+func NewLocalStackLogsEgressAPI(logCollector *LogCollector) *LocalStackLogsEgressAPI {
+	return &LocalStackLogsEgressAPI{
+		logCollector: logCollector,
+	}
+}
+
+// The interface StdLogsEgressAPI for the functions below is defined in the under cmd/localstack/logs_egress_api.go
+// The default implementation is a NoOpLogsEgressAPI
+
+func (s *LocalStackLogsEgressAPI) GetExtensionSockets() (io.Writer, io.Writer, error) {
+	// os.Stderr can not be used for the stderrWriter because stderr is for internal logging (not customer visible).
+	return io.MultiWriter(s.logCollector, os.Stdout), io.MultiWriter(s.logCollector, os.Stdout), nil
+}
+
+func (s *LocalStackLogsEgressAPI) GetRuntimeSockets() (io.Writer, io.Writer, error) {
+	// os.Stderr can not be used for the stderrWriter because stderr is for internal logging (not customer visible).
+	return io.MultiWriter(s.logCollector, os.Stdout), io.MultiWriter(s.logCollector, os.Stdout), nil
+}

cmd/localstack/main.go

@@ -148,11 +148,12 @@ func main() {
 		UserLogger().Debugln("Process running as non-root user.")
 	}
 
-	logCollector := NewLogCollector()
-
 	// file watcher for hot-reloading
 	fileWatcherContext, cancelFileWatcher := context.WithCancel(context.Background())
 
+	logCollector := NewLogCollector()
+	localStackLogsEgressApi := NewLocalStackLogsEgressAPI(logCollector)
+
 	// build sandbox
 	sandbox := rapidcore.
 		NewSandboxBuilder().
@@ -164,7 +165,8 @@ func main() {
 			stopDnsServer()
 		}).
 		SetExtensionsFlag(true).
-		SetInitCachingFlag(true)
+		SetInitCachingFlag(true).
+		SetLogsEgressAPI(localStackLogsEgressApi)
 
 	// xray daemon
 	endpoint := "http://" + lsOpts.LocalstackIP + ":" + lsOpts.EdgePort