improve

Change-Id: I57331425089e18582f5938536c5026e703ce34c6
apache · Oct 27, 2020 · 381aefe · 381aefe
1 parent e81ea88
commit 381aefe
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/hugegraph-core/src/main/java/com/baidu/hugegraph/schema/builder/AbstractBuilder.java b/hugegraph-core/src/main/java/com/baidu/hugegraph/schema/builder/AbstractBuilder.java
@@ -30,9 +30,11 @@
 import com.baidu.hugegraph.schema.EdgeLabel;
 import com.baidu.hugegraph.schema.IndexLabel;
 import com.baidu.hugegraph.schema.PropertyKey;
+import com.baidu.hugegraph.schema.SchemaElement;
 import com.baidu.hugegraph.schema.VertexLabel;
 import com.baidu.hugegraph.type.HugeType;
 import com.baidu.hugegraph.type.define.GraphMode;
+import com.baidu.hugegraph.type.define.SchemaStatus;
 import com.baidu.hugegraph.util.E;
 import com.baidu.hugegraph.util.LockUtil;
 
@@ -77,6 +79,11 @@ protected <V> V lockCheckAndCreateSchema(HugeType type, String name,
         }
     }
 
+    protected void updateSchemaStatus(SchemaElement element,
+                                      SchemaStatus status) {
+        this.transaction.updateSchemaStatus(element, status);
+    }
+
     protected void checkSchemaIdIfRestoringMode(HugeType type, Id id) {
         if (this.transaction.graphMode() == GraphMode.RESTORING) {
             E.checkArgument(id != null,

diff --git a/hugegraph-core/src/main/java/com/baidu/hugegraph/schema/builder/IndexLabelBuilder.java b/hugegraph-core/src/main/java/com/baidu/hugegraph/schema/builder/IndexLabelBuilder.java
@@ -231,8 +231,7 @@ public IndexLabel.CreatedIndexLabel createWithTask() {
                 return new IndexLabel.CreatedIndexLabel(indexLabel,
                                                         rebuildTask);
             } catch (Throwable e) {
-                indexLabel.status(SchemaStatus.INVALID);
-                this.graph().addIndexLabel(schemaLabel, indexLabel);
+                this.updateSchemaStatus(indexLabel, SchemaStatus.INVALID);
                 throw e;
             }
         });