lanyuanxiaoyao/hudi
1
0
Fork 0
Code Activity

[HUDI-709] Add unit test for UtilsCommand (#1686)

Browse Source
This commit is contained in:
hongdd
2020-06-18 19:54:14 +08:00
committed by GitHub
parent 2a04647f5e
commit 5099a91edd
2 changed files with 88 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
hudi-cli/src/main/java/org/apache/hudi/cli/commands/UtilsCommand.java
View File

@@ -18,6 +18,7 @@
package org.apache.hudi.cli.commands;
import org.apache.hudi.common.util.StringUtils;
import org.springframework.shell.core.CommandMarker;
import org.springframework.shell.core.annotation.CliCommand;
import org.springframework.shell.core.annotation.CliOption;
@@ -30,9 +31,15 @@ import org.springframework.stereotype.Component;
public class UtilsCommand implements CommandMarker {
@CliCommand(value = "utils loadClass", help = "Load a class")
public String loadClass(@CliOption(key = {"class"}, help = "Check mode") final String clazz) throws Exception {
Class klass = Class.forName(clazz);
return klass.getProtectionDomain().getCodeSource().getLocation().toExternalForm();
public String loadClass(@CliOption(key = {"class"}, help = "Check mode") final String clazz) {
if (StringUtils.isNullOrEmpty(clazz)) {
return "Class to be loaded can not be null!";
}
try {
Class klass = Class.forName(clazz);
return klass.getProtectionDomain().getCodeSource().getLocation().toExternalForm();
} catch (ClassNotFoundException e) {
return String.format("Class %s not found!", clazz);
}
}
}