Capture HTTP requests after daemon address is obained

Author: juffalow

src/index.ts

@@ -35,6 +35,7 @@ async function start(): Promise<void> {
       const address = await services.ServiceDiscovery.discoverInstance(namespace, name);
 
       (services.Trace as any).setDaemonAddress(address);
+      (services.Trace as any).captureHTTPRequests();
     }
 
     server.listen(config.port, () => {

src/services/trace/AWSXRay.ts

@@ -5,9 +5,6 @@ import https from 'https';
 class AWSXRay implements Services.Trace {
   constructor (plugins: string) {
     this.setPlugins(plugins);
-
-    XRay.captureHTTPsGlobal(http, true);
-    XRay.captureHTTPsGlobal(https, true);
   }
 
   public openSegment(defaultName: string) {
@@ -63,6 +60,11 @@ class AWSXRay implements Services.Trace {
 
     XRay.config(xrayPlugins); 
   }
+
+  public captureHTTPRequests(): void {
+    XRay.captureHTTPsGlobal(http, true);
+    XRay.captureHTTPsGlobal(https, true);
+  }
 }
 
 export default AWSXRay;

src/worker.ts

@@ -76,6 +76,7 @@ async function start() {
     const address = await services.ServiceDiscovery.discoverInstance(namespace, name);
 
     (services.Trace as any).setDaemonAddress(address);
+    (services.Trace as any).captureHTTPRequests();
   }
 
   logger.info('Upload worker is running');