Chore: Add context to org (#40685)
* Add context to org * Rebase * Fix rebase
This commit is contained in:
@@ -1,6 +1,7 @@
|
||||
package datasources
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"io/ioutil"
|
||||
"os"
|
||||
@@ -17,7 +18,7 @@ type configReader struct {
|
||||
log log.Logger
|
||||
}
|
||||
|
||||
func (cr *configReader) readConfig(path string) ([]*configs, error) {
|
||||
func (cr *configReader) readConfig(ctx context.Context, path string) ([]*configs, error) {
|
||||
var datasources []*configs
|
||||
|
||||
files, err := ioutil.ReadDir(path)
|
||||
@@ -39,7 +40,7 @@ func (cr *configReader) readConfig(path string) ([]*configs, error) {
|
||||
}
|
||||
}
|
||||
|
||||
err = cr.validateDefaultUniqueness(datasources)
|
||||
err = cr.validateDefaultUniqueness(ctx, datasources)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
@@ -88,7 +89,7 @@ func (cr *configReader) parseDatasourceConfig(path string, file os.FileInfo) (*c
|
||||
return v0.mapToDatasourceFromConfig(apiVersion.APIVersion), nil
|
||||
}
|
||||
|
||||
func (cr *configReader) validateDefaultUniqueness(datasources []*configs) error {
|
||||
func (cr *configReader) validateDefaultUniqueness(ctx context.Context, datasources []*configs) error {
|
||||
defaultCount := map[int64]int{}
|
||||
for i := range datasources {
|
||||
if datasources[i].Datasources == nil {
|
||||
@@ -100,7 +101,7 @@ func (cr *configReader) validateDefaultUniqueness(datasources []*configs) error
|
||||
ds.OrgID = 1
|
||||
}
|
||||
|
||||
if err := cr.validateAccessAndOrgID(ds); err != nil {
|
||||
if err := cr.validateAccessAndOrgID(ctx, ds); err != nil {
|
||||
return fmt.Errorf("failed to provision %q data source: %w", ds.Name, err)
|
||||
}
|
||||
|
||||
@@ -122,8 +123,8 @@ func (cr *configReader) validateDefaultUniqueness(datasources []*configs) error
|
||||
return nil
|
||||
}
|
||||
|
||||
func (cr *configReader) validateAccessAndOrgID(ds *upsertDataSourceFromConfig) error {
|
||||
if err := utils.CheckOrgExists(ds.OrgID); err != nil {
|
||||
func (cr *configReader) validateAccessAndOrgID(ctx context.Context, ds *upsertDataSourceFromConfig) error {
|
||||
if err := utils.CheckOrgExists(ctx, ds.OrgID); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user