Add Environment & Services support

lib/api/index.js

@@ -6,7 +6,8 @@ const process = require("process"),
   pkg = require("../../package.json"),
   debug = require("debug")(`${pkg.name}:event`),
   LibhoneyImpl = require("./libhoney"),
-  MockImpl = require("./mock");
+  MockImpl = require("./mock"),
+  utils = require("../util");
 
 const { honeycomb, aws, w3c, util } = propagation;
 
@@ -38,6 +39,9 @@ module.exports = {
     }
     debug(`using impl: ${impl == LibhoneyImpl ? "libhoney-event" : "mock"}`);
     apiImpl = new impl(opts);
+
+    // tell honeycomb propagator whether we should propagate dataset
+    honeycomb.setPropagateDataset(utils.isClassic(opts.writeKey));
   },
 
   traceActive() {

lib/api/index.test.js

@@ -10,17 +10,40 @@ jest.mock("../deterministic_sampler");
 
 beforeEach(() => {
   api._resetForTesting();
-  api.configure({ impl: "libhoney-event", transmission: "mock", writeKey: "abc123" });
+  api.configure({
+    impl: "libhoney-event",
+    transmission: "mock",
+    writeKey: "e38be416d0d68f9ed1e96432ac1a3380",
+  });
 });
 
-test("libhoney default config", () => {
+test("libhoney default config - classic", () => {
   const honey = api._apiForTesting().honey;
   expect(honey.transmission.constructorArg.apiHost).toBe("https://api.honeycomb.io");
   expect(honey.transmission.constructorArg.dataset).toBe("nodejs");
-  expect(honey.transmission.constructorArg.writeKey).toBe("abc123");
+  expect(honey.transmission.constructorArg.writeKey).toBe("e38be416d0d68f9ed1e96432ac1a3380");
+  expect(honey.transmission.constructorArg.userAgentAddition).toBe(
+    `honeycomb-beeline/${pkg.version}`
+  );
+  expect(honey._builder._fields["service_name"]).toBe("unknown_service:nodejs");
+});
+
+test("libhoney default config - non-classoc", () => {
+  api._resetForTesting();
+  api.configure({
+    impl: "libhoney-event",
+    transmission: "mock",
+    writeKey: "d68f9ed1e96432ac1a3380",
+  });
+
+  const honey = api._apiForTesting().honey;
+  expect(honey.transmission.constructorArg.apiHost).toBe("https://api.honeycomb.io");
+  expect(honey.transmission.constructorArg.dataset).toBe("unknown_service");
+  expect(honey.transmission.constructorArg.writeKey).toBe("d68f9ed1e96432ac1a3380");
   expect(honey.transmission.constructorArg.userAgentAddition).toBe(
     `honeycomb-beeline/${pkg.version}`
   );
+  expect(honey._builder._fields["service_name"]).toBe("unknown_service:nodejs");
 });
 
 test("startTrace starts tracking and creates an initial event, finishTrace sends it", () => {

lib/api/libhoney.js

@@ -10,7 +10,8 @@ const libhoney = require("libhoney"),
   schema = require("../schema"),
   Span = require("./span"),
   pkg = require("../../package.json"),
-  debug = require("debug")(`${pkg.name}:event`);
+  debug = require("debug")(`${pkg.name}:event`),
+  util = require("../util");
 
 const defaultName = "nodejs";
 
@@ -81,7 +82,38 @@ module.exports = class LibhoneyEventAPI {
       debug(`using proxy ${proxy}`);
     }
 
-    this.defaultDataset = opts.dataset || process.env["HONEYCOMB_DATASET"] || defaultName;
+    if (opts.serviceName) {
+      opts.serviceName.trim();
+    } else {
+      console.warn("empty serviceName configuration option");
+      opts.serviceName = ["unknown_service", process.name ? process.name.trim() : "nodejs"].join(
+        ":"
+      );
+    }
+
+    if (!opts.writeKey) {
+      console.warn("empty writeKey configuration option");
+    }
+
+    if (util.isClassic(opts.writeKey)) {
+      let dataset = opts.dataset || process.env["HONEYCOMB_DATASET"];
+      if (!dataset) {
+        console.warn("empty dataset configuration option");
+      } else {
+        dataset = dataset.trim();
+      }
+
+      this.defaultDataset = dataset ? dataset : defaultName;
+    } else {
+      if (opts.dataset) {
+        console.warn("dataset should be empty - using service name instead");
+      }
+
+      // don't use process name for unknown services
+      this.defaultDataset = opts.serviceName.startsWith("unknown_service")
+        ? "unknown_service"
+        : opts.serviceName;
+    }
 
     const libhoneyOpts = getFilteredOptions(opts);
 

lib/propagation/honeycomb.js

@@ -9,6 +9,13 @@ exports.TRACE_HTTP_HEADER = TRACE_HTTP_HEADER;
 const VERSION = "1";
 exports.VERSION = VERSION;
 
+let propagateDataset = true;
+
+exports.setPropagateDataset = setPropagateDataset;
+function setPropagateDataset(enabled) {
+  propagateDataset = enabled;
+}
+
 // assumes a header of the form:
 
 // VERSION;PAYLOAD
@@ -38,7 +45,7 @@ function marshalTraceContextv1(context) {
   let dataset = context.dataset;
 
   let datasetClause = "";
-  if (dataset) {
+  if (propagateDataset && dataset) {
     datasetClause = `dataset=${encodeURIComponent(dataset)},`;
   }
 
@@ -84,7 +91,9 @@ function unmarshalTraceContextv1(payload) {
         parentSpanId = v;
         break;
       case "dataset":
-        dataset = decodeURIComponent(v);
+        if (propagateDataset) {  
+          dataset = decodeURIComponent(v);
+        }
         break;
       case "context":
         contextb64 = v;

lib/propagation/honeycomb.test.js

@@ -80,7 +80,7 @@ cases(
 );
 
 describe("roundtrip", () => {
-  test("works", () => {
+  test("classic", () => {
     let contextStr = honeycomb.marshalTraceContext(testContext);
     expect(honeycomb.unmarshalTraceContext(contextStr)).toEqual({
       traceId: "abcdef123456",
@@ -93,4 +93,18 @@ describe("roundtrip", () => {
       },
     });
   });
+
+  test("non-classic", () => {
+    honeycomb.setPropagateDataset(false);
+    let contextStr = honeycomb.marshalTraceContext(testContext);
+    expect(honeycomb.unmarshalTraceContext(contextStr)).toEqual({
+      traceId: "abcdef123456",
+      parentSpanId: "0102030405",
+      customContext: {
+        userID: 1,
+        errorMsg: "failed to sign on",
+        toRetry: true,
+      },
+    });
+  });
 });

lib/util.js

@@ -17,3 +17,9 @@ function captureStackTrace(skipFrames = 0, limitFrames = 10) {
   // the +1 here to get rid of the `Error\n` line at the top of the stacktrace.
   return frames.slice(1 + skipFrames).join("\n");
 }
+
+exports.isClassic = isClassic;
+
+function isClassic(writeKey) {
+  return !writeKey || writeKey.length == 32;
+}