Apply GCPair to TreeNode, TextNode

src/document/change/context.ts

@@ -22,7 +22,6 @@ import { CRDTRoot } from '@yorkie-js-sdk/src/document/crdt/root';
 import {
   CRDTContainer,
   CRDTElement,
-  CRDTGCElement,
 } from '@yorkie-js-sdk/src/document/crdt/element';
 import { Operation } from '@yorkie-js-sdk/src/document/operation/operation';
 import { ChangeID } from '@yorkie-js-sdk/src/document/change/change_id';
@@ -103,14 +102,6 @@ export class ChangeContext<P extends Indexable = Indexable> {
     this.root.registerRemovedElement(deleted);
   }
 
-  /**
-   * `registerElementHasRemovedNodes` register GC element has removed node for
-   * garbage collection.
-   */
-  public registerElementHasRemovedNodes(elem: CRDTGCElement): void {
-    this.root.registerElementHasRemovedNodes(elem);
-  }
-
   /**
    * `registerGCPair` registers the given pair to hash table.
    */

src/document/crdt/element.ts

@@ -157,11 +157,3 @@ export abstract class CRDTContainer extends CRDTElement {
    */
   abstract getByID(createdAt: TimeTicket): CRDTElement | undefined;
 }
-
-/**
- * `CRDTGCElement` represents element which has garbage collecting method.
- */
-export abstract class CRDTGCElement extends CRDTElement {
-  abstract getRemovedNodesLen(): number;
-  abstract purgeRemovedNodesBefore(ticket: TimeTicket): number;
-}

src/document/crdt/rga_tree_split.ts

@@ -25,6 +25,7 @@ import {
   TimeTicket,
   TimeTicketStruct,
 } from '@yorkie-js-sdk/src/document/time/ticket';
