mirror of https://github.com/milvus-io/milvus.git
36 lines
1.1 KiB
Go
36 lines
1.1 KiB
Go
package querynode
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/zilliztech/milvus-distributed/internal/util/flowgraph"
|
|
)
|
|
|
|
func (dsService *dataSyncService) newDmInputNode(ctx context.Context) *flowgraph.InputNode {
|
|
// query node doesn't need to consume any topic
|
|
insertStream, _ := dsService.msFactory.NewTtMsgStream(ctx)
|
|
dsService.dmStream = insertStream
|
|
|
|
maxQueueLength := Params.FlowGraphMaxQueueLength
|
|
maxParallelism := Params.FlowGraphMaxParallelism
|
|
|
|
node := flowgraph.NewInputNode(&insertStream, "dmInputNode", maxQueueLength, maxParallelism)
|
|
return node
|
|
}
|
|
|
|
func (dsService *dataSyncService) newDDInputNode(ctx context.Context) *flowgraph.InputNode {
|
|
consumeChannels := Params.DDChannelNames
|
|
consumeSubName := Params.MsgChannelSubName
|
|
|
|
ddStream, _ := dsService.msFactory.NewTtMsgStream(ctx)
|
|
ddStream.AsConsumer(consumeChannels, consumeSubName)
|
|
|
|
dsService.ddStream = ddStream
|
|
|
|
maxQueueLength := Params.FlowGraphMaxQueueLength
|
|
maxParallelism := Params.FlowGraphMaxParallelism
|
|
|
|
node := flowgraph.NewInputNode(&ddStream, "ddInputNode", maxQueueLength, maxParallelism)
|
|
return node
|
|
}
|