mirror of
https://github.com/snipe/snipe-it.git
synced 2026-08-18 11:15:42 +00:00
167 lines
5.6 KiB
PHP
167 lines
5.6 KiB
PHP
<?php
|
|
|
|
namespace Database\Seeders;
|
|
|
|
use App\Models\Company;
|
|
use App\Models\Department;
|
|
use App\Models\User;
|
|
use Database\Seeders\Concerns\ReportsMemory;
|
|
use Illuminate\Database\Eloquent\Factories\Sequence;
|
|
use Illuminate\Database\Seeder;
|
|
use Illuminate\Support\Facades\Log;
|
|
use Illuminate\Support\Facades\Storage;
|
|
|
|
class UserSeeder extends Seeder
|
|
{
|
|
use ReportsMemory;
|
|
|
|
/**
|
|
* Run the database seeds.
|
|
*
|
|
* @return void
|
|
*/
|
|
public function run()
|
|
{
|
|
User::truncate();
|
|
|
|
if (! Company::count()) {
|
|
$this->call(CompanySeeder::class);
|
|
}
|
|
|
|
$companyIds = Company::all()->pluck('id');
|
|
|
|
if (! Department::count()) {
|
|
$this->call(DepartmentSeeder::class);
|
|
}
|
|
|
|
$departmentIds = Department::all()->pluck('id');
|
|
|
|
// Named admins get multiple companies. They manage assets across several organisations.
|
|
foreach (['firstAdmin', 'snipeAdmin', 'testAdmin'] as $state) {
|
|
$user = User::factory()->{$state}()->withoutCompany()->create([
|
|
'department_id' => $departmentIds->random(),
|
|
]);
|
|
$ids = $companyIds->random(min(rand(2, 3), $companyIds->count()))->toArray();
|
|
$user->companies()->sync($ids);
|
|
$user->syncLegacyCompanyIdMirror();
|
|
}
|
|
|
|
// Superusers, one company each.
|
|
User::factory()->count(3)->superuser()
|
|
->withoutCompany()
|
|
->state(new Sequence(fn ($sequence) => [
|
|
'department_id' => $departmentIds->random(),
|
|
]))
|
|
->create()
|
|
->each(function (User $user) use ($companyIds) {
|
|
$user->companies()->sync([$companyIds->random()]);
|
|
$user->syncLegacyCompanyIdMirror();
|
|
});
|
|
|
|
// Admins, one company each.
|
|
User::factory()->count(3)->admin()
|
|
->withoutCompany()
|
|
->state(new Sequence(fn ($sequence) => [
|
|
'department_id' => $departmentIds->random(),
|
|
]))
|
|
->create()
|
|
->each(function (User $user) use ($companyIds) {
|
|
$user->companies()->sync([$companyIds->random()]);
|
|
$user->syncLegacyCompanyIdMirror();
|
|
});
|
|
|
|
// Regular users, three groups:
|
|
// ~30% (600) no company
|
|
// ~50% (1000) one company
|
|
// ~20% (400) two or three companies
|
|
//
|
|
// Chunked so we don't hold 1000-2000 User models in memory at once
|
|
// (each ->each() / foreach that follows a big ->create() otherwise
|
|
// materializes the whole collection). Reduced demo-server memory
|
|
// pressure that was pushing seeding into swap.
|
|
$chunk = 200;
|
|
|
|
$departmentState = fn () => new Sequence(fn ($sequence) => [
|
|
'department_id' => $departmentIds->random(),
|
|
]);
|
|
|
|
$this->reportMemory('UserSeeder start of regular-user batches');
|
|
|
|
memory_reset_peak_usage();
|
|
for ($i = 0; $i < 600 / $chunk; $i++) {
|
|
User::factory()->count($chunk)->viewAssets()
|
|
->withoutCompany()
|
|
->state($departmentState())
|
|
->create();
|
|
gc_collect_cycles();
|
|
}
|
|
$this->reportMemory('UserSeeder after 600 no-company users (chunked)');
|
|
|
|
memory_reset_peak_usage();
|
|
for ($i = 0; $i < 1000 / $chunk; $i++) {
|
|
User::factory()->count($chunk)->viewAssets()
|
|
->withoutCompany()
|
|
->state($departmentState())
|
|
->create()
|
|
->each(function (User $user) use ($companyIds) {
|
|
$user->companies()->sync([$companyIds->random()]);
|
|
$user->syncLegacyCompanyIdMirror();
|
|
});
|
|
gc_collect_cycles();
|
|
}
|
|
$this->reportMemory('UserSeeder after 1000 one-company users (chunked)');
|
|
|
|
memory_reset_peak_usage();
|
|
for ($i = 0; $i < 400 / $chunk; $i++) {
|
|
User::factory()->count($chunk)->viewAssets()
|
|
->withoutCompany()
|
|
->state($departmentState())
|
|
->create()
|
|
->each(function (User $user) use ($companyIds) {
|
|
$ids = $companyIds->random(min(rand(2, 3), $companyIds->count()))->toArray();
|
|
$user->companies()->sync($ids);
|
|
$user->syncLegacyCompanyIdMirror();
|
|
});
|
|
gc_collect_cycles();
|
|
}
|
|
$this->reportMemory('UserSeeder after 400 multi-company users (chunked)');
|
|
|
|
$src = public_path('/img/demo/avatars/');
|
|
$dst = 'avatars'.'/';
|
|
$del_files = Storage::files($dst);
|
|
|
|
foreach ($del_files as $del_file) { // iterate files
|
|
$file_to_delete = str_replace($src, '', $del_file);
|
|
Log::debug('Deleting: '.$file_to_delete);
|
|
try {
|
|
Storage::disk('public')->delete($dst.$del_file);
|
|
} catch (\Exception $e) {
|
|
Log::debug($e);
|
|
}
|
|
}
|
|
|
|
$add_files = glob($src.'/*.*');
|
|
foreach ($add_files as $add_file) {
|
|
$file_to_copy = str_replace($src, '', $add_file);
|
|
Log::debug('Copying: '.$file_to_copy);
|
|
try {
|
|
Storage::disk('public')->put($dst.$file_to_copy, file_get_contents($src.$file_to_copy));
|
|
} catch (\Exception $e) {
|
|
Log::debug($e);
|
|
}
|
|
}
|
|
|
|
$users = User::orderBy('id', 'asc')->take(20)->get();
|
|
$file_number = 1;
|
|
|
|
foreach ($users as $user) {
|
|
|
|
$user->avatar = $file_number.'.jpg';
|
|
$user->save();
|
|
$file_number++;
|
|
}
|
|
|
|
$this->reportMemory('UserSeeder end (all users + avatars complete)');
|
|
}
|
|
}
|