Method schemaUsingConn

runtime/pkg/rduckdb/generic.go:467–529

(ctx context.Context, ilike, name string, pageSize uint32, pageToken string, conn *sqlx.Conn)

}

func (m *generic) schemaUsingConn(ctx context.Context, ilike, name string, pageSize uint32, pageToken string, conn *sqlx.Conn) ([]*Table, string, error) {
	if ilike != "" && name != "" {
		return nil, "", fmt.Errorf("cannot specify both `ilike` and `name`")
	}

	var whereClause string
	var args []any
	if ilike != "" {
		whereClause = " AND t.table_name ilike ?"
		args = []any{ilike}
	} else if name != "" {
		whereClause = " AND t.table_name = ?"
		args = []any{name}
	}

	// Add pagination filter
	if pageToken != "" {
		var startAfterName string
		if err := pagination.UnmarshalPageToken(pageToken, &startAfterName); err != nil {
			return nil, "", fmt.Errorf("invalid page token: %w", err)
		}
		whereClause += " AND t.table_name > ?"
		args = append(args, startAfterName)
	}

	q := fmt.Sprintf(`
		SELECT
			coalesce(t.table_catalog, current_database()) AS "database",
			current_schema() AS "schema",
			t.table_name AS "name",
			t.table_type = 'VIEW' AS "view", 
			array_agg(c.column_name ORDER BY c.ordinal_position) AS "column_names",
			array_agg(c.data_type ORDER BY c.ordinal_position) AS "column_types",
			array_agg(c.is_nullable = 'YES' ORDER BY c.ordinal_position) AS "column_nullable"
		FROM information_schema.tables t
		JOIN information_schema.columns c 
			ON t.table_schema = c.table_schema 
			AND t.table_name = c.table_name
		WHERE database = current_database() 
			AND t.table_schema = current_schema()
			%s
		GROUP BY ALL
		ORDER BY t.table_name
		LIMIT ?
	`, whereClause)

	limit := pagination.ValidPageSize(pageSize, drivers.DefaultPageSize)
	args = append(args, limit+1)

	var res []*Table
	err := conn.SelectContext(ctx, &res, q, args...)
	if err != nil {
		return nil, "", err
	}

	next := ""
	if len(res) > limit {
		res = res[:limit]