фильтрация внутри `for` с соответствующими

Question

Читаю файл TSV шаблон и использовать используя что-то вроде этого:

case class Entry(entryType: Int, value: Int) 

def filterEntries(): Iterator[Entry] = { 
    for { 
    line <- scala.io.Source.fromFile("filename").getLines() 
    } yield new Entry(line.split("\t").map(x => x.toInt)) 
} 

Теперь я оба заинтересованы в фильтрации записей, чьи entryType установлены на 0 и пренебрегая линии с количество столбцов больше или меньше 2 (это не соответствует конструктору). Мне было интересно, есть ли идиоматический способ достичь этого, может быть использование сопоставления шаблонов и метода unapply в сопутствующем объекте. Единственное, о чем я могу думать, это использовать .filter на результирующем итераторе.

Я также принимаю решение, не связанное с циклом for, но которое возвращает Iterator[Entry]. Эти решения должны быть толерантными к искаженным входным данным.

Answer 1

Это больше состоянии артистических:

package object liner { 
    implicit class R(val sc: StringContext) { 
    object r { 
     def unapplySeq(s: String): Option[Seq[String]] = sc.parts.mkString.r unapplySeq s 
    } 
    } 
} 

package liner { 

    case class Entry(entryType: Int, value: Int) 

    object I { 
    def unapply(s: String): Option[Int] = util.Try(s.toInt).toOption 
    } 

    object Test extends App { 
    def lines = List("1 2", "3", "", " 4 5 ", "junk", "0, 100000", "6 7 8") 

    def entries = lines flatMap { 
     case r"""\s*${I(i)}(\d+)\s+${I(j)}(\d+)\s*""" if i != 0 => Some(Entry(i, j)) 
     case __________________________________________________ => None 
    } 
    Console println entries 
    } 
} 

Будем надеяться, что регулярное выражение Интерполятор сделает его в стандартный дистрибутив в ближайшее время, но это показывает, как легко подстроить. Также, надеюсь, интерполятор типа scanf позволит легко извлекать case f"$i%d".

Я только начал использовать «удлиненный шаблон» в шаблонах, чтобы выровнять стрелки.

Существует куколочный или, может быть, личиночный регулярное выражение макрос:

https://github.com/som-snytt/regextractor

Answer 2

Вы можете создавать переменные в голове для-понимания, а затем использовать караул:

редактирования: обеспечить длину массива

for { 
    line <- scala.io.Source.fromFile("filename").getLines() 
    arr = line.split("\t").map(x => x.toInt) 
    if arr.size == 2 && arr(0) != 0 
} yield new Entry(arr(0), arr(1)) 

Answer 3

Левая из <- или = в for-loop может быть полноценным шаблоном. Таким образом, вы можете написать следующее:

def filterEntries(): Iterator[Int] = for { 
    line <- scala.io.Source.fromFile("filename").getLines() 
    arr = line.split("\t").map(x => x.toInt) 
    if arr.size == 2 
    // now you may use pattern matching to extract the array 
    Array(entryType, value) = arr 
    if entryType == 0 
} yield Entry(entryType, value) 

Обратите внимание, что это решение будет бросать NumberFormatException, если поле не раскладывается в Int. Если вы этого не хотите, вам нужно будет инкапсулировать x.toInt с Try и совпадением шаблонов.

Answer 4

Я решил его, используя следующий код:

import scala.util.{Try, Success} 

val lines = List(
    "1\t2", 
    "1\t", 
    "2", 
    "hello", 
    "1\t3" 
) 

case class Entry(val entryType: Int, val value: Int) 
object Entry { 
    def unapply(line: String) = { 
    line.split("\t").map(x => Try(x.toInt)) match { 
     case Array(Success(entryType: Int), Success(value: Int)) => Some(Entry(entryType, value)) 
     case _ => 
     println("Malformed line: " + line) 
     None 
    } 
    } 
} 

for { 
    line <- lines 
    entryOption = Entry.unapply(line) 
    if entryOption.isDefined 
} yield entryOption.get