|
17 | 17 |
|
18 | 18 | package bio.overture.maestro.app.infra.adapter.inbound.messaging;
|
19 | 19 |
|
| 20 | +import static bio.overture.maestro.app.infra.adapter.inbound.messaging.IndexMessagesHelper.handleIndexRepository; |
| 21 | +import static bio.overture.maestro.app.infra.adapter.inbound.messaging.IndexMessagesHelper.handleIndexResult; |
| 22 | + |
20 | 23 | import bio.overture.maestro.domain.api.Indexer;
|
21 | 24 | import bio.overture.maestro.domain.api.exception.FailureData;
|
22 | 25 | import bio.overture.maestro.domain.api.message.*;
|
|
32 | 35 | import reactor.core.publisher.Flux;
|
33 | 36 | import reactor.core.publisher.Mono;
|
34 | 37 |
|
35 |
| -import static bio.overture.maestro.app.infra.adapter.inbound.messaging.IndexMessagesHelper.handleIndexRepository; |
36 |
| -import static bio.overture.maestro.app.infra.adapter.inbound.messaging.IndexMessagesHelper.handleIndexResult; |
37 |
| - |
38 | 38 | @Slf4j
|
39 | 39 | @EnableBinding(Sink.class)
|
40 | 40 | public class IndexingMessagesStreamListener {
|
41 | 41 |
|
42 |
| - private final Indexer indexer; |
43 |
| - |
44 |
| - public IndexingMessagesStreamListener(@NonNull Indexer indexer) { |
45 |
| - this.indexer = indexer; |
46 |
| - } |
47 |
| - |
48 |
| - @StreamListener(Sink.INPUT) |
49 |
| - public void handleAnalysisMessage(@Payload IndexMessage indexMessage) { |
50 |
| - if (isAnalysisReq(indexMessage)) { |
51 |
| - val indexAnalysisMessage = new IndexAnalysisMessage(indexMessage.getAnalysisId(), |
52 |
| - indexMessage.getStudyId(), |
53 |
| - indexMessage.getRepositoryCode(), |
54 |
| - indexMessage.getRemoveAnalysis()); |
55 |
| - handleIndexResult(() -> this.indexOrRemoveAnalysis(indexAnalysisMessage)); |
56 |
| - } else if (isStudyMsg(indexMessage)) { |
57 |
| - val indexStudyMessage = new IndexStudyMessage(indexMessage.getStudyId(), indexMessage.getRepositoryCode()); |
58 |
| - handleIndexResult(() -> this.indexStudy(indexStudyMessage)); |
59 |
| - } else if (isRepoMsg(indexMessage)) { |
60 |
| - val indexRepositoryMessage = new IndexRepositoryMessage(indexMessage.getRepositoryCode()); |
61 |
| - handleIndexRepository(() -> this.indexRepository(indexRepositoryMessage)); |
62 |
| - } else { |
63 |
| - throw new IllegalArgumentException("invalid message format"); |
64 |
| - } |
65 |
| - } |
66 |
| - |
67 |
| - private boolean isAnalysisReq(IndexMessage indexMessage) { |
68 |
| - return !StringUtils.isEmpty(indexMessage.getAnalysisId()) |
69 |
| - && !StringUtils.isEmpty(indexMessage.getStudyId()) |
70 |
| - && !StringUtils.isEmpty(indexMessage.getRepositoryCode()); |
71 |
| - } |
72 |
| - |
73 |
| - private boolean isStudyMsg(IndexMessage indexMessage) { |
74 |
| - return StringUtils.isEmpty(indexMessage.getAnalysisId()) |
75 |
| - && !StringUtils.isEmpty(indexMessage.getStudyId()) |
76 |
| - && !StringUtils.isEmpty(indexMessage.getRepositoryCode()); |
77 |
| - } |
78 |
| - |
79 |
| - private boolean isRepoMsg(IndexMessage indexMessage) { |
80 |
| - return StringUtils.isEmpty(indexMessage.getAnalysisId()) |
81 |
| - && StringUtils.isEmpty(indexMessage.getStudyId()) |
82 |
| - && !StringUtils.isEmpty(indexMessage.getRepositoryCode()); |
| 42 | + private final Indexer indexer; |
| 43 | + |
| 44 | + public IndexingMessagesStreamListener(@NonNull Indexer indexer) { |
| 45 | + this.indexer = indexer; |
| 46 | + } |
| 47 | + |
| 48 | + @StreamListener(Sink.INPUT) |
| 49 | + public void handleAnalysisMessage(@Payload IndexMessage indexMessage) { |
| 50 | + if (isAnalysisReq(indexMessage)) { |
| 51 | + val indexAnalysisMessage = |
| 52 | + new IndexAnalysisMessage( |
| 53 | + indexMessage.getAnalysisId(), |
| 54 | + indexMessage.getStudyId(), |
| 55 | + indexMessage.getRepositoryCode(), |
| 56 | + indexMessage.getRemoveAnalysis()); |
| 57 | + handleIndexResult(() -> this.indexOrRemoveAnalysis(indexAnalysisMessage)); |
| 58 | + } else if (isStudyMsg(indexMessage)) { |
| 59 | + val indexStudyMessage = |
| 60 | + new IndexStudyMessage(indexMessage.getStudyId(), indexMessage.getRepositoryCode()); |
| 61 | + handleIndexResult(() -> this.indexStudy(indexStudyMessage)); |
| 62 | + } else if (isRepoMsg(indexMessage)) { |
| 63 | + val indexRepositoryMessage = new IndexRepositoryMessage(indexMessage.getRepositoryCode()); |
| 64 | + handleIndexRepository(() -> this.indexRepository(indexRepositoryMessage)); |
| 65 | + } else { |
| 66 | + throw new IllegalArgumentException("invalid message format"); |
83 | 67 | }
|
84 |
| - |
85 |
| - private Flux<Tuple2<IndexAnalysisMessage, IndexResult>> indexOrRemoveAnalysis(IndexAnalysisMessage msg) { |
86 |
| - if (msg.getRemoveAnalysis()) { |
87 |
| - return Flux.from(removeAnalysis(msg)); |
88 |
| - } else { |
89 |
| - return indexAnalysis(msg); |
90 |
| - } |
| 68 | + } |
| 69 | + |
| 70 | + private boolean isAnalysisReq(IndexMessage indexMessage) { |
| 71 | + return !StringUtils.isEmpty(indexMessage.getAnalysisId()) |
| 72 | + && !StringUtils.isEmpty(indexMessage.getStudyId()) |
| 73 | + && !StringUtils.isEmpty(indexMessage.getRepositoryCode()); |
| 74 | + } |
| 75 | + |
| 76 | + private boolean isStudyMsg(IndexMessage indexMessage) { |
| 77 | + return StringUtils.isEmpty(indexMessage.getAnalysisId()) |
| 78 | + && !StringUtils.isEmpty(indexMessage.getStudyId()) |
| 79 | + && !StringUtils.isEmpty(indexMessage.getRepositoryCode()); |
| 80 | + } |
| 81 | + |
| 82 | + private boolean isRepoMsg(IndexMessage indexMessage) { |
| 83 | + return StringUtils.isEmpty(indexMessage.getAnalysisId()) |
| 84 | + && StringUtils.isEmpty(indexMessage.getStudyId()) |
| 85 | + && !StringUtils.isEmpty(indexMessage.getRepositoryCode()); |
| 86 | + } |
| 87 | + |
| 88 | + private Flux<Tuple2<IndexAnalysisMessage, IndexResult>> indexOrRemoveAnalysis( |
| 89 | + IndexAnalysisMessage msg) { |
| 90 | + if (msg.getRemoveAnalysis()) { |
| 91 | + return Flux.from(removeAnalysis(msg)); |
| 92 | + } else { |
| 93 | + return indexAnalysis(msg); |
91 | 94 | }
|
92 |
| - |
93 |
| - private Mono<Tuple2<IndexAnalysisMessage, IndexResult>> removeAnalysis(IndexAnalysisMessage msg) { |
94 |
| - return indexer.removeAnalysis(RemoveAnalysisCommand.builder() |
95 |
| - .analysisIdentifier(AnalysisIdentifier.builder() |
96 |
| - .studyId(msg.getStudyId()) |
97 |
| - .analysisId(msg.getAnalysisId()) |
98 |
| - .repositoryCode(msg.getRepositoryCode()) |
99 |
| - .build()) |
| 95 | + } |
| 96 | + |
| 97 | + private Flux<Tuple2<IndexAnalysisMessage, IndexResult>> removeAnalysis(IndexAnalysisMessage msg) { |
| 98 | + return indexer |
| 99 | + .removeAnalysis( |
| 100 | + RemoveAnalysisCommand.builder() |
| 101 | + .analysisIdentifier( |
| 102 | + AnalysisIdentifier.builder() |
| 103 | + .studyId(msg.getStudyId()) |
| 104 | + .analysisId(msg.getAnalysisId()) |
| 105 | + .repositoryCode(msg.getRepositoryCode()) |
| 106 | + .build()) |
100 | 107 | .build())
|
101 |
| - .map(out -> new Tuple2<>(msg, out)) |
102 |
| - .onErrorResume((e) -> catchUnhandledErrors(msg, e)); |
103 |
| - } |
| 108 | + .map(out -> new Tuple2<>(msg, out)) |
| 109 | + .onErrorResume((e) -> catchUnhandledErrors(msg, e)); |
| 110 | + } |
| 111 | + |
| 112 | + private Flux<Tuple2<IndexAnalysisMessage, IndexResult>> indexAnalysis(IndexAnalysisMessage msg) { |
| 113 | + return indexer |
| 114 | + .indexAnalysis( |
| 115 | + IndexAnalysisCommand.builder() |
| 116 | + .analysisIdentifier( |
| 117 | + AnalysisIdentifier.builder() |
| 118 | + .studyId(msg.getStudyId()) |
| 119 | + .analysisId(msg.getAnalysisId()) |
| 120 | + .repositoryCode(msg.getRepositoryCode()) |
| 121 | + .build()) |
| 122 | + .build()) |
| 123 | + .map(out -> new Tuple2<>(msg, out)) |
| 124 | + .onErrorResume((e) -> catchUnhandledErrors(msg, e)); |
| 125 | + } |
104 | 126 |
|
105 |
| - private Flux<Tuple2<IndexAnalysisMessage, IndexResult>> indexAnalysis(IndexAnalysisMessage msg) { |
106 |
| - return indexer.indexAnalysis(IndexAnalysisCommand.builder() |
107 |
| - .analysisIdentifier(AnalysisIdentifier.builder() |
| 127 | + private Flux<Tuple2<IndexStudyMessage, IndexResult>> indexStudy(IndexStudyMessage msg) { |
| 128 | + return indexer |
| 129 | + .indexStudy( |
| 130 | + IndexStudyCommand.builder() |
108 | 131 | .studyId(msg.getStudyId())
|
109 |
| - .analysisId(msg.getAnalysisId()) |
110 | 132 | .repositoryCode(msg.getRepositoryCode())
|
111 |
| - .build() |
112 |
| - ).build()) |
| 133 | + .build()) |
| 134 | + .map(out -> new Tuple2<>(msg, out)); |
| 135 | + } |
| 136 | + |
| 137 | + private Mono<Tuple2<IndexRepositoryMessage, IndexResult>> indexRepository( |
| 138 | + IndexRepositoryMessage msg) { |
| 139 | + return indexer |
| 140 | + .indexRepository( |
| 141 | + IndexStudyRepositoryCommand.builder().repositoryCode(msg.getRepositoryCode()).build()) |
113 | 142 | .map(out -> new Tuple2<>(msg, out))
|
114 | 143 | .onErrorResume((e) -> catchUnhandledErrors(msg, e));
|
115 |
| - } |
116 |
| - |
117 |
| - private Flux<Tuple2<IndexStudyMessage, IndexResult>> indexStudy(IndexStudyMessage msg) { |
118 |
| - return indexer.indexStudy(IndexStudyCommand.builder() |
119 |
| - .studyId(msg.getStudyId()) |
120 |
| - .repositoryCode(msg.getRepositoryCode()) |
121 |
| - .build()) |
122 |
| - .map(out -> new Tuple2<>(msg, out)); |
123 |
| - } |
124 |
| - |
125 |
| - private Mono<Tuple2<IndexRepositoryMessage, IndexResult>> indexRepository(IndexRepositoryMessage msg) { |
126 |
| - return indexer.indexRepository(IndexStudyRepositoryCommand.builder() |
127 |
| - .repositoryCode(msg.getRepositoryCode()) |
128 |
| - .build()) |
129 |
| - .map(out -> new Tuple2<>(msg, out)) |
130 |
| - .onErrorResume((e) -> catchUnhandledErrors(msg, e)); |
131 |
| - } |
132 |
| - |
133 |
| - private <T> Mono<Tuple2<T, IndexResult>> catchUnhandledErrors(T msg, Throwable e) { |
134 |
| - log.error("failed processing message: {} ", msg, e); |
135 |
| - val indexResult = IndexResult.builder() |
136 |
| - .successful(false) |
137 |
| - .failureData(FailureData.builder().build()) |
138 |
| - .build(); |
139 |
| - return Mono.just(new Tuple2<>(msg, indexResult)); |
140 |
| - } |
141 |
| - |
| 144 | + } |
| 145 | + |
| 146 | + private <T> Mono<Tuple2<T, IndexResult>> catchUnhandledErrors(T msg, Throwable e) { |
| 147 | + log.error("failed processing message: {} ", msg, e); |
| 148 | + val indexResult = |
| 149 | + IndexResult.builder().successful(false).failureData(FailureData.builder().build()).build(); |
| 150 | + return Mono.just(new Tuple2<>(msg, indexResult)); |
| 151 | + } |
142 | 152 | }
|
0 commit comments