client

type AssignmentService interface {
	// AssignmentDiscover is used to watches the assignment discovery.
	types.AssignmentDiscoverWatcher

	// UpdateReplicateConfiguration updates the replicate configuration to the milvus cluster.
	UpdateReplicateConfiguration(ctx context.Context, config *commonpb.ReplicateConfiguration) error

	// GetReplicateConfiguration returns the replicate configuration of the milvus cluster.
	GetReplicateConfiguration(ctx context.Context) (*replicateutil.ConfigHelper, error)

	// GetLatestAssignments returns the latest assignment discovery result.
	GetLatestAssignments(ctx context.Context) (*types.VersionedStreamingNodeAssignments, error)

	// UpdateWALBalancePolicy is used to update the WAL balance policy.
	// Return the WAL balance policy after the update.
	// Deprecated: This function is deprecated and will be removed in the future.
	UpdateWALBalancePolicy(ctx context.Context, req *types.UpdateWALBalancePolicyRequest) (*types.UpdateWALBalancePolicyResponse, error)
}

type BroadcastService interface {
	// Broadcast sends a broadcast message to the streaming service.
	Broadcast(ctx context.Context, msg message.BroadcastMutableMessage) (*types.BroadcastAppendResult, error)

	// Ack sends a broadcast ack to the streaming service.
	Ack(ctx context.Context, msg message.ImmutableMessage) error
}

type Client interface {
	// Broadcast access broadcast service.
	// Broadcast service will always be available.
	// When streaming service is enabled, the broadcast will use the streaming service.
	// When streaming service is disabled, the broadcast will use legacy msgstream.
	Broadcast() BroadcastService

	// Assignment access assignment service.
	// Assignment service will only be available when streaming service is enabled.
	Assignment() AssignmentService

	// Close close the client.
	Close()
}