[apache#5960] fix(CLI): Add register and link commands to CLI for model

Fix some error.
Abyss-lord · Jan 3, 2025 · 04a9339 · 04a9339
1 parent 5eeec9a
commit 04a9339
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/clients/cli/src/main/java/org/apache/gravitino/cli/commands/LinkModel.java b/clients/cli/src/main/java/org/apache/gravitino/cli/commands/LinkModel.java
@@ -101,6 +101,6 @@ public void handle() {
     }
 
     System.out.println(
-        "Link model " + model + " to " + uri + " with alias " + Arrays.toString(alias));
+        "Linked model " + model + " to " + uri + " with aliases " + Arrays.toString(alias));
   }
 }
diff --git a/clients/cli/src/main/java/org/apache/gravitino/cli/commands/RegisterModel.java b/clients/cli/src/main/java/org/apache/gravitino/cli/commands/RegisterModel.java
@@ -22,6 +22,7 @@
 import java.util.Map;
 import org.apache.gravitino.NameIdentifier;
 import org.apache.gravitino.cli.ErrorMessages;
+import org.apache.gravitino.cli.Main;
 import org.apache.gravitino.client.GravitinoClient;
 import org.apache.gravitino.exceptions.ModelAlreadyExistsException;
 import org.apache.gravitino.exceptions.NoSuchCatalogException;
@@ -96,6 +97,7 @@ public void handle() {
       System.out.println("Successful register " + registeredModel.name() + ".");
     } else {
       System.err.println("Failed to register model: " + model + ".");
+      Main.exit(-1);
     }
   }
 }