Javapoet:用更少的代码行编写同样的东西(addModifiers)

Question

所以，我有这段代码(使用Javapoet Lib(：

if (myBeautifulBoolean) <--------------------------
                            theClass = TypeSpec.classBuilder(classe.getName())
                                    .addModifiers(javax.lang.model.element.Modifier.valueOf(classe.getProte().toString().toUpperCase()),  Modifier.FINAL) <-------------------
                                    .superclass(father==null?ClassName.OBJECT:father)
                                    .addMethods(methods)
                                    .addFields(fields)
                                    .build();
                            else
                                theClass = TypeSpec.classBuilder(classe.getName())
                                        .addModifiers(javax.lang.model.element.Modifier.valueOf(classe.getProte().toString().toUpperCase())) <------------------
                                        .superclass(father==null?ClassName.OBJECT:father)
                                        .addMethods(methods)
                                        .addFields(fields)
                                        .build();

我希望它变成这样：

                                theClass = TypeSpec.classBuilder(classe.getName())
                                    .addModifiers(javax.lang.model.element.Modifier.valueOf(classe.getProte().toString().toUpperCase()),  myBeautifulBoolean?Modifier.FINAL:null) <----------
                                    .superclass(father==null?ClassName.OBJECT:father)
                                    .addMethods(methods)
                                    .addFields(fields)
                                    .build();

问题出在哪里？如果我写我的美丽布尔值？Modifier.FINAL：null，我得到一个异常，因为addmodifiers((的参数不能为空，而且没有什么比Modifier.NOTFINAL更像了

那么，有没有办法告诉代码"Ehi，如果布尔值为真，则添加一个参数，如果不是，则不要"？

Answer 1

addModifiers需要一个数组。 你可以做addModifiers(test ? new Modifier[] { mod, Modifier.FINAL} : new Modifier[] { mod })你可以用一个辅助方法让它更漂亮

public static <T> T[] arr(T... array) { return array; }
// later
    .addModifiers(test ? arr(mod, FINAL) : arr(mod))