add trace context

optimizely · Jan 19, 2024 · 055831a · 055831a
1 parent c07acc0
commit 055831a
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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
@@ -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)
 }