如何在不扫描整个表的情况下检索ets键列表?

时间:2016-02-01 03:29:54

标签: elixir ets

我使用ets通过elixir作为一个简单的内存持久层来存储和检索密钥,也用于偶尔的foldl,它涉及减少许多具有不同值的重复键。我正在使用行李选项。

是否有一种简单的O(1)方法来检索当前密钥的列表而无需进行更多涉及的表遍历或匹配或折叠?

欢迎Erlang或Elixir语法响应。

:ets.new(:cache, [:bag, :named_table, :protected])

我有一个原子键的静态映射,用我用来帮助插入的整数索引。但并非所有的钥匙都被使用..

chunk_key_map = %{2 => :chunk_key_2, ..... 28 => :chunk_key_28}

如果没有快捷方式,我知道我可以做一个:查找我的每个静态原子键值并测试!= []并生成我自己的列表,但是想看看ets是否支持这样的功能。

由于

2 个答案:

答案 0 :(得分:2)

要获取ets中的键列表而不触及其值,您可以这样使用ets:first/1ets:next/2函数的组合:

-export([keys/1]).

keys(TableName) ->
    FirstKey = ets:first(TableName),
        keys(TableName, FirstKey, [FirstKey]).

keys(_TableName, '$end_of_table', ['$end_of_table'|Acc]) ->
    Acc;
keys(TableName, CurrentKey, Acc) ->
    NextKey = ets:next(TableName, CurrentKey),
    keys(TableName, NextKey, [NextKey|Acc]).

导出的API为keys/1。它获取表名,获取它的第一个键,启动累加器作为状态,并在内部调用keys/2,它获取其他键并以递归方式累积它们。

请注意,bag表格类型没有订单,因此如果您的表格类型为bag,则keys/1的返回值不会被订购。

答案 1 :(得分:1)

所以我没有找到ets技术,而是在elixir中以恒定时间实现了键列表检索代码,因为我的键映射是静态的。

    list = Enum.reduce(2..28, [], fn head, acc -> 
            case :ets.lookup(:cache, Map.get(chunk_key_map, head)) do
                [] -> acc
                _ -> [acc, head]
            end
        end)

    List.flatten(list)

更新:根据回复,我采取了Hamidreza的ets 遍历逻辑并使用{​​{1}}将其包装到Elixir Stream中。

Stream.resource/3

然后我通过管道

运行流
defp get_ets_keys_lazy(table_name) when is_atom(table_name) do
    eot = :"$end_of_table"

    Stream.resource(
        fn -> [] end,

        fn acc ->
            case acc do
                [] -> 
                    case :ets.first(table_name) do
                        ^eot -> {:halt, acc}
                        first_key -> {[first_key], first_key}                       
                    end

                acc -> 
                    case :ets.next(table_name, acc) do  
                        ^eot -> {:halt, acc}
                        next_key -> {[next_key], next_key}
                    end
            end
        end,

        fn _acc -> :ok end
    )
end