diff options
Diffstat (limited to 'src/main/scala/reading/models/Filter.scala')
| -rw-r--r-- | src/main/scala/reading/models/Filter.scala | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/main/scala/reading/models/Filter.scala b/src/main/scala/reading/models/Filter.scala index 7ec6340..e4c8b93 100644 --- a/src/main/scala/reading/models/Filter.scala +++ b/src/main/scala/reading/models/Filter.scala @@ -32,10 +32,10 @@ object Filter { case FilterKind.Grade => Grade.withNameOption(nonFormattedName).map(apply[Grade]) } - def contains(filters: Seq[Filter], filter: Filter): Boolean = + def contains(filters: Set[Filter], filter: Filter): Boolean = filters.find(_ == filter).nonEmpty - def remove(fs: Seq[Filter], rf: Filter): Seq[Filter] = + def remove(fs: Set[Filter], rf: Filter): Set[Filter] = fs.filterNot { f => (f == rf || rf.kind == FilterKind.Grade && f.kind == FilterKind.Program @@ -56,7 +56,7 @@ object Filter { .map(f => (f, Books().filter(f.filter))) .toMap - def add(books: Seq[Book], filters: Seq[Filter]): Seq[Book] = + def add(books: Seq[Book], filters: Set[Filter]): Seq[Book] = filters.foldLeft(books)(add) def add(books: Seq[Book], filter: Filter): Seq[Book] = |