Method Dump

backend/plugin/db/databricks/dump.go:30–129

(ctx context.Context, writer io.Writer, _ *storepb.DatabaseSchemaMetadata)

)

func (d *Driver) Dump(ctx context.Context, writer io.Writer, _ *storepb.DatabaseSchemaMetadata) error {
	catalogMap, err := d.listCatologTables(ctx, "")
	if err != nil {
		return err
	}

	for catalogName, schemaMap := range catalogMap {
		if catalogName == sysCatalog {
			continue
		}
		if _, err := fmt.Fprintf(writer, "CREATE CATALOG %s\n", catalogName); err != nil {
			return err
		}

		for schemaName, tableList := range schemaMap {
			if schemaName == infoSchema {
				continue
			}
			if schemaName != dftSchema {
				if _, err := fmt.Fprintf(writer, "CREATE SCHEMA `%s`.`%s`\n", catalogName, schemaName); err != nil {
					return err
				}
			}

			viewDDL := strings.Builder{}
			mtViewDDL := strings.Builder{}
			extTblDDL := strings.Builder{}
			mngTblDDL := strings.Builder{}
			for _, tblUnion := range tableList {
				qualifiedName, err := getQualifiedTblName(catalogName, schemaName, tblUnion.name)
				if err != nil {
					return err
				}

				switch tblUnion.typeName {
				case catalog.TableTypeView:
					ddl, err := d.showCreateTable(ctx, qualifiedName)
					if err != nil {
						return err
					}
					formatDDL := fmt.Sprintf(schemaStmtFmt, "VIEW", qualifiedName, ddl)
					if _, err := viewDDL.WriteString(formatDDL); err != nil {
						return err
					}
				case catalog.TableTypeMaterializedView:
					tblAddInfo, err := d.descTbl(ctx, qualifiedName)
					if err != nil {
						return err
					}
					tblProperties, err := formatTblProperties(tblAddInfo)
					if err != nil {
						return err
					}
					formatDDL, err := genMaterializedViewDDL(qualifiedName, tblUnion.materialView, tblProperties)
					if err != nil {
						return err
					}
					if _, err := mtViewDDL.WriteString(formatDDL); err != nil {