parallex

defmodule Parallex do
  @moduledoc """
  Concurrent versions of some common `Enum` functions.
  These might be useful if the applied function is heavy.
  """

  @doc """
  Perform an `Enum.map/2` function using concurrent processes.

  ## Examples

      iex> Parallex.pmap([1, 2, 3, 4], &(&1*&1))
      [1, 4, 9, 16]

  """
  def pmap(collection, func) do
    collection
    |> Enum.map(&(Task.async(fn -> func.(&1) end)))
    |> Enum.map(&Task.await/1)
  end

  @doc """
  Perform an `Enum.filter/2` function using concurrent processes.

  ## Examples

      iex> Parallex.pfilter([1, 2, 3, 4], &(&1 < 4))
      [1, 2, 3]

  """
  def pfilter(collection, func) do
    iter = Enum.count(collection) - 1
    results = pmap(collection, func)

    Enum.reduce(0..iter, [],
      fn i, acc ->
        cond do
          Enum.at(results, i) -> [Enum.at(collection, i) | acc]
          true -> acc
        end
      end)
    |> Enum.reverse
  end
end