Search code examples
aws-lambdaamazon-cloudfrontaws-lambda-edge

Lambda@Edge routing between two origins based on cookie


Can some please tell me how to route to 2 different origins based on cookies using Lambda@Edge? I tried below code, but it is not working. Thanks in advance.

 'use strict';

exports.handler = (event, context, callback) => {
    const request = event.Records[0].cf.request;
    const headers = request.headers;
    const origin = request.origin;

    //Setup the two different origins
    const originA = "site1.example.com";
    const originB = "site2.example.com";


    //Determine whether the user has visited before based on a cookie value
    //Grab the 'origin' cookie if it's been set before
    if (headers.cookie) {
        for (let i = 0; i < headers.cookie.length; i++) {
            if (headers.cookie[i].value.indexOf('origin=A') >= 0) {
                console.log('Origin A cookie found');
                headers['host'] = [{key: 'host',          value: originA}];
                origin.s3.domainName = originA;
                break;
            } else if (headers.cookie[i].value.indexOf('origin=B') >= 0) {
                console.log('Origin B cookie found');
                headers['host'] = [{key: 'host',          value: originB}];
                origin.s3.domainName = originB;
                break;
            }
        }
    }    

    callback(null, request);
};

Solution

  • It looks like you are not using an s3 origin. Changing s3 to custom in your code will fix the problem.

    origin.custom.domainName = originA;
    origin.custom.domainName = originB;