+import { GCChild, GCPair, GCParent } from '@yorkie-js-sdk/src/document/crdt/gc';
 
 export interface ValueChange<T> {
   actor: ActorID;
@@ -243,9 +244,10 @@ export type RGATreeSplitPosRange = [RGATreeSplitPos, RGATreeSplitPos];
 /**
  * `RGATreeSplitNode` is a node of RGATreeSplit.
  */
-export class RGATreeSplitNode<
-  T extends RGATreeSplitValue,
-> extends SplayNode<T> {
+export class RGATreeSplitNode<T extends RGATreeSplitValue>
+  extends SplayNode<T>
+  implements GCChild
+{
   private id: RGATreeSplitNodeID;
   private removedAt?: TimeTicket;
 
@@ -491,6 +493,13 @@ export class RGATreeSplitNode<
     this.value = value.substring(0, offset) as T;
     return value.substring(offset, value.length) as T;
   }
+
+  /**
+   * `toIDString` returns a string that can be used as an ID for this position.
+   */
+  public toIDString(): string {
+    return this.id.toIDString();
+  }
 }
 
 /**
@@ -500,17 +509,15 @@ export class RGATreeSplitNode<
  * the block is split.
  *
  */
-export class RGATreeSplit<T extends RGATreeSplitValue> {
+export class RGATreeSplit<T extends RGATreeSplitValue> implements GCParent {
   private head: RGATreeSplitNode<T>;
   private treeByIndex: SplayTree<T>;
   private treeByID: LLRBTree<RGATreeSplitNodeID, RGATreeSplitNode<T>>;
-  private removedNodeMap: Map<string, RGATreeSplitNode<T>>;
 
   constructor() {
     this.head = RGATreeSplitNode.create(InitialRGATreeSplitNodeID);
     this.treeByIndex = new SplayTree();
     this.treeByID = new LLRBTree(RGATreeSplitNode.createComparator());
-    this.removedNodeMap = new Map();
 
     this.treeByIndex.insert(this.head);
     this.treeByID.put(this.head.getID(), this.head);
@@ -533,22 +540,30 @@ export class RGATreeSplit<T extends RGATreeSplitValue> {
    * @param editedAt - edited time
    * @param value - value
    * @param maxCreatedAtMapByActor - maxCreatedAtMapByActor
-   * @returns `[RGATreeSplitPos, Map<string, TimeTicket>, Array<Change>]`
+   * @returns `[RGATreeSplitPos, Map<string, TimeTicket>, Array<GCPair>, Array<Change>]`
    */
   public edit(
     range: RGATreeSplitPosRange,
     editedAt: TimeTicket,
     value?: T,
     maxCreatedAtMapByActor?: Map<string, TimeTicket>,
-  ): [RGATreeSplitPos, Map<string, TimeTicket>, Array<ValueChange<T>>] {
+  ): [
+    RGATreeSplitPos,
+    Map<string, TimeTicket>,
+    Array<GCPair>,
+    Array<ValueChange<T>>,
+  ] {
     // 01. split nodes with from and to
     const [toLeft, toRight] = this.findNodeWithSplit(range[1], editedAt);
     const [fromLeft, fromRight] = this.findNodeWithSplit(range[0], editedAt);
 
     // 02. delete between from and to
     const nodesToDelete = this.findBetween(fromRight, toRight);
-    const [changes, maxCreatedAtMap, removedNodeMapByNodeKey] =
-      this.deleteNodes(nodesToDelete, editedAt, maxCreatedAtMapByActor);
+    const [changes, maxCreatedAtMap, removedNodes] = this.deleteNodes(
+      nodesToDelete,
+      editedAt,
+      maxCreatedAtMapByActor,
+    );
 
     const caretID = toRight ? toRight.getID() : toLeft.getID();
     let caretPos = RGATreeSplitPos.of(caretID, 0);
@@ -580,11 +595,12 @@ export class RGATreeSplit<T extends RGATreeSplitValue> {
     }
 
     // 04. add removed node
-    for (const [key, removedNode] of removedNodeMapByNodeKey) {
-      this.removedNodeMap.set(key, removedNode);
+    const pairs: Array<GCPair> = [];
+    for (const [, removedNode] of removedNodes) {
+      pairs.push({ parent: this, child: removedNode });
     }
 
-    return [caretPos, maxCreatedAtMap, changes];
+    return [caretPos, maxCreatedAtMap, pairs, changes];
   }
 
   /**
@@ -865,7 +881,7 @@ export class RGATreeSplit<T extends RGATreeSplitValue> {
     );
 
     const createdAtMapByActor = new Map();
-    const removedNodeMap = new Map();
+    const removedNodes = new Map();
     // First we need to collect indexes for change.
     const changes = this.makeChanges(nodesToKeep, editedAt);
     for (const node of nodesToDelete) {
@@ -877,13 +893,13 @@ export class RGATreeSplit<T extends RGATreeSplitValue> {
       ) {
         createdAtMapByActor.set(actorID, node.getID().getCreatedAt());
       }
-      removedNodeMap.set(node.getID().toIDString(), node);
+      removedNodes.set(node.getID().toIDString(), node);
       node.remove(editedAt);
     }
     // Finally remove index nodes of tombstones.
     this.deleteIndexNodes(nodesToKeep);
 
-    return [changes, createdAtMapByActor, removedNodeMap];
+    return [changes, createdAtMapByActor, removedNodes];
   }
 
   private filterNodes(
@@ -988,34 +1004,13 @@ export class RGATreeSplit<T extends RGATreeSplitValue> {
   }
 
   /**
-   * `getRemovedNodesLen` returns size of removed nodes.
-   */
-  public getRemovedNodesLen(): number {
-    return this.removedNodeMap.size;
-  }
-
-  /**
-   * `purgeRemovedNodesBefore` physically purges nodes that have been removed.
+   * `purge` physically purges the given node from RGATreeSplit.
    */
-  public purgeRemovedNodesBefore(ticket: TimeTicket): number {
-    let count = 0;
-    for (const [, node] of this.removedNodeMap) {
-      if (node.getRemovedAt() && ticket.compare(node.getRemovedAt()!) >= 0) {
-        this.treeByIndex.delete(node);
-        this.purge(node);
-        this.treeByID.remove(node.getID());
-        this.removedNodeMap.delete(node.getID().toIDString());
-        count++;
-      }
+  public purge(node: GCChild): void {
+    if (!(node instanceof RGATreeSplitNode<T>)) {
+      return;
     }
 
-    return count;
-  }
-
-  /**
-   * `purge` physically purges the given node from RGATreeSplit.
-   */
-  public purge(node: RGATreeSplitNode<T>): void {
     const prev = node.getPrev();
     const next = node.getNext();
     const insPrev = node.getInsPrev();

src/document/crdt/rht.ts

@@ -269,6 +269,7 @@ export class RHT {
   public purge(child: RHTNode) {
     const node = this.nodeMapByKey.get(child.getKey());
     if (node == undefined || node.toIDString() != child.toIDString()) {
+      // TODO(hackerwins): Should we return an error when the child is not found?
       return;
     }