feat(generator): emit ResumableUploadCallSettings on stub and service settings

Author: whowes

sdk-platform-java/gapic-generator-java/src/main/java/com/google/api/generator/gapic/composer/comment/SettingsCommentComposer.java

@@ -41,13 +41,19 @@ public class SettingsCommentComposer {
   private static final String CLASS_HEADER_DEFAULT_ADDRESS_PORT_PATTERN =
       "The default service address (%s) and default port (%d) are used.";
   private static final String CLASS_HEADER_SAMPLE_CODE_PATTERN =
-      "For example, to set the [RetrySettings](https://cloud.google.com/java/docs/reference/gax/latest/com.google.api.gax.retrying.RetrySettings) of %s:";
+      "For example, to set the"
+          + " [RetrySettings](https://cloud.google.com/java/docs/reference/gax/latest/com.google.api.gax.retrying.RetrySettings)"
+          + " of %s:";
 
   private static final String CLASS_HEADER_LRO_SAMPLE_CODE_PATTERN =
-      "To configure the RetrySettings of a Long Running Operation method, create an OperationTimedPollAlgorithm object and update the RPC's polling algorithm. For example, to configure the RetrySettings for %s:";
+      "To configure the RetrySettings of a Long Running Operation method, create an"
+          + " OperationTimedPollAlgorithm object and update the RPC's polling algorithm. For"
+          + " example, to configure the RetrySettings for %s:";
 
   private static final String CLASS_HEADER_SAMPLE_CODE_SUFFIX =
-      "Please refer to the [Client Side Retry Guide](https://docs.cloud.google.com/java/docs/client-retries) for additional support in setting retries.";
+      "Please refer to the [Client Side Retry"
+          + " Guide](https://docs.cloud.google.com/java/docs/client-retries) for additional support"
+          + " in setting retries.";
 
   private static final String CLASS_HEADER_BUILDER_DESCRIPTION =
       "The builder of this class is recursive, so contained classes are themselves builders. When"
@@ -130,6 +136,23 @@ public static CommentStatement createCallSettingsGetterComment(
         isMethodInternal);
   }
 
+  public static CommentStatement createResumableUploadCallSettingsGetterComment(
+      String javaMethodName, boolean isMethodDeprecated, boolean isMethodInternal) {
+    JavaDocComment.Builder docBuilder =
+        JavaDocComment.builder()
+            .addComment(String.format(CALL_SETTINGS_METHOD_DOC_PATTERN, javaMethodName))
+            .addParagraph(
+                "Note that custom retry settings and headers configured via ApiCallContext"
+                    + " apply strictly to the initial session initiation request.");
+    if (isMethodDeprecated) {
+      docBuilder.setDeprecated(CommentComposer.DEPRECATED_METHOD_STRING);
+    }
+    if (isMethodInternal) {
+      docBuilder.setInternalOnly(CommentComposer.INTERNAL_ONLY_METHOD_STRING);
+    }
+    return CommentStatement.withComment(docBuilder.build());
+  }
+
   public static CommentStatement createBuilderClassComment(String outerClassName) {
     return toCommentStatement(String.format(BUILDER_CLASS_DOC_PATTERN, outerClassName));
   }
@@ -140,6 +163,23 @@ public static CommentStatement createCallSettingsBuilderGetterComment(
     return toCommentStatement(methodComment, isMethodDeprecated, isMethodInternal);
   }
 
