Ctrie

A concurrent hash-trie or Ctrie^[1] is a concurrent thread-safe lock-free implementation of a hash array mapped trie. It is used to implement the concurrent map abstraction. It has particularly scalable concurrent insert and remove operations and is memory-efficient^[2]. It is the first known concurrent data-structure that supports O(1), atomic, lock-free snapshots^[3].

Operation

The Ctrie data structure is a non-blocking concurrent hash array mapped trie based on single-word compare-and-swap instructions in a shared-memory system. It supports concurrent lookup, insert and remove operations. Just like the hash array mapped trie, it uses the entire 32-bit space for hash values thus having low risk of hashcode collisions. Each node may branch to up to 32 sub tries. To conserve memory, each node contains a 32 bits bitmap where each bit indicates the presence of a branch followed by an array of length equal to the Hamming weight of the bitmap.

Keys are inserted by doing an atomic compare-and-swap operation on the node which needs to be modified. To ensure that updates are done independently and in a proper order, a special indirection node (an I-node) is inserted between each regular node and its subtries.

The figure above illustrates the Ctrie insert operation. Trie A is empty - an atomic CAS instruction is used to swap the old node C1 with the new version of C1 which has the new key k1. If the CAS is not successful, the operation is restarted. If the CAS is successful, we obtain the trie B. This procedure is repeated when a new key k2 is added (trie C). If two hashcodes of the keys in the Ctrie collide as is the case with k2 and k3, the Ctrie must be extended with at least one more level - trie D has a new indirection node I2 with a new node C2 which holds both colliding keys. Further CAS instructions are done on the contents of the indirection nodes I1 and I2 - such CAS instructions can be done independently of each other, thus enabling concurrent updates with less contention.

The Ctrie is defined by the pointer to the root indirection node (or a root I-node). The following types of nodes are defined for the Ctrie:

 structure INode {
   main: CNode
 }
 
 structure CNode {
   bmp: integer
   array: Branch[2^W]
 }
 
 Branch: INode | SNode
 
 structure SNode {
   k: KeyType
   v: ValueType
 }

A C-node is a branching node. It typically contains up to 32 branches, so W above is 5. Each branch may either be a key-value pair (represented with an S-node) or another I-node. To avoid wasting 32 entries in the branching array when some branches may be empty, an integer bitmap is used to denote which bits are full and which are empty. The helper method flagpos is used to inspect the relevant hashcode bits for a given level and extract the value of the bit in the bitmap to see if its set or not - denoting whether there is a branch at that position or not. If there is a bit, it also computes its position in the branch array. The formula used to do this is:

 bit = bmp & (1 << ((hashcode >> level) & 0x1F))
 pos = bitcount((bit - 1) & bmp)

Note that the operations treat only the I-nodes as mutable nodes - all other nodes are never changed after being created and added to the Ctrie.

Below is an illustration of the pseudocode of the insert operation:

 def insert(k, v)
   r = READ(root)
   if iinsert(r, k, v, 0, null) = RESTART insert(k, v)

 def iinsert(i, k, v, lev, parent)
   cn = READ(i.main)
   flag, pos = flagpos(k.hc, lev, cn.bmp)
   if cn.bmp & flag = 0 {
     ncn = cn.inserted(pos, flag, SNode(k, v))
     if CAS(i.main, cn, ncn) return OK
     else return RESTART
   }
   cn.array(pos) match {
     case sin: INode => {
       return iinsert(sin, k, v, lev + W, i)
     case sn: SNode =>
       if sn.k ≠ k {
         nsn = SNode(k, v)
         nin = INode(CNode(sn, nsn, lev + W))
         ncn = cn.updated(pos, nin)
         if CAS(i.main, cn, ncn) return OK
         else return RESTART
       } else {
         ncn = cn.updated(pos, SNode(k, v))
         if CAS(i.main, cn, ncn) return OK
         else return RESTART
       }
   }

The inserted and updated methods on nodes return new versions of the C-node with a value inserted or updated at the specified position, respectively. Note that the insert operation above is tail-recursive, so it can be rewritten as a while loop. Other operations are described in more detail in the original paper on Ctries^[1].

The data-structure has been proven to be correct^[1] - Ctrie operations have been shown to have the atomicity, linearizability and lock-freedom properties. The lookup operation can be modified to guarantee wait-freedom.

Advantages of Ctries

Ctries have been shown to be comparable in performance with concurrent skip lists^[4], concurrent hash tables and similar data structures in terms of the lookup operation, being slightly slower than hash tables and faster than skip lists due to the lower level of indirections. However, they are far more scalable than most concurrent hash tables where the insertions are concerned^[1]. Most concurrent hash tables are bad at conserving memory - when the keys are removed from the hash table, the underlying array is not shrinked. Ctries have the property that the allocated memory is always a function of only the current number of keys in the data-structure ^[1].

Ctries have logarithmic complexity bounds of the basic operations, albeit with a low constant factor due to the high branching level (usually 32).

Ctries support a lock-free, linearizable, constant-time snapshot operation. This is a breakthrough in concurrent data-structure design, since existing concurrent data-structures do not support snapshots. The snapshot operation allows implementing lock-free, linearizable iterator, size and clear operations - existing concurrent data-structures^[5] have implementations which either use global locks or are correct only given that there are no concurrent modifications to the data-structure. In particular, Ctries have an O(1) iterator creation operation, O(1) clear operation, O(1) duplicate operation and an amortized O(1) size retrieval operation.

Problems with Ctries

Most concurrent data structures require dynamic memory allocation. The current implementation^[3] of the Ctrie is written for the JVM, where garbage collection is provided by the platform itself. While its possible to keep a concurrent memory pool for the nodes shared by all instances of Ctries in an application or use reference counting to properly deallocate nodes, no implementation so far dealt with manual memory management of nodes used in Ctries. Such a technique may be viable for managed environments as well, since it could lower the pressure on the GC.

Implementations

A Ctrie implementation^[3] is available on GitHub. It is a mutable thread-safe implementation which ensures progress and supports lock-free, linearizable, O(1) snapshots.

A data-structure similar to Ctries has been used in ScalaSTM^[6], a software transactional memory library for the JVM.

References

1. ^ ^{a b c d e} Prokopec, A. et al. (2011) Cache-Aware Lock-Free Concurrent Hash Tries. Technical Report, 2011.
2. ^ Prokopec, A. et al. (2011) Lock-Free Resizeable Concurrent Tries. The 24th International Workshop on Languages and Compilers for Parallel Computing, 2011.
3. ^ ^{a b c} Prokopec, A. JVM implementation on GitHub
4. ^ Java 6.0 ConcurrentSkipListMap
5. ^ Java 5.0 ConcurrentHashMap
6. ^ N. Bronson ScalaSTM