mirror of
https://gitee.com/milvus-io/milvus.git
synced 2024-12-05 13:28:49 +08:00
a984e46a29
issue: https://github.com/milvus-io/milvus/issues/32827 Signed-off-by: bigsheeper <yihao.dai@zilliz.com>
42 lines
1.4 KiB
Go
42 lines
1.4 KiB
Go
package broker
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/milvus-io/milvus-proto/go-api/v2/msgpb"
|
|
"github.com/milvus-io/milvus/internal/proto/datapb"
|
|
"github.com/milvus-io/milvus/internal/types"
|
|
"github.com/milvus-io/milvus/pkg/util/typeutil"
|
|
)
|
|
|
|
// Broker is the interface for datanode to interact with other components.
|
|
//
|
|
//go:generate mockery --name=Broker --structname=MockBroker --output=./ --filename=mock_broker.go --with-expecter --inpackage
|
|
type Broker interface {
|
|
DataCoord
|
|
}
|
|
|
|
type coordBroker struct {
|
|
*dataCoordBroker
|
|
}
|
|
|
|
func NewCoordBroker(dc types.DataCoordClient, serverID int64) Broker {
|
|
return &coordBroker{
|
|
dataCoordBroker: &dataCoordBroker{
|
|
client: dc,
|
|
serverID: serverID,
|
|
},
|
|
}
|
|
}
|
|
|
|
// DataCoord is the interface wraps `DataCoord` grpc call
|
|
type DataCoord interface {
|
|
AssignSegmentID(ctx context.Context, reqs ...*datapb.SegmentIDRequest) ([]typeutil.UniqueID, error)
|
|
ReportTimeTick(ctx context.Context, msgs []*msgpb.DataNodeTtMsg) error
|
|
GetSegmentInfo(ctx context.Context, segmentIDs []int64) ([]*datapb.SegmentInfo, error)
|
|
UpdateChannelCheckpoint(ctx context.Context, channelCPs []*msgpb.MsgPosition) error
|
|
SaveBinlogPaths(ctx context.Context, req *datapb.SaveBinlogPathsRequest) error
|
|
DropVirtualChannel(ctx context.Context, req *datapb.DropVirtualChannelRequest) (*datapb.DropVirtualChannelResponse, error)
|
|
UpdateSegmentStatistics(ctx context.Context, req *datapb.UpdateSegmentStatisticsRequest) error
|
|
}
|