Method SyncInstance

backend/plugin/db/databricks/sync.go:53–88

(ctx context.Context)

}

func (d *Driver) SyncInstance(ctx context.Context) (*db.InstanceMetadata, error) {
	instanceMetadata := &db.InstanceMetadata{}

	// fetch version.
	versionData, _, err := d.execStatementSync(ctx, "SELECT VERSION()", 0)
	if err != nil {
		return nil, err
	}
	if len(versionData) != 1 || len(versionData[0]) != 1 {
		return nil, errors.New("invalid version format")
	}
	splitVersion := strings.Split(versionData[0][0], " ")
	if len(splitVersion) != 2 {
		return nil, errors.New("invalid version format")
	}
	instanceMetadata.Version = splitVersion[0]

	// fetch table data from databricks.
	catalogMap, err := d.listCatologTables(ctx, "")
	if err != nil {
		return nil, err
	}

	for catalogName, schemaMap := range catalogMap {
		dbMetadataMeta := storepb.DatabaseSchemaMetadata{}
		schemas := convertToStorepbSchemas(schemaMap)
		dbMetadataMeta.Name = catalogName
		dbMetadataMeta.Schemas = schemas
		instanceMetadata.Databases = append(instanceMetadata.Databases, &dbMetadataMeta)
	}

	// fetch workspace users.
	// TODO(tommy): complete this part when Permissions API for Golang is implemented.

	return instanceMetadata, nil
}

// list all tables in the workspace when catalogName is set to "".
func (d *Driver) listCatologTables(ctx context.Context, targetCatalogName string) (databricksCatalogMap, error) {