systemdtailer.go

// Copyright © 2019 Banzai Cloud
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//    http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package v1alpha1

import (
	"fmt"
	"strings"

	"github.com/kube-logging/logging-operator/pkg/sdk/extensions/api/tailer"
	config "github.com/kube-logging/logging-operator/pkg/sdk/extensions/extensionsconfig"
)

func (s SystemdTailer) defaults() SystemdTailer {
	result := s
	// setting defaults
	if result.Path == "" {
		result.Path = "/var/log/journal"
	}
	if result.MaxEntries == 0 {
		result.MaxEntries = 1000
	}
	return result
}

// Command returns the desired command for the current systemdtailer
func (s SystemdTailer) Command(Name string) []string {
	s = s.defaults()
	command := []string{
		"/fluent-bit/bin/fluent-bit", "-i", "systemd",
		"-p", fmt.Sprintf("path=%s", s.Path),
		"-p", fmt.Sprintf("db=/var/pos/%s.db", Name),
		"-p", fmt.Sprintf("max_entries=%d", s.MaxEntries),
	}
	if s.SystemdFilter != "" {
		// check if filter includes journal field, if not default to _SYSTEMD_UNIT
		if !strings.Contains(s.SystemdFilter, "=") {
			s.SystemdFilter = fmt.Sprintf("_SYSTEMD_UNIT=%s", s.SystemdFilter)
		}
		command = append(command, "-p", fmt.Sprintf("systemd_filter=%s", s.SystemdFilter))
	}
	command = append(command,
		"-o", "file",
		"-p", "format=plain",
	)
	if !s.Verbose {
		command = append(command, "-qq")
	}
	command = append(command, config.HostTailer.VersionedFluentBitPathArgs("/dev/stdout")...)
	return command
}

// GeneralDescriptor returns the tailer.General general Tailer struct
func (s SystemdTailer) GeneralDescriptor() tailer.General {
	s = s.defaults()
	return tailer.General{Name: s.Name, Path: s.Path, Disabled: s.Disabled, ContainerBase: s.ContainerBase, Image: s.Image}
}