Skip to content

Commit 7f47d81

Browse files
feature_tracking : Update ExtractFeatures to receive both userUc and mergedUc for getOTelLoggingSupportedConfig. (#1899)
1 parent 7576cc8 commit 7f47d81

File tree

12 files changed

+165
-57
lines changed

12 files changed

+165
-57
lines changed

confgenerator/confgenerator_test.go

+5-5
Original file line numberDiff line numberDiff line change
@@ -255,7 +255,7 @@ func generateConfigs(pc platformConfig, testDir string) (got map[string]string,
255255
}
256256
}()
257257

258-
uc, err := confgenerator.MergeConfFiles(
258+
mergedUc, err := confgenerator.MergeConfFiles(
259259
ctx,
260260
filepath.Join("testdata", testDir, inputFileName),
261261
apps.BuiltInConfStructs,
@@ -266,7 +266,7 @@ func generateConfigs(pc platformConfig, testDir string) (got map[string]string,
266266
got[builtinConfigFileName] = apps.BuiltInConfStructs[pc.platform.Name()].String()
267267

268268
// Fluent Bit configs
269-
flbGeneratedConfigs, err := uc.GenerateFluentBitConfigs(ctx,
269+
flbGeneratedConfigs, err := mergedUc.GenerateFluentBitConfigs(ctx,
270270
pc.defaultLogsDir,
271271
pc.defaultStateDir,
272272
)
@@ -278,21 +278,21 @@ func generateConfigs(pc platformConfig, testDir string) (got map[string]string,
278278
}
279279

280280
// Otel configs
281-
otelGeneratedConfig, err := uc.GenerateOtelConfig(ctx)
281+
otelGeneratedConfig, err := mergedUc.GenerateOtelConfig(ctx)
282282
if err != nil {
283283
return
284284
}
285285
got["otel.yaml"] = otelGeneratedConfig
286286

287287
inputBytes, err := os.ReadFile(filepath.Join("testdata", testDir, inputFileName))
288288

289-
userConf, err := confgenerator.UnmarshalYamlToUnifiedConfig(ctx, inputBytes)
289+
userUc, err := confgenerator.UnmarshalYamlToUnifiedConfig(ctx, inputBytes)
290290
if err != nil {
291291
return
292292
}
293293

294294
// Feature Tracking
295-
extractedFeatures, err := confgenerator.ExtractFeatures(ctx, userConf)
295+
extractedFeatures, err := confgenerator.ExtractFeatures(ctx, userUc, mergedUc)
296296
if err != nil {
297297
return
298298
}

confgenerator/feature_tracking.go

+15-14
Original file line numberDiff line numberDiff line change
@@ -72,43 +72,43 @@ type CustomFeatures interface {
7272
// ExtractFeatures fields that containing a tracking tag will be tracked.
7373
// Automatic collection of bool or int fields. Any value that exists on tracking
7474
// tag will be used instead of value from UnifiedConfig.
75-
func ExtractFeatures(ctx context.Context, uc *UnifiedConfig) ([]Feature, error) {
76-
allFeatures := getOverriddenDefaultPipelines(uc)
77-
allFeatures = append(allFeatures, getSelfLogCollection(uc))
78-
allFeatures = append(allFeatures, getOTelLoggingSupportedConfig(ctx, uc))
75+
func ExtractFeatures(ctx context.Context, userUc, mergedUc *UnifiedConfig) ([]Feature, error) {
76+
allFeatures := getOverriddenDefaultPipelines(userUc)
77+
allFeatures = append(allFeatures, getSelfLogCollection(userUc))
78+
allFeatures = append(allFeatures, getOTelLoggingSupportedConfig(ctx, mergedUc))
7979

8080
var err error
8181
var tempTrackedFeatures []Feature
82-
if uc.HasMetrics() {
83-
tempTrackedFeatures, err = trackedMappedComponents("metrics", "receivers", uc.Metrics.Receivers)
82+
if userUc.HasMetrics() {
83+
tempTrackedFeatures, err = trackedMappedComponents("metrics", "receivers", userUc.Metrics.Receivers)
8484
if err != nil {
8585
return nil, err
8686
}
8787
allFeatures = append(allFeatures, tempTrackedFeatures...)
8888

89-
tempTrackedFeatures, err = trackedMappedComponents("metrics", "processors", uc.Metrics.Processors)
89+
tempTrackedFeatures, err = trackedMappedComponents("metrics", "processors", userUc.Metrics.Processors)
9090
if err != nil {
9191
return nil, err
9292
}
9393
allFeatures = append(allFeatures, tempTrackedFeatures...)
9494
}
9595

96-
if uc.HasLogging() {
97-
tempTrackedFeatures, err = trackedMappedComponents("logging", "receivers", uc.Logging.Receivers)
96+
if userUc.HasLogging() {
97+
tempTrackedFeatures, err = trackedMappedComponents("logging", "receivers", userUc.Logging.Receivers)
9898
if err != nil {
9999
return nil, err
100100
}
101101
allFeatures = append(allFeatures, tempTrackedFeatures...)
102102

103-
tempTrackedFeatures, err = trackedMappedComponents("logging", "processors", uc.Logging.Processors)
103+
tempTrackedFeatures, err = trackedMappedComponents("logging", "processors", userUc.Logging.Processors)
104104
if err != nil {
105105
return nil, err
106106
}
107107
allFeatures = append(allFeatures, tempTrackedFeatures...)
108108
}
109109

110-
if uc.HasCombined() {
111-
tempTrackedFeatures, err = trackedMappedComponents("combined", "receivers", uc.Combined.Receivers)
110+
if userUc.HasCombined() {
111+
tempTrackedFeatures, err = trackedMappedComponents("combined", "receivers", userUc.Combined.Receivers)
112112
if err != nil {
113113
return nil, err
114114
}
@@ -439,7 +439,8 @@ func getMetadata(field reflect.StructField) metadata {
439439
}
440440
}
441441

442-
func getOTelLoggingSupportedConfig(ctx context.Context, uc *UnifiedConfig) Feature {
442+
// TODO: b/399354366 - Cleanup when OTel Logging Support is fully released.
443+
func getOTelLoggingSupportedConfig(ctx context.Context, mergedUc *UnifiedConfig) Feature {
443444
feature := Feature{
444445
Module: "logging",
445446
Kind: "service",
@@ -448,7 +449,7 @@ func getOTelLoggingSupportedConfig(ctx context.Context, uc *UnifiedConfig) Featu
448449
Value: "false",
449450
}
450451

451-
if uc.OTelLoggingSupported(ctx) {
452+
if mergedUc.OTelLoggingSupported(ctx) {
452453
feature.Value = "true"
453454
}
454455

0 commit comments

Comments
 (0)