스칼라에서 맵과 필터가 null 인 경우

Question

지도와 널 건너 뛰기가 한 번의 호출로 표현되는 방식으로 다음 코드를 표현할 수 있습니까? 당신이

for { 
    item <- list 
    itemConverted = item.accept(this, arg).asInstanceOf[T] 
    itemNonNull = itemConverted if itemConverted != 0 
} yield itemNonNull 

이 가능할 것이다 collect을 사용하여 (이것은 원래의 표현에 더 많거나 적은 변환됩니다 불구하고), 좋아하지만 경우

list.map(_.accept(this, arg).asInstanceOf[T]).filter(_ != null) 

Answer 1

list flatMap { i => Option(i.accept(this, arg).asInstanceOf[T]) } 

또는 대안, 그것은 가능성이 두 배 accept를 부를 것이다 대부분의 주장에서 부분 함수에 대한 시험 때문에 :

list collect { 
    case i if i.accept(this, arg).asInstanceOf[T] != null => i.accept(this, arg).asInstanceOf[T] 
} 

이것을 방지하려면 메모 작성기 (또는 스마트 추출기)를 사용해야합니다.

Answer 2

성능에 대해 우려하고 있다면, 당신은 따라서 map 및 filter 목록이 아닌 두 개의 패스를 통해 한 번에 수행됩니다, .view

list.view.map(_.accept(this, arg).asInstanceOf[T]).filter(_ != null) 

view이 통과가 지연 될됩니다 추가 할 수 있습니다. 이 패턴을 재사용이 걱정되는 경우

, 당신은 당신의 자신의 도우미 함수 정의 할 수 있습니다 같아요

def mapNN[A,B](list: List[A])(f: A => B) = { 
    list.view.map(f(_)).filter(_ != null) 
} 

mapNN(list)(_.accept(this, arg).asInstanceOf[T]) 

테스트 ...

> mapNN(List(1,2,3))(x => if (x%2==0) x else null).toList 
res7: List[Any] = List(2)