#node #tree #arena #reference #refcell #mutability #data

yanked arena-tree

An arena-based tree, with RefCell for mutability

Uses old Rust 2015

0.3.0 May 24, 2015
0.2.0 May 23, 2015
0.1.0 May 19, 2015

#25 in #mutability

34 downloads per month

MIT license

16KB
295 lines

No Maintenance Intended

No Maintenance Intended

This repository exists to publish the results of some experiments that I did over a couple week-ends that one time. This is not a software project that is being maintained over time.

If you would like to use this code, consider copying into your own source repository under the terms of the MIT license.

rust-forest

Various implementation strategies for “DOM-like” tree data structures in Rust.

“DOM-like” here means that data structures can be used to represent the parsed content of an HTML or XML document, like the DOM does, but don’t necessarily have the exact same API as the DOM. That is:

  • A tree is made up of nodes.
  • Each node has zero or more child nodes, which are ordered.
  • Each node has a no more than one parent, the node that it is a child of.
  • A node without a parent is called a root.
  • As a consequence, each node may also have siblings: its parent’s other children, if any.
  • From any given node, access to its parent, previous sibling, next sibling, first child, and last child (if any) can take no more than O(1) time.
  • Each node also has data associated to it, which for the purpose of this project is purely generic. For an HTML document, the data would be either the text of a text node, or the name and attributes of an element node.
  • The tree is mutable: nodes (with their sub-trees) can be inserted or removed anywhere in the tree.

In particular, the need to access parents means we can not use the obvious structure of child nodes being owned directly by their parent:

struct Node<T> {
    data: T,
    children: Vec<Node<T>>,
    parent: /* ??? */
}

More generally, a tree with parent and sibling relationships (in addition to children relationships) can be viewed as a graph of a special kind, but still a graph that contains cycles. And Rust’s default ownership model does not support cycles easily. Therefore, we need a more involved strategy to manage the lifetime of nodes.

In Servo, DOM nodes are managed by the JavaScript garbage collector. Rust however currently does not have a tracing garbage collector of its own, and many Rust projects might want to manipulate DOM-like trees without embedding an entire JavaScript virtual machine.

This repository contains multiple crates that each implement a DOM-like data structure with a different approach and different trade-offs.

rctree

The lifetime of nodes is managed through reference counting. To avoid reference cycles which would cause memory leaks, the tree is asymmetric: each node holds optional strong references to its next sibling and first child, but only optional weak references to its parent, previous sibling, and last child.

Nodes are destroyed as soon as there is no strong reference left to them. The structure is such that holding a reference to the root is sufficient to keep the entire tree alive. However, if for example the only reference that exists from outside the tree is one that you use to traverse it, you will not be able to go back “up” the tree to ancestors and previous siblings after going “down”, as those nodes will have been destroyed.

Weak references to destroyed nodes are treated as if they were not set at all. (E.g. a node can become a root when its parent is destroyed.)

Since nodes are aliased (have multiple references to them), RefCell is used for interior mutability.

Advantages:

  • A single NodeRef user-visible type to manipulate the tree, with methods
  • Memory is freed as soon as it becomes unused (if parts of the tree are removed).

Disadvantages:

  • The tree can only be accessed from the thread is was created in.
  • Any tree manipulation, including read-only traversals, requires incrementing and decrementing reference counts, which causes run-time overhead
  • Nodes are allocated individually, which may cause memory fragmentation and hurt performance.

arena-tree

The lifetime of nodes is managed through an arena allocator.

Nodes are tied, through &'a T references, to the lifetime of the arena and are destroyed all at once when the arena is destroyed. The links between nodes are also &'a T references internally.

Since nodes are aliased (have multiple references to them), Cell is used for interior mutability.

Advantages:

  • Nodes still have methods, the arena can largely be ignored during tree manipulation (as long as it’s kept alive).
  • Less runtime overhead. (Allocation is fast, no reference count to increment or decrement.)
  • Nodes are allocated in large, continuous chunks of memory. This helps memory cache performance.

Disadvantages:

  • There is one more object (the arena) to keep around.
  • The tree can only be accessed from the thread is was created in.
  • Memory “leaks” when nodes are removed from the tree before the arena is destroyed. This can be alleviated by maintaining a free list of nodes that are not used anymore and can be reused, but only if these nodes have been freed explicitly. (Whereas reference counting and garbage collection would free unused nodes automatically.) But then nothing prevents references to freed nodes to be kept around. Another explicit mechanism (like tracking a generation number in each node) can be added to prevent (mostly) such references and accessing a freed or re-allocated node.

idtree

Similar to arena-tree, but the arena is simplified to a single Vec and numerical identifiers (indices in the vector) are used instead of &'a T references.

Advantages:

  • There is no RefCell, mutability is handled in a way much more idiomatic to Rust through unique (&mut) access to the arena.
  • The tree can be sent or shared across threads like a Vec. This enables e.g. parallel tree traversals.

Disadvantages:

  • Every access requires going through the arena, which can be cumbersome.
  • There is some runtime overhead over arena-tree because of bound checks.
  • A node ID from a given arena can be used in a different arena, which is likely to not cause an error and access an unrelated node.

lib.rs:

A DOM-like tree data structure based on &Node references.

Any non-trivial tree involves reference cycles (e.g. if a node has a first child, the parent of the child is that node). To enable this, nodes need to live in an arena allocator such as arena::TypedArena distrubuted with rustc (which is #[unstable] as of this writing) or typed_arena::Arena.

If you need mutability in the node’s data, make it a cell (Cell or RefCell) or use cells inside of it.

Example

extern crate typed_arena;
extern crate arena_tree;
use std::cell::RefCell;

let arena = typed_arena::Arena::new();
let a = arena.alloc(arena_tree::Node::new(RefCell::new(String::new())));
let b = arena.alloc(arena_tree::Node::new(RefCell::new(String::new())));
a.append(b);
b.data.borrow_mut().push_str("content");
assert_eq!(a.descendants().map(|node| node.data.borrow().clone()).collect::<Vec<_>>(), [
"".to_string(), "content".to_string()
]);

No runtime deps