mirror of
https://github.com/yu-i-i/overleaf-cep.git
synced 2026-06-08 00:29:04 +02:00
Merge pull request #18018 from overleaf/revert-17906-ab-split-test-assignments-optim-pt2
Revert "[web] Store anonymous users split test assignments in new format in session" GitOrigin-RevId: 2c1a95031a9d1d99b9dfef54eb4b80264a32ba0d
This commit is contained in:
Alexandre Bourdin
committed by
Copybot
Copybot
parent
f929d8d8a8
commit
23da3d55e6
@@ -9,68 +9,63 @@ const SplitTestUtils = require('./SplitTestUtils')
|
||||
const SplitTestUserGetter = require('./SplitTestUserGetter')
|
||||
|
||||
const CACHE_TOMBSTONE_SPLIT_TEST_NOT_ACTIVE_FOR_USER = null
|
||||
const TOKEN_SEP = ';'
|
||||
// this is safe to use as a separator adjacent to a base64 string because Mongo object IDs
|
||||
// do not generate any padding when converted (24 hex digits = 12 bytes => multiple of 6),
|
||||
// thus do not contain any trailing `=`
|
||||
const KEY_VALUE_SEP = '='
|
||||
const ID_VERSION_SEP = '_'
|
||||
const VARIANT_DATE_SEP = ':'
|
||||
|
||||
async function getAssignments(session) {
|
||||
await _convertAnonymousAssignmentsIfNeeded(session)
|
||||
|
||||
if (!session.sta) {
|
||||
if (!session.splitTests && !session.sta) {
|
||||
return undefined
|
||||
}
|
||||
|
||||
const assignments = {}
|
||||
const tokens = session.sta.split(TOKEN_SEP)
|
||||
const splitTests = Array.from((await SplitTestCache.get('')).values())
|
||||
for (const token of tokens) {
|
||||
try {
|
||||
if (!token.length) {
|
||||
continue
|
||||
}
|
||||
const [splitTestNameVersion, info] = token.split(KEY_VALUE_SEP)
|
||||
const [splitTestId64, versionStr] =
|
||||
splitTestNameVersion.split(ID_VERSION_SEP)
|
||||
// await _convertAnonymousAssignmentsIfNeeded(session)
|
||||
const assignments = _.clone(session.splitTests || {})
|
||||
if (session.sta) {
|
||||
const tokens = session.sta.split(';')
|
||||
const splitTests = Array.from((await SplitTestCache.get('')).values())
|
||||
for (const token of tokens) {
|
||||
try {
|
||||
if (!token.length) {
|
||||
continue
|
||||
}
|
||||
const [splitTestNameVersion, info] = token.split('=')
|
||||
const [splitTestId64, versionStr] = splitTestNameVersion.split('_')
|
||||
|
||||
const splitTest = splitTests.find(
|
||||
test => splitTestId64 === _convertIdToBase64(test._id)
|
||||
)
|
||||
if (!splitTest) {
|
||||
continue
|
||||
}
|
||||
const splitTest = splitTests.find(
|
||||
test =>
|
||||
test._id.toString() ===
|
||||
new ObjectId(Buffer.from(splitTestId64, 'base64')).toString()
|
||||
)
|
||||
if (!splitTest) {
|
||||
continue
|
||||
}
|
||||
|
||||
const splitTestName = splitTest.name
|
||||
const versionNumber = parseInt(versionStr)
|
||||
const [variantChar, timestampStr36] = info.split(VARIANT_DATE_SEP)
|
||||
const assignedAt = new Date(parseInt(timestampStr36, 36) * 1000)
|
||||
let variantName
|
||||
if (variantChar === 'd') {
|
||||
variantName = 'default'
|
||||
} else {
|
||||
const variantIndex = parseInt(variantChar)
|
||||
variantName =
|
||||
SplitTestUtils.getCurrentVersion(splitTest).variants[variantIndex]
|
||||
.name
|
||||
}
|
||||
const splitTestName = splitTest.name
|
||||
const versionNumber = parseInt(versionStr)
|
||||
const [variantChar, timestampStr36] = info.split(':')
|
||||
const assignedAt = new Date(parseInt(timestampStr36, 36) * 1000)
|
||||
let variantName
|
||||
if (variantChar === 'd') {
|
||||
variantName = 'default'
|
||||
} else {
|
||||
const variantIndex = parseInt(variantChar)
|
||||
variantName =
|
||||
SplitTestUtils.getCurrentVersion(splitTest).variants[variantIndex]
|
||||
.name
|
||||
}
|
||||
|
||||
if (!assignments[splitTestName]) {
|
||||
assignments[splitTestName] = []
|
||||
if (!assignments[splitTestName]) {
|
||||
assignments[splitTestName] = []
|
||||
}
|
||||
assignments[splitTestName].push({
|
||||
versionNumber,
|
||||
variantName,
|
||||
phase: 'release', // anonymous users can only be exposed to tests in release phase
|
||||
assignedAt,
|
||||
})
|
||||
} catch (error) {
|
||||
logger.error(
|
||||
{ err: error, token },
|
||||
'Failed to resolve anonymous split test assignment from session'
|
||||
)
|
||||
}
|
||||
assignments[splitTestName].push({
|
||||
versionNumber,
|
||||
variantName,
|
||||
phase: 'release', // anonymous users can only be exposed to tests in release phase
|
||||
assignedAt,
|
||||
})
|
||||
} catch (error) {
|
||||
logger.error(
|
||||
{ err: error, token },
|
||||
'Failed to resolve cached anonymous split test assignments from session'
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -78,23 +73,36 @@ async function getAssignments(session) {
|
||||
}
|
||||
|
||||
async function appendAssignment(session, assignment) {
|
||||
await _convertAnonymousAssignmentsIfNeeded(session)
|
||||
// await _convertAnonymousAssignmentsIfNeeded(session)
|
||||
|
||||
if (!session.splitTests) {
|
||||
session.splitTests = {}
|
||||
}
|
||||
if (!session.splitTests[assignment.splitTestName]) {
|
||||
session.splitTests[assignment.splitTestName] = []
|
||||
}
|
||||
|
||||
const assignments = await getAssignments(session)
|
||||
if (
|
||||
!_hasExistingAssignment(
|
||||
session,
|
||||
assignment.splitTestId,
|
||||
assignment.versionNumber
|
||||
)
|
||||
!_.find(assignments[assignment.splitTestName], {
|
||||
variantName: assignment.variantName,
|
||||
versionNumber: assignment.versionNumber,
|
||||
})
|
||||
) {
|
||||
if (!session.sta) {
|
||||
session.sta = ''
|
||||
}
|
||||
const splitTests = await SplitTestCache.get('')
|
||||
const splitTest = splitTests.get(assignment.splitTestName)
|
||||
const assignmentString = _buildAssignmentString(splitTest, assignment)
|
||||
const separator = session.sta.length > 0 ? TOKEN_SEP : ''
|
||||
session.sta += `${separator}${assignmentString}`
|
||||
// if (!session.sta) {
|
||||
// session.sta = ''
|
||||
// }
|
||||
// const splitTests = await SplitTestCache.get('')
|
||||
// const splitTest = splitTests.get(assignment.splitTestName)
|
||||
// const assignmentString = _buildAssignmentString(splitTest, assignment)
|
||||
// const separator = session.sta.length > 0 ? ';' : ''
|
||||
// session.sta += `${separator}${assignmentString}`
|
||||
session.splitTests[assignment.splitTestName].push({
|
||||
variantName: assignment.variantName,
|
||||
versionNumber: assignment.versionNumber,
|
||||
phase: assignment.phase,
|
||||
assignedAt: assignment.assignedAt,
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
@@ -174,63 +182,62 @@ function collectSessionStats(session) {
|
||||
JSON.stringify(session.cachedSplitTestAssignments).length
|
||||
)
|
||||
}
|
||||
if (session.sta) {
|
||||
if (session.splitTests) {
|
||||
Metrics.summary(
|
||||
'split_test_session_storage_count',
|
||||
(session.sta || '').split(';').length
|
||||
(session.sta || '').split(';').length +
|
||||
Object.keys(session.splitTests).length
|
||||
)
|
||||
Metrics.summary(
|
||||
'split_test_session_storage_size',
|
||||
(session.sta || '').length
|
||||
(session.sta || '').length + JSON.stringify(session.splitTests).length
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
async function _convertAnonymousAssignmentsIfNeeded(session) {
|
||||
if (session.splitTests) {
|
||||
const splitTests = await SplitTestCache.get('')
|
||||
if (!session.sta) {
|
||||
session.sta = ''
|
||||
}
|
||||
for (const [splitTestName, assignments] of Object.entries(
|
||||
session.splitTests
|
||||
)) {
|
||||
const splitTest = splitTests.get(splitTestName)
|
||||
for (const assignment of assignments) {
|
||||
const assignmentString = _buildAssignmentString(splitTest, assignment)
|
||||
const separator = session.sta.length > 0 ? TOKEN_SEP : ''
|
||||
session.sta += `${separator}${assignmentString}`
|
||||
}
|
||||
}
|
||||
delete session.splitTests
|
||||
}
|
||||
}
|
||||
// async function _convertAnonymousAssignmentsIfNeeded(session) {
|
||||
// if (typeof session.splitTests === 'object') {
|
||||
// const sessionAssignments = session.splitTests
|
||||
// const splitTests = await SplitTestCache.get('')
|
||||
// session.splitTests = ''
|
||||
// for (const [splitTestName, assignments] of Object.entries(
|
||||
// sessionAssignments
|
||||
// )) {
|
||||
// const splitTest = splitTests.get(splitTestName)
|
||||
// for (const assignment of assignments) {
|
||||
// const assignmentString = _buildAssignmentString(splitTest, assignment)
|
||||
// const separator = session.splitTests.length > 0 ? ';' : ''
|
||||
// session.splitTests += `${separator}${assignmentString}`
|
||||
// }
|
||||
// }
|
||||
// }
|
||||
// }
|
||||
|
||||
function _hasExistingAssignment(session, splitTest, versionNumber) {
|
||||
if (!session.sta) {
|
||||
return false
|
||||
}
|
||||
const index = session.sta.indexOf(
|
||||
`${_convertIdToBase64(splitTest._id)}${ID_VERSION_SEP}${versionNumber}=`
|
||||
)
|
||||
return index >= 0
|
||||
}
|
||||
// function _hasExistingAssignment(session, splitTest, versionNumber) {
|
||||
// if (!session.sta) {
|
||||
// return false
|
||||
// }
|
||||
// const index = session.sta.indexOf(
|
||||
// `${_convertIdToBase64(splitTest._id)}_${versionNumber}=`
|
||||
// )
|
||||
// return index >= 0
|
||||
// }
|
||||
|
||||
function _buildAssignmentString(splitTest, assignment) {
|
||||
const { versionNumber, variantName, assignedAt } = assignment
|
||||
const variants = SplitTestUtils.getCurrentVersion(splitTest).variants
|
||||
const splitTestId = _convertIdToBase64(splitTest._id)
|
||||
const variantChar =
|
||||
variantName === 'default'
|
||||
? 'd'
|
||||
: _.findIndex(variants, { name: variantName })
|
||||
const timestamp = Math.floor(assignedAt.getTime() / 1000).toString(36)
|
||||
return `${splitTestId}${ID_VERSION_SEP}${versionNumber}${KEY_VALUE_SEP}${variantChar}${VARIANT_DATE_SEP}${timestamp}`
|
||||
}
|
||||
// function _buildAssignmentString(splitTest, assignment) {
|
||||
// const { versionNumber, variantName, assignedAt } = assignment
|
||||
// const variants = SplitTestUtils.getCurrentVersion(splitTest).variants
|
||||
// const splitTestId = _convertIdToBase64(splitTest._id)
|
||||
// const variantChar =
|
||||
// variantName === 'default'
|
||||
// ? 'd'
|
||||
// : _.findIndex(variants, { name: variantName })
|
||||
// const timestamp = Math.floor(assignedAt.getTime() / 1000).toString(36)
|
||||
// return `${splitTestId}_${versionNumber}=${variantChar}:${timestamp}`
|
||||
// }
|
||||
|
||||
function _convertIdToBase64(id) {
|
||||
return new ObjectId(id).toString('base64')
|
||||
}
|
||||
// function _convertIdToBase64(id) {
|
||||
// return new ObjectId(id).toString('base64')
|
||||
// }
|
||||
|
||||
module.exports = {
|
||||
getAssignments: callbackify(getAssignments),
|
||||
|
||||
Reference in New Issue
Block a user