databricks-cli/cmd/account/log-delivery/log-delivery.go

426 lines
15 KiB
Go
Executable File

// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT.
package log_delivery
import (
"fmt"
"github.com/databricks/cli/cmd/root"
"github.com/databricks/cli/libs/cmdio"
"github.com/databricks/cli/libs/flags"
"github.com/databricks/databricks-sdk-go/service/billing"
"github.com/spf13/cobra"
)
// Slice with functions to override default command behavior.
// Functions can be added from the `init()` function in manually curated files in this directory.
var cmdOverrides []func(*cobra.Command)
func New() *cobra.Command {
cmd := &cobra.Command{
Use: "log-delivery",
Short: `These APIs manage log delivery configurations for this account.`,
Long: `These APIs manage log delivery configurations for this account. The two
supported log types for this API are _billable usage logs_ and _audit logs_.
This feature is in Public Preview. This feature works with all account ID
types.
Log delivery works with all account types. However, if your account is on the
E2 version of the platform or on a select custom plan that allows multiple
workspaces per account, you can optionally configure different storage
destinations for each workspace. Log delivery status is also provided to know
the latest status of log delivery attempts. The high-level flow of billable
usage delivery:
1. **Create storage**: In AWS, [create a new AWS S3 bucket] with a specific
bucket policy. Using Databricks APIs, call the Account API to create a
[storage configuration object](:method:Storage/Create) that uses the bucket
name. 2. **Create credentials**: In AWS, create the appropriate AWS IAM role.
For full details, including the required IAM role policies and trust
relationship, see [Billable usage log delivery]. Using Databricks APIs, call
the Account API to create a [credential configuration
object](:method:Credentials/Create) that uses the IAM role"s ARN. 3. **Create
log delivery configuration**: Using Databricks APIs, call the Account API to
[create a log delivery configuration](:method:LogDelivery/Create) that uses
the credential and storage configuration objects from previous steps. You can
specify if the logs should include all events of that log type in your account
(_Account level_ delivery) or only events for a specific set of workspaces
(_workspace level_ delivery). Account level log delivery applies to all
current and future workspaces plus account level logs, while workspace level
log delivery solely delivers logs related to the specified workspaces. You can
create multiple types of delivery configurations per account.
For billable usage delivery: * For more information about billable usage logs,
see [Billable usage log delivery]. For the CSV schema, see the [Usage page]. *
The delivery location is <bucket-name>/<prefix>/billable-usage/csv/, where
<prefix> is the name of the optional delivery path prefix you set up during
log delivery configuration. Files are named
workspaceId=<workspace-id>-usageMonth=<month>.csv. * All billable usage logs
apply to specific workspaces (_workspace level_ logs). You can aggregate usage
for your entire account by creating an _account level_ delivery configuration
that delivers logs for all current and future workspaces in your account. *
The files are delivered daily by overwriting the month's CSV file for each
workspace.
For audit log delivery: * For more information about about audit log delivery,
see [Audit log delivery], which includes information about the used JSON
schema. * The delivery location is
<bucket-name>/<delivery-path-prefix>/workspaceId=<workspaceId>/date=<yyyy-mm-dd>/auditlogs_<internal-id>.json.
Files may get overwritten with the same content multiple times to achieve
exactly-once delivery. * If the audit log delivery configuration included
specific workspace IDs, only _workspace-level_ audit logs for those workspaces
are delivered. If the log delivery configuration applies to the entire account
(_account level_ delivery configuration), the audit log delivery includes
workspace-level audit logs for all workspaces in the account as well as
account-level audit logs. See [Audit log delivery] for details. * Auditable
events are typically available in logs within 15 minutes.
[Audit log delivery]: https://docs.databricks.com/administration-guide/account-settings/audit-logs.html
[Billable usage log delivery]: https://docs.databricks.com/administration-guide/account-settings/billable-usage-delivery.html
[Usage page]: https://docs.databricks.com/administration-guide/account-settings/usage.html
[create a new AWS S3 bucket]: https://docs.databricks.com/administration-guide/account-api/aws-storage.html`,
GroupID: "billing",
Annotations: map[string]string{
"package": "billing",
},
}
// Apply optional overrides to this command.
for _, fn := range cmdOverrides {
fn(cmd)
}
return cmd
}
// start create command
// Slice with functions to override default command behavior.
// Functions can be added from the `init()` function in manually curated files in this directory.
var createOverrides []func(
*cobra.Command,
*billing.WrappedCreateLogDeliveryConfiguration,
)
func newCreate() *cobra.Command {
cmd := &cobra.Command{}
var createReq billing.WrappedCreateLogDeliveryConfiguration
var createJson flags.JsonFlag
// TODO: short flags
cmd.Flags().Var(&createJson, "json", `either inline JSON string or @path/to/file.json with request body`)
// TODO: complex arg: log_delivery_configuration
cmd.Use = "create"
cmd.Short = `Create a new log delivery configuration.`
cmd.Long = `Create a new log delivery configuration.
Creates a new Databricks log delivery configuration to enable delivery of the
specified type of logs to your storage location. This requires that you
already created a [credential object](:method:Credentials/Create) (which
encapsulates a cross-account service IAM role) and a [storage configuration
object](:method:Storage/Create) (which encapsulates an S3 bucket).
For full details, including the required IAM role policies and bucket
policies, see [Deliver and access billable usage logs] or [Configure audit
logging].
**Note**: There is a limit on the number of log delivery configurations
available per account (each limit applies separately to each log type
including billable usage and audit logs). You can create a maximum of two
enabled account-level delivery configurations (configurations without a
workspace filter) per type. Additionally, you can create two enabled
workspace-level delivery configurations per workspace for each log type, which
means that the same workspace ID can occur in the workspace filter for no more
than two delivery configurations per log type.
You cannot delete a log delivery configuration, but you can disable it (see
[Enable or disable log delivery
configuration](:method:LogDelivery/PatchStatus)).
[Configure audit logging]: https://docs.databricks.com/administration-guide/account-settings/audit-logs.html
[Deliver and access billable usage logs]: https://docs.databricks.com/administration-guide/account-settings/billable-usage-delivery.html`
cmd.Annotations = make(map[string]string)
cmd.Args = func(cmd *cobra.Command, args []string) error {
check := cobra.ExactArgs(0)
return check(cmd, args)
}
cmd.PreRunE = root.MustAccountClient
cmd.RunE = func(cmd *cobra.Command, args []string) (err error) {
ctx := cmd.Context()
a := root.AccountClient(ctx)
if cmd.Flags().Changed("json") {
err = createJson.Unmarshal(&createReq)
if err != nil {
return err
}
}
response, err := a.LogDelivery.Create(ctx, createReq)
if err != nil {
return err
}
return cmdio.Render(ctx, response)
}
// Disable completions since they are not applicable.
// Can be overridden by manual implementation in `override.go`.
cmd.ValidArgsFunction = cobra.NoFileCompletions
// Apply optional overrides to this command.
for _, fn := range createOverrides {
fn(cmd, &createReq)
}
return cmd
}
func init() {
cmdOverrides = append(cmdOverrides, func(cmd *cobra.Command) {
cmd.AddCommand(newCreate())
})
}
// start get command
// Slice with functions to override default command behavior.
// Functions can be added from the `init()` function in manually curated files in this directory.
var getOverrides []func(
*cobra.Command,
*billing.GetLogDeliveryRequest,
)
func newGet() *cobra.Command {
cmd := &cobra.Command{}
var getReq billing.GetLogDeliveryRequest
// TODO: short flags
cmd.Use = "get LOG_DELIVERY_CONFIGURATION_ID"
cmd.Short = `Get log delivery configuration.`
cmd.Long = `Get log delivery configuration.
Gets a Databricks log delivery configuration object for an account, both
specified by ID.
Arguments:
LOG_DELIVERY_CONFIGURATION_ID: Databricks log delivery configuration ID`
cmd.Annotations = make(map[string]string)
cmd.PreRunE = root.MustAccountClient
cmd.RunE = func(cmd *cobra.Command, args []string) (err error) {
ctx := cmd.Context()
a := root.AccountClient(ctx)
if len(args) == 0 {
promptSpinner := cmdio.Spinner(ctx)
promptSpinner <- "No LOG_DELIVERY_CONFIGURATION_ID argument specified. Loading names for Log Delivery drop-down."
names, err := a.LogDelivery.LogDeliveryConfigurationConfigNameToConfigIdMap(ctx, billing.ListLogDeliveryRequest{})
close(promptSpinner)
if err != nil {
return fmt.Errorf("failed to load names for Log Delivery drop-down. Please manually specify required arguments. Original error: %w", err)
}
id, err := cmdio.Select(ctx, names, "Databricks log delivery configuration ID")
if err != nil {
return err
}
args = append(args, id)
}
if len(args) != 1 {
return fmt.Errorf("expected to have databricks log delivery configuration id")
}
getReq.LogDeliveryConfigurationId = args[0]
response, err := a.LogDelivery.Get(ctx, getReq)
if err != nil {
return err
}
return cmdio.Render(ctx, response)
}
// Disable completions since they are not applicable.
// Can be overridden by manual implementation in `override.go`.
cmd.ValidArgsFunction = cobra.NoFileCompletions
// Apply optional overrides to this command.
for _, fn := range getOverrides {
fn(cmd, &getReq)
}
return cmd
}
func init() {
cmdOverrides = append(cmdOverrides, func(cmd *cobra.Command) {
cmd.AddCommand(newGet())
})
}
// start list command
// Slice with functions to override default command behavior.
// Functions can be added from the `init()` function in manually curated files in this directory.
var listOverrides []func(
*cobra.Command,
*billing.ListLogDeliveryRequest,
)
func newList() *cobra.Command {
cmd := &cobra.Command{}
var listReq billing.ListLogDeliveryRequest
// TODO: short flags
cmd.Flags().StringVar(&listReq.CredentialsId, "credentials-id", listReq.CredentialsId, `Filter by credential configuration ID.`)
cmd.Flags().Var(&listReq.Status, "status", `Filter by status ENABLED or DISABLED. Supported values: [DISABLED, ENABLED]`)
cmd.Flags().StringVar(&listReq.StorageConfigurationId, "storage-configuration-id", listReq.StorageConfigurationId, `Filter by storage configuration ID.`)
cmd.Use = "list"
cmd.Short = `Get all log delivery configurations.`
cmd.Long = `Get all log delivery configurations.
Gets all Databricks log delivery configurations associated with an account
specified by ID.`
cmd.Annotations = make(map[string]string)
cmd.Args = func(cmd *cobra.Command, args []string) error {
check := cobra.ExactArgs(0)
return check(cmd, args)
}
cmd.PreRunE = root.MustAccountClient
cmd.RunE = func(cmd *cobra.Command, args []string) (err error) {
ctx := cmd.Context()
a := root.AccountClient(ctx)
response, err := a.LogDelivery.ListAll(ctx, listReq)
if err != nil {
return err
}
return cmdio.Render(ctx, response)
}
// Disable completions since they are not applicable.
// Can be overridden by manual implementation in `override.go`.
cmd.ValidArgsFunction = cobra.NoFileCompletions
// Apply optional overrides to this command.
for _, fn := range listOverrides {
fn(cmd, &listReq)
}
return cmd
}
func init() {
cmdOverrides = append(cmdOverrides, func(cmd *cobra.Command) {
cmd.AddCommand(newList())
})
}
// start patch-status command
// Slice with functions to override default command behavior.
// Functions can be added from the `init()` function in manually curated files in this directory.
var patchStatusOverrides []func(
*cobra.Command,
*billing.UpdateLogDeliveryConfigurationStatusRequest,
)
func newPatchStatus() *cobra.Command {
cmd := &cobra.Command{}
var patchStatusReq billing.UpdateLogDeliveryConfigurationStatusRequest
var patchStatusJson flags.JsonFlag
// TODO: short flags
cmd.Flags().Var(&patchStatusJson, "json", `either inline JSON string or @path/to/file.json with request body`)
cmd.Use = "patch-status LOG_DELIVERY_CONFIGURATION_ID STATUS"
cmd.Short = `Enable or disable log delivery configuration.`
cmd.Long = `Enable or disable log delivery configuration.
Enables or disables a log delivery configuration. Deletion of delivery
configurations is not supported, so disable log delivery configurations that
are no longer needed. Note that you can't re-enable a delivery configuration
if this would violate the delivery configuration limits described under
[Create log delivery](:method:LogDelivery/Create).
Arguments:
LOG_DELIVERY_CONFIGURATION_ID: Databricks log delivery configuration ID
STATUS: Status of log delivery configuration. Set to ENABLED (enabled) or
DISABLED (disabled). Defaults to ENABLED. You can [enable or disable
the configuration](#operation/patch-log-delivery-config-status) later.
Deletion of a configuration is not supported, so disable a log delivery
configuration that is no longer needed.`
cmd.Annotations = make(map[string]string)
cmd.Args = func(cmd *cobra.Command, args []string) error {
if cmd.Flags().Changed("json") {
err := cobra.ExactArgs(1)(cmd, args)
if err != nil {
return fmt.Errorf("when --json flag is specified, provide only LOG_DELIVERY_CONFIGURATION_ID as positional arguments. Provide 'status' in your JSON input")
}
return nil
}
check := cobra.ExactArgs(2)
return check(cmd, args)
}
cmd.PreRunE = root.MustAccountClient
cmd.RunE = func(cmd *cobra.Command, args []string) (err error) {
ctx := cmd.Context()
a := root.AccountClient(ctx)
if cmd.Flags().Changed("json") {
err = patchStatusJson.Unmarshal(&patchStatusReq)
if err != nil {
return err
}
}
patchStatusReq.LogDeliveryConfigurationId = args[0]
if !cmd.Flags().Changed("json") {
_, err = fmt.Sscan(args[1], &patchStatusReq.Status)
if err != nil {
return fmt.Errorf("invalid STATUS: %s", args[1])
}
}
err = a.LogDelivery.PatchStatus(ctx, patchStatusReq)
if err != nil {
return err
}
return nil
}
// Disable completions since they are not applicable.
// Can be overridden by manual implementation in `override.go`.
cmd.ValidArgsFunction = cobra.NoFileCompletions
// Apply optional overrides to this command.
for _, fn := range patchStatusOverrides {
fn(cmd, &patchStatusReq)
}
return cmd
}
func init() {
cmdOverrides = append(cmdOverrides, func(cmd *cobra.Command) {
cmd.AddCommand(newPatchStatus())
})
}
// end service LogDelivery