feat(agent): send User-Agent

RisingStack · Sep 11, 2017 · d1f4554 · d1f4554
1 parent 13459ea
commit d1f4554
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 6 deletions.
diff --git a/lib/agent/api/index.js b/lib/agent/api/index.js
@@ -14,6 +14,7 @@ var BufferStream = require('./bufferStream')
 var bl = require('bl')
 var libPackage = require('../../../package')
 var zlib = require('zlib')
+var userAgent = 'trace/' + libPackage.version
 
 function CollectorApi (options) {
   this.COLLECTOR_API_SERVICE = url.resolve(options.collectorApiUrl, options.collectorApiServiceEndpoint)
@@ -49,7 +50,8 @@ CollectorApi.prototype._sendSync = function (destinationUrl, data, options) {
         'Authorization': 'Bearer ' + this.apiKey,
         'Content-Type': 'application/json',
         'X-Reporter-Version': libPackage.version,
-        'X-Reporter-Language': this.collectorLanguage
+        'X-Reporter-Language': this.collectorLanguage,
+        'User-Agent': userAgent
       }, options && options.headers),
       timeout: 1000
     })
@@ -77,7 +79,8 @@ CollectorApi.prototype._send = function (destinationUrl, data, callback, options
     'Authorization': 'Bearer ' + this.apiKey,
     'Content-Type': 'application/json',
     'X-Reporter-Version': libPackage.version,
-    'X-Reporter-Language': this.collectorLanguage
+    'X-Reporter-Language': this.collectorLanguage,
+    'User-Agent': userAgent
   }
 
   assign(headers, options.headers)
@@ -281,7 +284,8 @@ CollectorApi.prototype.getService = function (cb) {
       'Content-Type': 'application/json',
       'X-Reporter-Version': libPackage.version,
       'X-Reporter-Language': this.collectorLanguage,
-      'Content-Length': Buffer.byteLength(payload)
+      'Content-Length': Buffer.byteLength(payload),
+      'User-Agent': userAgent
     }
   }, function (res) {
     res.setEncoding('utf8')

diff --git a/lib/agent/api/index.spec.js b/lib/agent/api/index.spec.js
@@ -181,7 +181,8 @@ describe('The Trace CollectorApi module', function () {
         'Authorization': 'Bearer testApiKey',
         'Content-Type': 'application/json',
         'X-Reporter-Version': libPackage.version,
-        'X-Reporter-Language': defaultConfig.collectorLanguage
+        'X-Reporter-Language': defaultConfig.collectorLanguage,
+        'User-Agent': 'trace/' + libPackage.version
       }
     })
       .post(path, function (body) {
@@ -228,7 +229,8 @@ describe('The Trace CollectorApi module', function () {
         'Authorization': 'Bearer testApiKey',
         'Content-Type': 'application/json',
         'X-Reporter-Version': libPackage.version,
-        'X-Reporter-Language': defaultConfig.collectorLanguage
+        'X-Reporter-Language': defaultConfig.collectorLanguage,
+        'User-Agent': 'trace/' + libPackage.version
       }
     })
       .post(defaultConfig.collectorApiServiceEndpoint, JSON.stringify(data))
@@ -271,7 +273,8 @@ describe('The Trace CollectorApi module', function () {
         'Authorization': 'Bearer testApiKey',
         'Content-Type': 'application/json',
         'X-Reporter-Version': libPackage.version,
-        'X-Reporter-Language': defaultConfig.collectorLanguage
+        'X-Reporter-Language': defaultConfig.collectorLanguage,
+        'User-Agent': 'trace/' + libPackage.version
       }
     })
       .post(defaultConfig.collectorApiServiceEndpoint, JSON.stringify(data))