13

我希望能够在运行时动态更改计算管道,但似乎 GenStage 需要在编译时通过该subscribe_to: [...]机制定义计算图。有没有办法创建动态计算图?例如在下面,我想在运行时在我的管道图中的“减 7”和“减 4”顶点之间切换。

在此处输入图像描述

这可能使用 GenStage 吗?我可能会拥有非常复杂的管道,因此我需要一个能够以复杂方式扩展以更改图形的解决方案,而不是临时解决方案,例如在这种情况下,将要减去的整数参数化。我希望能够添加或删除整个子树,在子树之间切换,并将节点添加到图中,包括将它们拼接到包括主树在内的任何子树的中间。

请参阅下面的编辑

这是最初的生产者:

defmodule GenstageTest.Producer do
  use GenStage

  def start_link(initial \\ 1) do
    GenStage.start_link(__MODULE__, initial, name: __MODULE__)
  end

  def init(counter), do: {:producer, counter}

  def handle_demand(demand, state) do
    events = Enum.to_list(state..(state + demand - 1))
    {:noreply, events, state + demand}
  end
end

这是生产者消费者之一:

defmodule GenstageTest.PcTimesFive do
  use GenStage

  def start_link do
    GenStage.start_link(__MODULE__, :state_doesnt_matter, name: __MODULE__)
  end

  def init(state) do
    {:producer_consumer, state, subscribe_to: [GenstageTest.PcAddOne]}
  end

  def handle_events(events, _from, state) do
    numbers =
      events
      |> Enum.map(&(&1 * 5))
    {:noreply, numbers, state}
  end
end

这是最终消费者:

defmodule GenstageTest.Consumer do
  use GenStage

  def start_link do
    GenStage.start_link(__MODULE__, :state_doesnt_matter)
  end

  def init(state) do
    {:consumer, state, subscribe_to: [GenstageTest.PcDivTwo]}
  end

  def handle_events(events, _from, state) do
    for event <- events do
      IO.inspect({self(), event, state})
    end

    # As a consumer we never emit events
    {:noreply, [], state}
  end
end

I 这一切都以Elixir School Genstage 教程为蓝本。

所有模块和 mix.exs 都可以在 github上找到。

在@AquarHEAD L 部分回答后 3 天后编辑。

我已经设法让运行时订阅正常工作。下面分别修改一些producer、producer_consumers、consumers:

制片人:

defmodule GenstageTest.Producer do
  use GenStage

  def start_link(initial \\ 1) do
    GenStage.start_link(__MODULE__, initial, name: __MODULE__)
  end

  def init(counter), do: {:producer, counter}

  def handle_demand(demand, state) do
    events = Enum.to_list(state..(state + demand - 1))
    {:noreply, events, state + demand}
  end

  def handle_info({:doprint}, state) do
    IO.puts "yep"
    {:noreply, [], state}
  end

  def handle_info({:cancel, sublink}, state) do
    GenStage.cancel sublink, []
    {:noreply, [], state}
  end

end

生产者_消费者:

defmodule GenstageTest.PcAddOne do
  use GenStage

  def start_link do
    GenStage.start_link(__MODULE__, :state_doesnt_matter, name: __MODULE__)
  end

  def init(state) do
    {:producer_consumer, state}
  end

  def handle_events(events, _from, state) do
    numbers =
      events
      |> Enum.map(&(&1 + 1))
    {:noreply, numbers, state}
  end
end

消费者:

defmodule GenstageTest.Consumer do
  use GenStage

  def start_link do
    GenStage.start_link(__MODULE__, :state_doesnt_matter)
  end

  def init(state) do
    {:consumer, state}
  end

  def handle_events(events, _from, state) do
    for event <- events do
      IO.inspect event
      #File.write("/home/tbrowne/scratch/output.txt", 
      #  Kernel.inspect(event) <> " ", [:append])
      :timer.sleep(100)
    end

    # As a consumer we never emit events
    {:noreply, [], state}
  end
end

现在一旦这些都在 lib 目录中可用(记得添加 {:gen_stage, "~> 0.11"}到你的 mix.exs deps),或者复制并粘贴到 IEX 中,那么以下将完美地工作:

{:ok, p} = GenstageTest.Producer.start_link(0)
{:ok, a1} = GenstageTest.PcAddOne.start_link()
{:ok, c} = GenstageTest.Consumer.start_link()
{:ok, link1} = GenStage.sync_subscribe(a1, to: p, min_demand: 0, max_demand: 1, cancel: :transient)
{:ok, link2} = GenStage.sync_subscribe(c, to: a1, min_demand: 0, max_demand: 1, cancel: :transient)

现在的问题是,我仍然不知道如何取消订阅。有取消功能,也有停止功能GenStage.stop(c)例如似乎什么也没做,而我的各种尝试GenStage.cancel/3只会给出错误。

回顾一下,我现在需要的是能够停止某些阶段并用其他阶段替换它们。取消订阅的语法是什么,从哪里调用它?由于没有具体示例,因此在文档中没有很好地解释。

4

2 回答 2

5

您完全可以在运行时更改管道,查看 GenStage 文档中的第一个示例您也可以使用该:manual模式来精细控制需求。还有取消订阅的 API。我认为这些足以动态管理 GenStage 管道。

于 2018-10-26T15:30:13.853 回答
0

为什么不实现自己的GenStage.Dispatcher?这是行为

于 2018-11-01T23:16:07.033 回答