do for datasources too

wbrowne · wbrowne · commit fc93e9ae2997 · 2025-07-24T14:50:26.000+01:00
diff --git a/backend/datasource/instance_provider.go b/backend/datasource/instance_provider.go
@@ -30,7 +30,7 @@ type InstanceFactoryFunc func(ctx context.Context, settings backend.DataSourceIn
 // and providing that to instancemgmt.New.
 func NewInstanceManager(fn InstanceFactoryFunc) instancemgmt.InstanceManager {
 	ip := NewInstanceProvider(fn)
-	return instancemgmt.New(ip)
+	return instancemgmt.NewContextAwareInstanceManager(ip)
 }
 
 // NewInstanceProvider create a new data source instance provuder,

Original file line number	Diff line number	Diff line change
`@@ -30,7 +30,7 @@ type InstanceFactoryFunc func(ctx context.Context, settings backend.DataSourceIn`
`30`	`30`	`// and providing that to instancemgmt.New.`
`31`	`31`	`func NewInstanceManager(fn InstanceFactoryFunc) instancemgmt.InstanceManager {`
`32`	`32`	`ip := NewInstanceProvider(fn)`
`33`		`- return instancemgmt.New(ip)`
	`33`	`+ return instancemgmt.NewContextAwareInstanceManager(ip)`
`34`	`34`	`}`
`35`	`35`
`36`	`36`	`// NewInstanceProvider create a new data source instance provuder,`