@@ -157,8 +157,9 @@ public void updateSchemaCache() {
157157
158158 public void setLoadingMode () {
159159 String graph = this .client .graph ().graph ();
160+ String graphSpace = this .client .graph ().graphSpace ();
160161 try {
161- this .client .graphs ().mode (graph , GraphMode .LOADING );
162+ this .client .graphs ().mode (graphSpace , graph , GraphMode .LOADING );
162163 } catch (ServerException e ) {
163164 if (e .getMessage ().contains ("Can not deserialize value of type" )) {
164165 LOG .warn ("HugeGraphServer doesn't support loading mode" );
@@ -170,8 +171,9 @@ public void setLoadingMode() {
170171
171172 public void setRestoreMode () {
172173 String graph = this .client .graph ().graph ();
174+ String graphSpace = this .client .graph ().graphSpace ();
173175 try {
174- this .client .graphs ().mode (graph , GraphMode .RESTORING );
176+ this .client .graphs ().mode (graphSpace , graph , GraphMode .RESTORING );
175177 } catch (ServerException e ) {
176178 if (e .getMessage ().contains ("Can not deserialize value of type" )) {
177179 LOG .warn ("HugeGraphServer doesn't support loading mode" );
@@ -184,9 +186,10 @@ public void setRestoreMode() {
184186 public void unsetLoadingMode () {
185187 try {
186188 String graph = this .client .graph ().graph ();
187- GraphMode mode = this .client .graphs ().mode (graph );
189+ String graphSpace = this .client .graph ().graphSpace ();
190+ GraphMode mode = this .client .graphs ().mode (graphSpace , graph );
188191 if (mode .loading ()) {
189- this .client .graphs ().mode (graph , GraphMode .NONE );
192+ this .client .graphs ().mode (graphSpace , graph , GraphMode .NONE );
190193 }
191194 } catch (Exception e ) {
192195 throw new LoadException ("Failed to unset mode %s for server" ,
0 commit comments