mirror of
https://gitee.com/milvus-io/milvus.git
synced 2025-12-07 01:28:27 +08:00
004a1875dc
Introduce a batch subscription mechanism in msgdispatcher: the msgdispatcher now includes a vchannel watch task queue, where all vchannels in the queue will subscribe to the MQ only once and pull messages from the oldest vchannel checkpoint to the latest. issue: https://github.com/milvus-io/milvus/issues/39862 --------- Signed-off-by: bigsheeper <yihao.dai@zilliz.com>
34 lines
741 B
Go
34 lines
741 B
Go
package msgdispatcher
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"go.uber.org/zap"
|
|
|
|
"github.com/milvus-io/milvus-proto/go-api/v2/msgpb"
|
|
"github.com/milvus-io/milvus/pkg/v2/log"
|
|
"github.com/milvus-io/milvus/pkg/v2/mq/msgstream"
|
|
"github.com/milvus-io/milvus/pkg/v2/util/paramtable"
|
|
)
|
|
|
|
func TestSendTimeout(t *testing.T) {
|
|
target := newTarget(&StreamConfig{
|
|
VChannel: "test1",
|
|
Pos: &msgpb.MsgPosition{},
|
|
})
|
|
|
|
time.Sleep(paramtable.Get().MQCfg.MaxTolerantLag.GetAsDuration(time.Second))
|
|
|
|
counter := 0
|
|
for i := 0; i < 10; i++ {
|
|
err := target.send(&msgstream.MsgPack{})
|
|
if err != nil {
|
|
log.Error("send failed", zap.Int("idx", i), zap.Error(err))
|
|
counter++
|
|
}
|
|
}
|
|
assert.Equal(t, counter, 0)
|
|
}
|