cli/bundle/direct/dresources/all.go at 7641549f6db810dc87fb6404854b2dd2fbff3caf · databricks/cli · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
package dresources

import (
	"fmt"

	"github.com/databricks/databricks-sdk-go"
)

var SupportedResources = map[string]any{
	"jobs":                    (*ResourceJob)(nil),
	"pipelines":               (*ResourcePipeline)(nil),
	"experiments":             (*ResourceExperiment)(nil),
	"catalogs":                (*ResourceCatalog)(nil),
	"schemas":                 (*ResourceSchema)(nil),
	"external_locations":      (*ResourceExternalLocation)(nil),
	"volumes":                 (*ResourceVolume)(nil),
	"models":                  (*ResourceMlflowModel)(nil),
	"apps":                    (*ResourceApp)(nil),
	"sql_warehouses":          (*ResourceSqlWarehouse)(nil),
	"database_instances":      (*ResourceDatabaseInstance)(nil),
	"database_catalogs":       (*ResourceDatabaseCatalog)(nil),
	"synced_database_tables":  (*ResourceSyncedDatabaseTable)(nil),
	"postgres_projects":       (*ResourcePostgresProject)(nil),
	"postgres_branches":       (*ResourcePostgresBranch)(nil),
	"postgres_endpoints":      (*ResourcePostgresEndpoint)(nil),
	"alerts":                  (*ResourceAlert)(nil),
	"clusters":                (*ResourceCluster)(nil),
	"registered_models":       (*ResourceRegisteredModel)(nil),
	"dashboards":              (*ResourceDashboard)(nil),
	"secret_scopes":           (*ResourceSecretScope)(nil),
	"model_serving_endpoints": (*ResourceModelServingEndpoint)(nil),
	"quality_monitors":        (*ResourceQualityMonitor)(nil),

	// Permissions
	"jobs.permissions":                    (*ResourcePermissions)(nil),
	"pipelines.permissions":               (*ResourcePermissions)(nil),
	"apps.permissions":                    (*ResourcePermissions)(nil),
	"alerts.permissions":                  (*ResourcePermissions)(nil),
	"clusters.permissions":                (*ResourcePermissions)(nil),
	"database_instances.permissions":      (*ResourcePermissions)(nil),
	"postgres_projects.permissions":       (*ResourcePermissions)(nil),
	"experiments.permissions":             (*ResourcePermissions)(nil),
	"models.permissions":                  (*ResourceModelPermissions)(nil),
	"sql_warehouses.permissions":          (*ResourcePermissions)(nil),
	"secret_scopes.permissions":           (*ResourceSecretScopeAcls)(nil),
	"model_serving_endpoints.permissions": (*ResourcePermissions)(nil),
	"dashboards.permissions":              (*ResourcePermissions)(nil),

	// Grants
	"catalogs.grants":           (*ResourceGrants)(nil),
	"schemas.grants":            (*ResourceGrants)(nil),
	"external_locations.grants": (*ResourceGrants)(nil),
	"volumes.grants":            (*ResourceGrants)(nil),
	"registered_models.grants":  (*ResourceGrants)(nil),
}

func InitAll(client *databricks.WorkspaceClient) (map[string]*Adapter, error) {
	result := make(map[string]*Adapter)
	for resourceType, resource := range SupportedResources {
		adapter, err := NewAdapter(resource, resourceType, client)
		if err != nil {
			return nil, fmt.Errorf("%s: %w", resourceType, err)
		}
		result[resourceType] = adapter
	}
	return result, nil
}