tethys/database/DisableForeignKeys.php

80 lines
1.8 KiB
PHP
Raw Normal View History

2018-09-06 15:58:54 +00:00
<?php
namespace Database;
use Illuminate\Support\Facades\DB;
/**
* Class DisablesForeignKeys.
*/
trait DisableForeignKeys
{
/**
* @var array
*/
private $commands = [
'mysql' => [
'enable' => 'SET FOREIGN_KEY_CHECKS=1;',
'disable' => 'SET FOREIGN_KEY_CHECKS=0;',
],
'sqlite' => [
'enable' => 'PRAGMA foreign_keys = ON;',
'disable' => 'PRAGMA foreign_keys = OFF;',
],
'sqlsrv' => [
'enable' => 'EXEC sp_msforeachtable @command1="print \'?\'", @command2="ALTER TABLE ? WITH CHECK CHECK CONSTRAINT all";',
'disable' => 'EXEC sp_msforeachtable "ALTER TABLE ? NOCHECK CONSTRAINT all";',
],
'pgsql' => [
'enable' => 'SET CONSTRAINTS ALL IMMEDIATE;',
'disable' => 'SET CONSTRAINTS ALL DEFERRED;',
],
];
/**
* Disable foreign key checks for current db driver.
*/
protected function disableForeignKeys()
{
DB::statement($this->getDisableStatement());
}
/**
* Enable foreign key checks for current db driver.
*/
protected function enableForeignKeys()
{
DB::statement($this->getEnableStatement());
}
/**
* Return current driver enable command.
*
* @return mixed
*/
private function getEnableStatement()
{
return $this->getDriverCommands()['enable'];
}
/**
* Return current driver disable command.
*
* @return mixed
*/
private function getDisableStatement()
{
return $this->getDriverCommands()['disable'];
}
/**
* Returns command array for current db driver.
*
* @return mixed
*/
private function getDriverCommands()
{
return $this->commands[DB::getDriverName()];
}
}