PhabricatorFactDimension.php
e3a1a32444d0master

PhabricatorFactDimension.php
View Options

			<?php

			abstract class PhabricatorFactDimension extends PhabricatorFactDAO {

			abstract protected function getDimensionColumnName();

			final public function newDimensionID($key) {
			$map = $this->newDimensionMap(array($key));
			return idx($map, $key);
			}

			final public function newDimensionMap(array $keys, $create = false) {
			if (!$keys) {
			return array();
			}

			$conn = $this->establishConnection('r');
			$column = $this->getDimensionColumnName();

			$rows = queryfx_all(
			$conn,
			'SELECT id, %C FROM %T WHERE %C IN (%Ls)',
			$column,
			$this->getTableName(),
			$column,
			$keys);
			$rows = ipull($rows, 'id', $column);

			$map = array();
			$need = array();
			foreach ($keys as $key) {
			if (isset($rows[$key])) {
			$map[$key] = (int)$rows[$key];
			} else {
			$need[] = $key;
			}
			}

			if (!$need) {
			return $map;
			}

			if (!$create) {
			return $map;
			}

			$sql = array();
			foreach ($need as $key) {
			$sql[] = qsprintf(
			$conn,
			'(%s)',
			$key);
			}

			foreach (PhabricatorLiskDAO::chunkSQL($sql) as $chunk) {
			queryfx(
			$conn,
			'INSERT IGNORE INTO %T (%C) VALUES %Q',
			$this->getTableName(),
			$column,
			$chunk);
			}

			$rows = queryfx_all(
			$conn,
			'SELECT id, %C FROM %T WHERE %C IN (%Ls)',
			$column,
			$this->getTableName(),
			$column,
			$need);
			$rows = ipull($rows, 'id', $column);

			foreach ($need as $key) {
			if (isset($rows[$key])) {
			$map[$key] = (int)$rows[$key];
			} else {
			throw new Exception(
			pht(
			'Failed to load or generate dimension ID ("%s") for dimension '.
			'key "%s".',
			get_class($this),
			$key));
			}
			}

			return $map;
			}

			}