auth fix: init store password must be not empty

apache · Mar 22, 2021 · cdf5640 · cdf5640
1 parent 4c4b770
commit cdf5640
Showing 1 changed file with 17 additions and 9 deletions.
diff --git a/hugegraph-api/src/main/java/com/baidu/hugegraph/auth/StandardAuthenticator.java b/hugegraph-api/src/main/java/com/baidu/hugegraph/auth/StandardAuthenticator.java
@@ -65,17 +65,25 @@ private void initAdminUser() throws Exception {
     }
 
     private String inputPassword() {
+        String password = "";
         String prompt = "Please input the admin password:";
+        String alertMsg = "password must be not empty";
         Console console = System.console();
-        if (console != null) {
-            char[] chars = console.readPassword(prompt);
-            return new String(chars);
-        } else {
-            System.out.print(prompt);
-            @SuppressWarnings("resource") // just wrapper of System.in
-            Scanner scanner = new Scanner(System.in);
-            return scanner.nextLine();
-        }
+        do {
+            if (console != null) {
+                char[] chars = console.readPassword(prompt);
+                password = new String(chars);
+            } else {
+                System.out.print(prompt);
+                @SuppressWarnings("resource") // just wrapper of System.in
+                Scanner scanner = new Scanner(System.in);
+                password = scanner.nextLine();
+            }
+            if (password.isEmpty()) {
+                System.out.println(alertMsg);
+            }
+        } while (password.isEmpty());
+        return password;
     }
 
     @Override