Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Update all users' quota monthly #122

Merged
merged 2 commits into from
Mar 30, 2025
Merged

Update all users' quota monthly #122

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
6276088
feat: update all users' quota monthly
yenche123 Mar 30, 2025
be3207b
chore: add a keyword about kicking
yenche123 Mar 30, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion liubai-backends/liubai-laf/cloud-functions/ai-entrance.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -616,7 +616,7 @@ class AiDirective {
}

private static isKickBot(text: string) {
const prefix = ["踢掉", "踢掉", "Kick", "Remove"]
const prefix = ["踢掉", "踢掉", "Kick", "Remove", "T调"]
const botMatched = this._getCommandedBot(prefix, text)
return botMatched
}
Expand Down
92 changes: 11 additions & 81 deletions liubai-backends/liubai-laf/cloud-functions/clock-one-hr.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -102,91 +102,21 @@ export async function updateUserQuota() {
}
console.warn("the moment is the 1st day of the month and in midnight")

// 2. construct query
// 2. get to update
const w2 = {
oState: "NORMAL",
subscription: {
isOn: "Y",
},
quota: _.exists(true),
}
const u2 = {
"quota.aiConversationCount": 0,
"quota.aiClusterCount": 0,
}
const uCol = db.collection("User")
const res2 = await uCol.where(w2).update(u2, { multi: true })
console.log("updateUserQuota res2: ", res2)

// 3. to get
const MAX_RUN_TIMES = 50
const NUM_PER_RUN = 50
let runTimes = 0
while(runTimes < MAX_RUN_TIMES) {
let q3 = uCol.where(w2)
q3 = q3.limit(NUM_PER_RUN).orderBy("insertedStamp", "asc")
if(runTimes > 0) {
q3 = q3.skip(runTimes * NUM_PER_RUN)
}
const res3 = await q3.get<Table_User>()
const list3 = res3.data
const len3 = list3.length
console.log("updateUserQuota len3: ", len3)
if(len3 < 1) break

// 4. get updated ids
const updatedIds: string[] = []
for(let i=0; i<len3; i++) {
const v = list3[i]
const userId = v._id
const sub = v.subscription
if(!sub) continue
const isSubscribed = checkIfUserSubscribed(v)
if(!isSubscribed) continue

if(sub.isLifelong) {
updatedIds.push(userId)
continue
}

console.log(`see ${userId} subscription: `)
console.log(sub)

// 5. calculate the duration between now and the expireStamp
const expireStamp = sub.expireStamp ?? 1
const stamp5 = localizeStamp(expireStamp)
const d5 = new Date(stamp5)
const diffMonths5 = differenceInCalendarMonths(d5, d1)
console.log("months between now and expireStamp: ", diffMonths5)


// 6. calcuate the duration between now and the chargedStamp
const chargedStamp = sub.chargedStamp ?? 1
const stamp6 = localizeStamp(chargedStamp)
const d6 = new Date(stamp6)
const diffMonths6 = differenceInCalendarMonths(d1, d6)
console.log("months between chargedStamp and now: ", diffMonths6)

// 7. ignore if the expiredStamp is within this month
// and the timing when the user charged is within the last month
if(diffMonths5 === 0 && diffMonths6 <= 1) {
continue
}

updatedIds.push(userId)
}

// 5. update quota
if(updatedIds.length > 0) {
const q5 = uCol.where({ _id: _.in(updatedIds) })
const u5: Partial<UserQuota> = {
aiConversationCount: 0,
aiClusterCount: 0,
}
const res5 = await q5.update({ quota: u5 })
console.log("updateUserQuota res5: ")
console.log(res5)
}

// 6. break if len3 is less than NUM_PER_RUN
if(len3 < NUM_PER_RUN) {
break
}
runTimes++
}
// 3. send report
const reporter = new LiuReporter()
reporter.sendAny("Monthly Update User Quota", res2)

}

Expand Down