-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathdb.js
48 lines (43 loc) · 1.52 KB
/
db.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
const mongoose = require("mongoose");
require('dotenv').config();
const mongoURI = process.env.MONGODB_URI;
const User = require("./models/User");
const connectToMongo = () => {
try {
mongoose.set("strictQuery", false);
mongoose.connect(mongoURI, {
useNewUrlParser: true,
useUnifiedTopology: true,
});
console.log("MongoDB connected");
// addDefaultUsernames();
} catch (error) {
console.log(error);
process.exit();
}
};
async function addDefaultUsernames() {
try {
// Find all users without a username
const usersWithoutUsername = await User.find({ username: { $exists: false } });
// Update each user with a default username
for (const user of usersWithoutUsername) {
user.username = generateDefaultUsername(user.email); // You can generate username based on email or any other data
await user.save();
console.log(`Added username ${user.username} for user ${user._id}`);
}
console.log('All users updated with default usernames');
mongoose.connection.close();
} catch (error) {
console.error('Error adding default usernames:', error);
mongoose.connection.close();
}
}
// Function to generate a default username based on email
function generateDefaultUsername(email) {
// For simplicity, you can just use a part of the email as the username
const username = email.split('@')[0];
return username;
}
module.exports = connectToMongo;
// mongodb+srv://naitikpatel2002:Naitik%[email protected]/profolio?retryWrites=true&w=majority