opentelemetry-collector/processor/batchprocessor/factory.go

81 lines
2.4 KiB
Go

// Copyright The OpenTelemetry Authors
//
// 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 batchprocessor // import "go.opentelemetry.io/collector/processor/batchprocessor"
import (
"context"
"time"
"go.opentelemetry.io/collector/component"
"go.opentelemetry.io/collector/config"
"go.opentelemetry.io/collector/consumer"
)
const (
// The value of "type" key in configuration.
typeStr = "batch"
defaultSendBatchSize = uint32(8192)
defaultTimeout = 200 * time.Millisecond
)
// NewFactory returns a new factory for the Batch processor.
func NewFactory() component.ProcessorFactory {
return component.NewProcessorFactory(
typeStr,
createDefaultConfig,
component.WithTracesProcessor(createTracesProcessor),
component.WithMetricsProcessor(createMetricsProcessor),
component.WithLogsProcessor(createLogsProcessor))
}
func createDefaultConfig() config.Processor {
return &Config{
ProcessorSettings: config.NewProcessorSettings(config.NewComponentID(typeStr)),
SendBatchSize: defaultSendBatchSize,
Timeout: defaultTimeout,
}
}
func createTracesProcessor(
_ context.Context,
set component.ProcessorCreateSettings,
cfg config.Processor,
nextConsumer consumer.Traces,
) (component.TracesProcessor, error) {
level := set.MetricsLevel
return newBatchTracesProcessor(set, nextConsumer, cfg.(*Config), level)
}
func createMetricsProcessor(
_ context.Context,
set component.ProcessorCreateSettings,
cfg config.Processor,
nextConsumer consumer.Metrics,
) (component.MetricsProcessor, error) {
level := set.MetricsLevel
return newBatchMetricsProcessor(set, nextConsumer, cfg.(*Config), level)
}
func createLogsProcessor(
_ context.Context,
set component.ProcessorCreateSettings,
cfg config.Processor,
nextConsumer consumer.Logs,
) (component.LogsProcessor, error) {
level := set.MetricsLevel
return newBatchLogsProcessor(set, nextConsumer, cfg.(*Config), level)
}