Add method to set fluentbit hook

pkg/log/log.go

@@ -4,6 +4,7 @@ import (
 	"context"
 	"encoding/json"
 	"io"
+	"net/url"
 	"os"
 	"strings"
 	"time"
@@ -32,6 +33,12 @@ const (
 	LevelEnvName = "LOG_LEVEL"
 )
 
+var (
+	ErrEndpointNotSet = errors.New("fluentbit endpoint not set")
+	ErrPathSet        = errors.New("fluentbit endpoint path is set")
+	ErrNonTCPEndpoint = errors.New("fluentbit endpoint scheme must be tcp")
+)
+
 // OutputSink describes the current output sink.
 type OutputSink uint8
 
@@ -79,6 +86,22 @@ func SetOutput(sink OutputSink) error {
 	}
 }
 
+// SetFluenbitOutput sets the fluentbit output
+func SetFluentbitOutput(url *url.URL) error {
+	if url == nil {
+		return ErrEndpointNotSet
+	}
+	if url.Scheme != "tcp" {
+		return ErrNonTCPEndpoint
+	}
+	if url.Path != "" {
+		return ErrPathSet
+	}
+	hook := NewFluentbitHook(url.Host)
+	log.AddHook(hook)
+	return nil
+}
+
 var envVarFields field.Fields
 
 // initEnvVarFields populates envVarFields with values from the host's environment.