Add Header provider to Google Batch client

Signed-off-by: Paolo Di Tommaso <paolo.ditommaso@gmail.com>
nextflow-io · Jan 31, 2023 · 2097992 · 2097992
1 parent a7282d6
commit 2097992
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/plugins/nf-google/src/main/nextflow/cloud/google/batch/client/BatchClient.groovy b/plugins/nf-google/src/main/nextflow/cloud/google/batch/client/BatchClient.groovy
@@ -17,6 +17,7 @@
 package nextflow.cloud.google.batch.client
 
 import com.google.api.gax.core.CredentialsProvider
+import com.google.api.gax.rpc.FixedHeaderProvider
 import com.google.auth.Credentials
 import com.google.cloud.batch.v1.BatchServiceClient
 import com.google.cloud.batch.v1.BatchServiceSettings
@@ -64,7 +65,12 @@ class BatchClient {
         final provider = createCredentialsProvider(config)
         if( provider ) {
             log.debug "[GOOGLE BATCH] Creating service client with config credentials"
-            final settings = BatchServiceSettings.newBuilder().setCredentialsProvider(provider).build()
+            final userAgent = FixedHeaderProvider.create('user-agent', 'Nextflow')
+            final settings = BatchServiceSettings
+                            .newBuilder()
+                            .setHeaderProvider(userAgent)
+                            .setCredentialsProvider(provider)
+                            .build()
             return BatchServiceClient.create(settings)
         }
         else {