From c8112c5c70208359e3e9c3e12acbef0067cf22fa Mon Sep 17 00:00:00 2001
From: Paschalis Tsilias <tpaschalis@users.noreply.github.com>
Date: Thu, 27 Jul 2023 18:53:28 +0300
Subject: [PATCH] flow: Add 'write_to' argument to logging block (#4620)

Signed-off-by: Paschalis Tsilias <paschalis.tsilias@grafana.com>
---
 CHANGELOG.md                                  |  3 ++
 .../flow/reference/config-blocks/logging.md   | 11 ++++++
 pkg/flow/logging/logger.go                    | 35 +++++++++++++++++--
 pkg/flow/logging/options.go                   |  3 +-
 4 files changed, 49 insertions(+), 3 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 29e920e1ecd1..8438fa6d5a67 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -18,6 +18,9 @@ Main (unreleased)
 
 - Clustering: Enable peer discovery with the go-discover package. (@tpaschalis)
 
+- Flow: Allow the `logging` configuration block to tee the Agent's logs to one
+  or more loki.* components. (@tpaschalis)
+
 - New Grafana Agent Flow components:
 
   - `prometheus.exporter.gcp` - scrape GCP metrics (@tburgessdev)
diff --git a/docs/sources/flow/reference/config-blocks/logging.md b/docs/sources/flow/reference/config-blocks/logging.md
index bad9493833e2..c86d61e7a4a2 100644
--- a/docs/sources/flow/reference/config-blocks/logging.md
+++ b/docs/sources/flow/reference/config-blocks/logging.md
@@ -26,6 +26,7 @@ Name | Type | Description | Default | Required
 ---- | ---- | ----------- | ------- | --------
 `level` | `string` | Level at which log lines should be written | `"info"` | no
 `format` | `string` | Format to use for writing log lines | `"logfmt"` | no
+`write_to` | `list(LogsReceiver)` | List of receivers to send log entries to | | no
 
 ### Log level
 
@@ -45,6 +46,16 @@ The following strings are recognized as valid log line formats:
 
 [logfmt]: https://brandur.org/logfmt
 
+### Log receivers
+
+The `write_to` argument allows the Agent to tee its log entries to one or more
+`loki.*` component log receivers in addition to the default [location][].
+This, for example can be the export of a `loki.write` component to ship log
+entries directly to Loki, or a `loki.relabel` component to add a certain label
+first.
+
+[location]: #log-location
+
 ## Log location
 
 Grafana Agent writes all logs to `stderr`.
diff --git a/pkg/flow/logging/logger.go b/pkg/flow/logging/logger.go
index 3ce4ebf95878..d1d9c99c5a3e 100644
--- a/pkg/flow/logging/logger.go
+++ b/pkg/flow/logging/logger.go
@@ -4,9 +4,13 @@ import (
 	"fmt"
 	"io"
 	"sync"
+	"time"
 
 	"github.com/go-kit/log"
 	"github.com/go-kit/log/level"
+	"github.com/grafana/agent/component/common/loki"
+	"github.com/grafana/loki/pkg/logproto"
+	"github.com/prometheus/common/model"
 )
 
 // Logger implements the github.com/go-kit/log.Logger interface. It supports
@@ -50,12 +54,18 @@ func (l *Logger) Update(o Options) error {
 
 func buildLogger(w io.Writer, o Options) (log.Logger, error) {
 	var l log.Logger
+	var wr io.Writer
+	wr = w
+
+	if len(o.WriteTo) > 0 {
+		wr = io.MultiWriter(w, &lokiWriter{o.WriteTo})
+	}
 
 	switch o.Format {
 	case FormatLogfmt:
-		l = log.NewLogfmtLogger(log.NewSyncWriter(w))
+		l = log.NewLogfmtLogger(log.NewSyncWriter(wr))
 	case FormatJSON:
-		l = log.NewJSONLogger(log.NewSyncWriter(w))
+		l = log.NewJSONLogger(log.NewSyncWriter(wr))
 	default:
 		return nil, fmt.Errorf("unrecognized log format %q", o.Format)
 	}
@@ -65,3 +75,24 @@ func buildLogger(w io.Writer, o Options) (log.Logger, error) {
 	l = log.With(l, "ts", log.DefaultTimestampUTC)
 	return l, nil
 }
+
+type lokiWriter struct {
+	f []loki.LogsReceiver
+}
+
+func (fw *lokiWriter) Write(p []byte) (int, error) {
+	for _, receiver := range fw.f {
+		select {
+		case receiver.Chan() <- loki.Entry{
+			Labels: model.LabelSet{"component": "agent"},
+			Entry: logproto.Entry{
+				Timestamp: time.Now(),
+				Line:      string(p),
+			},
+		}:
+		default:
+			return 0, fmt.Errorf("lokiWriter failed to forward entry, channel was blocked")
+		}
+	}
+	return len(p), nil
+}
diff --git a/pkg/flow/logging/options.go b/pkg/flow/logging/options.go
index b7e61590caaa..d09995bd5b58 100644
--- a/pkg/flow/logging/options.go
+++ b/pkg/flow/logging/options.go
@@ -5,6 +5,7 @@ import (
 	"fmt"
 
 	"github.com/go-kit/log/level"
+	"github.com/grafana/agent/component/common/loki"
 	"github.com/grafana/agent/pkg/river"
 )
 
@@ -13,7 +14,7 @@ type Options struct {
 	Level  Level  `river:"level,attr,optional"`
 	Format Format `river:"format,attr,optional"`
 
-	// TODO: log sink parameter (e.g., to use the Windows Event logger)
+	WriteTo []loki.LogsReceiver `river:"write_to,attr,optional"`
 }
 
 // DefaultOptions holds defaults for creating a Logger.