jersey2 allow headers overrides - PLUS 498cef428

swagger-api · Aug 8, 2024 · 0c5872a · 0c5872a
1 parent 2c5329a
commit 0c5872a
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/modules/swagger-codegen/src/main/resources/Java/libraries/jersey2/api.mustache b/modules/swagger-codegen/src/main/resources/Java/libraries/jersey2/api.mustache
@@ -27,6 +27,7 @@ import java.util.Map;
 {{#operations}}
 public class {{classname}} {
   private ApiClient {{localVariablePrefix}}apiClient;
+  private Map<String, String> headers;
 
   public {{classname}}() {
     this(Configuration.getDefaultApiClient());
@@ -44,6 +45,10 @@ public class {{classname}} {
     this.{{localVariablePrefix}}apiClient = apiClient;
   }
 
+  public void setHeadersOverrides(Map<String, String> headers) {
+    this.headers = headers;
+  }
+
   {{#operation}}
   /**
    * {{summary}}
@@ -138,8 +143,18 @@ public class {{classname}} {
 
     {{#returnType}}
     GenericType<{{{returnType}}}> {{localVariablePrefix}}localVarReturnType = new GenericType<{{{returnType}}}>() {};
+
+    if (headers != null) {
+      localVarHeaderParams.putAll(headers);
+    }
+
     return {{localVariablePrefix}}apiClient.invokeAPI({{localVariablePrefix}}localVarPath, "{{httpMethod}}", {{localVariablePrefix}}localVarQueryParams, {{localVariablePrefix}}localVarPostBody, {{localVariablePrefix}}localVarHeaderParams, {{localVariablePrefix}}localVarFormParams, {{localVariablePrefix}}localVarAccept, {{localVariablePrefix}}localVarContentType, {{localVariablePrefix}}localVarAuthNames, {{localVariablePrefix}}localVarReturnType);
     {{/returnType}}{{^returnType}}
+
+    if (headers != null) {
+      localVarHeaderParams.putAll(headers);
+    }
+
     return {{localVariablePrefix}}apiClient.invokeAPI({{localVariablePrefix}}localVarPath, "{{httpMethod}}", {{localVariablePrefix}}localVarQueryParams, {{localVariablePrefix}}localVarPostBody, {{localVariablePrefix}}localVarHeaderParams, {{localVariablePrefix}}localVarFormParams, {{localVariablePrefix}}localVarAccept, {{localVariablePrefix}}localVarContentType, {{localVariablePrefix}}localVarAuthNames, null);
     {{/returnType}}
   }

diff --git a/modules/swagger-codegen/src/main/resources/Java/libraries/okhttp-gson/api.mustache b/modules/swagger-codegen/src/main/resources/Java/libraries/okhttp-gson/api.mustache
@@ -58,6 +58,7 @@ import java.util.Map;
 {{#operations}}
 public class {{classname}} {
     private ApiClient {{localVariablePrefix}}apiClient;
+    private Map<String, String> headers;
 
     public {{classname}}() {
         this(Configuration.getDefaultApiClient());
@@ -75,6 +76,10 @@ public class {{classname}} {
         this.{{localVariablePrefix}}apiClient = apiClient;
     }
 
+    public void setHeadersOverrides(Map<String, String> headers) {
+        this.headers = headers;
+    }
+
     {{#operation}}
     /**
      * Build call for {{operationId}}{{#allParams}}
@@ -139,6 +144,9 @@ public class {{classname}} {
         }
 
         String[] {{localVariablePrefix}}localVarAuthNames = new String[] { {{#authMethods}}"{{name}}"{{#hasMore}}, {{/hasMore}}{{/authMethods}} };
+        if (headers != null) {
+            localVarHeaderParams.putAll(headers);
+        }
         return {{localVariablePrefix}}apiClient.buildCall({{localVariablePrefix}}localVarPath, "{{httpMethod}}", {{localVariablePrefix}}localVarQueryParams, {{localVariablePrefix}}localVarCollectionQueryParams, {{localVariablePrefix}}localVarPostBody, {{localVariablePrefix}}localVarHeaderParams, {{localVariablePrefix}}localVarFormParams, {{localVariablePrefix}}localVarAuthNames, progressRequestListener);
     }