Developer
32 lines
927 B
JavaScript
32 lines
927 B
JavaScript
const mongoose = require('mongoose');
|
|
const User = require('../src/models/User');
|
|
const UserEquity = require('../src/models/UserEquity');
|
|
|
|
async function fixPlatformCount() {
|
|
try {
|
|
await mongoose.connect(process.env.MONGODB_URI || 'mongodb://localhost:27017/member-assistant');
|
|
console.log('MongoDB connected');
|
|
|
|
const users = await User.find({}).select('_id');
|
|
console.log(`Found ${users.length} users`);
|
|
|
|
let updated = 0;
|
|
for (const user of users) {
|
|
const count = await UserEquity.countDocuments({ owner: user._id });
|
|
await User.findByIdAndUpdate(user._id, { platformCount: count });
|
|
updated++;
|
|
if (updated % 100 === 0) {
|
|
console.log(`Progress: ${updated}/${users.length}`);
|
|
}
|
|
}
|
|
|
|
console.log(`Done! Updated ${updated} users.`);
|
|
process.exit(0);
|
|
} catch (error) {
|
|
console.error('Error:', error);
|
|
process.exit(1);
|
|
}
|
|
}
|
|
|
|
fixPlatformCount();
|