Haskell中的无限列表并行过滤器

Question

我想在 Haskell 的无限列表中找到第一个匹配的元素。

此代码正在运行：

findPassword passwordHash = (head . filter (checkPassword passwordHash)) allStrings

checkPassword 真的很长（因为它是 SHA1 哈希）

checkPassword hash string = (sha1 string) == hash

allStrings 只是所有可能字符串的列表：

allStrings = [ c : s | s <- "" : allStrings, c <- ['a'..'z'] ++ ['0'..'9'] ]

我希望这段代码并行运行，但如果我用 parFilter 替换过滤器：

import qualified Control.Parallel.Strategies as S
parFilter p = S.withStrategy (S.evalBuffer 1000 S.rseq) . filter p

它不起作用……你有什么想法吗？这段代码也使用了大量内存，但这是另一个问题。 完整的脚本在这里https://github.com/ThibaudDauce/habreaker

Answer 1

我很确定您想使用 parBuffer 而不是 evalBuffer。

请参阅此 SO 答案以获得很好的解释：

How to choose between parList and parBuffer?

这是一些演示代码：

import qualified Data.Map.Strict as M
import Control.Parallel.Strategies
import System.Environment
import Debug.Trace

fib 0 = 0
fib 1 = 1
fib n = fib (n-2) + fib (n-1)

fib' n | trace "calling fib" False = undefined
fib' n = fib n

theList = cycle [30,31,32]

firstN :: Int -> [Int]
firstN n = take n $ filter even $ map fib' theList

firstNpar :: Int -> Int -> [Int]
firstNpar n k = take n $ filter even $ runEval $ parBuffer k rseq $ map fib' theList

main = do
  (argn : argk : _) <- getArgs
  let n = read argn
  case argk of
    "" -> print $ firstN n
    _  -> let k = read argk in
          print $ firstNpar n k

示例运行：

prog 20 2 +RTS -N2         -- I only have two cores
prog 20 ''                 -- run single threaded