Add Hierarch.Ecto.LTree to support comman primary key(bigint)

Author: fahchen

lib/ecto/ltree.ex

@@ -0,0 +1,14 @@
+defmodule Hierarch.Ecto.LTree do
+  @behaviour Ecto.Type
+
+  def type, do: :ltree
+
+  @doc false
+  def cast(label), do: {:ok, label}
+
+  @doc false
+  def dump(label), do: {:ok, to_string(label)}
+
+  @doc false
+  def load(label), do: {:ok, label}
+end

lib/hierarch.ex

@@ -209,7 +209,7 @@ defmodule Hierarch do
 
       defp get_primary_key(schema) do
         [{id_column, value}] = Ecto.primary_key(schema)
-        {id_column, value}
+        {id_column, to_string(value)}
       end
     end
   end

priv/repo/migrations/20180628144013_create_organizations.exs

@@ -0,0 +1,14 @@
+defmodule Dummy.Repo.Migrations.CreateOrganizations do
+  use Ecto.Migration
+
+  def change do
+    execute "CREATE EXTENSION IF NOT EXISTS ltree" # Enables Ltree action
+
+    create table(:organizations) do
+      add :name, :string
+      add :ancestry, :ltree, null: false, default: ""
+    end
+
+    create index(:organizations, [:ancestry], using: "GIST")
+  end
+end

test/hierarch/bigint_primary_key_test.exs

@@ -0,0 +1,45 @@
+defmodule BigintPrimaryKeyTest do
+  use Hierarch.TestCase
+
+  setup_all do
+    organizations = create_organizations()
+
+    on_exit fn ->
+      Repo.delete_all(Organization)
+    end
+
+    {:ok, organizations}
+  end
+
+  test "returns the parent", organizations do
+    b = Map.get(organizations, "A.B")
+    c = Map.get(organizations, "A.B.C")
+
+    discendants = c
+                  |> Organization.parent
+                  |> Repo.one
+    assert discendants == b
+  end
+
+  test "returns sbilings", organizations do
+    b = Map.get(organizations, "A.B")
+    d = Map.get(organizations, "A.D")
+
+    sblings = b
+              |> Organization.sblings
+              |> Repo.all
+    assert sblings == [d]
+  end
+
+  test "return discendants", organizations do
+    a = Map.get(organizations, "A")
+    b = Map.get(organizations, "A.B")
+    c = Map.get(organizations, "A.B.C")
+    d = Map.get(organizations, "A.D")
+
+    discendants = a
+                  |> Organization.discendants
+                  |> Repo.all
+    assert discendants == [b, c, d]
+  end
+end

test/support/organization.ex

@@ -0,0 +1,10 @@
+defmodule Dummy.Organization do
+  @moduledoc false
+  use Ecto.Schema
+  use Hierarch, path_column: :ancestry
+
+  schema "organizations" do
+    field :name, :string
+    field :ancestry, Hierarch.Ecto.LTree
+  end
+end

test/test_helper.exs

@@ -4,7 +4,7 @@ defmodule Hierarch.TestCase do
   using(opts) do
     quote do
       use ExUnit.Case, unquote(opts)
-      alias Dummy.{Repo, Catelog}
+      alias Dummy.{Repo, Catelog, Organization}
 
       def create_catelogs do
         catelogs_list = [
@@ -31,6 +31,23 @@ defmodule Hierarch.TestCase do
           Map.put acc, name, catelog
         end)
       end
+
+      def create_organizations do
+        organizations_list = [
+          "A",
+          "A.B",
+          "A.B.C",
+          "A.D",
+        ]
+
+        Enum.reduce(organizations_list, %{}, fn (name, acc) ->
+          parent_name = Hierarch.LTree.parent_path(name)
+          parent = Map.get(acc, parent_name)
+
+          organization = Organization.build_child_of(parent, %{name: name}) |> Repo.insert!
+          Map.put acc, name, organization
+        end)
+      end
     end
   end