milvus/internal/datacoord/datanode_helper.go
congqixia 768535f31c
Add positionProvider implementation assertion (#9261)
Signed-off-by: Congqi Xia <congqi.xia@zilliz.com>
2021-10-05 20:58:04 +08:00

45 lines
1.5 KiB
Go

// Copyright (C) 2019-2020 Zilliz. All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance
// with the License. You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software distributed under the License
// is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
// or implied. See the License for the specific language governing permissions and limitations under the License.
package datacoord
import (
"github.com/milvus-io/milvus/internal/proto/datapb"
)
type vchannel struct {
CollectionID UniqueID
DmlChannel string
}
// positionProvider provides vchannel pair related position pairs
type positionProvider interface {
GetVChanPositions(vchans []vchannel, seekFromStartPosition bool) ([]*datapb.VchannelInfo, error)
}
var _ positionProvider = (*dummyPosProvider)(nil)
type dummyPosProvider struct{}
//GetVChanPositions implements positionProvider
func (dp dummyPosProvider) GetVChanPositions(vchans []vchannel, seekFromStartPosition bool) ([]*datapb.VchannelInfo, error) {
pairs := make([]*datapb.VchannelInfo, 0, len(vchans))
for _, vchan := range vchans {
pairs = append(pairs, &datapb.VchannelInfo{
CollectionID: vchan.CollectionID,
ChannelName: vchan.DmlChannel,
FlushedSegments: []int64{},
UnflushedSegments: []*datapb.SegmentInfo{},
})
}
return pairs, nil
}