Merge pull request #151 from Codelone/liyanl-dev

add null rank check
vesoft-inc · Sep 22, 2021 · b886279 · b886279
2 parents 9f414be + c0fed61
commit b886279
Showing 1 changed file with 2 additions and 0 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
@@ -156,6 +156,8 @@ class EdgeProcessor(data: DataFrame,
 
             val ranking: Long = if (edgeConfig.rankingField.isDefined) {
               val rankIndex = row.schema.fieldIndex(edgeConfig.rankingField.get)
+              assert(rankIndex >= 0 && !row.isNullAt(rankIndex),
+                s"rank must exist and cannot be null, your row data is $row")
               row.get(rankIndex).toString.toLong
             } else {
               0