milvus/reader/read_node/reader.go
bigsheeper 721998050b Add context to query node
Signed-off-by: bigsheeper <yihao.dai@zilliz.com>
2020-10-14 11:07:25 +08:00

36 lines
629 B
Go

package reader
import (
"context"
"github.com/czs007/suvlim/reader/message_client"
"log"
"sync"
)
func StartQueryNode(ctx context.Context, pulsarURL string) {
mc := message_client.MessageClient{}
mc.InitClient(ctx, pulsarURL)
mc.ReceiveMessage()
qn := CreateQueryNode(ctx, 0, 0, &mc)
// Segments Services
go qn.SegmentManagementService()
go qn.SegmentStatisticService()
wg := sync.WaitGroup{}
err := qn.InitFromMeta()
if err != nil {
log.Printf("Init query node from meta failed")
return
}
wg.Add(3)
go qn.RunMetaService(&wg)
go qn.RunInsertDelete(&wg)
go qn.RunSearch(&wg)
wg.Wait()
qn.Close()
}