Differential D21570 Diff 51344 resources/sql/autopatches/20210215.changeset.02.phid-populate.php

Changeset View

Standalone View

resources/sql/autopatches/20210215.changeset.02.phid-populate.php

	<?php			<?php

	$phid_type = DifferentialChangesetPHIDType::TYPECONST;			$phid_type = DifferentialChangesetPHIDType::TYPECONST;

	$changeset_table = new DifferentialChangeset();			$changeset_table = new DifferentialChangeset();

	$conn = $changeset_table->establishConnection('w');			$conn = $changeset_table->establishConnection('w');
	$table_name = $changeset_table->getTableName();			$table_name = $changeset_table->getTableName();

	$iterator = new LiskRawMigrationIterator($conn, $table_name);			$chunk_size = 4096;
	foreach ($iterator as $changeset_row) {
				$temporary_table = 'tmp_20210215_changeset_id_map';

				queryfx(
				$conn,
				'CREATE TEMPORARY TABLE %T (
				changeset_id INT UNSIGNED NOT NULL AUTO_INCREMENT PRIMARY KEY,
				changeset_phid VARBINARY(64) NOT NULL)',
				$temporary_table);

				$table_iterator = id(new LiskRawMigrationIterator($conn, $table_name))
				->setPageSize($chunk_size);

				$chunk_iterator = new PhutilChunkedIterator($table_iterator, $chunk_size);
				foreach ($chunk_iterator as $chunk) {

				$map = array();
				foreach ($chunk as $changeset_row) {
	$phid = $changeset_row['phid'];			$phid = $changeset_row['phid'];

	if (strlen($phid)) {			if (strlen($phid)) {
	continue;			continue;
	}			}

	$phid = PhabricatorPHID::generateNewPHID($phid_type);			$phid = PhabricatorPHID::generateNewPHID($phid_type);
				$id = $changeset_row['id'];

				$map[(int)$id] = $phid;
				}

				if (!$map) {
				continue;
				}

				$sql = array();
				foreach ($map as $changeset_id => $changeset_phid) {
				$sql[] = qsprintf(
				$conn,
				'(%d, %s)',
				$changeset_id,
				$changeset_phid);
				}

				queryfx(
				$conn,
				'TRUNCATE TABLE %T',
				$temporary_table);

				queryfx(
				$conn,
				'INSERT INTO %T (changeset_id, changeset_phid) VALUES %LQ',
				$temporary_table,
				$sql);

	queryfx(			queryfx(
	$conn,			$conn,
	'UPDATE %T SET phid = %s WHERE id = %d',			'UPDATE %T c JOIN %T x ON c.id = x.changeset_id
				SET c.phid = x.changeset_phid',
	$table_name,			$table_name,
	$phid,			$temporary_table);
	$changeset_row['id']);
	}			}