forked from open-telemetry/opentelemetry-collector
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add factory and new-style config for Batch processor
This is part of remaining migration to new configuration format. - Added zap.Logger to ProcessorFactory functions because it is needed by this processor (and can be used if needed in other processors in the future). - Added Name/SetName to Processor interface because it is needed by this processor (this now also makes Processor uniform with Receiver and Exporter interfaces). Github issue: open-telemetry#52 Testing done: make
- Loading branch information
Tigran Najaryan
committed
Jun 25, 2019
1 parent
026061c
commit ffa30bd
Showing
16 changed files
with
322 additions
and
10 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,45 @@ | ||
// Copyright 2019, 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 nodebatcher | ||
|
||
import ( | ||
"time" | ||
|
||
"github.com/open-telemetry/opentelemetry-service/internal/configmodels" | ||
) | ||
|
||
// ConfigV2 defines configuration for batch processor. | ||
type ConfigV2 struct { | ||
configmodels.ProcessorSettings `mapstructure:",squash"` | ||
|
||
// Timeout sets the time after which a batch will be sent regardless of size | ||
Timeout *time.Duration `mapstructure:"timeout,omitempty"` | ||
|
||
// SendBatchSize is the size of a batch which after hit, will trigger it to be sent. | ||
SendBatchSize *int `mapstructure:"send-batch-size,omitempty"` | ||
|
||
// NumTickers sets the number of tickers to use to divide the work of looping | ||
// over batch buckets. This is an advanced configuration option. | ||
NumTickers int `mapstructure:"num-tickers,omitempty"` | ||
|
||
// TickTime sets time interval at which the tickers tick. This is an advanced | ||
// configuration option. | ||
TickTime *time.Duration `mapstructure:"tick-time,omitempty"` | ||
|
||
// RemoveAfterTicks is the number of ticks that must pass without a span arriving | ||
// from a node after which the batcher for that node will be deleted. This is an | ||
// advanved configuration option. | ||
RemoveAfterTicks *int `mapstructure:"remove-after-ticks,omitempty"` | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,63 @@ | ||
// Copyright 2019, 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 nodebatcher | ||
|
||
import ( | ||
"path" | ||
"testing" | ||
"time" | ||
|
||
"github.com/stretchr/testify/assert" | ||
"github.com/stretchr/testify/require" | ||
|
||
"github.com/open-telemetry/opentelemetry-service/internal/configmodels" | ||
"github.com/open-telemetry/opentelemetry-service/internal/configv2" | ||
"github.com/open-telemetry/opentelemetry-service/internal/factories" | ||
) | ||
|
||
var _ = configv2.RegisterTestFactories() | ||
|
||
func TestLoadConfig(t *testing.T) { | ||
|
||
factory := factories.GetProcessorFactory(typeStr) | ||
|
||
config, err := configv2.LoadConfigFile(t, path.Join(".", "testdata", "config.yaml")) | ||
|
||
require.Nil(t, err) | ||
require.NotNil(t, config) | ||
|
||
p0 := config.Processors["batch"] | ||
assert.Equal(t, p0, factory.CreateDefaultConfig()) | ||
|
||
p1 := config.Processors["batch/2"] | ||
|
||
timeout := time.Second * 10 | ||
tickTime := time.Second * 5 | ||
removeAfterTicks := 20 | ||
sendBatchSize := 1000 | ||
|
||
assert.Equal(t, p1, | ||
&ConfigV2{ | ||
ProcessorSettings: configmodels.ProcessorSettings{ | ||
TypeVal: "batch", | ||
NameVal: "batch/2", | ||
}, | ||
Timeout: &timeout, | ||
NumTickers: 10, | ||
RemoveAfterTicks: &removeAfterTicks, | ||
SendBatchSize: &sendBatchSize, | ||
TickTime: &tickTime, | ||
}) | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,105 @@ | ||
// Copyright 2019, 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 nodebatcher | ||
|
||
import ( | ||
"go.uber.org/zap" | ||
|
||
"github.com/open-telemetry/opentelemetry-service/consumer" | ||
"github.com/open-telemetry/opentelemetry-service/internal/configmodels" | ||
"github.com/open-telemetry/opentelemetry-service/internal/factories" | ||
"github.com/open-telemetry/opentelemetry-service/processor" | ||
) | ||
|
||
var _ = factories.RegisterProcessorFactory(&processorFactory{}) | ||
|
||
const ( | ||
// The value of "type" key in configuration. | ||
typeStr = "batch" | ||
) | ||
|
||
// processorFactory is the factory for batch processor. | ||
type processorFactory struct { | ||
} | ||
|
||
// Type gets the type of the config created by this factory. | ||
func (f *processorFactory) Type() string { | ||
return typeStr | ||
} | ||
|
||
// CreateDefaultConfig creates the default configuration for processor. | ||
func (f *processorFactory) CreateDefaultConfig() configmodels.Processor { | ||
removeAfterTicks := int(defaultRemoveAfterCycles) | ||
sendBatchSize := int(defaultSendBatchSize) | ||
tickTime := defaultTickTime | ||
timeout := defaultTimeout | ||
|
||
return &ConfigV2{ | ||
ProcessorSettings: configmodels.ProcessorSettings{ | ||
TypeVal: typeStr, | ||
NameVal: typeStr, | ||
}, | ||
RemoveAfterTicks: &removeAfterTicks, | ||
SendBatchSize: &sendBatchSize, | ||
NumTickers: defaultNumTickers, | ||
TickTime: &tickTime, | ||
Timeout: &timeout, | ||
} | ||
} | ||
|
||
// CreateTraceProcessor creates a trace processor based on this config. | ||
func (f *processorFactory) CreateTraceProcessor( | ||
logger *zap.Logger, | ||
nextConsumer consumer.TraceConsumer, | ||
c configmodels.Processor, | ||
) (processor.TraceProcessor, error) { | ||
cfg := c.(*ConfigV2) | ||
|
||
var batchingOptions []Option | ||
if cfg.Timeout != nil { | ||
batchingOptions = append(batchingOptions, WithTimeout(*cfg.Timeout)) | ||
} | ||
if cfg.NumTickers > 0 { | ||
batchingOptions = append( | ||
batchingOptions, WithNumTickers(cfg.NumTickers), | ||
) | ||
} | ||
if cfg.TickTime != nil { | ||
batchingOptions = append( | ||
batchingOptions, WithTickTime(*cfg.TickTime), | ||
) | ||
} | ||
if cfg.SendBatchSize != nil { | ||
batchingOptions = append( | ||
batchingOptions, WithSendBatchSize(*cfg.SendBatchSize), | ||
) | ||
} | ||
if cfg.RemoveAfterTicks != nil { | ||
batchingOptions = append( | ||
batchingOptions, WithRemoveAfterTicks(*cfg.RemoveAfterTicks), | ||
) | ||
} | ||
|
||
return NewBatcher(cfg.NameVal, logger, nextConsumer, batchingOptions...), nil | ||
} | ||
|
||
// CreateMetricsProcessor creates a metrics processor based on this config. | ||
func (f *processorFactory) CreateMetricsProcessor( | ||
logger *zap.Logger, | ||
nextConsumer consumer.MetricsConsumer, | ||
cfg configmodels.Processor, | ||
) (processor.MetricsProcessor, error) { | ||
return nil, factories.ErrDataTypeIsNotSupported | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,49 @@ | ||
// Copyright 2019, 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 nodebatcher | ||
|
||
import ( | ||
"testing" | ||
|
||
"go.uber.org/zap" | ||
|
||
"github.com/stretchr/testify/assert" | ||
"github.com/stretchr/testify/require" | ||
|
||
"github.com/open-telemetry/opentelemetry-service/internal/factories" | ||
) | ||
|
||
func TestCreateDefaultConfig(t *testing.T) { | ||
factory := factories.GetProcessorFactory(typeStr) | ||
require.NotNil(t, factory) | ||
|
||
cfg := factory.CreateDefaultConfig() | ||
assert.NotNil(t, cfg, "failed to create default config") | ||
} | ||
|
||
func TestCreateProcessor(t *testing.T) { | ||
factory := factories.GetProcessorFactory(typeStr) | ||
require.NotNil(t, factory) | ||
|
||
cfg := factory.CreateDefaultConfig() | ||
|
||
tp, err := factory.CreateTraceProcessor(zap.NewNop(), nil, cfg) | ||
assert.NotNil(t, tp) | ||
assert.NoError(t, err, "cannot create trace processor") | ||
|
||
mp, err := factory.CreateMetricsProcessor(zap.NewNop(), nil, cfg) | ||
assert.Nil(t, mp) | ||
assert.Error(t, err, "should not be able to create metric processor") | ||
} |
20 changes: 20 additions & 0 deletions
20
internal/collector/processor/nodebatcher/testdata/config.yaml
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,20 @@ | ||
receivers: | ||
examplereceiver: | ||
|
||
processors: | ||
batch: | ||
batch/2: | ||
timeout: 10s | ||
send-batch-size: 1000 | ||
num-tickers: 10 | ||
tick-time: 5s | ||
remove-after-ticks: 20 | ||
|
||
exporters: | ||
exampleexporter: | ||
|
||
pipelines: | ||
traces: | ||
receivers: [examplereceiver] | ||
processors: [batch/2] | ||
exporters: [exampleexporter] |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.