12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397 |
- <?php
- // Copyright (C) 2010-2017 Combodo SARL
- //
- // This file is part of iTop.
- //
- // iTop is free software; you can redistribute it and/or modify
- // it under the terms of the GNU Affero General Public License as published by
- // the Free Software Foundation, either version 3 of the License, or
- // (at your option) any later version.
- //
- // iTop is distributed in the hope that it will be useful,
- // but WITHOUT ANY WARRANTY; without even the implied warranty of
- // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- // GNU Affero General Public License for more details.
- //
- // You should have received a copy of the GNU Affero General Public License
- // along with iTop. If not, see <http://www.gnu.org/licenses/>
- require_once('dbobjectiterator.php');
- /**
- * Object set management
- *
- * @copyright Copyright (C) 2010-2017 Combodo SARL
- * @license http://opensource.org/licenses/AGPL-3.0
- */
- /**
- * A set of persistent objects, could be heterogeneous as long as the objects in the set have a common ancestor class
- *
- * @package iTopORM
- */
- class DBObjectSet implements iDBObjectSetIterator
- {
- /**
- * @var array
- */
- protected $m_aAddedIds; // Ids of objects added (discrete lists)
- /**
- * @var hash array of (row => array of (classalias) => object/null) storing the objects added "in memory"
- */
- protected $m_aAddedObjects;
- /**
- * @var array
- */
- protected $m_aArgs;
- /**
- * @var array
- */
- protected $m_aAttToLoad;
- /**
- * @var array
- */
- protected $m_aOrderBy;
- /**
- * @var bool True when the filter has been used OR the set is built step by step (AddObject...)
- */
- public $m_bLoaded;
- /**
- * @var int Total number of rows for the query without LIMIT. null if unknown yet
- */
- protected $m_iNumTotalDBRows;
- /**
- * @var int Total number of rows LOADED in $this->m_oSQLResult via a SQL query. 0 by default
- */
- protected $m_iNumLoadedDBRows;
- /**
- * @var int
- */
- protected $m_iCurrRow;
- /*
- * @var DBSearch
- */
- protected $m_oFilter;
- /**
- * @var mysqli_result
- */
- protected $m_oSQLResult;
- /**
- * Create a new set based on a Search definition.
- *
- * @param DBSearch $oFilter The search filter defining the objects which are part of the set (multiple columns/objects per row are supported)
- * @param hash $aOrderBy Array of '[<classalias>.]attcode' => bAscending
- * @param hash $aArgs Values to substitute for the search/query parameters (if any). Format: param_name => value
- * @param hash $aExtendedDataSpec
- * @param int $iLimitCount Maximum number of rows to load (i.e. equivalent to MySQL's LIMIT start, count)
- * @param int $iLimitStart Index of the first row to load (i.e. equivalent to MySQL's LIMIT start, count)
- */
- public function __construct(DBSearch $oFilter, $aOrderBy = array(), $aArgs = array(), $aExtendedDataSpec = null, $iLimitCount = 0, $iLimitStart = 0)
- {
- $this->m_oFilter = $oFilter->DeepClone();
- $this->m_aAddedIds = array();
- $this->m_aOrderBy = $aOrderBy;
- $this->m_aArgs = $aArgs;
- $this->m_aAttToLoad = null;
- $this->m_aExtendedDataSpec = $aExtendedDataSpec;
- $this->m_iLimitCount = $iLimitCount;
- $this->m_iLimitStart = $iLimitStart;
- $this->m_iNumTotalDBRows = null;
- $this->m_iNumLoadedDBRows = 0;
- $this->m_bLoaded = false;
- $this->m_aAddedObjects = array();
- $this->m_iCurrRow = 0;
- $this->m_oSQLResult = null;
- }
- public function __destruct()
- {
- if (is_object($this->m_oSQLResult))
- {
- $this->m_oSQLResult->free();
- }
- }
- public function __toString()
- {
- $sRet = '';
- $this->Rewind();
- $sRet .= "Set (".$this->m_oFilter->ToOQL().")<br/>\n";
- $sRet .= "Query: <pre style=\"font-size: smaller; display:inline;\">".$this->m_oFilter->MakeSelectQuery().")</pre>\n";
-
- $sRet .= $this->Count()." records<br/>\n";
- if ($this->Count() > 0)
- {
- $sRet .= "<ul class=\"treeview\">\n";
- while ($oObj = $this->Fetch())
- {
- $sRet .= "<li>".$oObj->__toString()."</li>\n";
- }
- $sRet .= "</ul>\n";
- }
- return $sRet;
- }
- public function __clone()
- {
- $this->m_oFilter = $this->m_oFilter->DeepClone();
- $this->m_iNumTotalDBRows = null; // Total number of rows for the query without LIMIT. null if unknown yet
- $this->m_iNumLoadedDBRows = 0; // Total number of rows LOADED in $this->m_oSQLResult via a SQL query. 0 by default
- $this->m_bLoaded = false; // true when the filter has been used OR the set is built step by step (AddObject...)
- $this->m_iCurrRow = 0;
- $this->m_oSQLResult = null;
- }
- /**
- * Called when unserializing a DBObjectSet
- */
- public function __wakeup()
- {
- $this->m_iNumTotalDBRows = null; // Total number of rows for the query without LIMIT. null if unknown yet
- $this->m_iNumLoadedDBRows = 0; // Total number of rows LOADED in $this->m_oSQLResult via a SQL query. 0 by default
- $this->m_bLoaded = false; // true when the filter has been used OR the set is built step by step (AddObject...)
- $this->m_iCurrRow = 0;
- $this->m_oSQLResult = null;
- }
- public function SetShowObsoleteData($bShow)
- {
- $this->m_oFilter->SetShowObsoleteData($bShow);
- }
- public function GetShowObsoleteData()
- {
- return $this->m_oFilter->GetShowObsoleteData();
- }
- /**
- * Specify the subset of attributes to load (for each class of objects) before performing the SQL query for retrieving the rows from the DB
- *
- * @param hash $aAttToLoad Format: alias => array of attribute_codes
- *
- * @return void
- */
- public function OptimizeColumnLoad($aAttToLoad)
- {
- if (is_null($aAttToLoad))
- {
- $this->m_aAttToLoad = null;
- }
- else
- {
- // Complete the attribute list with the attribute codes
- $aAttToLoadWithAttDef = array();
- foreach($this->m_oFilter->GetSelectedClasses() as $sClassAlias => $sClass)
- {
- $aAttToLoadWithAttDef[$sClassAlias] = array();
- if (array_key_exists($sClassAlias, $aAttToLoad))
- {
- $aAttList = $aAttToLoad[$sClassAlias];
- foreach($aAttList as $sAttToLoad)
- {
- $oAttDef = MetaModel::GetAttributeDef($sClass, $sAttToLoad);
- $aAttToLoadWithAttDef[$sClassAlias][$sAttToLoad] = $oAttDef;
- if ($oAttDef->IsExternalKey())
- {
- // Add the external key friendly name anytime
- $oFriendlyNameAttDef = MetaModel::GetAttributeDef($sClass, $sAttToLoad.'_friendlyname');
- $aAttToLoadWithAttDef[$sClassAlias][$sAttToLoad.'_friendlyname'] = $oFriendlyNameAttDef;
- }
- }
- }
- // Add the friendly name anytime
- $oFriendlyNameAttDef = MetaModel::GetAttributeDef($sClass, 'friendlyname');
- $aAttToLoadWithAttDef[$sClassAlias]['friendlyname'] = $oFriendlyNameAttDef;
- // Make sure that the final class is requested anytime, whatever the specification (needed for object construction!)
- if (!MetaModel::IsStandaloneClass($sClass) && !array_key_exists('finalclass', $aAttToLoadWithAttDef[$sClassAlias]))
- {
- $aAttToLoadWithAttDef[$sClassAlias]['finalclass'] = MetaModel::GetAttributeDef($sClass, 'finalclass');
- }
- }
- $this->m_aAttToLoad = $aAttToLoadWithAttDef;
- }
- }
- /**
- * Create a set (in-memory) containing just the given object
- *
- * @param DBobject $oObject
- *
- * @return DBObjectSet The singleton set
- */
- static public function FromObject($oObject)
- {
- $oRetSet = self::FromScratch(get_class($oObject));
- $oRetSet->AddObject($oObject);
- return $oRetSet;
- }
- /**
- * Create an empty set (in-memory), for the given class (and its subclasses) of objects
- *
- * @param string $sClass The class (or an ancestor) for the objects to be added in this set
- *
- * @return DBObjectSet The empty set
- */
- static public function FromScratch($sClass)
- {
- $oFilter = new DBObjectSearch($sClass);
- $oFilter->AddConditionExpression(new FalseExpression());
- $oRetSet = new self($oFilter);
- $oRetSet->m_bLoaded = true; // no DB load
- $oRetSet->m_iNumTotalDBRows = 0; // Nothing from the DB
- return $oRetSet;
- }
- /**
- * Create a set (in-memory) with just one column (i.e. one object per row) and filled with the given array of objects
- *
- * @param string $sClass The class of the objects (must be a common ancestor to all objects in the set)
- * @param array $aObjects The list of objects to add into the set
- *
- * @return DBObjectSet
- */
- static public function FromArray($sClass, $aObjects)
- {
- $oRetSet = self::FromScratch($sClass);
- $oRetSet->AddObjectArray($aObjects, $sClass);
- return $oRetSet;
- }
- /**
- * Create a set in-memory with several classes of objects per row (with one alias per "column")
- *
- * Limitation:
- * The filter/OQL query representing such a set can not be rebuilt (only the first column will be taken into account)
- *
- * @param hash $aClasses Format: array of (alias => class)
- * @param hash $aObjects Format: array of (array of (classalias => object))
- *
- * @return DBObjectSet
- */
- static public function FromArrayAssoc($aClasses, $aObjects)
- {
- // In a perfect world, we should create a complete tree of DBObjectSearch,
- // but as we lack most of the information related to the objects,
- // let's create one search definition corresponding only to the first column
- $sClass = reset($aClasses);
- $sAlias = key($aClasses);
- $oFilter = new DBObjectSearch($sClass, $sAlias);
- $oRetSet = new self($oFilter);
- $oRetSet->m_bLoaded = true; // no DB load
- $oRetSet->m_iNumTotalDBRows = 0; // Nothing from the DB
-
- foreach($aObjects as $rowIndex => $aObjectsByClassAlias)
- {
- $oRetSet->AddObjectExtended($aObjectsByClassAlias);
- }
- return $oRetSet;
- }
- static public function FromLinkSet($oObject, $sLinkSetAttCode, $sExtKeyToRemote)
- {
- $oLinkAttCode = MetaModel::GetAttributeDef(get_class($oObject), $sLinkSetAttCode);
- $oExtKeyAttDef = MetaModel::GetAttributeDef($oLinkAttCode->GetLinkedClass(), $sExtKeyToRemote);
- $sTargetClass = $oExtKeyAttDef->GetTargetClass();
- $oLinkSet = $oObject->Get($sLinkSetAttCode);
- $aTargets = array();
- while ($oLink = $oLinkSet->Fetch())
- {
- $aTargets[] = MetaModel::GetObject($sTargetClass, $oLink->Get($sExtKeyToRemote));
- }
- return self::FromArray($sTargetClass, $aTargets);
- }
- /**
- * Note: After calling this method, the set cursor will be at the end of the set. You might to rewind it.
- *
- * @param bool $bWithId
- * @return array
- */
- public function ToArray($bWithId = true)
- {
- $aRet = array();
- $this->Rewind();
- while ($oObject = $this->Fetch())
- {
- if ($bWithId)
- {
- $aRet[$oObject->GetKey()] = $oObject;
- }
- else
- {
- $aRet[] = $oObject;
- }
- }
- return $aRet;
- }
- public function ToArrayOfValues()
- {
- if (!$this->m_bLoaded) $this->Load();
- $this->Rewind();
- $aSelectedClasses = $this->m_oFilter->GetSelectedClasses();
- $aRet = array();
- $iRow = 0;
- while($aObjects = $this->FetchAssoc())
- {
- foreach($aObjects as $sClassAlias => $oObject)
- {
- if (is_null($oObject))
- {
- $aRet[$iRow][$sClassAlias.'.'.'id'] = null;
- }
- else
- {
- $aRet[$iRow][$sClassAlias.'.'.'id'] = $oObject->GetKey();
- }
- if (is_null($oObject))
- {
- $sClass = $aSelectedClasses[$sClassAlias];
- }
- else
- {
- $sClass = get_class($oObject);
- }
- foreach(MetaModel::ListAttributeDefs($sClass) as $sAttCode => $oAttDef)
- {
- if ($oAttDef->IsScalar())
- {
- $sAttName = $sClassAlias.'.'.$sAttCode;
- if (is_null($oObject))
- {
- $aRet[$iRow][$sAttName] = null;
- }
- else
- {
- $aRet[$iRow][$sAttName] = $oObject->Get($sAttCode);
- }
- }
- }
- }
- $iRow++;
- }
- return $aRet;
- }
- /**
- * Note: After calling this method, the set cursor will be at the end of the set. You might to rewind it.
- *
- * @param string $sAttCode
- * @param bool $bWithId
- * @return array
- */
- public function GetColumnAsArray($sAttCode, $bWithId = true)
- {
- $aRet = array();
- $this->Rewind();
- while ($oObject = $this->Fetch())
- {
- if ($bWithId)
- {
- $aRet[$oObject->GetKey()] = $oObject->Get($sAttCode);
- }
- else
- {
- $aRet[] = $oObject->Get($sAttCode);
- }
- }
- return $aRet;
- }
- /**
- * Retrieve the DBSearch corresponding to the objects present in this set
- *
- * Limitation:
- * This method will NOT work for sets with several columns (i.e. several objects per row)
- *
- * @return DBObjectSearch
- */
- public function GetFilter()
- {
- // Make sure that we carry on the parameters of the set with the filter
- $oFilter = $this->m_oFilter->DeepClone();
- $oFilter->SetShowObsoleteData(true);
- // Note: the arguments found within a set can be object (but not in a filter)
- // That's why PrepareQueryArguments must be invoked there
- $oFilter->SetInternalParams(array_merge($oFilter->GetInternalParams(), $this->m_aArgs));
-
- if (count($this->m_aAddedIds) == 0)
- {
- return $oFilter;
- }
- else
- {
- $oIdListExpr = ListExpression::FromScalars(array_keys($this->m_aAddedIds));
- $oIdExpr = new FieldExpression('id', $oFilter->GetClassAlias());
- $oIdInList = new BinaryExpression($oIdExpr, 'IN', $oIdListExpr);
- $oFilter->MergeConditionExpression($oIdInList);
- return $oFilter;
- }
- }
- /**
- * The (common ancestor) class of the objects in the first column of this set
- *
- * @return string The class of the objects in the first column
- */
- public function GetClass()
- {
- return $this->m_oFilter->GetClass();
- }
- /**
- * The alias for the class of the objects in the first column of this set
- *
- * @return string The alias of the class in the first column
- */
- public function GetClassAlias()
- {
- return $this->m_oFilter->GetClassAlias();
- }
- /**
- * The list of all classes (one per column) which are part of this set
- *
- * @return hash Format: alias => class
- */
- public function GetSelectedClasses()
- {
- return $this->m_oFilter->GetSelectedClasses();
- }
- /**
- * The root class (i.e. highest ancestor in the MeaModel class hierarchy) for the first column on this set
- *
- * @return string The root class for the objects in the first column of the set
- */
- public function GetRootClass()
- {
- return MetaModel::GetRootClass($this->GetClass());
- }
- /**
- * The arguments used for building this set
- *
- * @return hash Format: parameter_name => value
- */
- public function GetArgs()
- {
- return $this->m_aArgs;
- }
- /**
- * Sets the limits for loading the rows from the DB. Equivalent to MySQL's LIMIT start,count clause.
- * @param int $iLimitCount The number of rows to load
- * @param int $iLimitStart The index of the first row to load
- */
- public function SetLimit($iLimitCount, $iLimitStart = 0)
- {
- $this->m_iLimitCount = $iLimitCount;
- $this->m_iLimitStart = $iLimitStart;
- }
- /**
- * Sets the sort order for loading the rows from the DB. Changing the order by causes a Reload.
- *
- * @param hash $aOrderBy Format: [alias.]attcode => boolean (true = ascending, false = descending)
- */
- public function SetOrderBy($aOrderBy)
- {
- if ($this->m_aOrderBy != $aOrderBy)
- {
- $this->m_aOrderBy = $aOrderBy;
- if ($this->m_bLoaded)
- {
- $this->m_bLoaded = false;
- $this->Load();
- }
- }
- }
- /**
- * Sets the sort order for loading the rows from the DB. Changing the order by causes a Reload.
- *
- * @param hash $aAliases Format: alias => boolean (true = ascending, false = descending). If omitted, then it defaults to all the selected classes
- */
- public function SetOrderByClasses($aAliases = null)
- {
- if ($aAliases === null)
- {
- $aAliases = array();
- foreach ($this->GetSelectedClasses() as $sAlias => $sClass)
- {
- $aAliases[$sAlias] = true;
- }
- }
- $aAttributes = array();
- foreach ($aAliases as $sAlias => $bClassDirection)
- {
- foreach (MetaModel::GetOrderByDefault($this->m_oFilter->GetClass($sAlias)) as $sAttCode => $bAttributeDirection)
- {
- $bDirection = $bClassDirection ? $bAttributeDirection : !$bAttributeDirection;
- $aAttributes[$sAlias.'.'.$sAttCode] = $bDirection;
- }
- }
- $this->SetOrderBy($aAttributes);
- }
- /**
- * Returns the 'count' limit for loading the rows from the DB
- *
- * @return int
- */
- public function GetLimitCount()
- {
- return $this->m_iLimitCount;
- }
- /**
- * Returns the 'start' limit for loading the rows from the DB
- *
- * @return int
- */
- public function GetLimitStart()
- {
- return $this->m_iLimitStart;
- }
- /**
- * Get the sort order used for loading this set from the database
- *
- * Limitation: the sort order has no effect on objects added in-memory
- *
- * @return hash Format: field_code => boolean (true = ascending, false = descending)
- */
- public function GetRealSortOrder()
- {
- // Get the class default sort order if not specified with the API
- //
- if (empty($this->m_aOrderBy))
- {
- return MetaModel::GetOrderByDefault($this->m_oFilter->GetClass());
- }
- else
- {
- return $this->m_aOrderBy;
- }
- }
- /**
- * Loads the set from the database. Actually performs the SQL query to retrieve the records from the DB.
- */
- public function Load()
- {
- if ($this->m_bLoaded) return;
- // Note: it is mandatory to set this value now, to protect against reentrance
- $this->m_bLoaded = true;
- if ($this->m_iLimitCount > 0)
- {
- $sSQL = $this->m_oFilter->MakeSelectQuery($this->GetRealSortOrder(), $this->m_aArgs, $this->m_aAttToLoad, $this->m_aExtendedDataSpec, $this->m_iLimitCount, $this->m_iLimitStart);
- }
- else
- {
- $sSQL = $this->m_oFilter->MakeSelectQuery($this->GetRealSortOrder(), $this->m_aArgs, $this->m_aAttToLoad, $this->m_aExtendedDataSpec);
- }
-
- if (is_object($this->m_oSQLResult))
- {
- // Free previous resultset if any
- $this->m_oSQLResult->free();
- $this->m_oSQLResult = null;
- }
- $this->m_iNumTotalDBRows = null;
-
- $this->m_oSQLResult = CMDBSource::Query($sSQL);
- if ($this->m_oSQLResult === false) return;
-
- if (($this->m_iLimitCount == 0) && ($this->m_iLimitStart == 0))
- {
- $this->m_iNumTotalDBRows = $this->m_oSQLResult->num_rows;
- }
- $this->m_iNumLoadedDBRows = $this->m_oSQLResult->num_rows;
- }
- /**
- * The total number of rows in this set. Independently of the SetLimit used for loading the set and taking into account the rows added in-memory.
- *
- * May actually perform the SQL query SELECT COUNT... if the set was not previously loaded, or loaded with a SetLimit
- *
- * @return int The total number of rows for this set.
- */
- public function Count()
- {
- if (is_null($this->m_iNumTotalDBRows))
- {
- $sSQL = $this->m_oFilter->MakeSelectQuery(array(), $this->m_aArgs, null, null, 0, 0, true);
- $resQuery = CMDBSource::Query($sSQL);
- if (!$resQuery) return 0;
-
- $aRow = CMDBSource::FetchArray($resQuery);
- CMDBSource::FreeResult($resQuery);
- $this->m_iNumTotalDBRows = $aRow['COUNT'];
- }
- return $this->m_iNumTotalDBRows + count($this->m_aAddedObjects); // Does it fix Trac #887 ??
- }
-
- /**
- * Number of rows available in memory (loaded from DB + added in memory)
- *
- * @return number The number of rows available for Fetch'ing
- */
- protected function CountLoaded()
- {
- return $this->m_iNumLoadedDBRows + count($this->m_aAddedObjects);
- }
- /**
- * Fetch the object (with the given class alias) at the current position in the set and move the cursor to the next position.
- *
- * @param string $sRequestedClassAlias The class alias to fetch (if there are several objects/classes per row)
- * @return DBObject The fetched object or null when at the end
- */
- public function Fetch($sRequestedClassAlias = '')
- {
- if (!$this->m_bLoaded) $this->Load();
- if ($this->m_iCurrRow >= $this->CountLoaded())
- {
- return null;
- }
-
- if (strlen($sRequestedClassAlias) == 0)
- {
- $sRequestedClassAlias = $this->m_oFilter->GetClassAlias();
- }
- if ($this->m_iCurrRow < $this->m_iNumLoadedDBRows)
- {
- // Pick the row from the database
- $aRow = CMDBSource::FetchArray($this->m_oSQLResult);
- foreach ($this->m_oFilter->GetSelectedClasses() as $sClassAlias => $sClass)
- {
- if ($sRequestedClassAlias == $sClassAlias)
- {
- if (is_null($aRow[$sClassAlias.'id']))
- {
- $oRetObj = null;
- }
- else
- {
- $oRetObj = MetaModel::GetObjectByRow($sClass, $aRow, $sClassAlias, $this->m_aAttToLoad, $this->m_aExtendedDataSpec);
- }
- break;
- }
- }
- }
- else
- {
- // Pick the row from the objects added *in memory*
- $oRetObj = $this->m_aAddedObjects[$this->m_iCurrRow - $this->m_iNumLoadedDBRows][$sRequestedClassAlias];
- }
- $this->m_iCurrRow++;
- return $oRetObj;
- }
- /**
- * Fetch the whole row of objects (if several classes have been specified in the query) and move the cursor to the next position
- *
- * @return hash A hash with the format 'classAlias' => $oObj representing the current row of the set. Returns null when at the end.
- */
- public function FetchAssoc()
- {
- if (!$this->m_bLoaded) $this->Load();
- if ($this->m_iCurrRow >= $this->CountLoaded())
- {
- return null;
- }
-
- if ($this->m_iCurrRow < $this->m_iNumLoadedDBRows)
- {
- // Pick the row from the database
- $aRow = CMDBSource::FetchArray($this->m_oSQLResult);
- $aRetObjects = array();
- foreach ($this->m_oFilter->GetSelectedClasses() as $sClassAlias => $sClass)
- {
- if (is_null($aRow[$sClassAlias.'id']))
- {
- $oObj = null;
- }
- else
- {
- $oObj = MetaModel::GetObjectByRow($sClass, $aRow, $sClassAlias, $this->m_aAttToLoad, $this->m_aExtendedDataSpec);
- }
- $aRetObjects[$sClassAlias] = $oObj;
- }
- }
- else
- {
- // Pick the row from the objects added *in memory*
- $aRetObjects = array();
- foreach ($this->m_oFilter->GetSelectedClasses() as $sClassAlias => $sClass)
- {
- $aRetObjects[$sClassAlias] = $this->m_aAddedObjects[$this->m_iCurrRow - $this->m_iNumLoadedDBRows][$sClassAlias];
- }
- }
- $this->m_iCurrRow++;
- return $aRetObjects;
- }
- /**
- * Position the cursor (for iterating in the set) to the first position (equivalent to Seek(0))
- */
- public function Rewind()
- {
- if ($this->m_bLoaded)
- {
- $this->Seek(0);
- }
- }
- /**
- * Position the cursor (for iterating in the set) to the given position
- *
- * @param int $iRow
- */
- public function Seek($iRow)
- {
- if (!$this->m_bLoaded) $this->Load();
- $this->m_iCurrRow = min($iRow, $this->Count());
- if ($this->m_iCurrRow < $this->m_iNumLoadedDBRows)
- {
- $this->m_oSQLResult->data_seek($this->m_iCurrRow);
- }
- return $this->m_iCurrRow;
- }
- /**
- * Add an object to the current set (in-memory only, nothing is written to the database)
- *
- * Limitation:
- * Sets with several objects per row are NOT supported
- *
- * @param DBObject $oObject The object to add
- * @param string $sClassAlias The alias for the class of the object
- */
- public function AddObject($oObject, $sClassAlias = '')
- {
- if (!$this->m_bLoaded) $this->Load();
- if (strlen($sClassAlias) == 0)
- {
- $sClassAlias = $this->m_oFilter->GetClassAlias();
- }
- $iNextPos = count($this->m_aAddedObjects);
- $this->m_aAddedObjects[$iNextPos][$sClassAlias] = $oObject;
- if (!is_null($oObject))
- {
- $this->m_aAddedIds[$oObject->GetKey()] = true;
- }
- }
- /**
- * Add a hash containig objects into the current set.
- *
- * The expected format for the hash is: $aObjectArray[$idx][$sClassAlias] => $oObject
- * Limitation:
- * The aliases MUST match the ones used in the current set
- * Only the ID of the objects associated to the first alias (column) is remembered.. in case we have to rebuild a filter
- *
- * @param hash $aObjectArray
- */
- protected function AddObjectExtended($aObjectArray)
- {
- if (!$this->m_bLoaded) $this->Load();
- $iNextPos = count($this->m_aAddedObjects);
-
- $sFirstAlias = $this->m_oFilter->GetClassAlias();
- foreach ($aObjectArray as $sClassAlias => $oObject)
- {
- $this->m_aAddedObjects[$iNextPos][$sClassAlias] = $oObject;
-
- if (!is_null($oObject) && ($sFirstAlias == $sClassAlias))
- {
- $this->m_aAddedIds[$oObject->GetKey()] = true;
- }
- }
- }
- /**
- * Add an array of objects into the current set
- *
- * Limitation:
- * Sets with several classes per row are not supported (use AddObjectExtended instead)
- *
- * @param array $aObjects The array of objects to add
- * @param string $sClassAlias The Alias of the class for the added objects
- */
- public function AddObjectArray($aObjects, $sClassAlias = '')
- {
- if (!$this->m_bLoaded) $this->Load();
- // #@# todo - add a check on the object class ?
- foreach ($aObjects as $oObj)
- {
- $this->AddObject($oObj, $sClassAlias);
- }
- }
- /**
- * Append a given set to the current object. (This method used to be named Merge)
- *
- * Limitation:
- * The added objects are not checked for duplicates (i.e. one cann add several times the same object, or add an object already present in the set).
- *
- * @param DBObjectSet $oObjectSet The set to append
- * @throws CoreException
- */
- public function Append(DBObjectSet $oObjectSet)
- {
- if ($this->GetRootClass() != $oObjectSet->GetRootClass())
- {
- throw new CoreException("Could not merge two objects sets if they don't have the same root class");
- }
- if (!$this->m_bLoaded) $this->Load();
- $oObjectSet->Seek(0);
- while ($oObject = $oObjectSet->Fetch())
- {
- $this->AddObject($oObject);
- }
- }
- /**
- * Create a set containing the objects present in both the current set and another specified set
- *
- * Limitations:
- * Will NOT work if only a subset of the sets was loaded with SetLimit.
- * Works only with sets made of objects loaded from the database since the comparison is based on the objects identifiers
- *
- * @param DBObjectSet $oObjectSet The set to intersect with. The current position inside the set will be lost (= at the end)
- * @throws CoreException
- * @return DBObjectSet A new set of objects, containing the objects present in both sets (based on their identifier)
- */
- public function CreateIntersect(DBObjectSet $oObjectSet)
- {
- if ($this->GetRootClass() != $oObjectSet->GetRootClass())
- {
- throw new CoreException("Could not 'intersect' two objects sets if they don't have the same root class");
- }
- if (!$this->m_bLoaded) $this->Load();
- $aId2Row = array();
- $iCurrPos = $this->m_iCurrRow; // Save the cursor
- $idx = 0;
- while($oObj = $this->Fetch())
- {
- $aId2Row[$oObj->GetKey()] = $idx;
- $idx++;
- }
-
- $oNewSet = DBObjectSet::FromScratch($this->GetClass());
- $oObjectSet->Seek(0);
- while ($oObject = $oObjectSet->Fetch())
- {
- if (array_key_exists($oObject->GetKey(), $aId2Row))
- {
- $oNewSet->AddObject($oObject);
- }
- }
- $this->Seek($iCurrPos); // Restore the cursor
- return $oNewSet;
- }
- /**
- * Compare two sets of objects to determine if their content is identical or not.
- *
- * Limitation:
- * Works only for sets of 1 column (i.e. one class of object selected)
- *
- * @param DBObjectSet $oObjectSet
- * @param array $aExcludeColumns The list of columns to exclude frop the comparison
- * @return boolean True if the sets are identical, false otherwise
- */
- public function HasSameContents(DBObjectSet $oObjectSet, $aExcludeColumns = array())
- {
- $oComparator = new DBObjectSetComparator($this, $oObjectSet, $aExcludeColumns);
- return $oComparator->SetsAreEquivalent();
- }
- /**
- * Build a new set (in memory) made of objects of the given set which are NOT present in the current set
- *
- * Limitations:
- * The objects inside the set must be written in the database since the comparison is based on their identifiers
- * Sets with several objects per row are NOT supported
- *
- * @param DBObjectSet $oObjectSet
- * @throws CoreException
- *
- * @return DBObjectSet The "delta" set.
- */
- public function CreateDelta(DBObjectSet $oObjectSet)
- {
- if ($this->GetRootClass() != $oObjectSet->GetRootClass())
- {
- throw new CoreException("Could not 'delta' two objects sets if they don't have the same root class");
- }
- if (!$this->m_bLoaded) $this->Load();
- $aId2Row = array();
- $iCurrPos = $this->m_iCurrRow; // Save the cursor
- $idx = 0;
- while($oObj = $this->Fetch())
- {
- $aId2Row[$oObj->GetKey()] = $idx;
- $idx++;
- }
- $oNewSet = DBObjectSet::FromScratch($this->GetClass());
- $oObjectSet->Seek(0);
- while ($oObject = $oObjectSet->Fetch())
- {
- if (!array_key_exists($oObject->GetKey(), $aId2Row))
- {
- $oNewSet->AddObject($oObject);
- }
- }
- $this->Seek($iCurrPos); // Restore the cursor
- return $oNewSet;
- }
- /**
- * Will be deprecated soon - use MetaModel::GetRelatedObjectsDown/Up instead to take redundancy into account
- */
- public function GetRelatedObjects($sRelCode, $iMaxDepth = 99)
- {
- $aRelatedObjs = array();
- $aVisited = array(); // optimization for consecutive calls of MetaModel::GetRelatedObjects
- $this->Seek(0);
- while ($oObject = $this->Fetch())
- {
- $aMore = $oObject->GetRelatedObjects($sRelCode, $iMaxDepth, $aVisited);
- foreach ($aMore as $sClass => $aRelated)
- {
- foreach ($aRelated as $iObj => $oObj)
- {
- if (!isset($aRelatedObjs[$sClass][$iObj]))
- {
- $aRelatedObjs[$sClass][$iObj] = $oObj;
- }
- }
- }
- }
- return $aRelatedObjs;
- }
- /**
- * Compute the "RelatedObjects" (forward or "down" direction) for the set
- * for the specified relation
- *
- * @param string $sRelCode The code of the relation to use for the computation
- * @param int $iMaxDepth Maximum recursion depth
- * @param boolean $bEnableReduncancy Whether or not to take into account the redundancy
- *
- * @return RelationGraph The graph of all the related objects
- */
- public function GetRelatedObjectsDown($sRelCode, $iMaxDepth = 99, $bEnableRedundancy = true)
- {
- $oGraph = new RelationGraph();
- $this->Rewind();
- while($oObj = $this->Fetch())
- {
- $oGraph->AddSourceObject($oObj);
- }
- $oGraph->ComputeRelatedObjectsDown($sRelCode, $iMaxDepth, $bEnableRedundancy);
- return $oGraph;
- }
-
- /**
- * Compute the "RelatedObjects" (reverse or "up" direction) for the set
- * for the specified relation
- *
- * @param string $sRelCode The code of the relation to use for the computation
- * @param int $iMaxDepth Maximum recursion depth
- * @param boolean $bEnableReduncancy Whether or not to take into account the redundancy
- *
- * @return RelationGraph The graph of all the related objects
- */
- public function GetRelatedObjectsUp($sRelCode, $iMaxDepth = 99, $bEnableRedundancy = true)
- {
- $oGraph = new RelationGraph();
- $this->Rewind();
- while($oObj = $this->Fetch())
- {
- $oGraph->AddSinkObject($oObj);
- }
- $oGraph->ComputeRelatedObjectsUp($sRelCode, $iMaxDepth, $bEnableRedundancy);
- return $oGraph;
- }
-
- /**
- * Builds an object that contains the values that are common to all the objects
- * in the set. If for a given attribute, objects in the set have various values
- * then the resulting object will contain null for this value.
- * @param $aValues Hash Output: the distribution of the values, in the set, for each attribute
- * @return DBObject The object with the common values
- */
- public function ComputeCommonObject(&$aValues)
- {
- $sClass = $this->GetClass();
- $aList = MetaModel::ListAttributeDefs($sClass);
- $aValues = array();
- foreach($aList as $sAttCode => $oAttDef)
- {
- if ($oAttDef->IsScalar())
- {
- $aValues[$sAttCode] = array();
- }
- }
- $this->Rewind();
- while($oObj = $this->Fetch())
- {
- foreach($aList as $sAttCode => $oAttDef)
- {
- if ($oAttDef->IsScalar() && $oAttDef->IsWritable())
- {
- $currValue = $oObj->Get($sAttCode);
- if (is_object($currValue)) continue; // Skip non scalar values...
- if(!array_key_exists($currValue, $aValues[$sAttCode]))
- {
- $aValues[$sAttCode][$currValue] = array('count' => 1, 'display' => $oObj->GetAsHTML($sAttCode));
- }
- else
- {
- $aValues[$sAttCode][$currValue]['count']++;
- }
- }
- }
- }
-
- foreach($aValues as $sAttCode => $aMultiValues)
- {
- if (count($aMultiValues) > 1)
- {
- uasort($aValues[$sAttCode], 'HashCountComparison');
- }
- }
-
-
- // Now create an object that has values for the homogenous values only
- $oCommonObj = new $sClass(); // @@ What if the class is abstract ?
- $aComments = array();
- $iFormId = cmdbAbstractObject::GetNextFormId(); // Identifier that prefixes all the form fields
- $sReadyScript = '';
- $aDependsOn = array();
- $sFormPrefix = '2_';
- foreach($aList as $sAttCode => $oAttDef)
- {
- if ($oAttDef->IsScalar() && $oAttDef->IsWritable())
- {
- if ($oAttDef->GetEditClass() == 'One Way Password')
- {
- $oCommonObj->Set($sAttCode, null);
- }
- else
- {
- $iCount = count($aValues[$sAttCode]);
- if ($iCount == 1)
- {
- // Homogenous value
- reset($aValues[$sAttCode]);
- $aKeys = array_keys($aValues[$sAttCode]);
- $currValue = $aKeys[0]; // The only value is the first key
- $oCommonObj->Set($sAttCode, $currValue);
- }
- else
- {
- // Non-homogenous value
- $oCommonObj->Set($sAttCode, null);
- }
- }
- }
- }
- $this->Rewind();
- return $oCommonObj;
- }
- /**
- * List the constant fields (and their value) in the given query
- * @return Hash [Alias][AttCode] => value
- */
- public function ListConstantFields()
- {
- // The complete list of arguments will include magic arguments (e.g. current_user->attcode)
- $aScalarArgs = MetaModel::PrepareQueryArguments($this->m_oFilter->GetInternalParams(), $this->m_aArgs);
- $aConst = $this->m_oFilter->ListConstantFields();
-
- foreach($aConst as $sClassAlias => $aVals)
- {
- foreach($aVals as $sCode => $oExpr)
- {
- if (is_object($oExpr)) // Array_merge_recursive creates an array when the same key is present multiple times... ignore them
- {
- $oScalarExpr = $oExpr->GetAsScalar($aScalarArgs);
- $aConst[$sClassAlias][$sCode] = $oScalarExpr->GetValue();
- }
- }
- }
- return $aConst;
- }
-
- public function ApplyParameters()
- {
- $aAllArgs = MetaModel::PrepareQueryArguments($this->m_oFilter->GetInternalParams(), $this->m_aArgs);
- $this->m_oFilter->ApplyParameters($aAllArgs);
- }
- }
- /**
- * Helper function to perform a custom sort of a hash array
- */
- function HashCountComparison($a, $b) // Sort descending on 'count'
- {
- if ($a['count'] == $b['count'])
- {
- return 0;
- }
- return ($a['count'] > $b['count']) ? -1 : 1;
- }
- /**
- * Helper class to compare the content of two DBObjectSets based on the fingerprints of the contained objects
- * The FIRST SET MUST BE LOADED FROM THE DATABASE, the second one can be a set of objects in memory
- * When computing the actual differences, the algorithm tries to preserve as much as possible the EXISTING
- * objects (i.e. prefers 'modified' to 'removed' + 'added')
- *
- * LIMITATIONS:
- * - only DBObjectSets with one column (i.e. one class of object selected) are supported
- * - the first set must be the one loaded from the database
- */
- class DBObjectSetComparator
- {
- protected $aFingerprints1;
- protected $aFingerprints2;
- protected $aIDs1;
- protected $aIDs2;
- protected $aExcludedColumns;
- /**
- * @var iDBObjectSetIterator
- */
- protected $oSet1;
- /**
- * @var iDBObjectSetIterator
- */
- protected $oSet2;
- protected $sAdditionalKeyColumn;
- protected $aAdditionalKeys;
-
- /**
- * Initializes the comparator
- * @param iDBObjectSetIterator $oSet1 The first set of objects to compare, or null
- * @param iDBObjectSetIterator $oSet2 The second set of objects to compare, or null
- * @param array $aExcludedColumns The list of columns (= attribute codes) to exclude from the comparison
- * @param string $sAdditionalKeyColumn The attribute code of an additional column to be considered as a key indentifying the object (useful for n:n links)
- */
- public function __construct(iDBObjectSetIterator $oSet1, iDBObjectSetIterator $oSet2, $aExcludedColumns = array(), $sAdditionalKeyColumn = null)
- {
- $this->aFingerprints1 = null;
- $this->aFingerprints2 = null;
- $this->aIDs1 = array();
- $this->aIDs2 = array();
- $this->aExcludedColumns = $aExcludedColumns;
- $this->sAdditionalKeyColumn = $sAdditionalKeyColumn;
- $this->aAdditionalKeys = null;
- $this->oSet1 = $oSet1;
- $this->oSet2 = $oSet2;
- }
-
- /**
- * Builds the lists of fingerprints and initializes internal structures, if it was not already done
- */
- protected function ComputeFingerprints()
- {
- if ($this->aFingerprints1 === null)
- {
- $this->aFingerprints1 = array();
- $this->aFingerprints2 = array();
- $this->aAdditionalKeys = array();
-
- if ($this->oSet1 !== null)
- {
- $this->oSet1->Rewind();
- while($oObj = $this->oSet1->Fetch())
- {
- $sFingerprint = $oObj->Fingerprint($this->aExcludedColumns);
- $this->aFingerprints1[$sFingerprint] = $oObj;
- if (!$oObj->IsNew())
- {
- $this->aIDs1[$oObj->GetKey()] = $oObj;
- }
- }
- $this->oSet1->Rewind();
- }
-
- if ($this->oSet2 !== null)
- {
- $this->oSet2->Rewind();
- while($oObj = $this->oSet2->Fetch())
- {
- $sFingerprint = $oObj->Fingerprint($this->aExcludedColumns);
- $this->aFingerprints2[$sFingerprint] = $oObj;
- if (!$oObj->IsNew())
- {
- $this->aIDs2[$oObj->GetKey()] = $oObj;
- }
-
- if ($this->sAdditionalKeyColumn !== null)
- {
- $this->aAdditionalKeys[$oObj->Get($this->sAdditionalKeyColumn)] = $oObj;
- }
- }
- $this->oSet2->Rewind();
- }
- }
- }
-
- /**
- * Tells if the sets are equivalent or not. Returns as soon as the first difference is found.
- * @return boolean true if the set have an equivalent content, false otherwise
- */
- public function SetsAreEquivalent()
- {
- if (($this->oSet1 === null) && ($this->oSet2 === null))
- {
- // Both sets are empty, they are equal
- return true;
- }
- else if (($this->oSet1 === null) || ($this->oSet2 === null))
- {
- // one of them is empty, they are different
- return false;
- }
-
- if (($this->oSet1->GetRootClass() != $this->oSet2->GetRootClass()) || ($this->oSet1->Count() != $this->oSet2->Count())) return false;
-
- $this->ComputeFingerprints();
-
- // Check that all objects in Set1 are also in Set2
- foreach($this->aFingerprints1 as $sFingerprint => $oObj)
- {
- if (!array_key_exists($sFingerprint, $this->aFingerprints2))
- {
- return false;
- }
- }
-
- // Vice versa
- // Check that all objects in Set2 are also in Set1
- foreach($this->aFingerprints2 as $sFingerprint => $oObj)
- {
- if (!array_key_exists($sFingerprint, $this->aFingerprints1))
- {
- return false;
- }
- }
-
- return true;
- }
-
- /**
- * Get the list of differences between the two sets. In ordeer to write back into the database only the minimum changes
- * THE FIRST SET MUST BE THE ONE LOADED FROM THE DATABASE
- * Returns a hash: 'added' => DBObject(s), 'removed' => DBObject(s), 'modified' => DBObjects(s)
- * @return Ambigous <int:DBObject: , unknown>
- */
- public function GetDifferences()
- {
- $aResult = array('added' => array(), 'removed' => array(), 'modified' => array());
- $this->ComputeFingerprints();
-
- // Check that all objects in Set1 are also in Set2
- foreach($this->aFingerprints1 as $sFingerprint => $oObj)
- {
- // Beware: the elements from the first set MUST come from the database, otherwise the result will be irrelevant
- if ($oObj->IsNew()) throw new Exception('Cannot compute differences when elements from the first set are NOT in the database');
- if (array_key_exists($oObj->GetKey(), $this->aIDs2) && ($this->aIDs2[$oObj->GetKey()]->IsModified()))
- {
- // The very same object exists in both set, but was modified since its load
- $aResult['modified'][$oObj->GetKey()] = $this->aIDs2[$oObj->GetKey()];
- }
- else if (($this->sAdditionalKeyColumn !== null) && array_key_exists($oObj->Get($this->sAdditionalKeyColumn), $this->aAdditionalKeys))
- {
- // Special case for n:n links where the link is recreated between the very same 2 objects, but some of its attributes are modified
- // Let's consider this as a "modification" instead of "deletion" + "creation" in order to have a "clean" history for the objects
- $oDestObj = $this->aAdditionalKeys[$oObj->Get($this->sAdditionalKeyColumn)];
- $oCloneObj = $this->CopyFrom($oObj, $oDestObj);
- $aResult['modified'][$oObj->GetKey()] = $oCloneObj;
- // Mark this as processed, so that the pass on aFingerprints2 below ignores this object
- $sNewFingerprint = $oDestObj->Fingerprint($this->aExcludedColumns);
- $this->aFingerprints2[$sNewFingerprint] = $oCloneObj;
- }
- else if (!array_key_exists($sFingerprint, $this->aFingerprints2))
- {
- $aResult['removed'][] = $oObj;
- }
- }
-
- // Vice versa
- // Check that all objects in Set2 are also in Set1
- foreach($this->aFingerprints2 as $sFingerprint => $oObj)
- {
- if (array_key_exists($oObj->GetKey(), $this->aIDs1) && ($oObj->IsModified()))
- {
- // Already marked as modified above
- //$aResult['modified'][$oObj->GetKey()] = $oObj;
- }
- else if (!array_key_exists($sFingerprint, $this->aFingerprints1))
- {
- $aResult['added'][] = $oObj;
- }
- }
- return $aResult;
- }
-
- /**
- * Helpr to clone (in memory) an object and to apply to it the values taken from a second object
- * @param DBObject $oObjToClone
- * @param DBObject $oObjWithValues
- * @return DBObject The modified clone
- */
- protected function CopyFrom($oObjToClone, $oObjWithValues)
- {
- $oObj = MetaModel::GetObject(get_class($oObjToClone), $oObjToClone->GetKey());
- foreach(MetaModel::ListAttributeDefs(get_class($oObj)) as $sAttCode => $oAttDef)
- {
- if (!in_array($sAttCode, $this->aExcludedColumns) && $oAttDef->IsWritable())
- {
- $oObj->Set($sAttCode, $oObjWithValues->Get($sAttCode));
- }
- }
- return $oObj;
- }
- }
|