-
Notifications
You must be signed in to change notification settings - Fork 0
/
ThesaurusInteractionExtensions.java
54 lines (47 loc) · 2.51 KB
/
ThesaurusInteractionExtensions.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
package app.mesmedicaments.interactions.ansm.thesaurus;
import java.util.Set;
import java.util.stream.Collectors;
import java.util.stream.Stream;
import lombok.experimental.UtilityClass;
@UtilityClass
public final class ThesaurusInteractionExtensions {
static public Set<ThesaurusInteractionSubstanceSubstance> flatten(
Set<ThesaurusInteraction<?, ?>> set) {
return flatten(set.stream());
}
static public Set<ThesaurusInteractionSubstanceSubstance> flatten(
Stream<ThesaurusInteraction<?, ?>> stream) {
return stream.<ThesaurusInteractionSubstanceSubstance>flatMap(i -> {
if (i instanceof ThesaurusInteractionClasseClasse)
return flatten((ThesaurusInteractionClasseClasse) i).stream();
if (i instanceof ThesaurusInteractionClasseSubstance)
return flatten((ThesaurusInteractionClasseSubstance) i).stream();
if (i instanceof ThesaurusInteractionSubstanceSubstance)
return Stream.of((ThesaurusInteractionSubstanceSubstance) i);
throw new RuntimeException("Unsupported class: " + i.getClass().getSimpleName());
}).collect(Collectors.toSet());
}
static public Set<ThesaurusInteractionSubstanceSubstance> flatten(
ThesaurusInteractionClasseSubstance interaction) {
final var substance = interaction.getRightElement();
final var level = interaction.getLevel();
final var description = interaction.getDescription();
final var conduiteATenir = interaction.getConduiteATenir();
return interaction.getLeftElement().getSubstances().stream()
.map(s -> new ThesaurusInteractionSubstanceSubstance(s, substance, level,
description, conduiteATenir))
.collect(Collectors.toSet());
}
static public Set<ThesaurusInteractionSubstanceSubstance> flatten(
ThesaurusInteractionClasseClasse interaction) {
final var leftClasse = interaction.getLeftElement();
final var level = interaction.getLevel();
final var description = interaction.getDescription();
final var conduiteATenir = interaction.getConduiteATenir();
return interaction.getRightElement().getSubstances().stream()
.map(s -> new ThesaurusInteractionClasseSubstance(leftClasse, s, level, description,
conduiteATenir))
.map(ThesaurusInteractionExtensions::flatten).flatMap(Set::stream)
.collect(Collectors.toSet());
}
}