doctrine2/lib/Doctrine/Mapper/Joined.php

<?php 

class Doctrine_Mapper_Joined extends Doctrine_Mapper
{
    
    /**
     * inserts a record into database
     *
     * @param Doctrine_Record $record   record to be inserted
     * @return boolean
     * @todo Move to Doctrine_Table (which will become Doctrine_Mapper).
     */
    public function insert(Doctrine_Record $record)
    {
        $table = $this->_table;
                    
        $dataSet = $this->_formatDataSet($record);
        $component = $table->getComponentName();

        $classes = $table->getOption('joinedParents');
        array_unshift($classes, $component);

        foreach (array_reverse($classes) as $k => $parent) {
            if ($k === 0) {
                $rootRecord = new $parent();
                $rootRecord->merge($dataSet[$parent]);
                parent::insert($rootRecord);
                $record->assignIdentifier($rootRecord->identifier());
            } else {
                foreach ((array) $rootRecord->identifier() as $id => $value) {
                    $dataSet[$parent][$id] = $value;
                }
                $this->_conn->insert($this->_conn->getTable($parent), $dataSet[$parent]);
            }
        }

        return true;
    }
    
    /**
     * updates given record
     *
     * @param Doctrine_Record $record   record to be updated
     * @return boolean                  whether or not the update was successful
     * @todo Move to Doctrine_Table (which will become Doctrine_Mapper).
     */
    public function update(Doctrine_Record $record)
    {
        $event = new Doctrine_Event($record, Doctrine_Event::RECORD_UPDATE);
        $record->preUpdate($event);
        $table = $this->_table;
        $this->getRecordListener()->preUpdate($event);

        if ( ! $event->skipOperation) {
            $identifier = $record->identifier();                     
            $dataSet = $this->_formatDataSet($record);
            $component = $table->getComponentName();
            $classes = $table->getOption('joinedParents');
            array_unshift($classes, $component);

            foreach ($record as $field => $value) {
                if ($value instanceof Doctrine_Record) {
                    if ( ! $value->exists()) {
                        $value->save();
                    }
                    $record->set($field, $value->getIncremented());
                }
            }

            foreach (array_reverse($classes) as $class) {
                $parentTable = $this->_conn->getTable($class);
                $this->_conn->update($parentTable, $dataSet[$class], $identifier);
            }
            
            $record->assignIdentifier(true);
        }
        
        $this->getRecordListener()->postUpdate($event);
        $record->postUpdate($event);

        return true;
    }
    
    /**
     *
     *
     */
    public function getCustomJoins()
    {
        return $this->_table->getOption('joinedParents');
    }
    
    /**
     *
     */
    public function getDiscriminatorColumn($domainClassName)
    {
        $joinedParents = $this->_table->getOption('joinedParents');
        if (count($joinedParents) <= 0) {
            $inheritanceMap = $this->_table->getOption('inheritanceMap');
        } else {
            $inheritanceMap = $this->_conn->getTable(array_pop($joinedParents))->getOption('inheritanceMap');
        }
        return isset($inheritanceMap[$domainClassName]) ? $inheritanceMap[$domainClassName] : array();
    }
    
    /**
     *
     */
    public function getFieldNames()
    {
        if ($this->_fieldNames) {
            return $this->_fieldNames;
        }
        
        $fieldNames = $this->_table->getFieldNames();
        foreach ($this->_table->getOption('joinedParents') as $parent) {
            $fieldNames = array_merge($this->_conn->getTable($parent)->getFieldNames(),
                    $fieldNames);
        }
        $this->_fieldNames = $fieldNames;
        
        return $fieldNames;
    }
    
    /**
     * 
     */
    protected function _formatDataSet(Doctrine_Record $record)
    {
        $table = $this->_table;
        $dataSet = array();
        $component = $table->getComponentName();
        $array = $record->getPrepared();
        
        $classes = array_merge(array($component), $this->_table->getOption('joinedParents'));
        
        foreach ($classes as $class) {
            $table = $this->_conn->getTable($class);
            foreach ($table->getColumns() as $columnName => $definition) {
                if (isset($definition['primary'])) {
                    continue;
                }
                $fieldName = $table->getFieldName($columnName);
                $dataSet[$class][$fieldName] = isset($array[$fieldName]) ? $array[$fieldName] : null;
            }
        }
        
        return $dataSet;
    }
}