8fce8d5f23
* Make updateFooMetrics functions manually testable * Add logging, test script to metrics update functions * Improve on `batchedWaitAll` for update functions
80 lines
2.3 KiB
TypeScript
80 lines
2.3 KiB
TypeScript
import * as functions from 'firebase-functions'
|
|
import * as admin from 'firebase-admin'
|
|
import { groupBy, sum, sumBy } from 'lodash'
|
|
|
|
import { getValues, log, logMemory, mapAsync } from './utils'
|
|
import { Contract } from '../../common/contract'
|
|
import { Bet } from '../../common/bet'
|
|
import { User } from '../../common/user'
|
|
import { calculatePayout } from '../../common/calculate'
|
|
|
|
const firestore = admin.firestore()
|
|
|
|
const computeInvestmentValue = (
|
|
bets: Bet[],
|
|
contractsDict: { [k: string]: Contract }
|
|
) => {
|
|
return sumBy(bets, (bet) => {
|
|
const contract = contractsDict[bet.contractId]
|
|
if (!contract || contract.isResolved) return 0
|
|
if (bet.sale || bet.isSold) return 0
|
|
|
|
const payout = calculatePayout(contract, bet, 'MKT')
|
|
return payout - (bet.loanAmount ?? 0)
|
|
})
|
|
}
|
|
|
|
const computeTotalPool = (
|
|
user: User,
|
|
contractsDict: { [k: string]: Contract }
|
|
) => {
|
|
const creatorContracts = Object.values(contractsDict).filter(
|
|
(contract) => contract.creatorId === user.id
|
|
)
|
|
const pools = creatorContracts.map((contract) =>
|
|
sum(Object.values(contract.pool))
|
|
)
|
|
return sum(pools)
|
|
}
|
|
|
|
export const updateUserMetricsCore = async () => {
|
|
const [users, contracts, bets] = await Promise.all([
|
|
getValues<User>(firestore.collection('users')),
|
|
getValues<Contract>(firestore.collection('contracts')),
|
|
firestore.collectionGroup('bets').get(),
|
|
])
|
|
log(
|
|
`Loaded ${users.length} users, ${contracts.length} contracts, and ${bets.docs.length} bets.`
|
|
)
|
|
logMemory()
|
|
|
|
const contractsDict = Object.fromEntries(
|
|
contracts.map((contract) => [contract.id, contract])
|
|
)
|
|
|
|
const betsByUser = groupBy(
|
|
bets.docs.map((doc) => doc.data() as Bet),
|
|
(bet) => bet.userId
|
|
)
|
|
|
|
await mapAsync(users, async (user) => {
|
|
const investmentValue = computeInvestmentValue(
|
|
betsByUser[user.id] ?? [],
|
|
contractsDict
|
|
)
|
|
const creatorVolume = computeTotalPool(user, contractsDict)
|
|
const totalValue = user.balance + investmentValue
|
|
const totalPnL = totalValue - user.totalDeposits
|
|
return await firestore.collection('users').doc(user.id).update({
|
|
totalPnLCached: totalPnL,
|
|
creatorVolumeCached: creatorVolume,
|
|
})
|
|
})
|
|
log(`Updated metrics for ${users.length} users.`)
|
|
}
|
|
|
|
export const updateUserMetrics = functions
|
|
.runWith({ memory: '1GB' })
|
|
.pubsub.schedule('every 15 minutes')
|
|
.onRun(updateUserMetricsCore)
|