MultipleUpdateStrategy.php

<?php

/**
 * @link https://github.com/ruskid/yii2-csv-importer#README
 */

namespace ruskid\csvimporter;

use yii\base\Exception;
use yii\db\Query;
use Yii;

/**
 * Update from CSV. This will create|update rows of an ActiveRecord table.
 * A csv line is considered a new record if its key does not match any AR table row key.
 */
class MultipleUpdateStrategy extends MultipleImportStrategy implements ImportInterface {

	/**
	 * ActiveRecord class name
	 * @var string
	 */
	public $className;

	/**
	 * @var \Closure a function that returns a value that will be used to index the corresponding
	 * csv file line. The signature of the function should be the following: `function ($line)`.
	 * This value must be unique for each csv line.
	 * Where `$line` is an array representing one line of the csv file.
	 */
	public $csvKey;

	/**
	 * @var \Closure a function that returns a value that will be used to index the corresponding
	 * AR table row. The signature of the function should be the following: `function ($row)`.
	 * This value must be unique for each table row.
	 * Where `$row` is an array representing one row of the AR table.
	 */
	public $rowKey;

	/**
	 * AR query batch size.
	 * @var integer
	 */
	public $queryBatchSize = 1000;

	/**
	 * @throws Exception
	 */
	public function __construct() {
		$arguments = func_get_args();
		if (!empty($arguments)) {
			if (isset($arguments[0]['className']) && !isset($arguments[0]['tableName'])) {
				$arguments[0]['tableName'] = (new $arguments[0]['className'])->tableName();
			}
		}
		call_user_func_array(array($this, 'parent::__construct'), $arguments);
		$requiredFields = ['csvKey', 'rowKey', 'className'];
		foreach ($requiredFields as $field) {
			if ($this->{$field} === null) {
				throw new Exception(__CLASS__ . " $field is required.");
			}
		}
	}

	/**
	 * Will multiple import|update data into table. WARNING: this function deletes $data content to save memory.
	 * @param array $data CSV data passed by reference to save memory.
	 * @return [integer] records affected ['new' => integer, 'updated' => integer, 'unchanged' => integer]
	 */
	public function import(&$data) {
		$records = ['new' => 0, 'updated' => 0, 'unchanged' => 0];

		// Re-index csv data
		$dataReindexed = [];
		foreach ($data as $k => $line) {
			$dataReindexed[call_user_func($this->csvKey, $line)] = $line;
			unset($data[$k]);
		}

		$query = (new Query())->from($this->tableName);

		foreach ($query->each() as $row) {
			$key = call_user_func($this->rowKey, $row);

			if (key_exists($key, $dataReindexed)) {
				// remove this line from new records to be inserted
				$line = $dataReindexed[$key];
				unset($dataReindexed[$key]);

				// table row should be updated or unchanged
				$skipImport = isset($this->skipImport) ? call_user_func($this->skipImport, $line) : false;
				if(!$skipImport) {
					$values = $this->getLineValues($line);
					if ($this->changed($row, $values)) {
						$records['updated'] += $this->updateRecord($row, $values);
					} else {
						$records['unchanged'] ++;
					}
				} else {
					$records['unchanged'] ++;
				}
			}
		}
		
		// save new records
		$records['new'] = parent::import($dataReindexed);
		return $records;
	}

	/**
	 * Will get values for csv line
	 * @param string $line csv line
	 * @return array
	 */
	public function getLineValues($line) {
		foreach ($this->configs as $config) {
			$value = call_user_func($config['value'], $line);
			$values[$config['attribute']] = $value;
		}
		return $values;
	}

	/**
	 * Returns true if $row needs to be updated
	 * @param array $row
	 * @param array $values
	 * @return boolean
	 */
	public function changed($row, $values) {
		foreach ($values as $k => $value) {
			if ($row[$k] != $value) {
				return true;
			}
		}
		return false;
	}

	/**
	 * Updates $row with $values
	 * @param type $row
	 * @param type $values
	 * @return integer the number of affected rows
	 */
	public function updateRecord($row, $values) {
		return Yii::$app->db->createCommand()->update($this->tableName, $values, $row)->execute();
	}

}