process the read permission white list for root user

client/src/main/java/com/vesoft/nebula/client/storage/StorageClient.java

@@ -25,6 +25,7 @@
 import com.vesoft.nebula.storage.ScanEdgeRequest;
 import com.vesoft.nebula.storage.ScanVertexRequest;
 import com.vesoft.nebula.storage.VertexProp;
+
 import java.io.Serializable;
 import java.io.UnsupportedEncodingException;
 import java.util.ArrayList;
@@ -34,6 +35,7 @@
 import java.util.List;
 import java.util.Map;
 import java.util.Set;
+
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
@@ -57,7 +59,7 @@ public class StorageClient implements Serializable {
  private String graphAddress = null;
 
  // the write list for users with read permission
- private Map<String, List<String>> spaceLabelWriteList = new HashMap<>();
+ private Map<String, List<String>> spaceLabelWriteList = null;
  private String version = null;
 
  /**
@@ -1211,6 +1213,12 @@ private void authUser() throws AuthFailedException, IOErrorException,
  version);
  AuthResult authResult = graphConnection.authenticate(user, password);
  long sessionId = authResult.getSessionId();
+
+ if (user.equals("root")) {
+ return;
+ }
+
+ spaceLabelWriteList = new HashMap<>();
  ResultSet resultSet = new ResultSet(
  graphConnection.execute(sessionId, "DESC USER " + user),
  authResult.getTimezoneOffset());
@@ -1266,6 +1274,9 @@ private void authUser() throws AuthFailedException, IOErrorException,
  * @return true if spaceName and label in the WriteList
  */
  private boolean checkWriteList(String spaceName, String label) {
+ if (spaceLabelWriteList == null) {
+ return true;
+ }
  if (!spaceLabelWriteList.containsKey(spaceName)) {
  return false;
  }

client/src/test/java/com/vesoft/nebula/client/storage/StorageClientTest.java

@@ -18,10 +18,12 @@
 import com.vesoft.nebula.client.storage.scan.ScanVertexResult;
 import com.vesoft.nebula.client.storage.scan.ScanVertexResultIterator;
 import com.vesoft.nebula.client.util.ProcessUtil;
+
 import java.io.IOException;
 import java.io.UnsupportedEncodingException;
 import java.util.Arrays;
 import java.util.List;
+
 import org.junit.After;
 import org.junit.Assert;
 import org.junit.Before;
@@ -60,6 +62,9 @@ public void testStorageClientWithVersionInWhiteList() {
  List<HostAddress> address = Arrays.asList(new HostAddress(ip, 9559));
  StorageClient storageClient = new StorageClient(address);
  try {
+ storageClient.setGraphAddress("127.0.0.1:9669");
+ storageClient.setUser("root");
+ storageClient.setPassword("nebula");
  storageClient.setVersion("3.0.0");
  assert (storageClient.connect());
 
@@ -76,6 +81,9 @@ public void testStorageClientWithVersionNotInWhiteList() {
  List<HostAddress> address = Arrays.asList(new HostAddress(ip, 9559));
  StorageClient storageClient = new StorageClient(address);
  try {
+ storageClient.setGraphAddress("127.0.0.1:9669");
+ storageClient.setUser("root");
+ storageClient.setPassword("nebula");
  storageClient.setVersion("INVALID_VERSION");
  storageClient.connect();
  assert false;