add key sharing

Author: Ian-Nara

performance-testing/uid2-operator/k6-token-generate-refresh-identitymap-keysharing.js

@@ -5,7 +5,7 @@ import { check } from 'k6';
 import http from 'k6/http';
 
 const vus = 500;
-const baseUrl = "http://uid2-prod-opr-use2-alb-698161474.us-east-2.elb.amazonaws.com";
+const baseUrl = "http://uid2-prod-opr-use2-alb-1657081774.us-east-2.elb.amazonaws.com";
 const clientSecret = "";
 const clientKey = "";
 

performance-testing/uid2-operator/k6-token-generate-refresh-identitymap.js

@@ -5,7 +5,7 @@ import { check } from 'k6';
 import http from 'k6/http';
 
 const vus = 500;
-const baseUrl = "http://uid2-prod-opr-use2-alb-698161474.us-east-2.elb.amazonaws.com";
+const baseUrl = "http://uid2-prod-opr-use2-alb-1657081774.us-east-2.elb.amazonaws.com";
 const clientSecret = "";
 const clientKey = "";
 
@@ -48,6 +48,14 @@ export const options = {
       ],
       gracefulRampDown: '0s',
     },
+    // keySharingWarmup: {
+    //   executor: 'ramping-vus',
+    //   exec: 'keySharing',
+    //   stages: [
+    //     { duration: '30s', target: keySharingVUs}
+    //   ],
+    //   gracefulRampDown: '0s',
+    // },
     // Actual testing scenarios
     tokenGenerate: {
       executor: 'constant-vus',
@@ -73,6 +81,14 @@ export const options = {
       gracefulStop: '0s',
       startTime: '30s',
     }/*,
+    keySharing:{
+      executor: 'constant-vus',
+      exec: 'keySharing',
+      vus: keySharingVUs,
+      duration: testDuration,
+      gracefulStop: '0s',
+      startTime: '30s',
+    }
     identityMapLargeBatchSequential: {
       executor: 'constant-vus',
       exec: 'identityMapLargeBatch',
@@ -213,6 +229,24 @@ export function identityBuckets(data) {
   execute(bucketData, true);
 }
 
+export async function keySharing(data) {
+  const endpoint = '/v2/key/sharing';
+  if (data.keySharing == null) {
+    var newData = await generateKeySharingRequestWithTime();
+    data.keySharing = newData;
+  } else if (data.keySharing.time < (Date.now() - 45000)) {
+    data.keySharing = await generateKeySharingRequestWithTime();
+  }
+
+  var requestBody = data.keySharing.requestBody;
+  var keySharingData = {
+    endpoint: endpoint,
+    requestBody: requestBody,
+  }
+
+  execute(keySharingData, true);
+}
+
 // Helpers
 async function createReqWithTimestamp(timestampArr, obj) {
   var envelope = getEnvelopeWithTimestamp(timestampArr, obj);
@@ -401,6 +435,11 @@ async function generateIdentityMapRequestWithTime(emailCount) {
   return await generateRequestWithTime(emails);
 }
 
+async function generateKeySharingRequestWithTime() {
+  let requestData = { };
+  return await generateRequestWithTime(requestData);
+}
+
 const generateSinceTimestampStr = () => {
   var date = new Date(Date.now() - 2 * 24 * 60 * 60 * 1000 /* 2 days ago */);
   var year = date.getFullYear();

performance-testing/uid2-operator/k6-uid2-operator.js

@@ -48,6 +48,14 @@ export const options = {
       duration: '30s',
       gracefulStop: '0s',
     },
+    keySharingWarmup: {
+      executor: 'ramping-vus',
+      exec: 'keySharing',
+      stages: [
+        { duration: '30s', target: keySharingVUs}
+      ],
+      gracefulRampDown: '0s',
+    },
     // Actual testing scenarios
     tokenGenerate: {
       executor: 'constant-vus',
@@ -73,6 +81,14 @@ export const options = {
       gracefulStop: '0s',
       startTime: '660s',
     },
+    keySharing:{
+      executor: 'constant-vus',
+      exec: 'keySharing',
+      vus: keySharingVUs,
+      duration: testDuration,
+      gracefulStop: '0s',
+      startTime: '30s',
+    },
     identityMapLargeBatchSequential: {
       executor: 'constant-vus',
       exec: 'identityMapLargeBatch',
@@ -237,6 +253,24 @@ export function identityBuckets(data) {
   execute(bucketData, true);
 }
 
+export async function keySharing(data) {
+  const endpoint = '/v2/key/sharing';
+  if (data.keySharing == null) {
+    var newData = await generateKeySharingRequestWithTime();
+    data.keySharing = newData;
+  } else if (data.keySharing.time < (Date.now() - 45000)) {
+    data.keySharing = await generateKeySharingRequestWithTime();
+  }
+
+  var requestBody = data.keySharing.requestBody;
+  var keySharingData = {
+    endpoint: endpoint,
+    requestBody: requestBody,
+  }
+
+  execute(keySharingData, true);
+}
+
 // Helpers
 function selectRequestData(requestData) {
   var elementToUse = requestData[0];
@@ -440,6 +474,11 @@ async function generateFutureGenerateRequests(count) {
   return await generateFutureRequests(count, obj)
 }
 
+async function generateKeySharingRequestWithTime() {
+  let requestData = { };
+  return await generateRequestWithTime(requestData);
+}
+
 const generateSinceTimestampStr = () => {
   var date = new Date(Date.now() - 2 * 24 * 60 * 60 * 1000 /* 2 days ago */);
   var year = date.getFullYear();