From 3e64bd767174164d2a77ecb8dfd1aea548f7ff1a Mon Sep 17 00:00:00 2001
From: Anqi <anqi.wang@vesoft.com>
Date: Tue, 2 Jan 2024 10:44:46 +0800
Subject: [PATCH] add ssl for auth user when scan data

---
 .../nebula/client/storage/StorageClient.java    | 17 +++++++++++------
 .../client/storage/StorageClientTest.java       |  9 ++++-----
 2 files changed, 15 insertions(+), 11 deletions(-)

diff --git a/client/src/main/java/com/vesoft/nebula/client/storage/StorageClient.java b/client/src/main/java/com/vesoft/nebula/client/storage/StorageClient.java
index 15de6304b..b6cf73fd9 100644
--- a/client/src/main/java/com/vesoft/nebula/client/storage/StorageClient.java
+++ b/client/src/main/java/com/vesoft/nebula/client/storage/StorageClient.java
@@ -25,7 +25,6 @@
 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;
@@ -35,7 +34,6 @@
 import java.util.List;
 import java.util.Map;
 import java.util.Set;
-
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
@@ -1207,10 +1205,17 @@ private void authUser() throws AuthFailedException, IOErrorException,
         if (graphAddrAndPort.length != 2) {
             throw new IllegalArgumentException("the graph address is invalid.");
         }
-        graphConnection.open(new HostAddress(graphAddrAndPort[0].trim(),
-                        Integer.valueOf(graphAddrAndPort[1].trim())), timeout, false,
-                new HashMap<>(),
-                version);
+        if (sslParam == null) {
+            graphConnection.open(new HostAddress(graphAddrAndPort[0].trim(),
+                            Integer.valueOf(graphAddrAndPort[1].trim())), timeout, false,
+                    new HashMap<>(),
+                    version);
+        } else {
+            graphConnection.open(new HostAddress(graphAddrAndPort[0].trim(),
+                            Integer.valueOf(graphAddrAndPort[1].trim())), timeout, sslParam, false,
+                    new HashMap<>(),
+                    version);
+        }
         AuthResult authResult = graphConnection.authenticate(user, password);
         long sessionId = authResult.getSessionId();
 
diff --git a/client/src/test/java/com/vesoft/nebula/client/storage/StorageClientTest.java b/client/src/test/java/com/vesoft/nebula/client/storage/StorageClientTest.java
index 0205fe29d..8a0387581 100644
--- a/client/src/test/java/com/vesoft/nebula/client/storage/StorageClientTest.java
+++ b/client/src/test/java/com/vesoft/nebula/client/storage/StorageClientTest.java
@@ -8,7 +8,6 @@
 import com.vesoft.nebula.client.graph.data.CASignedSSLParam;
 import com.vesoft.nebula.client.graph.data.HostAddress;
 import com.vesoft.nebula.client.graph.data.SSLParam;
-import com.vesoft.nebula.client.graph.data.SelfSignedSSLParam;
 import com.vesoft.nebula.client.storage.data.EdgeRow;
 import com.vesoft.nebula.client.storage.data.EdgeTableRow;
 import com.vesoft.nebula.client.storage.data.VertexRow;
@@ -17,13 +16,9 @@
 import com.vesoft.nebula.client.storage.scan.ScanEdgeResultIterator;
 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;
@@ -428,6 +423,10 @@ public void testCASignedSSL() {
                     "src/test/resources/ssl/client.crt",
                     "src/test/resources/ssl/client.key");
             sslClient = new StorageClient(address, 1000, 1, 1, true, sslParam);
+            sslClient.setGraphAddress("127.0.0.1:8669");
+            sslClient.setUser("root");
+            sslClient.setPassword("nebula");
+            sslClient.setVersion("3.0.0");
             sslClient.connect();
 
             ScanVertexResultIterator resultIterator = sslClient.scanVertex(