Add WithContext for configuration repo get call (#318)

* add context Signed-off-by: troychiu <[email protected]> * spell error Signed-off-by: troychiu <[email protected]> --------- Signed-off-by: troychiu <[email protected]>
flyteorg · Jun 12, 2024 · f601edb · f601edb
1 parent 0683d18
commit f601edb
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/flyteadmin/pkg/repositories/gormimpl/configuration_repo.go b/flyteadmin/pkg/repositories/gormimpl/configuration_repo.go
@@ -25,7 +25,7 @@ func (r *ConfigurationRepo) GetActive(ctx context.Context) (models.Configuration
 	var metadata models.ConfigurationDocumentMetadata
 	timer := r.metrics.GetDuration.Start()
 	// Get the only one active configuration document
-	tx := r.db.Where(&models.ConfigurationDocumentMetadata{
+	tx := r.db.WithContext(ctx).Where(&models.ConfigurationDocumentMetadata{
 		Active: true,
 	}).Take(&metadata)
 	timer.Stop()

diff --git a/flytepropeller/pkg/controller/nodes/array/event_recorder_test.go b/flytepropeller/pkg/controller/nodes/array/event_recorder_test.go
@@ -37,7 +37,7 @@ func newBufferedEventRecorder() *bufferedEventRecorder {
 }
 
 func TestGetPluginLogs(t *testing.T) {
-	// intitialize log plugin
+	// initialize log plugin
 	logConfig := &logs.LogConfig{
 		Templates: []tasklog.TemplateLogPlugin{
 			tasklog.TemplateLogPlugin{