Zhen Ye 3bcdd92915
enhance: add broadcast for streaming service (#39020)
issue: #38399 

- Add new rpc for transfer broadcast to streaming coord
- Add broadcast service at streaming coord to make broadcast message
sent automicly

Signed-off-by: chyezh <chyezh@outlook.com>
2025-01-09 16:24:55 +08:00

33 lines
900 B
Go

package client
import (
"github.com/milvus-io/milvus/internal/streamingcoord/client/assignment"
"github.com/milvus-io/milvus/internal/streamingcoord/client/broadcast"
"github.com/milvus-io/milvus/internal/util/streamingutil/service/lazygrpc"
"github.com/milvus-io/milvus/internal/util/streamingutil/service/resolver"
)
// clientImpl is the implementation of Client.
type clientImpl struct {
conn lazygrpc.Conn
rb resolver.Builder
assignmentService *assignment.AssignmentServiceImpl
broadcastService *broadcast.BroadcastServiceImpl
}
func (c *clientImpl) Broadcast() BroadcastService {
return c.broadcastService
}
// Assignment access assignment service.
func (c *clientImpl) Assignment() AssignmentService {
return c.assignmentService
}
// Close close the client.
func (c *clientImpl) Close() {
c.assignmentService.Close()
c.conn.Close()
c.rb.Close()
}