如何使用 picocli 处理多类型选项

问题描述

我正在将现有应用程序转换为使用 picocli。现有选项之一如下所示：

-t,--threads           [1,n] for fixed thread pool,'cpus' for number of cpus,'cached' for cached

这允许选项是一个正整数或几个特殊字符串之一。现有代码将其视为字符串，如果它不是特殊字符串之一，则将其传递给 Integer.parseInt。

当然，我可以用 picocli 做同样的事情，但我想知道是否有更好的方法来处理这个问题。例如，允许将多个字段用于同一选项的东西，根据传递的内容填写适当的字段？这也可能允许我对可能的字符串选项使用枚举。

解决方法

一个想法是为此创建一个类，例如 ThreadPoolSize，它封装了固定数值或动态值的枚举。您需要为此数据类型创建自定义转换器。

那么你可以如下定义选项：

@Option(names = { "-t","--threads" },converter = ThreadPoolSizeConverter.class,description = "[1,n] for fixed thread pool,'cpus' for number of cpus,'cached' for cached")
ThreadPoolSize threadPoolSize;

线程池大小和转换器的自定义数据类型可能如下所示：

class ThreadPoolSize {
    enum Dynamic { cpus,cached }

    int fixed = -1;  // if -1,then use the dynamic value
    Dynamic dynamic; // if null,then use the fixed value
}

class ThreadPoolSizeConverter implements CommandLine.ITypeConverter<ThreadPoolSize> {

    @Override
    public ThreadPoolSize convert(String value) throws Exception {
        ThreadPoolSize result = new ThreadPoolSize();
        try {
            result.fixed = Integer.parseInt(value);
            if (result.fixed < 1) {
                throw new CommandLine.TypeConversionException("Invalid value " +
                        value + ": must be 1 or more.");
            }
        } catch (NumberFormatException nan) {
            try {
                result.dynamic = ThreadPoolSize.Dynamic.valueOf(
                        value.toLowerCase(Locale.ENGLISH));
            } catch (IllegalArgumentException ex) {
                throw new CommandLine.TypeConversionException("Invalid value " +
                        value + ": must be one of " + 
                        Arrays.toString(ThreadPoolSize.Dynamic.values()));
            }
        }
        return result;
    }
}