1 Star 0 Fork 0

zhangjungang/beats

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
factory.go 3.74 KB
一键复制 编辑 原始数据 按行查看 历史
package eventlog
import (
"fmt"
"sort"
"strings"
"github.com/elastic/beats/libbeat/common"
"github.com/joeshaw/multierror"
)
var commonConfigKeys = []string{"api", "name", "fields", "fields_under_root", "tags"}
// ConfigCommon is the common configuration data used to instantiate a new
// EventLog. Each implementation is free to support additional configuration
// options.
type ConfigCommon struct {
API string `config:"api"` // Name of the API to use. Optional.
Name string `config:"name"` // Name of the event log or channel.
common.EventMetadata `config:",inline"` // Fields and tags to add to each event.
}
type validator interface {
Validate() error
}
func readConfig(
data map[string]interface{},
config interface{},
validKeys []string,
) error {
c, err := common.NewConfigFrom(data)
if err != nil {
return fmt.Errorf("Failed reading config. %v", err)
}
if err := c.Unpack(config); err != nil {
return fmt.Errorf("Failed unpacking config. %v", err)
}
var errs multierror.Errors
if len(validKeys) > 0 {
sort.Strings(validKeys)
// Check for invalid keys.
for k := range data {
k = strings.ToLower(k)
i := sort.SearchStrings(validKeys, k)
if i >= len(validKeys) || validKeys[i] != k {
errs = append(errs, fmt.Errorf("Invalid event log key '%s' "+
"found. Valid keys are %s", k, strings.Join(validKeys, ", ")))
}
}
}
if v, ok := config.(validator); ok {
if err := v.Validate(); err != nil {
errs = append(errs, err)
}
}
return errs.Err()
}
// Producer produces a new event log instance for reading event log records.
type producer func(map[string]interface{}) (EventLog, error)
// Channels lists the available channels (event logs).
type channels func() ([]string, error)
// eventLogInfo is the registration info associated with an event log API.
type eventLogInfo struct {
apiName string
priority int
producer producer
channels func() ([]string, error)
}
// eventLogs is a map of priorities to eventLogInfo. The lower numbers have
// higher priorities.
var eventLogs = make(map[int]eventLogInfo)
// Register registers an EventLog API. Only the APIs that are available for the
// runtime OS should be registered. Each API must have a unique priority.
func Register(apiName string, priority int, producer producer, channels channels) {
info, exists := eventLogs[priority]
if exists {
panic(fmt.Sprintf("%s API is already registered with priority %d. "+
"Cannot register %s", info.apiName, info.priority, apiName))
}
eventLogs[priority] = eventLogInfo{
apiName: apiName,
priority: priority,
producer: producer,
channels: channels,
}
}
// New creates and returns a new EventLog instance based on the given config
// and the registered EventLog producers.
func New(options map[string]interface{}) (EventLog, error) {
if len(eventLogs) == 0 {
return nil, fmt.Errorf("No event log API is available on this system")
}
var config ConfigCommon
if err := readConfig(options, &config, nil); err != nil {
return nil, err
}
// A specific API is being requested (usually done for testing).
if config.API != "" {
for _, v := range eventLogs {
debugf("Checking %s", v.apiName)
if strings.EqualFold(v.apiName, config.API) {
debugf("Using %s API for event log %s", v.apiName, config.Name)
e, err := v.producer(options)
return e, err
}
}
return nil, fmt.Errorf("%s API is not available", config.API)
}
// Use the API with the highest priority.
keys := make([]int, 0, len(eventLogs))
for key := range eventLogs {
keys = append(keys, key)
}
sort.Ints(keys)
eventLog := eventLogs[keys[0]]
debugf("Using highest priority API, %s, for event log %s",
eventLog.apiName, config.Name)
e, err := eventLog.producer(options)
return e, err
}
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/zhangjungang/beats.git
git@gitee.com:zhangjungang/beats.git
zhangjungang
beats
beats
v5.6.13

搜索帮助

0d507c66 1850385 C8b1a773 1850385