Fix end trace

Author: eduardoemery

packages/opencensus-core/src/trace/model/tracer.ts

@@ -19,6 +19,7 @@ import {Trace} from './trace'
 import {Span} from './span' 
 import {PluginLoader} from './plugins/pluginloader'
 import {debug} from '../internal/util'
+import {Stackdriver} from '../exporters/stackdriver'
 
 export type Func<T> = (...args: any[]) => T;
 
@@ -29,6 +30,7 @@ export class TraceManager {
     private _active: boolean;
     private contextManager: cls.Namespace;
     private pluginLoader: PluginLoader;
+    private exporter
 
     //TODO: temp solution 
     private endedTraces: Trace[] = [];
@@ -37,6 +39,7 @@ export class TraceManager {
         this._active = false;
         this.contextManager = cls.createNamespace();
         this.pluginLoader = new PluginLoader(this);
+        this.exporter = new Stackdriver('opencensus-cesar');
     }
 
     public get currentTrace(): Trace  {
@@ -64,10 +67,13 @@ export class TraceManager {
         return newTrace;
     }
 
-    public endTrace():void {
+    public endTrace(): void {
         if (!this.currentTrace) return debug('cannot end trace - no active trace found')
         this.currentTrace.end();
+        this.exporter.emit(this.currentTrace);
         this.addEndedTrace(this.currentTrace);
+        //this.clearCurrentTrace();
+        this.startTrace();
     }
 
     public clearCurrentTrace() {
@@ -84,6 +90,12 @@ export class TraceManager {
         return newSpan;
     }
 
+    /*public endSpan(span: Span) {
+        debug('END SPAN', span);
+        span.end();
+        this.exporter.emit(this.currentTrace);
+    }*/
+
 
     private addEndedTrace(trace: Trace) {
         if (this.active) {

src/exporters/stackdriver.ts

@@ -1,8 +1,6 @@
 import {debug} from '../internal/util'
 import {google} from 'googleapis'
 import { JWT } from 'google-auth-library';
-//import { auth } from 'google-auth-library'
-//import { request } from 'http';
 const cloudTrace = google.cloudtrace('v1')
 var uuidv4 = require('uuid/v4');
 
@@ -14,25 +12,28 @@ export class Stackdriver {
         this.projectId = projectId;
     }
 
-    emit() {
+    emit(trace) {
+        debug('CURRENT TRACE ', trace)
         let resource = this.generateResource("cesar-opencensus",
                                 "2018-03-21T19:36:49.074Z",
                                 "2018-03-21T19:36:49.242Z")
-        this.authorize(function(projectId, authClient, resource) {
-            let request = {
-                projectId: projectId,
-                resource: resource,
-                auth: authClient
+        this.authorize(this.sendTrace, resource);
+    }
+
+    sendTrace(projectId, authClient, resource) {
+        let request = {
+            projectId: projectId,
+            resource: resource,
+            auth: authClient
+        }
+        cloudTrace.projects.patchTraces(request, function(err) {
+            if (err) {
+                debug(err);
+                return;
+            } else {
+                debug(JSON.stringify(request.resource));
             }
-            cloudTrace.projects.patchTraces(request, function(err) {
-                if (err) {
-                    debug(err);
-                    return;
-                } else {
-                    debug(JSON.stringify(request.resource));
-                }
-            })
-        }, resource);
+        })
     }
 
     private generateTraceId() {