22 createInstance ,
33 enums as OptimizelyEnums ,
44} from '@optimizely/optimizely-sdk/dist/optimizely.lite.es' ;
5- import { getCookie , setHeaderCookie } from './cookies' ;
5+
6+ import * as cookie from 'cookie' ;
67
78import {
89 dispatchEvent ,
@@ -36,8 +37,10 @@ exports.handler = async (event, _context, callback) => {
3637 }
3738
3839 let headers = { } ;
40+ let cookies = { } ;
3941 try {
4042 headers = request . headers ;
43+ cookies = headers . cookie ;
4144 } catch ( error ) {
4245 console . log (
4346 '[OPTIMIZELY] WARNING: Unable to get headers object from request object. This may be because the event is not a CloudFront event.'
@@ -47,14 +50,14 @@ exports.handler = async (event, _context, callback) => {
4750 try {
4851 // 1. User ID: Get the user ID from the cookie if it exists - otherwise generate a new user ID.
4952 console . log ( '[OPTIMIZELY] Getting user ID...' ) ;
50- let userId = getCookie ( headers , COOKIE_NAME_OPTIMIZELY_USER_ID ) ;
53+ let userId = cookies [ COOKIE_NAME_OPTIMIZELY_USER_ID ] || '' ;
5154
52- if ( ! userId ) {
55+ if ( userId === '' ) {
5356 userId = generateRandomUserId ( ) ;
54- headers = setHeaderCookie (
55- headers ,
56- ` ${ COOKIE_NAME_OPTIMIZELY_USER_ID } = ${ userId } `
57- ) ;
57+ headers = {
58+ ... headers ,
59+ 'Set-Cookie' : cookie . serialize ( COOKIE_NAME_OPTIMIZELY_USER_ID , userId ) ,
60+ } ;
5861 }
5962
6063 console . log ( `[OPTIMIZELY] Using User ID: ${ userId } ) ;
@@ -77,12 +80,12 @@ exports.handler = async (event, _context, callback) => {
7780 * Optional event dispatcher. Please uncomment the following line if you want to dispatch an impression event to optimizely logx backend.
7881 * When enabled, an event is dispatched asynchronously. It does not impact the response time for a particular worker but it may
7982 * add to the total compute time of the Lambda function and can impact AWS billing.
80- *
83+ *
8184 * The event dispatcher attached below is a sample implementation of fire-and-forget event dispatching
8285 * in Lambda, however if your needs are more complex, you can implement your own event dispatcher
8386 * and integrate with Step Functions, SQS, or other services.
8487 */
85- // eventDispatcher: dispatchEvent,
88+ // eventDispatcher: { dispatchEvent } ,
8689 } ) ;
8790
8891 // 4. Experimentation Logic: Get a specific experiment result for this particular User ID.
@@ -106,7 +109,8 @@ exports.handler = async (event, _context, callback) => {
106109 // === For a Single Flag === //
107110 const decision = optimizelyUserContext . decide ( '<YOUR_FLAG_HERE>' ) ; // TODO: Replace with your flag name.
108111 console . log (
109- `[OPTIMIZELY] The Flag ${ decision . flagKey } ${ decision . enabled ? 'Enabled' : 'Not Enabled'
112+ `[OPTIMIZELY] The Flag ${ decision . flagKey } ${
113+ decision . enabled ? 'Enabled' : 'Not Enabled'
110114 } for the user ${ decision . userContext . getUserId ( ) }
111115 ) ;
112116
@@ -134,7 +138,10 @@ exports.handler = async (event, _context, callback) => {
134138 // 5. Result: Return the result to the caller via appending headers or cookies to the callback function.
135139
136140 // 5a. Cookies
137- // headers = setHeaderCookie(headers, `${COOKIE_NAME_OPTIMIZELY_USER_ID}=${userId}`);
141+ // headers = {
142+ // ...headers,
143+ // 'Set-Cookie': cookie.serialize(COOKIE_NAME_OPTIMIZELY_USER_ID, userId)
144+ // }
138145
139146 // 5b. Headers
140147 // headers['<YOUR_FLAG_HERE>-decision'] = [
0 commit comments