sql_warehouses.go

package resources

import (
	"context"
	"net/url"

	"github.com/databricks/cli/libs/log"
	"github.com/databricks/databricks-sdk-go"
	"github.com/databricks/databricks-sdk-go/marshal"
	"github.com/databricks/databricks-sdk-go/service/sql"
)

type SqlWarehousePermissionLevel string

type SqlWarehousePermission struct {
	Level SqlWarehousePermissionLevel `json:"level"`

	UserName             string `json:"user_name,omitempty"`
	ServicePrincipalName string `json:"service_principal_name,omitempty"`
	GroupName            string `json:"group_name,omitempty"`
}

type SqlWarehouse struct {
	BaseResource

	Permissions []SqlWarehousePermission `json:"permissions,omitempty"`

	sql.CreateWarehouseRequest
}

func (sw *SqlWarehouse) UnmarshalJSON(b []byte) error {
	return marshal.Unmarshal(b, sw)
}

func (sw SqlWarehouse) MarshalJSON() ([]byte, error) {
	return marshal.Marshal(sw)
}

func (sw *SqlWarehouse) Exists(ctx context.Context, w *databricks.WorkspaceClient, id string) (bool, error) {
	_, err := w.Warehouses.GetById(ctx, id)
	if err != nil {
		log.Debugf(ctx, "sql warehouse %s does not exist", id)
		return false, err
	}
	return true, nil
}

func (*SqlWarehouse) ResourceDescription() ResourceDescription {
	return ResourceDescription{
		SingularName:  "sql_warehouse",
		PluralName:    "sql_warehouses",
		SingularTitle: "Sql Warehouse",
		PluralTitle:   "SQL Warehouses",
	}
}

func (sw *SqlWarehouse) InitializeURL(baseURL url.URL) {
	if sw.ID == "" {
		return
	}
	baseURL.Path = "sql/warehouses/" + sw.ID
	sw.URL = baseURL.String()
}

func (sw *SqlWarehouse) GetName() string {
	return sw.Name
}

func (sw *SqlWarehouse) GetURL() string {
	return sw.URL
}