improve

Change-Id: I3addbb2ffd44b53c4edd0241327ba6d4f4895126
apache · Jun 8, 2021 · c01e998 · c01e998
1 parent e12a45c
commit c01e998
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/hugegraph-api/src/main/java/com/baidu/hugegraph/api/graph/EdgeAPI.java b/hugegraph-api/src/main/java/com/baidu/hugegraph/api/graph/EdgeAPI.java
@@ -104,8 +104,8 @@ public String create(@Context GraphManager manager,
                         "Invalid target vertex label '%s'");
         }
 
-        Vertex srcVertex = getVertex(g, jsonEdge.source, null);
-        Vertex tgtVertex = getVertex(g, jsonEdge.target, null);
+        Vertex srcVertex = getVertex(g, jsonEdge.source, jsonEdge.sourceLabel);
+        Vertex tgtVertex = getVertex(g, jsonEdge.target, jsonEdge.targetLabel);
 
         Edge edge = commit(g, () -> {
             return srcVertex.addEdge(jsonEdge.label, tgtVertex,