lastomato commented on a change in pull request #13395:
URL: https://github.com/apache/beam/pull/13395#discussion_r530526890
##########
File path:
sdks/java/io/google-cloud-platform/src/main/java/org/apache/beam/sdk/io/gcp/healthcare/FhirIO.java
##########
@@ -1388,4 +1410,208 @@ public void deidentify(ProcessContext context)
}
}
}
+
+ /** The type Search. */
+ public static class Search extends PTransform<PCollection<KV<String,
Map<String, String>>>, FhirIO.Search.Result> {
+ private static final Logger LOG = LoggerFactory.getLogger(Search.class);
+
+ private final ValueProvider<String> fhirStore;
+
+ /**
+ * Instantiates a new Search.
+ *
+ * @param fhirStore the fhir store
+ */
+ Search(ValueProvider<String> fhirStore) {
+ this.fhirStore = fhirStore;
+ }
+
+ /**
+ * Instantiates a new Search.
+ *
+ * @param fhirStore the fhir store
+ */
+ Search(String fhirStore) {
+ this.fhirStore = StaticValueProvider.of(fhirStore);
+ }
+
+ /** The type Result. */
+ public static class Result implements POutput, PInput {
+ private PCollection<String> resources;
+
+ private PCollection<HealthcareIOError<String>> failedSearches;
+ PCollectionTuple pct;
+
+ /**
+ * Create FhirIO.Search.Result form PCollectionTuple with OUT and
DEAD_LETTER tags.
+ *
+ * @param pct the pct
+ * @return the search result
+ * @throws IllegalArgumentException the illegal argument exception
+ */
+ static FhirIO.Search.Result of(PCollectionTuple pct) throws
IllegalArgumentException {
+ if (pct.getAll()
+ .keySet()
+ .containsAll((Collection<?>)
TupleTagList.of(OUT).and(DEAD_LETTER))) {
+ return new FhirIO.Search.Result(pct);
+ } else {
+ throw new IllegalArgumentException(
+ "The PCollection tuple must have the FhirIO.Search.OUT "
+ + "and FhirIO.Search.DEAD_LETTER tuple tags");
+ }
+ }
+
+ private Result(PCollectionTuple pct) {
+ this.pct = pct;
+ this.resources = pct.get(OUT);
+ this.failedSearches =
+
pct.get(DEAD_LETTER).setCoder(HealthcareIOErrorCoder.of(StringUtf8Coder.of()));
Review comment:
It should be OK.
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
