Method getTableInfo

backend/plugin/db/hive/sync.go:238–323

(ctx context.Context, tableName string, databaseName string)

)

func (d *Driver) getTableInfo(ctx context.Context, tableName string, databaseName string) (
	*TableInfo,
	error,
) {
	tableInfo := &TableInfo{}

	query := fmt.Sprintf("DESCRIBE FORMATTED `%s`.`%s`", databaseName, tableName)
	rows, err := d.db.QueryContext(ctx, query)
	if err != nil {
		return nil, errors.Wrapf(err, "failed to describe table %s", tableName)
	}
	defer rows.Close()

	section := columnSection
	for rows.Next() {
		var colName, dataType, comment string
		if err := rows.Scan(&colName, &dataType, &comment); err != nil {
			return nil, errors.Wrap(err, "failed to scan row")
		}

		// The first rows are column metadata, followed by "# Detailed Table Information" and "# Storage Information"
		switch {
		case strings.HasPrefix(colName, "# col_name"):
			continue
		case strings.HasPrefix(colName, "# Detailed Table Information"):
			section = tableInfoSection
		case strings.HasPrefix(colName, "# Storage Information"):
			section = storageSection
		case strings.HasPrefix(colName, "# View Information"):
			section = viewSection
		default:
			// No action needed for other cases
		}
		switch section {
		case columnSection:
			if colName != "" && dataType != "" {
				// Column metadata.
				position := len(tableInfo.colMetadatas) + 1
				tableInfo.colMetadatas = append(tableInfo.colMetadatas, &storepb.ColumnMetadata{
					Name:     colName,
					Type:     dataType,
					Comment:  comment,
					Position: int32(position),
				})
			}
		case tableInfoSection:
			switch {
			case trimField(colName) == "Table Type:":
				tableInfo.tableType = trimField(dataType)
			case trimField(dataType) == "numRows":
				n, err := strconv.Atoi(trimField(comment))
				if err != nil {
					return nil, errors.Wrapf(err, "failed to parse row count")
				}
				tableInfo.numRows = n
			case trimField(dataType) == "totalSize":
				n, err := strconv.Atoi(trimField(comment))
				if err != nil {