mirror of
https://gitee.com/milvus-io/milvus.git
synced 2024-12-04 04:49:08 +08:00
b414800d49
* update Seek Signed-off-by: yudong.cai <yudong.cai@zilliz.com> * update Seek for mqTtMsgStream Signed-off-by: yudong.cai <yudong.cai@zilliz.com> * use Retry in Seek Signed-off-by: yudong.cai <yudong.cai@zilliz.com> * fix static-check Signed-off-by: yudong.cai <yudong.cai@zilliz.com>
126 lines
2.8 KiB
Go
126 lines
2.8 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 msgstream
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
)
|
|
|
|
type SimpleMsgStream struct {
|
|
msgChan chan *MsgPack
|
|
|
|
msgCount int
|
|
msgCountMtx sync.RWMutex
|
|
}
|
|
|
|
func (ms *SimpleMsgStream) Start() {
|
|
}
|
|
|
|
func (ms *SimpleMsgStream) Close() {
|
|
}
|
|
|
|
func (ms *SimpleMsgStream) Chan() <-chan *MsgPack {
|
|
return ms.msgChan
|
|
}
|
|
|
|
func (ms *SimpleMsgStream) AsProducer(channels []string) {
|
|
}
|
|
|
|
func (ms *SimpleMsgStream) AsConsumer(channels []string, subName string) {
|
|
}
|
|
|
|
func (ms *SimpleMsgStream) ComputeProduceChannelIndexes(tsMsgs []TsMsg) [][]int32 {
|
|
return nil
|
|
}
|
|
|
|
func (ms *SimpleMsgStream) SetRepackFunc(repackFunc RepackFunc) {
|
|
}
|
|
|
|
func (ms *SimpleMsgStream) getMsgCount() int {
|
|
ms.msgCountMtx.RLock()
|
|
defer ms.msgCountMtx.RUnlock()
|
|
|
|
return ms.msgCount
|
|
}
|
|
|
|
func (ms *SimpleMsgStream) increaseMsgCount(delta int) {
|
|
ms.msgCountMtx.Lock()
|
|
defer ms.msgCountMtx.Unlock()
|
|
|
|
ms.msgCount += delta
|
|
}
|
|
|
|
func (ms *SimpleMsgStream) decreaseMsgCount(delta int) {
|
|
ms.increaseMsgCount(-delta)
|
|
}
|
|
|
|
func (ms *SimpleMsgStream) Produce(pack *MsgPack) error {
|
|
defer ms.increaseMsgCount(1)
|
|
|
|
ms.msgChan <- pack
|
|
|
|
return nil
|
|
}
|
|
|
|
func (ms *SimpleMsgStream) Broadcast(pack *MsgPack) error {
|
|
return nil
|
|
}
|
|
|
|
func (ms *SimpleMsgStream) GetProduceChannels() []string {
|
|
return nil
|
|
}
|
|
|
|
func (ms *SimpleMsgStream) Consume() *MsgPack {
|
|
if ms.getMsgCount() <= 0 {
|
|
return nil
|
|
}
|
|
|
|
defer ms.decreaseMsgCount(1)
|
|
|
|
return <-ms.msgChan
|
|
}
|
|
|
|
func (ms *SimpleMsgStream) Seek(offset []*MsgPosition) error {
|
|
return nil
|
|
}
|
|
|
|
func NewSimpleMsgStream() *SimpleMsgStream {
|
|
return &SimpleMsgStream{
|
|
msgChan: make(chan *MsgPack, 1024),
|
|
msgCount: 0,
|
|
}
|
|
}
|
|
|
|
type SimpleMsgStreamFactory struct {
|
|
}
|
|
|
|
func (factory *SimpleMsgStreamFactory) SetParams(params map[string]interface{}) error {
|
|
return nil
|
|
}
|
|
|
|
func (factory *SimpleMsgStreamFactory) NewMsgStream(ctx context.Context) (MsgStream, error) {
|
|
return NewSimpleMsgStream(), nil
|
|
}
|
|
|
|
func (factory *SimpleMsgStreamFactory) NewTtMsgStream(ctx context.Context) (MsgStream, error) {
|
|
return NewSimpleMsgStream(), nil
|
|
}
|
|
|
|
func (factory *SimpleMsgStreamFactory) NewQueryMsgStream(ctx context.Context) (MsgStream, error) {
|
|
return NewSimpleMsgStream(), nil
|
|
}
|
|
|
|
func NewSimpleMsgStreamFactory() *SimpleMsgStreamFactory {
|
|
return &SimpleMsgStreamFactory{}
|
|
}
|