feat(bigtable): Expose pinger in go client (#12683)

sushanb · web-flow · commit a330457cfe04 · 2025-08-22T16:51:41.000-04:00
* feat(bigtable): Expose pinger in go client

* address rview

* fix operation completion

* fix operation completion 2
diff --git a/bigtable/bigtable.go b/bigtable/bigtable.go
@@ -912,6 +912,40 @@ func handleExecuteStreamEnd(stream btpb.Bigtable_ExecuteQueryClient, trailerMD *
 	return nil
 }
 
+// Pinger pings the server and warms up the connection.
+func (c *Client) Pinger(ctx context.Context) (err error) {
+	md := metadata.Join(metadata.Pairs(
+		resourcePrefixHeader, c.fullInstanceName(),
+		requestParamsHeader, c.reqParamsHeaderValInstance(),
+	), c.newFeatureFlags())
+
+	ctx = mergeOutgoingMetadata(ctx, md)
+	ctx = trace.StartSpan(ctx, "cloud.google.com/go/bigtable/PingAndWarm")
+	defer func() { trace.EndSpan(ctx, err) }()
+	mt := c.newBuiltinMetricsTracer(ctx, "", false)
+	defer mt.recordOperationCompletion()
+
+	err = c.pingerWithMetadata(ctx, mt)
+	statusCode, statusErr := convertToGrpcStatusErr(err)
+	mt.currOp.setStatus(statusCode.String())
+	return statusErr
+}
+
+func (c *Client) pingerWithMetadata(ctx context.Context, mt *builtinMetricsTracer) (err error) {
+	req := &btpb.PingAndWarmRequest{
+		Name:         c.fullInstanceName(),
+		AppProfileId: c.appProfile,
+	}
+	err = gaxInvokeWithRecorder(ctx, mt, "PingAndWarm", func(ctx context.Context, headerMD, trailerMD *metadata.MD, _ gax.CallSettings) error {
+		var err error
+		_, err = c.client.PingAndWarm(ctx, req, grpc.Header(headerMD), grpc.Trailer(trailerMD))
+		return err
+	})
+
+	return err
+
+}
+
 func (ti *tableImpl) ReadRows(ctx context.Context, arg RowSet, f func(Row) bool, opts ...ReadOption) error {
 	return ti.Table.ReadRows(ctx, arg, f, opts...)
 }
diff --git a/bigtable/integration_test.go b/bigtable/integration_test.go
@@ -239,6 +239,22 @@ func TestIntegration_ConditionalMutations(t *testing.T) {
 	}
 }
 
+func TestIntegration_Pinger(t *testing.T) {
+	ctx := context.Background()
+	testEnv, client, _, _, _, cleanup, err := setupIntegration(ctx, t)
+	if err != nil {
+		t.Fatal(err)
+	}
+	t.Cleanup(func() { cleanup() })
+	if !testEnv.Config().UseProd {
+		t.Skip("emulator doesn't support PingAndWarm")
+	}
+	if err := client.Pinger(ctx); err != nil {
+		t.Fatalf("pinger failed. got %v, want %v", err, nil)
+	}
+
+}
+
 func TestIntegration_PartialReadRows(t *testing.T) {
 	ctx := context.Background()
 	_, _, _, table, _, cleanup, err := setupIntegration(ctx, t)
