繁体   English   中英

当谓词计算为true时,F#停止Seq.map

[英]F# stop Seq.map when a predicate evaluates true

我正在以类似的方式生成序列:

migrators
|> Seq.map (fun m -> m())

migrator功能最终会返回一个受歧视的联合,如:

type MigratorResult =
| Success of string * TimeSpan
| Error of string * Exception

我想在遇到第一个Error停止map ,但我需要在最终序列中包含Error

我有以下内容向用户显示最终消息

match results |> List.rev with
| [] -> "No results equals no migrators"
| head :: _ ->
   match head with
   | Success (dt, t) -> "All migrators succeeded"
   | Error (dt, ex) -> "Migration halted owing to error"

所以我需要:

  1. 一种映射步骤生成Error时停止映射的方法
  2. 将错误作为添加到序列的最后一个元素的方法

我很欣赏除了map之外可能会有一个不同的序列方法,我是F#的新手并且在线搜索还没有产生任何东西!

我想这里有多种方法,但一种方法是使用展开:

migrators 
|> Seq.unfold (fun ms ->
    match ms with
    | m :: tl -> 
        match m () with
        | Success res -> Some (Success res, tl)
        | Error res   -> Some (Error res, [])
    | [] -> None)
|> List.ofSeq

注意最后的List.ofSeq ,就在那里实现序列。 另一种方法是使用序列理解,有些人可能会说它会产生更清晰的代码。

Tomaš暗示的丑陋事物是1)可变状态,2)操纵底层枚举器。 返回到包括谓词保持时的高阶函数将如下所示:

module Seq =
    let takeUntil pred (xs : _ seq) = seq{
        use en = xs.GetEnumerator()
        let flag = ref true
        while !flag && en.MoveNext() do
            flag := not <| pred en.Current
            yield en.Current }

seq{1..10} |> Seq.takeUntil (fun x -> x % 5 = 0)
|> Seq.toList
// val it : int list = [1; 2; 3; 4; 5]

对于您的特定应用程序,您可以将DU的大小写映射到布尔值。

(migrators : seq<MigratorResult>)
|> Seq.takeUntil (function Success _ -> false | Error _ -> true)

我认为@scrwtp的答案可能是最好的方法,如果您的输入相当小(并且您可以将其转换为F#列表以使用模式匹配)。 我将再添加一个版本,当您的输入只是一个序列并且您不想将其转换为列表时,该版本有效。

从本质上讲,你想要做一些几乎像Seq.takeWhile ,但它最后会给你一个额外的项目(谓词失败的那个)。

要使用更简单的示例,以下内容将返回序列中的所有数字,直到可被5整除的数字:

let nums = [ 2 .. 10 ]

nums
|> Seq.map (fun m -> m % 5)
|> Seq.takeWhile (fun n -> n <> 0)

所以,你基本上只需要提前看一个元素 - 为此,你可以使用Seq.pairwise ,它为你提供序列中的当前和下一个元素“

nums
|> Seq.map (fun m -> m % 5)
|> Seq.pairwise                          // Get sequence of pairs with the next value
|> Seq.takeWhile (fun (p, n) -> p <> 0)  // Look at the next value for test
|> Seq.mapi (fun i (p, n) ->             // For the first item, we return both
    if i = 0 then [p;n] else [n])        //   for all other, we return the second
|> Seq.concat

这里唯一的丑陋是你需要使用mapiconcat再次压平序列。

这不是很好,所以做一件好事就是定义你自己的高阶函数,如Seq.takeUntilAfter ,它封装了你需要的行为(隐藏了所有丑陋的东西)。 然后你的代码可以只使用这个函数,看起来很好看和可读(你可以尝试其他实现方法)。

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM