feat: add ability to set a global vector sink

This commit is contained in:
Jose Diaz-Gonzalez
2021-01-04 01:11:49 -05:00
parent e1425a85e4
commit 43d6a2ad94
5 changed files with 51 additions and 4 deletions

View File

@@ -120,6 +120,9 @@ func valueToConfig(appName string, value string) (vectorSink, error) {
t := fmt.Sprintf("type=%s", u.Scheme)
i := fmt.Sprintf("inputs[]=docker-source:%s", appName)
if appName == "--global" {
i = "inputs[]=docker-global-source"
}
initialQuery := fmt.Sprintf("%s&%s", t, i)
query := u.RawQuery
if query == "" {
@@ -171,6 +174,21 @@ func writeVectorConfig() error {
data.Sinks[fmt.Sprintf("docker-sink:%s", appName)] = sink
}
value := common.PropertyGet("logs", "--global", "vector-sink")
if value != "" {
sink, err := valueToConfig("--global", value)
if err != nil {
return err
}
data.Sources["docker-global-source"] = vectorSource{
Type: "docker_logs",
IncludeLabels: []string{"com.dokku.app-name"},
}
data.Sinks["docker-global-sink"] = sink
}
b, err := json.MarshalIndent(data, "", " ")
if err != nil {
return err

View File

@@ -13,7 +13,9 @@ var (
}
// GlobalProperties is a map of all valid global logs properties
GlobalProperties = map[string]bool{}
GlobalProperties = map[string]bool{
"vector-sink": true,
}
)
const VectorImage = "timberio/vector:0.11.X-debian"

View File

@@ -11,7 +11,8 @@ func ReportSingleApp(appName, infoFlag string) error {
}
flags := map[string]common.ReportFunc{
"--logs-vector-sink": reportVectorSink,
"--logs-vector-sink": reportVectorSink,
"--logs-global-vector-sink": reportGlobalVectorSink,
}
trimPrefix := false
@@ -23,3 +24,7 @@ func ReportSingleApp(appName, infoFlag string) error {
func reportVectorSink(appName string) string {
return common.PropertyGet("logs", appName, "vector-sink")
}
func reportGlobalVectorSink(appName string) string {
return common.PropertyGet("logs", "--global", "vector-sink")
}

View File

@@ -34,10 +34,16 @@ func main() {
}
case "set":
args := flag.NewFlagSet("logs:set", flag.ExitOnError)
global := args.Bool("global", false, "--global: set a global property")
args.Parse(os.Args[2:])
appName := args.Arg(0)
property := args.Arg(1)
value := args.Arg(2)
if *global {
appName = "--global"
property = args.Arg(0)
value = args.Arg(1)
}
err = logs.CommandSet(appName, property, value)
case "vector-logs":
args := flag.NewFlagSet("logs:vector-logs", flag.ExitOnError)