+  public static CommentStatement createResumableUploadCallSettingsBuilderGetterComment(
+      String javaMethodName, boolean isMethodDeprecated, boolean isMethodInternal) {
+    JavaDocComment.Builder docBuilder =
+        JavaDocComment.builder()
+            .addComment(String.format(CALL_SETTINGS_BUILDER_METHOD_DOC_PATTERN, javaMethodName))
+            .addParagraph(
+                "Note that custom retry settings and headers configured via ApiCallContext"
+                    + " apply strictly to the initial session initiation request.");
+    if (isMethodDeprecated) {
+      docBuilder.setDeprecated(CommentComposer.DEPRECATED_METHOD_STRING);
+    }
+    if (isMethodInternal) {
+      docBuilder.setInternalOnly(CommentComposer.INTERNAL_ONLY_METHOD_STRING);
+    }
+    return CommentStatement.withComment(docBuilder.build());
+  }
+
   public static List<CommentStatement> createClassHeaderComments(
       String configuredClassName,
       String defaultHost,

sdk-platform-java/gapic-generator-java/src/main/java/com/google/api/generator/gapic/composer/common/AbstractServiceSettingsClassComposer.java

@@ -25,6 +25,7 @@
 import com.google.api.gax.rpc.ClientSettings;
 import com.google.api.gax.rpc.OperationCallSettings;
 import com.google.api.gax.rpc.PagedCallSettings;
+import com.google.api.gax.rpc.ResumableUploadCallSettings;
 import com.google.api.gax.rpc.ServerStreamingCallSettings;
 import com.google.api.gax.rpc.StreamingCallSettings;
 import com.google.api.gax.rpc.StubSettings;
@@ -311,12 +312,18 @@ private static List<MethodDefinition> createSettingsGetterMethods(
   // Add method header comment statements and annotations.
   private static MethodDefinition methodBuilderHelper(
       Method protoMethod, MethodDefinition.Builder methodBuilder, String javaMethodName) {
-    return methodBuilder
-        .setHeaderCommentStatements(
-            SettingsCommentComposer.createCallSettingsGetterComment(
+    CommentStatement commentStatement =
+        protoMethod.isResumableUpload()
+            ? SettingsCommentComposer.createResumableUploadCallSettingsGetterComment(
+                getMethodNameFromSettingsVarName(javaMethodName),
+                protoMethod.isDeprecated(),
+                protoMethod.isInternalApi())
+            : SettingsCommentComposer.createCallSettingsGetterComment(
                 getMethodNameFromSettingsVarName(javaMethodName),
                 protoMethod.isDeprecated(),
-                protoMethod.isInternalApi()))
+                protoMethod.isInternalApi());
+    return methodBuilder
+        .setHeaderCommentStatements(commentStatement)
         .setAnnotations(createMethodAnnotations(protoMethod))
         .build();
   }
@@ -794,13 +801,19 @@ private static List<MethodDefinition> createNestedBuilderSettingsGetterMethods(
       String javaMethodName = String.format("%sSettings", javaStyleName);
       MethodDefinition.Builder methodBuilder =
           methodMakerFn.apply(getCallSettingsBuilderType(protoMethod, typeStore), javaMethodName);
+      CommentStatement commentStatement =
+          protoMethod.isResumableUpload()
+              ? SettingsCommentComposer.createResumableUploadCallSettingsBuilderGetterComment(
+                  getMethodNameFromSettingsVarName(javaMethodName),
+                  protoMethod.isDeprecated(),
+                  protoMethod.isInternalApi())
+              : SettingsCommentComposer.createCallSettingsBuilderGetterComment(
+                  getMethodNameFromSettingsVarName(javaMethodName),
+                  protoMethod.isDeprecated(),
+                  protoMethod.isInternalApi());
       javaMethods.add(
           methodBuilder
-              .setHeaderCommentStatements(
-                  SettingsCommentComposer.createCallSettingsBuilderGetterComment(
-                      getMethodNameFromSettingsVarName(javaMethodName),
-                      protoMethod.isDeprecated(),
-                      protoMethod.isInternalApi()))
+              .setHeaderCommentStatements(commentStatement)
               .setAnnotations(createMethodAnnotations(protoMethod))
               .build());
 
@@ -856,6 +869,7 @@ private static TypeStore createStaticTypes() {
             Operation.class,
             OperationCallSettings.class,
             PagedCallSettings.class,
+            ResumableUploadCallSettings.class,
             ServerStreamingCallSettings.class,
             StreamingCallSettings.class,
             StubSettings.class,
@@ -933,7 +947,13 @@ private static TypeNode getCallSettingsTypeHelper(
       Method protoMethod, TypeStore typeStore, boolean isBuilder) {
     Class<?> callSettingsClazz =
         isBuilder ? UnaryCallSettings.Builder.class : UnaryCallSettings.class;
-    if (protoMethod.isPaged()) {
+    if (protoMethod.isResumableUpload()) {
+      return TypeNode.withReference(
+          ConcreteReference.withClazz(
+              isBuilder
+                  ? ResumableUploadCallSettings.Builder.class
+                  : ResumableUploadCallSettings.class));
+    } else if (protoMethod.isPaged()) {
       callSettingsClazz = isBuilder ? PagedCallSettings.Builder.class : PagedCallSettings.class;
     } else if (protoMethod.isBatching()) {
       callSettingsClazz =