liferoad commented on code in PR #35876:
URL: https://github.com/apache/beam/pull/35876#discussion_r2279289086
##########
sdks/go/pkg/beam/io/xlang/bigtableio/bigtable.go:
##########
@@ -135,3 +138,36 @@ func Read(s beam.Scope, projectId string, instanceId
string, table string, opts
outs := xlschema.Transform(s, btConfig, readURN,
xlschema.ExpansionAddr(addr),
xlschema.UnnamedOutputType(typex.New(reflect.TypeOf(Row{}))))
return outs[beam.UnnamedOutputTag()]
}
+
+type WriteOption func(*writeConfig)
+type writeConfig struct {
+ addr string
+}
+
+// WriteExpansionAddr specifies the address of a persistent expansion service
to use for a Write
+// transform. If this is not provided, or if an empty string is provided, the
transform will
+// automatically start an appropriate expansion service instead.
+func WriteExpansionAddr(addr string) WriteOption {
+ return func(wc *writeConfig) {
+ wc.addr = addr
+ }
+}
+
+// Write writes rows to a Bigtable table.
+func Write(s beam.Scope, projectId string, instanceId string, table string,
col beam.PCollection, opts ...WriteOption) {
Review Comment:
can we add one test?
--
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.
To unsubscribe, e-mail: github-unsubscr...@beam.apache.org
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
