mirror of https://github.com/databricks/cli.git
25 lines
448 B
Go
25 lines
448 B
Go
|
package variable
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
|
||
|
|
"github.com/databricks/databricks-sdk-go"
|
||
|
|
)
|
||
|
|
|
||
|
|
type lookupQuery struct {
|
||
|
|
name string
|
||
|
|
}
|
||
|
|
|
||
|
|
func (l *lookupQuery) Resolve(ctx context.Context, w *databricks.WorkspaceClient) (string, error) {
|
||
|
|
entity, err := w.Queries.GetByDisplayName(ctx, l.name)
|
||
|
|
if err != nil {
|
||
|
|
return "", err
|
||
|
|
}
|
||
|
|
return fmt.Sprint(entity.Id), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (l *lookupQuery) String() string {
|
||
|
|
return fmt.Sprintf("query: %s", l.name)
|
||
|
|
}
|