未验证 提交 ebb9b24b 编写于 作者: J Jiquan Long 提交者: GitHub

Improve name of flowgraph node (#14538)

Signed-off-by: Ndragondriver <jiquan.long@zilliz.com>
上级 d5e6adf3
......@@ -18,6 +18,7 @@ package datanode
import (
"context"
"fmt"
"sync"
"sync/atomic"
......@@ -68,7 +69,7 @@ type ddNode struct {
// Name returns node name, implementing flowgraph.Node
func (ddn *ddNode) Name() string {
return "ddNode"
return fmt.Sprintf("ddNode-%d-%s", ddn.collectionID, ddn.vchannelName)
}
// Operate handles input messages, implementing flowgrpah.Node
......
......@@ -18,6 +18,7 @@ package datanode
import (
"context"
"fmt"
"testing"
"github.com/stretchr/testify/assert"
......@@ -90,7 +91,7 @@ func TestFlowGraph_DDNode_newDDNode(te *testing.T) {
for _, seg := range ddNode.flushedSegments {
flushedSegIDs = append(flushedSegIDs, seg.ID)
}
assert.Equal(t, "ddNode", ddNode.Name())
assert.Equal(t, fmt.Sprintf("ddNode-%d-%s", ddNode.collectionID, ddNode.vchannelName), ddNode.Name())
assert.Equal(t, test.inCollID, ddNode.collectionID)
assert.Equal(t, len(test.inFlushedSegs), len(ddNode.flushedSegments))
assert.ElementsMatch(t, test.inFlushedSegs, flushedSegIDs)
......
......@@ -91,7 +91,7 @@ func newDelDataBuf() *DelDataBuf {
}
func (dn *deleteNode) Name() string {
return "deleteNode"
return "deleteNode-" + dn.channelName
}
func (dn *deleteNode) Close() {
......
......@@ -121,7 +121,7 @@ func TestFlowGraphDeleteNode_newDeleteNode(te *testing.T) {
assert.Nil(t, err)
assert.NotNil(t, dn)
assert.Equal(t, "deleteNode", dn.Name())
assert.Equal(t, "deleteNode-"+dn.channelName, dn.Name())
dn.Close()
})
}
......
......@@ -57,6 +57,7 @@ func newDmInputNode(ctx context.Context, seekPos *internalpb.MsgPosition, dmNode
log.Debug("datanode Seek successfully", zap.String("Channel Name", seekPos.GetChannelName()), zap.Duration("elapse", time.Since(start)))
}
node := flowgraph.NewInputNode(insertStream, "dmInputNode", dmNodeConfig.maxQueueLength, dmNodeConfig.maxParallelism)
name := fmt.Sprintf("dmInputNode-%d-%s", dmNodeConfig.collectionID, dmNodeConfig.vChannelName)
node := flowgraph.NewInputNode(insertStream, name, dmNodeConfig.maxQueueLength, dmNodeConfig.maxParallelism)
return node, nil
}
......@@ -142,7 +142,7 @@ func (bd *BufferData) updateSize(no int64) {
}
func (ibNode *insertBufferNode) Name() string {
return "ibNode"
return "ibNode-" + ibNode.channelName
}
func (ibNode *insertBufferNode) Close() {
......
......@@ -17,6 +17,8 @@
package querynode
import (
"fmt"
"github.com/opentracing/opentracing-go"
"go.uber.org/zap"
......@@ -36,7 +38,7 @@ type filterDeleteNode struct {
// Name returns the name of filterDeleteNode
func (fddNode *filterDeleteNode) Name() string {
return "fdNode"
return fmt.Sprintf("fdNode-%d", fddNode.collectionID)
}
// Operate handles input messages, to filter invalid delete messages
......
......@@ -17,6 +17,8 @@
package querynode
import (
"fmt"
"github.com/opentracing/opentracing-go"
"go.uber.org/zap"
......@@ -36,7 +38,7 @@ type filterDmNode struct {
// Name returns the name of filterDmNode
func (fdmNode *filterDmNode) Name() string {
return "fdmNode"
return fmt.Sprintf("fdmNode-%d", fdmNode.collectionID)
}
// Operate handles input messages, to filter invalid insert messages
......
......@@ -18,6 +18,7 @@ package querynode
import (
"context"
"fmt"
"go.uber.org/zap"
......@@ -37,7 +38,7 @@ type serviceTimeNode struct {
// Name returns the name of serviceTimeNode
func (stNode *serviceTimeNode) Name() string {
return "stNode"
return fmt.Sprintf("stNode-%d-%s", stNode.collectionID, stNode.vChannel)
}
// Close would close serviceTimeNode
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册