Viewing file:  Batch.php (13.48 KB) -rw-r--r--
Select action/file-type:
 (+) |  (+) |  (+) | Code (+) | Session (+) |  (+) | SDB (+) |  (+) |  (+) |  (+) |  (+) |  (+) |
<?php declare(strict_types=1);
namespace Mavinoo\Batch;
use Mavinoo\Batch\Common\Common;
use Illuminate\Database\Eloquent\Model;
use Illuminate\Database\DatabaseManager;
use Illuminate\Support\Carbon;
class Batch implements BatchInterface
{
/**
* @var DatabaseManager
*/
protected $db;
public function __construct(DatabaseManager $db)
{
$this->db = $db;
}
/**
* <h2>Update multiple rows.</h2>
*
* Example:<br>
* ```
* $userInstance = new \App\Models\User;
* $value = [
* [
* 'id' => 1,
* 'status' => 'active',
* 'nickname' => 'Mohammad'
* ],
* [
* 'id' => 5,
* 'status' => 'deactive',
* 'nickname' => 'Ghanbari'
* ],
* [
* 'id' => 7,
* 'balance' => ['+', 500]
* ]
* ];
* $index = 'id';
* Batch::update($userInstance, $value, $index);
* ```
*
* @param \Illuminate\Database\Eloquent\Model $table
* @param array $values
* @param string $index
* @param bool $raw
* @return bool|int
* @updatedBy Ibrahim Sakr <ebrahimes@gmail.com>
*/
public function update(Model $table, array $values, string $index = null, bool $raw = false)
{
$final = [];
$ids = [];
if (!count($values)) {
return false;
}
if (!isset($index) || empty($index)) {
$index = $table->getKeyName();
}
$driver = $table->getConnection()->getDriverName();
foreach ($values as $key => $val) {
$ids[] = $val[$index];
if ($table->usesTimestamps()) {
$updatedAtColumn = $table->getUpdatedAtColumn();
if (!isset($val[$updatedAtColumn])) {
$val[$updatedAtColumn] = Carbon::now()->format($table->getDateFormat());
}
}
foreach (array_keys($val) as $field) {
if ($field !== $index) {
// If increment / decrement
if (gettype($val[$field]) == 'array') {
// If array has two values
if (!array_key_exists(0, $val[$field]) || !array_key_exists(1, $val[$field])) {
throw new \ArgumentCountError('Increment/Decrement array needs to have 2 values, a math operator (+, -, *, /, %) and a number');
}
// Check first value
if (gettype($val[$field][0]) != 'string' || !in_array($val[$field][0], ['+', '-', '*', '/', '%'])) {
throw new \TypeError('First value in Increment/Decrement array needs to be a string and a math operator (+, -, *, /, %)');
}
// Check second value
if (!is_numeric($val[$field][1])) {
throw new \TypeError('Second value in Increment/Decrement array needs to be numeric');
}
// Increment / decrement
if (Common::disableBacktick($driver)) {
$value = $field . $val[$field][0] . $val[$field][1];
} else {
$value = '`' . $field . '`' . $val[$field][0] . $val[$field][1];
}
} else {
// Only update
$finalField = $raw ? Common::mysql_escape($val[$field]) : "'" . Common::mysql_escape($val[$field]) . "'";
$value = (is_null($val[$field]) ? 'NULL' : $finalField);
}
if (Common::disableBacktick($driver))
$final[$field][] = 'WHEN ' . $index . ' = \'' . $val[$index] . '\' THEN ' . $value . ' ';
else
$final[$field][] = 'WHEN `' . $index . '` = \'' . $val[$index] . '\' THEN ' . $value . ' ';
}
}
}
if (Common::disableBacktick($driver)) {
$cases = '';
foreach ($final as $k => $v) {
$cases .= '"' . $k . '" = (CASE ' . implode("\n", $v) . "\n"
. 'ELSE "' . $k . '" END), ';
}
$query = "UPDATE \"" . $this->getFullTableName($table) . '" SET ' . substr($cases, 0, -2) . " WHERE \"$index\" IN('" . implode("','", $ids) . "');";
} else {
$cases = '';
foreach ($final as $k => $v) {
$cases .= '`' . $k . '` = (CASE ' . implode("\n", $v) . "\n"
. 'ELSE `' . $k . '` END), ';
}
$query = "UPDATE `" . $this->getFullTableName($table) . "` SET " . substr($cases, 0, -2) . " WHERE `$index` IN(" . '"' . implode('","', $ids) . '"' . ");";
}
return $this->db->connection($this->getConnectionName($table))->update($query);
}
/**
* Update multiple rows
* @param Model $table
* @param array $values
* @param string $index
* @param string|null $index2
* @param bool $raw
* @return bool|int
* @updatedBy Ibrahim Sakr <ebrahimes@gmail.com>
*
* @desc
* Example
* $table = 'users';
* $value = [
* [
* 'id' => 1,
* 'status' => 'active',
* 'nickname' => 'Mohammad'
* ] ,
* [
* 'id' => 5,
* 'status' => 'deactive',
* 'nickname' => 'Ghanbari'
* ] ,
* ];
* $index = 'id';
* $index2 = 'user_id';
*
*/
public function updateWithTwoIndex(Model $table, array $values, string $index = null, string $index2 = null, bool $raw = false)
{
$final = [];
$ids = [];
$driver = $table->getConnection()->getDriverName();
if (!count($values)) {
return false;
}
if (!isset($index) || empty($index)) {
$index = $table->getKeyName();
}
foreach ($values as $key => $val) {
$ids[] = $val[$index];
$ids2[] = $val[$index2];
foreach (array_keys($val) as $field) {
if ($field !== $index || $field !== $index2) {
$finalField = $raw ? Common::mysql_escape($val[$field]) : "'" . Common::mysql_escape($val[$field]) . "'";
$value = (is_null($val[$field]) ? 'NULL' : $finalField);
if (Common::disableBacktick($driver)) {
$final[$field][] = 'WHEN (' . $index . ' = \'' . Common::mysql_escape($val[$index]) . '\' AND ' . $index2 . ' = \'' . $val[$index2] . '\') THEN ' . $value . ' ';
} else {
$final[$field][] = 'WHEN (`' . $index . '` = "' . Common::mysql_escape($val[$index]) . '" AND `' . $index2 . '` = "' . $val[$index2] . '") THEN ' . $value . ' ';
}
}
}
}
if (Common::disableBacktick($driver)) {
$cases = '';
foreach ($final as $k => $v) {
$cases .= '"' . $k . '" = (CASE ' . implode("\n", $v) . "\n"
. 'ELSE "' . $k . '" END), ';
}
$query = "UPDATE \"" . $this->getFullTableName($table) . '" SET ' . substr($cases, 0, -2) . " WHERE \"$index\" IN('" . implode("','", $ids) . "') AND \"$index2\" IN('" . implode("','", $ids2) . "');";
//$query = "UPDATE \"" . $this->getFullTableName($table) . "\" SET " . substr($cases, 0, -2) . " WHERE \"$index\" IN(" . '"' . implode('","', $ids) . '")' . " AND \"$index2\" IN(" . '"' . implode('","', $ids2) . '"' . " );";
} else {
$cases = '';
foreach ($final as $k => $v) {
$cases .= '`' . $k . '` = (CASE ' . implode("\n", $v) . "\n"
. 'ELSE `' . $k . '` END), ';
}
$query = "UPDATE `" . $this->getFullTableName($table) . "` SET " . substr($cases, 0, -2) . " WHERE `$index` IN(" . '"' . implode('","', $ids) . '")' . " AND `$index2` IN(" . '"' . implode('","', $ids2) . '"' . " );";
}
return $this->db->connection($this->getConnectionName($table))->update($query);
}
/**
* Insert Multi rows.
*
* @param Model $table
* @param array $columns
* @param array $values
* @param int $batchSize
* @param bool $insertIgnore
* @return bool|mixed
* @throws \Throwable
* @updatedBy Ibrahim Sakr <ebrahimes@gmail.com>
* @desc
* Example
*
* $table = 'users';
* $columns = [
* 'firstName',
* 'lastName',
* 'email',
* 'isActive',
* 'status',
* ];
* $values = [
* [
* 'Mohammad',
* 'Ghanbari',
* 'emailSample_1@gmail.com',
* '1',
* '0',
* ] ,
* [
* 'Saeed',
* 'Mohammadi',
* 'emailSample_2@gmail.com',
* '1',
* '0',
* ] ,
* [
* 'Avin',
* 'Ghanbari',
* 'emailSample_3@gmail.com',
* '1',
* '0',
* ] ,
* ];
* $batchSize = 500; // insert 500 (default), 100 minimum rows in one query
*/
public function insert(Model $table, array $columns, array $values, int $batchSize = 500, bool $insertIgnore = false)
{
// no need for the old validation since we now use type hint that supports from php 7.0
// but I kept this one
if (count($columns) !== count(current($values))) {
return false;
}
$query = [];
$minChunck = 100;
$totalValues = count($values);
$batchSizeInsert = ($totalValues < $batchSize && $batchSize < $minChunck) ? $minChunck : $batchSize;
$totalChunk = ($batchSizeInsert < $minChunck) ? $minChunck : $batchSizeInsert;
$values = array_chunk($values, $totalChunk, true);
if ($table->usesTimestamps()) {
$createdAtColumn = $table->getCreatedAtColumn();
$updatedAtColumn = $table->getUpdatedAtColumn();
$now = Carbon::now()->format($table->getDateFormat());
$addCreatedAtValue = false;
$addUpdatedAtValue = false;
if (!in_array($createdAtColumn, $columns)) {
$addCreatedAtValue = true;
array_push($columns, $createdAtColumn);
}
if (!in_array($updatedAtColumn, $columns)) {
$addUpdatedAtValue = true;
array_push($columns, $updatedAtColumn);
}
foreach ($values as $key => $value) {
foreach ($value as $rowKey => $row) {
if ($addCreatedAtValue) {
array_push($values[$key][$rowKey], $now);
}
if ($addUpdatedAtValue) {
array_push($values[$key][$rowKey], $now);
}
}
}
}
$driver = $table->getConnection()->getDriverName();
if (Common::disableBacktick($driver)) {
foreach ($columns as $key => $column) {
$columns[$key] = '"' . Common::mysql_escape($column) . '"';
}
} else {
foreach ($columns as $key => $column) {
$columns[$key] = '`' . Common::mysql_escape($column) . '`';
}
}
foreach ($values as $value) {
$valueArray = [];
foreach ($value as $data) {
foreach ($data as $key => $item) {
$item = is_null($item) ? 'NULL' : "'" . Common::mysql_escape($item) . "'";
$data[$key] = $item;
}
$valueArray[] = '(' . implode(',', $data) . ')';
}
$valueString = implode(', ', $valueArray);
$ignoreStmt = $insertIgnore ? ' IGNORE ' : '';
if (Common::disableBacktick($driver)) {
$query[] = 'INSERT ' . $ignoreStmt . ' INTO "' . $this->getFullTableName($table) . '" (' . implode(',', $columns) . ") VALUES $valueString;";
} else {
$query[] = 'INSERT ' . $ignoreStmt . ' INTO `' . $this->getFullTableName($table) . '` (' . implode(',', $columns) . ") VALUES $valueString;";
}
}
if (count($query)) {
return $this->db->transaction(function () use ($totalValues, $totalChunk, $query, $table) {
$totalQuery = 0;
foreach ($query as $value) {
$totalQuery += $this->db->connection($this->getConnectionName($table))->statement($value) ? 1 : 0;
}
return [
'totalRows' => $totalValues,
'totalBatch' => $totalChunk,
'totalQuery' => $totalQuery
];
});
}
return false;
}
/**
* Get full table name.
*
* @param Model $model
* @return string
* @author Ibrahim Sakr <ebrahimes@gmail.com>
*/
private function getFullTableName(Model $model)
{
return $model->getConnection()->getTablePrefix() . $model->getTable();
}
/**
* Get connection name.
*
* @param Model $model
* @return string|null
* @author Ibrahim Sakr <ebrahimes@gmail.com>
*/
private function getConnectionName(Model $model)
{
if (!is_null($cn = $model->getConnectionName())) {
return $cn;
}
return $model->getConnection()->getName();
}
}
|