Function resolveDataSourceID

backend/api/v1/sql_service.go:2206–2239

(ctx context.Context, instance *store.InstanceMessage, dataSourceID string, statement string, allowAdminDataSource bool)

}

func resolveDataSourceID(ctx context.Context, instance *store.InstanceMessage, dataSourceID string, statement string, allowAdminDataSource bool) (string, error) {
	if dataSourceID != "" {
		return dataSourceID, nil
	}

	var adminDataSourceID string
	var readOnlyDataSourceID string
	readOnlyCount := 0
	for _, dataSource := range instance.Metadata.GetDataSources() {
		switch dataSource.GetType() {
		case storepb.DataSourceType_ADMIN:
			adminDataSourceID = dataSource.GetId()
		case storepb.DataSourceType_READ_ONLY:
			readOnlyCount++
			readOnlyDataSourceID = dataSource.GetId()
		default:
		}
	}

	if allowAdminDataSource && adminDataSourceID != "" && requiresAdminDataSource(ctx, instance.Metadata.GetEngine(), statement) {
		return adminDataSourceID, nil
	}

	switch {
	case readOnlyCount == 1:
		return readOnlyDataSourceID, nil
	case readOnlyCount > 1:
		return "", connect.NewError(connect.CodeFailedPrecondition, errors.New("instance has multiple read-only data sources, please specify data_source_id explicitly"))
	case adminDataSourceID != "":
		return adminDataSourceID, nil
	default:
		return "", connect.NewError(connect.CodeFailedPrecondition, errors.New("instance has no admin data source"))
	}
}

func requiresAdminDataSource(ctx context.Context, engine storepb.Engine, statement string) bool {
	readOnly, _, err := parserbase.ValidateSQLForEditor(engine, statement)