randomise pool choice when no cookie is present

James Prendergast · James Prendergast · commit 04be2980de66 · 2020-02-27T11:30:24.000+11:00
diff --git a/edge-functions/src/originrequest.js b/edge-functions/src/originrequest.js
@@ -6,7 +6,7 @@ const origins = {
 };
 const cookieName = 'pool';
 
-function parseCookies(headers) {
+const parseCookies = (headers) => {
     const parsedCookie = {};
     if (headers.cookie) {
         headers.cookie[0].value.split(';').forEach((cookie) => {
@@ -19,6 +19,8 @@ function parseCookies(headers) {
     return parsedCookie;
 }
 
+const rollTheDice = () => Math.floor(Math.random()*3) === 0 ? 'b' : 'a';
+
 exports.handler = (event, context, callback) => {
     const request = event.Records[0].cf.request;
     const headers = request.headers;
@@ -31,6 +33,8 @@ exports.handler = (event, context, callback) => {
 
     if(parsedCookies && parsedCookies[cookieName]){
         targetPool = parsedCookies[cookieName];
+    }else{
+        targetPool = rollTheDice();
     }
 
     let s3Origin = origins[targetPool];
