Merge pull request #164 from Thericecookers/master

bugfix: Reverse edge has wrong partitionId
vesoft-inc · Oct 18, 2021 · 6c2700f · 6c2700f
2 parents 4cd75f0 + f41f41d
commit 6c2700f
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/nebula-exchange/src/main/scala/com/vesoft/nebula/exchange/processor/EdgeProcessor.scala b/nebula-exchange/src/main/scala/com/vesoft/nebula/exchange/processor/EdgeProcessor.scala
@@ -175,7 +175,8 @@ class EdgeProcessor(data: DataFrame,
               hostAddrs.append(new HostAddress(addr.getHostText, addr.getPort))
             }
 
-            val partitionId = NebulaUtils.getPartitionId(srcId, partitionNum, vidType)
+            val srcPartitionId = NebulaUtils.getPartitionId(srcId, partitionNum, vidType)
+            val dstPartitionId = NebulaUtils.getPartitionId(dstId, partitionNum, vidType)
             val codec       = new NebulaCodecImpl()
 
             import java.nio.ByteBuffer
@@ -199,13 +200,13 @@ class EdgeProcessor(data: DataFrame,
               dstId.getBytes()
             }
             val positiveEdgeKey = codec.edgeKeyByDefaultVer(spaceVidLen,
-                                                            partitionId,
+                                                            srcPartitionId,
                                                             srcBytes,
                                                             edgeItem.getEdge_type,
                                                             ranking,
                                                             dstBytes)
             val reverseEdgeKey = codec.edgeKeyByDefaultVer(spaceVidLen,
-                                                           partitionId,
+                                                           dstPartitionId,
                                                            dstBytes,
                                                            -edgeItem.getEdge_type,
                                                            ranking,