Method getTableInfo

plugins/destination/duckdb/client/migrate.go:243–296

(ctx context.Context, tableName string)

}

func (c *Client) getTableInfo(ctx context.Context, tableName string) (*schema.Table, error) {
	info := tableInfo{}
	rows, err := c.db.QueryContext(ctx, fmt.Sprintf(sqlTableInfo, tableName))
	if err != nil {
		if strings.Contains(err.Error(), fmt.Sprintf("Table with name %s does not exist!", tableName)) {
			// Table doesn't exist
			return nil, nil
		}
		return nil, err
	}
	defer rows.Close()
	for rows.Next() {
		colInfo := columnInfo{}
		if err := rows.Scan(
			&colInfo.index,
			&colInfo.name,
			&colInfo.typ,
			&colInfo.notNull,
			&colInfo.defaultValue,
			&colInfo.pk); err != nil {
			return nil, err
		}
		colInfo.typ = strings.ToLower(colInfo.typ)
		colInfo.unique, err = c.isColumnUnique(ctx, tableName, colInfo.name)
		if err != nil {
			return nil, err
		}
		info.columns = append(info.columns, colInfo)
	}
	if err := rows.Err(); err != nil {
		return nil, err
	}

	if len(info.columns) == 0 {
		// Table doesn't exist
		return nil, nil
	}

	columns := make(schema.ColumnList, len(info.columns))
	for i, col := range info.columns {
		columns[i] = schema.Column{
			Name:       col.name,
			Type:       duckDBToArrow(col.typ),
			NotNull:    col.notNull,
			PrimaryKey: col.pk,
			Unique:     col.unique,
		}
	}

	return &schema.Table{
		Name:    tableName,
		Columns: columns,
	}, nil
}