UA-Fediland/pixelfed
2
0
Fork 0
mirror of https://github.com/pixelfed/pixelfed.git synced 2024-11-22 06:21:27 +00:00
Code Issues Projects Releases Packages Wiki Activity

Isert cities in chunks to increase performance

Browse source
This commit is contained in:
monkeyless 2019-08-09 05:38:48 +02:00
parent 450492db53
commit 27952883b6
1 changed files with 25 additions and 26 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

51
app/Console/Commands/ImportCities.php
View file

@ -1,5 +1,4 @@
<?php
namespace App\Console\Commands;
use Illuminate\Console\Command;
@ -14,15 +13,13 @@ class ImportCities extends Command
*
* @var string
*/
protected $signature = 'import:cities';
protected $signature = 'import:cities {chunk=1000}';
/**
* The console command description.
*
* @var string
*/
protected $description = 'Import Cities to database';
/**
* Create a new command instance.
*
@ -32,7 +29,6 @@ class ImportCities extends Command
{
parent::__construct();
}
/**
* Execute the console command.
*
@ -41,38 +37,41 @@ class ImportCities extends Command
public function handle()
{
$path = storage_path('app/cities.json');
if(!is_file($path)) {
if (!is_file($path)) {
$this->error('Missing storage/app/cities.json file!');
return;
}
if(Place::count() > 10) {
$this->error('Cities already imported, aborting operation...');
return;
}
// if (Place::count() > 10) {
// $this->error('Cities already imported, aborting operation...');
// return;
// }
$this->info('Importing city data into database ...');
$cities = file_get_contents($path);
$cities = json_decode($cities);
$count = count($cities);
$this->info("Found {$count} cities to insert ...");
$bar = $this->output->createProgressBar($count);
$cityCount = count($cities);
$this->info("Found {$cityCount} cities to insert ...");
$bar = $this->output->createProgressBar($cityCount);
$bar->start();
$buffer = [];
$count = 0;
foreach ($cities as $city) {
$country = $city->country == 'XK' ? 'Kosovo' : (new \League\ISO3166\ISO3166)->alpha2($city->country)['name'];
DB::transaction(function () use ($city, $country) {
$place = new Place();
$place->name = $city->name;
$place->slug = Str::slug($city->name);
$place->country = $country;
$place->lat = $city->lat;
$place->long = $city->lng;
$place->save();
});
$bar->advance();
$buffer[] = ["name" => $city->name, "slug" => Str::slug($city->name), "country" => $country, "lat" => $city->lat, "long" => $city->lng];
$count++;
if ($count % $this->argument('chunk') == 0) {
$this->insertBuffer($buffer, $count);
$bar->advance(count($buffer));
$buffer = [];
}
}
$this->insertBuffer($buffer, $count);
$bar->finish();
$this->info('Successfully imported ' . $count . ' entries.');
return;
}
private function insertBuffer($buffer, $count)
{
DB::table('places')->insert($buffer);
}
}