Set entityId on flag evaluation

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@eppo/js-client-sdk-common",
-  "version": "4.13.0",
+  "version": "4.13.1",
   "description": "Common library for Eppo JavaScript SDKs (web, react native, and node)",
   "main": "dist/index.js",
   "files": [

src/client/eppo-client.spec.ts

@@ -79,6 +79,7 @@ describe('EppoClient E2E test', () => {
   const mockFlag: Flag = {
     key: flagKey,
     enabled: true,
+    entityId: 123,
     variationType: VariationType.STRING,
     variations: { a: variationA, b: variationB },
     allocations: [
@@ -195,6 +196,16 @@ describe('EppoClient E2E test', () => {
       client.setAssignmentLogger(mockLogger);
       expect(td.explain(mockLogger.logAssignment).callCount).toEqual(MAX_EVENT_QUEUE_SIZE);
     });
+
+    it('should log assignment event with entityId', () => {
+      const mockLogger = td.object<IAssignmentLogger>();
+      const client = new EppoClient({ flagConfigurationStore: storage });
+      client.setAssignmentLogger(mockLogger);
+      client.getStringAssignment(flagKey, 'subject-to-be-logged', {}, 'default-value');
+      expect(td.explain(mockLogger.logAssignment).callCount).toEqual(1);
+      const loggedAssignmentEvent = td.explain(mockLogger.logAssignment).calls[0].args[0];
+      expect(loggedAssignmentEvent.entityId).toEqual(123);
+    });
   });
 
   describe('check type match', () => {

src/client/eppo-client.ts

@@ -1287,7 +1287,7 @@ export default class EppoClient {
     });
   }
 
-  private maybeLogAssignment(result: FlagEvaluation & { entityId?: number }) {
+  private maybeLogAssignment(result: FlagEvaluation) {
     const {
       flagKey,
       format,

src/client/eppo-precomputed-client.ts

@@ -237,6 +237,7 @@ export default class EppoPrecomputedClient {
       allocationKey: precomputedFlag.allocationKey ?? '',
       extraLogging: precomputedFlag.extraLogging ?? {},
       doLog: precomputedFlag.doLog,
+      entityId: null,
     };
 
     try {

src/evaluator.ts

@@ -30,6 +30,7 @@ export interface FlagEvaluationWithoutDetails {
   extraLogging: Record<string, string>;
   // whether to log assignment event
   doLog: boolean;
+  entityId: number | null;
 }
 
 export interface FlagEvaluation extends FlagEvaluationWithoutDetails {
@@ -122,6 +123,7 @@ export class Evaluator {
                 extraLogging: split.extraLogging ?? {},
                 doLog: allocation.doLog,
                 flagEvaluationDetails,
+                entityId: flag.entityId ?? null,
               };
             }
           }
@@ -227,6 +229,7 @@ export function noneResult(
     extraLogging: {},
     doLog: false,
     flagEvaluationDetails,
+    entityId: null,
   };
 }
 
@@ -288,5 +291,6 @@ export function overrideResult(
     format: '',
     allocationKey: overrideAllocationKey,
     extraLogging: {},
+    entityId: null,
   };
 }