zipkin-tracer.js

// FIX: this uses an injected msg property __tracer__ to encode the
// zipkin tracking ids. This is incorrect, the message meta data should be used
// instead. However, it needs to be verified that Seeneca will maintain the meta
// context properly (which it should, but this needs to be tested).


var Zipkin = require('zipkin-simple')
var Tracer = new Zipkin({
  host: '127.0.0.1',
  port: 9411
})
var serverInwardData;
var traceData;

function internal_action (msg) {
  return msg.role === 'seneca' ||
    msg.role === 'transport' ||
    msg.role === 'options' ||
    msg.role === 'mesh' ||
    msg.init
}

function get_pattern(msg, meta) {
  return meta ? meta.pattern : (msg.meta$ ? msg.meta$.pattern : null)
}

function get_plugin(msg, meta) {
  return meta ? meta.plugin_name : (msg.meta$ ? msg.meta$.plugin_name : null)
}


function client_inward (ctx, msg, meta) {
  var service = ctx.seneca.options().tag
  var pin = get_pattern(msg, meta)
 
  var trace_data = Tracer.get_child(serverInwardData || traceData || ctx.seneca.fixedargs.__tracer__)
  serverInwardData = undefined;


  trace_data = Tracer.send_client_send(trace_data, {
    service: service,
    name: pin
  })

   ctx.__tracer__ = msg.__tracer__ = ctx.seneca.fixedargs.__tracer__ = trace_data
}


function server_inward (ctx, msg, meta) {
  var service = ctx.seneca.options().tag
  var pin = get_pattern(msg, meta)

  var trace_data = Tracer.send_server_recv(msg.__tracer__, {
    service: service,
    name: pin
  })

  serverInwardData = traceData = ctx.__tracer__ = msg.__tracer__ = ctx.seneca.fixedargs.__tracer__ = trace_data
}

function client_outward (ctx, msg, meta) {
  var service = ctx.seneca.options().tag
  var pin = get_pattern(msg, meta)
  var trace_data = ctx.__tracer__

  Tracer.send_client_recv(trace_data, {
    service: service,
    name: pin
  })
}

function server_outward (ctx, msg, meta) {
  var service = ctx.seneca.options().tag
  var pin = get_pattern(msg, meta)
  var trace_data = ctx.__tracer__

  Tracer.send_server_send(trace_data, {
    service: service,
    name: pin
  })
}

function zipkin_inward (ctx, data) {
  if (internal_action(data.msg)) {
    return
  }

  var msg = data.msg
  var meta = data.meta
  
  ctx.server = msg.transport$ && get_plugin(msg, meta) !== 'client$'
  if (ctx.server) {
    return server_inward(ctx, msg, meta)
  }

  client_inward(ctx, msg, meta)
}

function zipkin_outward (ctx, data) {
  if (internal_action(data.msg)) {
    return
  }

  if (!ctx.__tracer__) {
    return
  }

  var msg = data.msg
  var meta = data.meta
  
  if (ctx.server) {
    return server_outward(ctx, msg, meta)
  }

  client_outward(ctx, msg, meta)
}

function injectTracerData(data){
	data.sampled = Tracer.opts.sampling === 1;
	serverInwardData = data;

}

function tracer_plugin (options) {
  serverInwardData = null;
  traceData = null;
  if(options.exposeInjectTracerData){
	options.exposeInjectTracerData(injectTracerData);
  }
  Tracer.options(options)
  var seneca = this
  seneca.fixedargs.__tracer__ = null;

  seneca.inward(zipkin_inward)
  seneca.outward(zipkin_outward)
}

module.exports = tracer_plugin