process the read permission white list for root user

Author: Nicole00

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;