From 5989be591da366e82d1480b4980b315fbb59800b Mon Sep 17 00:00:00 2001 From: homedirectory Date: Thu, 12 May 2022 19:32:23 +0300 Subject: [PATCH] implemented a version of addModifiers in FieldSpec and TypeSpec that accepts an Iterable type --- src/main/java/com/squareup/javapoet/FieldSpec.java | 7 +++++++ src/main/java/com/squareup/javapoet/TypeSpec.java | 7 +++++++ 2 files changed, 14 insertions(+) diff --git a/src/main/java/com/squareup/javapoet/FieldSpec.java b/src/main/java/com/squareup/javapoet/FieldSpec.java index f530d6ecb..99157780a 100644 --- a/src/main/java/com/squareup/javapoet/FieldSpec.java +++ b/src/main/java/com/squareup/javapoet/FieldSpec.java @@ -158,6 +158,13 @@ public Builder addModifiers(Modifier... modifiers) { return this; } + public Builder addModifiers(Iterable modifiers) { + for (Modifier modifier : modifiers) { + this.modifiers.add(modifier); + } + return this; + } + public Builder initializer(String format, Object... args) { return initializer(CodeBlock.of(format, args)); } diff --git a/src/main/java/com/squareup/javapoet/TypeSpec.java b/src/main/java/com/squareup/javapoet/TypeSpec.java index 6cd2ea636..7e77ae418 100644 --- a/src/main/java/com/squareup/javapoet/TypeSpec.java +++ b/src/main/java/com/squareup/javapoet/TypeSpec.java @@ -474,6 +474,13 @@ public Builder addModifiers(Modifier... modifiers) { return this; } + public Builder addModifiers(Iterable modifiers) { + for (Modifier modifier : modifiers) { + this.modifiers.add(modifier); + } + return this; + } + public Builder addTypeVariables(Iterable typeVariables) { checkArgument(typeVariables != null, "typeVariables == null"); for (TypeVariableName typeVariable : typeVariables) {