nhyatt/bitwarden
1
0
Fork 0
mirror of https://github.com/bitwarden/server.git synced 2025-05-08 21:22:20 -05:00
Code

Seed many users

Browse Source
This commit is contained in:
Hinton 2025-04-06 22:42:45 +09:00
parent 0abf4da126
commit 360aef2521
No known key found for this signature in database
GPG Key ID: 5F7295599C5D965C
1 changed files with 14 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
util/Seeder/Commands/GenerateCommand.cs
View File

@ -1,4 +1,5 @@
using Bit.Seeder.Factories; using Bit.Infrastructure.EntityFramework.Models;
using Bit.Seeder.Factories;
using Bit.Seeder.Settings; using Bit.Seeder.Settings;
using Bit.SharedWeb.Utilities; using Bit.SharedWeb.Utilities;
using Microsoft.AspNetCore.DataProtection; using Microsoft.AspNetCore.DataProtection;
@ -23,6 +24,16 @@ public class GenerateCommand
var user = UserSeeder.CreateUser($"admin@{domain}"); var user = UserSeeder.CreateUser($"admin@{domain}");
var orgUser = organization.CreateOrganizationUser(user); var orgUser = organization.CreateOrganizationUser(user);
var additionalUsers = new List<User>();
var additionalOrgUsers = new List<OrganizationUser>();
for (var i = 0; i < users; i++)
{
var additionalUser = UserSeeder.CreateUser($"user{i}@{domain}");
additionalUsers.Add(additionalUser);
additionalOrgUsers.Add(organization.CreateOrganizationUser(additionalUser));
}
using (var scope = serviceProvider.CreateScope()) using (var scope = serviceProvider.CreateScope())
{ {
var scopedServices = scope.ServiceProvider; var scopedServices = scope.ServiceProvider;
@ -31,6 +42,8 @@ public class GenerateCommand
db.Add(organization); db.Add(organization);
db.Add(user); db.Add(user);
db.Add(orgUser); db.Add(orgUser);
db.AddRange(additionalUsers);
db.AddRange(additionalOrgUsers);
db.SaveChanges(); db.SaveChanges();
} }