[
https://issues.apache.org/jira/browse/BEAM-14483?focusedWorklogId=774375&page=com.atlassian.jira.plugin.system.issuetabpanels:worklog-tabpanel#worklog-774375
]
ASF GitHub Bot logged work on BEAM-14483:
-----------------------------------------
Author: ASF GitHub Bot
Created on: 25/May/22 06:55
Start Date: 25/May/22 06:55
Worklog Time Spent: 10m
Work Description: ihji commented on code in PR #17696:
URL: https://github.com/apache/beam/pull/17696#discussion_r881279584
##########
sdks/java/extensions/python/src/main/java/org/apache/beam/sdk/extensions/python/transforms/PythonMap.java:
##########
@@ -0,0 +1,69 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements. See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership. The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.apache.beam.sdk.extensions.python.transforms;
+
+import org.apache.beam.sdk.coders.Coder;
+import org.apache.beam.sdk.extensions.python.PythonExternalTransform;
+import org.apache.beam.sdk.transforms.PTransform;
+import org.apache.beam.sdk.util.PythonCallableSource;
+import org.apache.beam.sdk.values.PCollection;
+import org.checkerframework.checker.nullness.qual.Nullable;
+
+public class PythonMap<InputT, OutputT>
+ extends PTransform<PCollection<? extends InputT>, PCollection<OutputT>> {
+
+ private PythonCallableSource pythonFunction;
+ private @Nullable String expansionService;
+ private Coder<?> outputCoder;
+ private static final String PYTHON_MAP_FN_TRANSFORM = "apache_beam.Map";
+ private static final String PYTHON_FLATMAP_FN_TRANSFORM =
"apache_beam.FlatMap";
+ private String pythonTransform;
+
+ private PythonMap(
+ PythonCallableSource pythonFunction, Coder<?> outputCoder, String
pythonTransform) {
+ this.pythonFunction = pythonFunction;
+ this.outputCoder = outputCoder;
+ this.pythonTransform = pythonTransform;
+ }
+
+ public static <InputT, OutputT> PythonMap<InputT, OutputT> viaMapFn(
+ PythonCallableSource pythonFunction, Coder<?> outputCoder) {
+ return new PythonMap<>(pythonFunction, outputCoder,
PYTHON_MAP_FN_TRANSFORM);
+ }
+
+ public static <InputT, OutputT> PythonMap<InputT, OutputT> viaFlatMapFn(
+ PythonCallableSource pythonFunction, Coder<?> outputCoder) {
+ return new PythonMap<>(pythonFunction, outputCoder,
PYTHON_FLATMAP_FN_TRANSFORM);
+ }
+
+ public PythonMap<InputT, OutputT> withExpansionService(String
expansionService) {
+ this.expansionService = expansionService;
+ return this;
+ }
+
+ @Override
+ public PCollection<OutputT> expand(PCollection<? extends InputT> input) {
+ PythonExternalTransform<PCollection<? extends InputT>,
PCollection<OutputT>> pythonMapElements =
+ (expansionService == null)
Review Comment:
FYI, we don't need a ternary operation with null check if we use an empty
string as a default value for `expansionService`.
##########
sdks/java/extensions/python/src/main/java/org/apache/beam/sdk/extensions/python/PythonExternalTransform.java:
##########
@@ -191,6 +196,23 @@ public PythonExternalTransform<InputT, OutputT>
withTypeHint(
return this;
}
+ public PythonExternalTransform<InputT, OutputT> withOutputCoders(
+ Map<String, Coder<?>> outputCoders) {
+ if (this.outputCoders.size() > 0) {
+ throw new IllegalArgumentException("Output coders were already
specified");
+ }
+ this.outputCoders.putAll(outputCoders);
+ return this;
+ }
+
+ public PythonExternalTransform<InputT, OutputT> withOutputCoder(Coder<?>
outputCoder) {
Review Comment:
ditto
##########
sdks/java/extensions/python/src/main/java/org/apache/beam/sdk/extensions/python/PythonExternalTransform.java:
##########
@@ -191,6 +196,23 @@ public PythonExternalTransform<InputT, OutputT>
withTypeHint(
return this;
}
+ public PythonExternalTransform<InputT, OutputT> withOutputCoders(
Review Comment:
Do we need a documentation since it's a public method?
Issue Time Tracking
-------------------
Worklog Id: (was: 774375)
Time Spent: 1h 50m (was: 1h 40m)
> Add Java cross-language transforms for invoking Python Map and FlatMap
> ----------------------------------------------------------------------
>
> Key: BEAM-14483
> URL: https://issues.apache.org/jira/browse/BEAM-14483
> Project: Beam
> Issue Type: New Feature
> Components: cross-language
> Reporter: Chamikara Madhusanka Jayalath
> Assignee: Chamikara Madhusanka Jayalath
> Priority: P1
> Time Spent: 1h 50m
> Remaining Estimate: 0h
>
--
This message was sent by Atlassian Jira
(v8.20.7#820007)
