diff --git a/pkg/handlers/get_datafile.go b/pkg/handlers/get_datafile.go index 394f1a67..1a51ae8f 100644 --- a/pkg/handlers/get_datafile.go +++ b/pkg/handlers/get_datafile.go @@ -36,7 +36,7 @@ func GetDatafile(w http.ResponseWriter, r *http.Request) { logger := middleware.GetLogger(r) - datafile := optlyClient.GetOptimizelyConfig().GetDatafile() + datafile := optlyClient.WithTraceContext(r.Context()).GetOptimizelyConfig().GetDatafile() var raw map[string]interface{} if err = json.Unmarshal([]byte(datafile), &raw); err != nil { RenderError(err, http.StatusInternalServerError, w, r) diff --git a/pkg/handlers/get_datafile_test.go b/pkg/handlers/get_datafile_test.go index 6d3f6c04..e90c89d4 100644 --- a/pkg/handlers/get_datafile_test.go +++ b/pkg/handlers/get_datafile_test.go @@ -1,5 +1,5 @@ /**************************************************************************** - * Copyright 2021,2023-2024 Optimizely, Inc. and contributors * + * Copyright 2021,2023 Optimizely, Inc. and contributors * * * * Licensed under the Apache License, Version 2.0 (the "License"); * * you may not use this file except in compliance with the License. * diff --git a/pkg/handlers/optimizely_config.go b/pkg/handlers/optimizely_config.go index 23ba32ed..cb79aa4b 100644 --- a/pkg/handlers/optimizely_config.go +++ b/pkg/handlers/optimizely_config.go @@ -35,7 +35,7 @@ func OptimizelyConfig(w http.ResponseWriter, r *http.Request) { logger := middleware.GetLogger(r) - conf := optlyClient.GetOptimizelyConfig() + conf := optlyClient.WithTraceContext(r.Context()).GetOptimizelyConfig() logger.Info().Msg("Successfully returned OptimizelyConfig") render.JSON(w, r, conf) }