From 5bd204109fc7047cde08165ef88dc69b3368005f Mon Sep 17 00:00:00 2001 From: Derek McGowan Date: Fri, 1 Mar 2024 14:18:46 -0800 Subject: [PATCH] Remove grpc from Client connection interface Signed-off-by: Derek McGowan --- client/client.go | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/client/client.go b/client/client.go index b8c0566ca..4d44e42e2 100644 --- a/client/client.go +++ b/client/client.go @@ -737,8 +737,9 @@ func (c *Client) VersionService() versionservice.VersionClient { return versionservice.NewVersionClient(c.conn) } -// Conn returns the underlying GRPC connection object -func (c *Client) Conn() *grpc.ClientConn { +// Conn returns the underlying RPC connection object +// Either *grpc.ClientConn or *ttrpc.Conn +func (c *Client) Conn() any { c.connMu.Lock() defer c.connMu.Unlock() return c.conn