// b_tree.proto
message Tree {
  Any value = 1;
  Tree left = 2;
  Tree right = 3;
}

public class BinaryTree<T extends Message> {
  private final BTree.Tree tree;
  private final Class<? extends T> clazz;

  public BinaryTree(final T value, final BinaryTree left, final BinaryTree right) {
    this(value.getClass(), BTree.Tree.newBuilder()
        .setValue(Any.pack(value))
        .setLeft(left.left)
        .setRight(right.right)
        .build());
  }

  private BinaryTree(final Class<T> clazz, final BTree.Tree tree) {
    this.clazz = clazz;
    this.tree = tree;
  }

  public T getValue() {
    return tree.getValue().unpack(clazz);
  }

  public BinaryTree<T> getLeft() {
    return new BinaryTree<>(clazz, tree.left);
  }

  public BinaryTree<T> getRight() {
    return new BinaryTree<>(clazz, tree.right);
  }

  public static <T> BinaryTree<T> unpack(final Class<T> clazz, final byte[] input) throws Exception {
    return new BinaryTree<>(clazz, BTree.Tree.parseFrom(input));
  }

  public byte[] pack() {
    return tree.toByteArray();
  }
}

FamDav fucked around with this message at 07:12 on Sep 5, 2018

FamDav fucked around with this message at 02:39 on Oct 14, 2018