phpvms/app/Services/DatabaseService.php

95 lines
2.3 KiB
PHP
Raw Normal View History

2017-06-25 00:09:27 +08:00
<?php
namespace App\Services;
use Carbon\Carbon;
use Illuminate\Database\QueryException;
use Illuminate\Support\Facades\DB;
2018-02-21 12:33:09 +08:00
use Symfony\Component\Yaml\Yaml;
use Webpatser\Uuid\Uuid;
2017-06-25 00:09:27 +08:00
2017-06-25 02:20:24 +08:00
class DatabaseService extends BaseService
{
2017-06-25 02:30:13 +08:00
protected $time_fields = [
2017-06-25 02:23:30 +08:00
'created_at',
'updated_at'
];
2017-06-25 02:30:13 +08:00
protected $uuid_tables = [
2017-12-31 04:13:47 +08:00
'acars',
2017-06-25 02:20:24 +08:00
'flights',
2017-06-25 02:20:39 +08:00
'pireps',
2017-06-25 02:20:24 +08:00
];
2017-06-25 00:09:27 +08:00
protected function time(): string
{
return Carbon::now('UTC')->format('Y-m-d H:i:s');
}
/**
* @param $yaml_file
* @param bool $ignore_errors
* @return array
* @throws \Exception
*/
public function seed_from_yaml_file($yaml_file, $ignore_errors=false): array
{
2017-06-25 02:20:24 +08:00
$yml = file_get_contents($yaml_file);
return $this->seed_from_yaml($yml, $ignore_errors);
2017-06-25 02:20:24 +08:00
}
/**
* @param $yml
* @param bool $ignore_errors
* @return array
* @throws \Exception
*/
public function seed_from_yaml($yml, $ignore_errors=false): array
2017-06-25 02:20:24 +08:00
{
$imported = [];
2017-06-25 02:20:24 +08:00
$yml = Yaml::parse($yml);
foreach ($yml as $table => $rows) {
$imported[$table] = 0;
foreach ($rows as $row) {
2017-06-25 02:20:24 +08:00
# see if this table uses a UUID as the PK
# if no ID is specified
if(in_array($table, $this->uuid_tables)) {
if(!array_key_exists('id', $row)) {
$row['id'] = Uuid::generate()->string;
}
}
# encrypt any password fields
if(array_key_exists('password', $row)) {
$row['password'] = bcrypt($row['password']);
}
# if any time fields are == to "now", then insert the right time
2017-06-25 02:23:30 +08:00
foreach($this->time_fields as $tf) {
2017-12-24 01:58:17 +08:00
if(array_key_exists($tf, $row) && strtolower($row[$tf]) === 'now') {
$row[$tf] = $this->time();
}
}
try {
DB::table($table)->insert($row);
++$imported[$table];
} catch(QueryException $e) {
if($ignore_errors) {
continue;
}
throw $e;
}
}
}
return $imported;
}
2017-06-25 00:09:27 +08:00
}