Completed
Pull Request — 4.0 (#55)
by Huberty
03:08
created

TDBMService::delete()   D

Complexity

Conditions 10
Paths 8

Size

Total Lines 38
Code Lines 26

Duplication

Lines 0
Ratio 0 %

Importance

Changes 5
Bugs 0 Features 0
Metric Value
c 5
b 0
f 0
dl 0
loc 38
rs 4.8196
cc 10
eloc 26
nc 8
nop 1

How to fix   Complexity   

Long Method

Small methods make your code easier to understand, in particular if combined with a good name. Besides, if your method is small, finding a good name is usually much easier.

For example, if you find yourself adding comments to a method's body, this is usually a good sign to extract the commented part to a new method, and use the comment as a starting point when coming up with a good name for this new method.

Commonly applied refactorings include:

1
<?php
2
3
/*
4
 Copyright (C) 2006-2016 David Négrier - THE CODING MACHINE
5
6
This program is free software; you can redistribute it and/or modify
7
it under the terms of the GNU General Public License as published by
8
the Free Software Foundation; either version 2 of the License, or
9
(at your option) any later version.
10
11
This program is distributed in the hope that it will be useful,
12
but WITHOUT ANY WARRANTY; without even the implied warranty of
13
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14
GNU General Public License for more details.
15
16
You should have received a copy of the GNU General Public License
17
along with this program; if not, write to the Free Software
18
Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
19
*/
20
21
namespace Mouf\Database\TDBM;
22
23
use Doctrine\Common\Cache\Cache;
24
use Doctrine\Common\Cache\VoidCache;
25
use Doctrine\DBAL\Connection;
26
use Doctrine\DBAL\Schema\Column;
27
use Doctrine\DBAL\Schema\ForeignKeyConstraint;
28
use Doctrine\DBAL\Schema\Schema;
29
use Doctrine\DBAL\Schema\Table;
30
use Mouf\Database\MagicQuery;
31
use Mouf\Database\SchemaAnalyzer\SchemaAnalyzer;
32
use Mouf\Database\TDBM\Filters\OrderBySQLString;
33
use Mouf\Database\TDBM\Utils\TDBMDaoGenerator;
34
35
/**
36
 * The TDBMService class is the main TDBM class. It provides methods to retrieve TDBMObject instances
37
 * from the database.
38
 *
39
 * @author David Negrier
40
 * @ExtendedAction {"name":"Generate DAOs", "url":"tdbmadmin/", "default":false}
41
 */
42
class TDBMService
43
{
44
    const MODE_CURSOR = 1;
45
    const MODE_ARRAY = 2;
46
47
    /**
48
     * The database connection.
49
     *
50
     * @var Connection
51
     */
52
    private $connection;
53
54
    /**
55
     * @var SchemaAnalyzer
56
     */
57
    private $schemaAnalyzer;
58
59
    /**
60
     * @var MagicQuery
61
     */
62
    private $magicQuery;
63
64
    /**
65
     * @var TDBMSchemaAnalyzer
66
     */
67
    private $tdbmSchemaAnalyzer;
68
69
    /**
70
     * @var string
71
     */
72
    private $cachePrefix;
73
74
    /**
75
     * Cache of table of primary keys.
76
     * Primary keys are stored by tables, as an array of column.
77
     * For instance $primary_key['my_table'][0] will return the first column of the primary key of table 'my_table'.
78
     *
79
     * @var string[]
80
     */
81
    private $primaryKeysColumns;
82
83
    /**
84
     * Service storing objects in memory.
85
     * Access is done by table name and then by primary key.
86
     * If the primary key is split on several columns, access is done by an array of columns, serialized.
87
     *
88
     * @var StandardObjectStorage|WeakrefObjectStorage
89
     */
90
    private $objectStorage;
91
92
    /**
93
     * The fetch mode of the result sets returned by `getObjects`.
94
     * Can be one of: TDBMObjectArray::MODE_CURSOR or TDBMObjectArray::MODE_ARRAY or TDBMObjectArray::MODE_COMPATIBLE_ARRAY.
95
     *
96
     * In 'MODE_ARRAY' mode (default), the result is an array. Use this mode by default (unless the list returned is very big).
97
     * In 'MODE_CURSOR' mode, the result is a Generator which is an iterable collection that can be scanned only once (only one "foreach") on it,
98
     * and it cannot be accessed via key. Use this mode for large datasets processed by batch.
99
     * In 'MODE_COMPATIBLE_ARRAY' mode, the result is an old TDBMObjectArray (used up to TDBM 3.2).
100
     * You can access the array by key, or using foreach, several times.
101
     *
102
     * @var int
103
     */
104
    private $mode = self::MODE_ARRAY;
105
106
    /**
107
     * Table of new objects not yet inserted in database or objects modified that must be saved.
108
     *
109
     * @var \SplObjectStorage of DbRow objects
110
     */
111
    private $toSaveObjects;
112
113
    /**
114
     * The content of the cache variable.
115
     *
116
     * @var array<string, mixed>
117
     */
118
    private $cache;
119
120
    /**
121
     * Map associating a table name to a fully qualified Bean class name.
122
     *
123
     * @var array
124
     */
125
    private $tableToBeanMap = [];
126
127
    /**
128
     * @var \ReflectionClass[]
129
     */
130
    private $reflectionClassCache = array();
131
132
    /**
133
     * @param Connection     $connection     The DBAL DB connection to use
134
     * @param Cache|null     $cache          A cache service to be used
135
     * @param SchemaAnalyzer $schemaAnalyzer The schema analyzer that will be used to find shortest paths...
136
     *                                       Will be automatically created if not passed.
137
     */
138
    public function __construct(Connection $connection, Cache $cache = null, SchemaAnalyzer $schemaAnalyzer = null)
139
    {
140
        if (extension_loaded('weakref')) {
141
            $this->objectStorage = new WeakrefObjectStorage();
142
        } else {
143
            $this->objectStorage = new StandardObjectStorage();
0 ignored issues
show
Documentation Bug introduced by
It seems like new \Mouf\Database\TDBM\StandardObjectStorage() of type object<Mouf\Database\TDBM\StandardObjectStorage> is incompatible with the declared type object<Mouf\Database\TDBM\WeakrefObjectStorage> of property $objectStorage.

Our type inference engine has found an assignment to a property that is incompatible with the declared type of that property.

Either this assignment is in error or the assigned type should be added to the documentation/type hint for that property..

Loading history...
144
        }
145
        $this->connection = $connection;
146
        if ($cache !== null) {
147
            $this->cache = $cache;
148
        } else {
149
            $this->cache = new VoidCache();
150
        }
151
        if ($schemaAnalyzer) {
152
            $this->schemaAnalyzer = $schemaAnalyzer;
153
        } else {
154
            $this->schemaAnalyzer = new SchemaAnalyzer($this->connection->getSchemaManager(), $this->cache, $this->getConnectionUniqueId());
155
        }
156
157
        $this->magicQuery = new MagicQuery($this->connection, $this->cache, $this->schemaAnalyzer);
158
159
        $this->tdbmSchemaAnalyzer = new TDBMSchemaAnalyzer($connection, $this->cache, $this->schemaAnalyzer);
160
        $this->cachePrefix = $this->tdbmSchemaAnalyzer->getCachePrefix();
161
162
        $this->toSaveObjects = new \SplObjectStorage();
163
    }
164
165
    /**
166
     * Returns the object used to connect to the database.
167
     *
168
     * @return Connection
169
     */
170
    public function getConnection()
171
    {
172
        return $this->connection;
173
    }
174
175
    /**
176
     * Creates a unique cache key for the current connection.
177
     *
178
     * @return string
179
     */
180
    private function getConnectionUniqueId()
181
    {
182
        return hash('md4', $this->connection->getHost().'-'.$this->connection->getPort().'-'.$this->connection->getDatabase().'-'.$this->connection->getDriver()->getName());
183
    }
184
185
    /**
186
     * Sets the default fetch mode of the result sets returned by `findObjects`.
187
     * Can be one of: TDBMObjectArray::MODE_CURSOR or TDBMObjectArray::MODE_ARRAY.
188
     *
189
     * In 'MODE_ARRAY' mode (default), the result is a ResultIterator object that behaves like an array. Use this mode by default (unless the list returned is very big).
190
     * In 'MODE_CURSOR' mode, the result is a ResultIterator object. If you scan it many times (by calling several time a foreach loop), the query will be run
191
     * several times. In cursor mode, you cannot access the result set by key. Use this mode for large datasets processed by batch.
192
     *
193
     * @param int $mode
194
     *
195
     * @return $this
196
     *
197
     * @throws TDBMException
198
     */
199
    public function setFetchMode($mode)
200
    {
201 View Code Duplication
        if ($mode !== self::MODE_CURSOR && $mode !== self::MODE_ARRAY) {
0 ignored issues
show
Duplication introduced by
This code seems to be duplicated across your project.

Duplicated code is one of the most pungent code smells. If you need to duplicate the same code in three or more different places, we strongly encourage you to look into extracting the code into a single class or operation.

You can also find more detailed suggestions in the “Code” section of your repository.

Loading history...
202
            throw new TDBMException("Unknown fetch mode: '".$this->mode."'");
203
        }
204
        $this->mode = $mode;
205
206
        return $this;
207
    }
208
209
    /**
210
     * Returns a TDBMObject associated from table "$table_name".
211
     * If the $filters parameter is an int/string, the object returned will be the object whose primary key = $filters.
212
     * $filters can also be a set of TDBM_Filters (see the findObjects method for more details).
213
     *
214
     * For instance, if there is a table 'users', with a primary key on column 'user_id' and a column 'user_name', then
215
     * 			$user = $tdbmService->getObject('users',1);
216
     * 			echo $user->name;
217
     * will return the name of the user whose user_id is one.
218
     *
219
     * If a table has a primary key over several columns, you should pass to $id an array containing the the value of the various columns.
220
     * For instance:
221
     * 			$group = $tdbmService->getObject('groups',array(1,2));
222
     *
223
     * Note that TDBMObject performs caching for you. If you get twice the same object, the reference of the object you will get
224
     * will be the same.
225
     *
226
     * For instance:
227
     * 			$user1 = $tdbmService->getObject('users',1);
228
     * 			$user2 = $tdbmService->getObject('users',1);
229
     * 			$user1->name = 'John Doe';
230
     * 			echo $user2->name;
231
     * will return 'John Doe'.
232
     *
233
     * You can use filters instead of passing the primary key. For instance:
234
     * 			$user = $tdbmService->getObject('users',new EqualFilter('users', 'login', 'jdoe'));
235
     * This will return the user whose login is 'jdoe'.
236
     * Please note that if 2 users have the jdoe login in database, the method will throw a TDBM_DuplicateRowException.
237
     *
238
     * Also, you can specify the return class for the object (provided the return class extends TDBMObject).
239
     * For instance:
240
     *  	$user = $tdbmService->getObject('users',1,'User');
241
     * will return an object from the "User" class. The "User" class must extend the "TDBMObject" class.
242
     * Please be sure not to override any method or any property unless you perfectly know what you are doing!
243
     *
244
     * @param string $table_name   The name of the table we retrieve an object from.
245
     * @param mixed  $filters      If the filter is a string/integer, it will be considered as the id of the object (the value of the primary key). Otherwise, it can be a filter bag (see the filterbag parameter of the findObjects method for more details about filter bags)
246
     * @param string $className    Optional: The name of the class to instanciate. This class must extend the TDBMObject class. If none is specified, a TDBMObject instance will be returned.
247
     * @param bool   $lazy_loading If set to true, and if the primary key is passed in parameter of getObject, the object will not be queried in database. It will be queried when you first try to access a column. If at that time the object cannot be found in database, an exception will be thrown.
248
     *
249
     * @return TDBMObject
250
     */
251
/*	public function getObject($table_name, $filters, $className = null, $lazy_loading = false) {
252
253
        if (is_array($filters) || $filters instanceof FilterInterface) {
254
            $isFilterBag = false;
255
            if (is_array($filters)) {
256
                // Is this a multiple primary key or a filter bag?
257
                // Let's have a look at the first item of the array to decide.
258
                foreach ($filters as $filter) {
259
                    if (is_array($filter) || $filter instanceof FilterInterface) {
260
                        $isFilterBag = true;
261
                    }
262
                    break;
263
                }
264
            } else {
265
                $isFilterBag = true;
266
            }
267
268
            if ($isFilterBag == true) {
269
                // If a filter bag was passer in parameter, let's perform a findObjects.
270
                $objects = $this->findObjects($table_name, $filters, null, null, null, $className);
271
                if (count($objects) == 0) {
272
                    return null;
273
                } elseif (count($objects) > 1) {
274
                    throw new DuplicateRowException("Error while querying an object for table '$table_name': ".count($objects)." rows have been returned, but we should have received at most one.");
275
                }
276
                // Return the first and only object.
277
                if ($objects instanceof \Generator) {
278
                    return $objects->current();
279
                } else {
280
                    return $objects[0];
281
                }
282
            }
283
        }
284
        $id = $filters;
285
        if ($this->connection == null) {
286
            throw new TDBMException("Error while calling TdbmService->getObject(): No connection has been established on the database!");
287
        }
288
        $table_name = $this->connection->toStandardcase($table_name);
289
290
        // If the ID is null, let's throw an exception
291
        if ($id === null) {
292
            throw new TDBMException("The ID you passed to TdbmService->getObject is null for the object of type '$table_name'. Objects primary keys cannot be null.");
293
        }
294
295
        // If the primary key is split over many columns, the IDs are passed in an array. Let's serialize this array to store it.
296
        if (is_array($id)) {
297
            $id = serialize($id);
298
        }
299
300
        if ($className === null) {
301
            if (isset($this->tableToBeanMap[$table_name])) {
302
                $className = $this->tableToBeanMap[$table_name];
303
            } else {
304
                $className = "Mouf\\Database\\TDBM\\TDBMObject";
305
            }
306
        }
307
308
        if ($this->objectStorage->has($table_name, $id)) {
309
            $obj = $this->objectStorage->get($table_name, $id);
310
            if (is_a($obj, $className)) {
311
                return $obj;
312
            } else {
313
                throw new TDBMException("Error! The object with ID '$id' for table '$table_name' has already been retrieved. The type for this object is '".get_class($obj)."'' which is not a subtype of '$className'");
314
            }
315
        }
316
317
        if ($className != "Mouf\\Database\\TDBM\\TDBMObject" && !is_subclass_of($className, "Mouf\\Database\\TDBM\\TDBMObject")) {
318
            if (!class_exists($className)) {
319
                throw new TDBMException("Error while calling TDBMService->getObject: The class ".$className." does not exist.");
320
            } else {
321
                throw new TDBMException("Error while calling TDBMService->getObject: The class ".$className." should extend TDBMObject.");
322
            }
323
        }
324
        $obj = new $className($this, $table_name, $id);
325
326
        if ($lazy_loading == false) {
327
            // If we are not doing lazy loading, let's load the object:
328
            $obj->_dbLoadIfNotLoaded();
329
        }
330
331
        $this->objectStorage->set($table_name, $id, $obj);
332
333
        return $obj;
334
    }*/
335
336
    /**
337
     * Removes the given object from database.
338
     * This cannot be called on an object that is not attached to this TDBMService
339
     * (will throw a TDBMInvalidOperationException).
340
     *
341
     * @param AbstractTDBMObject $object the object to delete.
342
     *
343
     * @throws TDBMException
344
     * @throws TDBMInvalidOperationException
345
     */
346
    public function delete(AbstractTDBMObject $object)
347
    {
348
        switch ($object->_getStatus()) {
349
            case TDBMObjectStateEnum::STATE_DELETED:
350
                // Nothing to do, object already deleted.
351
                return;
352
            case TDBMObjectStateEnum::STATE_DETACHED:
353
                throw new TDBMInvalidOperationException('Cannot delete a detached object');
354
            case TDBMObjectStateEnum::STATE_NEW:
355
                $this->deleteManyToManyRelationships($object);
356
                foreach ($object->_getDbRows() as $dbRow) {
357
                    $this->removeFromToSaveObjectList($dbRow);
358
                }
359
                break;
360
            case TDBMObjectStateEnum::STATE_DIRTY:
361
                foreach ($object->_getDbRows() as $dbRow) {
362
                    $this->removeFromToSaveObjectList($dbRow);
363
                }
364
                // And continue deleting...
0 ignored issues
show
Unused Code Comprehensibility introduced by
43% of this comment could be valid code. Did you maybe forget this after debugging?

Sometimes obsolete code just ends up commented out instead of removed. In this case it is better to remove the code once you have checked you do not need it.

The code might also have been commented out for debugging purposes. In this case it is vital that someone uncomments it again or your project may behave in very unexpected ways in production.

This check looks for comments that seem to be mostly valid code and reports them.

Loading history...
365
            case TDBMObjectStateEnum::STATE_NOT_LOADED:
366
            case TDBMObjectStateEnum::STATE_LOADED:
367
                $this->deleteManyToManyRelationships($object);
368
                // Let's delete db rows, in reverse order.
369
                foreach (array_reverse($object->_getDbRows()) as $dbRow) {
370
                    $tableName = $dbRow->_getDbTableName();
371
                    $primaryKeys = $dbRow->_getPrimaryKeys();
372
                    $this->connection->delete($tableName, $primaryKeys);
373
                    $this->objectStorage->remove($dbRow->_getDbTableName(), $this->getObjectHash($primaryKeys));
374
                }
375
                break;
376
            // @codeCoverageIgnoreStart
377
            default:
378
                throw new TDBMInvalidOperationException('Unexpected status for bean');
379
            // @codeCoverageIgnoreEnd
380
        }
381
382
        $object->_setStatus(TDBMObjectStateEnum::STATE_DELETED);
383
    }
384
385
    /**
386
     * Removes all many to many relationships for this object.
387
     *
388
     * @param AbstractTDBMObject $object
389
     */
390
    private function deleteManyToManyRelationships(AbstractTDBMObject $object)
391
    {
392
        foreach ($object->_getDbRows() as $tableName => $dbRow) {
393
            $pivotTables = $this->tdbmSchemaAnalyzer->getPivotTableLinkedToTable($tableName);
394
            foreach ($pivotTables as $pivotTable) {
395
                $remoteBeans = $object->_getRelationships($pivotTable);
396
                foreach ($remoteBeans as $remoteBean) {
397
                    $object->_removeRelationship($pivotTable, $remoteBean);
398
                }
399
            }
400
        }
401
        $this->persistManyToManyRelationships($object);
402
    }
403
404
    /**
405
     * This function removes the given object from the database. It will also remove all objects relied to the one given
406
     * by parameter before all.
407
     *
408
     * Notice: if the object has a multiple primary key, the function will not work.
409
     *
410
     * @param AbstractTDBMObject $objToDelete
411
     */
412
    public function deleteCascade(AbstractTDBMObject $objToDelete)
413
    {
414
        $this->deleteAllConstraintWithThisObject($objToDelete);
415
        $this->delete($objToDelete);
416
    }
417
418
    /**
419
     * This function is used only in TDBMService (private function)
420
     * It will call deleteCascade function foreach object relied with a foreign key to the object given by parameter.
421
     *
422
     * @param AbstractTDBMObject $obj
423
     */
424
    private function deleteAllConstraintWithThisObject(AbstractTDBMObject $obj)
425
    {
426
        $dbRows = $obj->_getDbRows();
427
        foreach ($dbRows as $dbRow) {
428
            $tableName = $dbRow->_getDbTableName();
429
            $pks = array_values($dbRow->_getPrimaryKeys());
430
            if (!empty($pks)) {
431
                $incomingFks = $this->tdbmSchemaAnalyzer->getIncomingForeignKeys($tableName);
432
433
                foreach ($incomingFks as $incomingFk) {
434
                    $filter = array_combine($incomingFk->getLocalColumns(), $pks);
435
436
                    $results = $this->findObjects($incomingFk->getLocalTableName(), $filter);
437
438
                    foreach ($results as $bean) {
439
                        $this->deleteCascade($bean);
440
                    }
441
                }
442
            }
443
        }
444
    }
445
446
    /**
447
     * This function performs a save() of all the objects that have been modified.
448
     */
449
    public function completeSave()
450
    {
451
        foreach ($this->toSaveObjects as $dbRow) {
452
            $this->save($dbRow->getTDBMObject());
453
        }
454
    }
455
456
    /**
457
     * Takes in input a filter_bag (which can be about anything from a string to an array of TDBMObjects... see above from documentation),
458
     * and gives back a proper Filter object.
459
     *
460
     * @param mixed $filter_bag
461
     * @param int   $counter
462
     *
463
     * @return array First item: filter string, second item: parameters.
464
     *
465
     * @throws TDBMException
466
     */
467
    public function buildFilterFromFilterBag($filter_bag, $counter = 1)
468
    {
469
        if ($filter_bag === null) {
470
            return ['', []];
471
        } elseif (is_string($filter_bag)) {
472
            return [$filter_bag, []];
473
        } elseif (is_array($filter_bag)) {
474
            $sqlParts = [];
475
            $parameters = [];
476
            foreach ($filter_bag as $column => $value) {
477
                if (is_int($column)) {
478
                    list($subSqlPart, $subParameters) = $this->buildFilterFromFilterBag($value, $counter);
479
                    $sqlParts[] = $subSqlPart;
480
                    $parameters += $subParameters;
481
                } else {
482
                    $paramName = 'tdbmparam'.$counter;
483
                    if (is_array($value)) {
484
                        $sqlParts[] = $this->connection->quoteIdentifier($column).' IN :'.$paramName;
485
                    } else {
486
                        $sqlParts[] = $this->connection->quoteIdentifier($column).' = :'.$paramName;
487
                    }
488
                    $parameters[$paramName] = $value;
489
                    ++$counter;
490
                }
491
            }
492
493
            return [implode(' AND ', $sqlParts), $parameters];
494
        } elseif ($filter_bag instanceof AbstractTDBMObject) {
495
            $sqlParts = [];
496
            $parameters = [];
497
            $dbRows = $filter_bag->_getDbRows();
498
            $dbRow = reset($dbRows);
499
            $primaryKeys = $dbRow->_getPrimaryKeys();
500
501
            foreach ($primaryKeys as $column => $value) {
502
                $paramName = 'tdbmparam'.$counter;
503
                $sqlParts[] = $this->connection->quoteIdentifier($dbRow->_getDbTableName()).'.'.$this->connection->quoteIdentifier($column).' = :'.$paramName;
504
                $parameters[$paramName] = $value;
505
                ++$counter;
506
            }
507
508
            return [implode(' AND ', $sqlParts), $parameters];
509
        } elseif ($filter_bag instanceof \Iterator) {
510
            return $this->buildFilterFromFilterBag(iterator_to_array($filter_bag), $counter);
511
        } else {
512
            throw new TDBMException('Error in filter. An object has been passed that is neither a SQL string, nor an array, nor a bean, nor null.');
513
        }
514
    }
515
516
    /**
517
     * Takes in input an order_bag (which can be about anything from a string to an array of OrderByColumn objects... see above from documentation),
518
     * and gives back an array of OrderByColumn / OrderBySQLString objects.
519
     *
520
     * @param unknown_type $orderby_bag
521
     *
522
     * @return array
523
     */
524
    public function buildOrderArrayFromOrderBag($orderby_bag)
525
    {
526
        // Fourth, let's apply the same steps to the orderby_bag
527
        // 4-1 orderby_bag should be an array, if it is a singleton, let's put it in an array.
528
529
        if (!is_array($orderby_bag)) {
530
            $orderby_bag = array($orderby_bag);
531
        }
532
533
        // 4-2, let's take all the objects out of the orderby bag, and let's make objects from them
534
        $orderby_bag2 = array();
535
        foreach ($orderby_bag as $thing) {
536
            if (is_a($thing, 'Mouf\\Database\\TDBM\\Filters\\OrderBySQLString')) {
537
                $orderby_bag2[] = $thing;
538
            } elseif (is_a($thing, 'Mouf\\Database\\TDBM\\Filters\\OrderByColumn')) {
539
                $orderby_bag2[] = $thing;
540
            } elseif (is_string($thing)) {
541
                $orderby_bag2[] = new OrderBySQLString($thing);
542
            } elseif ($thing !== null) {
543
                throw new TDBMException('Error in orderby bag in getObjectsByFilter. An object has been passed that is neither a OrderBySQLString, nor a OrderByColumn, nor a string, nor null.');
544
            }
545
        }
546
547
        return $orderby_bag2;
548
    }
549
550
    /**
551
     * @param string $table
552
     *
553
     * @return string[]
554
     */
555
    public function getPrimaryKeyColumns($table)
556
    {
557
        if (!isset($this->primaryKeysColumns[$table])) {
558
            $this->primaryKeysColumns[$table] = $this->tdbmSchemaAnalyzer->getSchema()->getTable($table)->getPrimaryKeyColumns();
559
560
            // TODO TDBM4: See if we need to improve error reporting if table name does not exist.
561
562
            /*$arr = array();
563
            foreach ($this->connection->getPrimaryKey($table) as $col) {
564
                $arr[] = $col->name;
565
            }
566
            // The primaryKeysColumns contains only the column's name, not the DB_Column object.
567
            $this->primaryKeysColumns[$table] = $arr;
568
            if (empty($this->primaryKeysColumns[$table]))
569
            {
570
                // Unable to find primary key.... this is an error
571
                // Let's try to be precise in error reporting. Let's try to find the table.
572
                $tables = $this->connection->checkTableExist($table);
573
                if ($tables === true)
574
                throw new TDBMException("Could not find table primary key for table '$table'. Please define a primary key for this table.");
575
                elseif ($tables !== null) {
576
                    if (count($tables)==1)
577
                    $str = "Could not find table '$table'. Maybe you meant this table: '".$tables[0]."'";
578
                    else
579
                    $str = "Could not find table '$table'. Maybe you meant one of those tables: '".implode("', '",$tables)."'";
580
                    throw new TDBMException($str);
581
                }
582
            }*/
583
        }
584
585
        return $this->primaryKeysColumns[$table];
586
    }
587
588
    /**
589
     * This is an internal function, you should not use it in your application.
590
     * This is used internally by TDBM to add an object to the object cache.
591
     *
592
     * @param DbRow $dbRow
593
     */
594
    public function _addToCache(DbRow $dbRow)
595
    {
596
        $primaryKey = $this->getPrimaryKeysForObjectFromDbRow($dbRow);
597
        $hash = $this->getObjectHash($primaryKey);
598
        $this->objectStorage->set($dbRow->_getDbTableName(), $hash, $dbRow);
599
    }
600
601
    /**
602
     * This is an internal function, you should not use it in your application.
603
     * This is used internally by TDBM to remove the object from the list of objects that have been
604
     * created/updated but not saved yet.
605
     *
606
     * @param DbRow $myObject
607
     */
608
    private function removeFromToSaveObjectList(DbRow $myObject)
609
    {
610
        unset($this->toSaveObjects[$myObject]);
611
    }
612
613
    /**
614
     * This is an internal function, you should not use it in your application.
615
     * This is used internally by TDBM to add an object to the list of objects that have been
616
     * created/updated but not saved yet.
617
     *
618
     * @param AbstractTDBMObject $myObject
619
     */
620
    public function _addToToSaveObjectList(DbRow $myObject)
621
    {
622
        $this->toSaveObjects[$myObject] = true;
623
    }
624
625
    /**
626
     * Generates all the daos and beans.
627
     *
628
     * @param string $daoFactoryClassName The classe name of the DAO factory
629
     * @param string $daonamespace        The namespace for the DAOs, without trailing \
630
     * @param string $beannamespace       The Namespace for the beans, without trailing \
631
     * @param bool   $storeInUtc          If the generated daos should store the date in UTC timezone instead of user's timezone.
632
     * @param string $composerFile        If it's set, location of custom Composer file. Relative to project root.
633
     *
634
     * @return \string[] the list of tables
635
     */
636
    public function generateAllDaosAndBeans($daoFactoryClassName, $daonamespace, $beannamespace, $storeInUtc, $composerFile = null)
637
    {
638
        $tdbmDaoGenerator = new TDBMDaoGenerator($this->schemaAnalyzer, $this->tdbmSchemaAnalyzer->getSchema(), $this->tdbmSchemaAnalyzer);
639
        if (null !== $composerFile) {
640
            $tdbmDaoGenerator->setComposerFile(__DIR__.'/../../../../../../../'.$composerFile);
641
        }
642
643
        return $tdbmDaoGenerator->generateAllDaosAndBeans($daoFactoryClassName, $daonamespace, $beannamespace, $storeInUtc);
644
    }
645
646
    /**
647
     * @param array<string, string> $tableToBeanMap
648
     */
649
    public function setTableToBeanMap(array $tableToBeanMap)
650
    {
651
        $this->tableToBeanMap = $tableToBeanMap;
652
    }
653
654
    /**
655
     * Saves $object by INSERTing or UPDAT(E)ing it in the database.
656
     *
657
     * @param AbstractTDBMObject $object
658
     *
659
     * @throws TDBMException
660
     */
661
    public function save(AbstractTDBMObject $object)
662
    {
663
        $status = $object->_getStatus();
664
665
        // Let's attach this object if it is in detached state.
666
        if ($status === TDBMObjectStateEnum::STATE_DETACHED) {
667
            $object->_attach($this);
668
            $status = $object->_getStatus();
669
        }
670
671
        if ($status === TDBMObjectStateEnum::STATE_NEW) {
672
            $dbRows = $object->_getDbRows();
673
674
            $unindexedPrimaryKeys = array();
675
676
            foreach ($dbRows as $dbRow) {
677
                $tableName = $dbRow->_getDbTableName();
678
679
                $schema = $this->tdbmSchemaAnalyzer->getSchema();
680
                $tableDescriptor = $schema->getTable($tableName);
681
682
                $primaryKeyColumns = $this->getPrimaryKeyColumns($tableName);
683
684
                if (empty($unindexedPrimaryKeys)) {
685
                    $primaryKeys = $this->getPrimaryKeysForObjectFromDbRow($dbRow);
686
                } else {
687
                    // First insert, the children must have the same primary key as the parent.
688
                    $primaryKeys = $this->_getPrimaryKeysFromIndexedPrimaryKeys($tableName, $unindexedPrimaryKeys);
689
                    $dbRow->_setPrimaryKeys($primaryKeys);
690
                }
691
692
                $references = $dbRow->_getReferences();
693
694
                // Let's save all references in NEW or DETACHED state (we need their primary key)
695
                foreach ($references as $fkName => $reference) {
696
                    $refStatus = $reference->_getStatus();
697
                    if ($refStatus === TDBMObjectStateEnum::STATE_NEW || $refStatus === TDBMObjectStateEnum::STATE_DETACHED) {
698
                        $this->save($reference);
699
                    }
700
                }
701
702
                $dbRowData = $dbRow->_getDbRow();
703
704
                // Let's see if the columns for primary key have been set before inserting.
705
                // We assume that if one of the value of the PK has been set, the PK is set.
706
                $isPkSet = !empty($primaryKeys);
707
708
                /*if (!$isPkSet) {
709
                    // if there is no autoincrement and no pk set, let's go in error.
710
                    $isAutoIncrement = true;
711
712
                    foreach ($primaryKeyColumns as $pkColumnName) {
713
                        $pkColumn = $tableDescriptor->getColumn($pkColumnName);
714
                        if (!$pkColumn->getAutoincrement()) {
715
                            $isAutoIncrement = false;
716
                        }
717
                    }
718
719
                    if (!$isAutoIncrement) {
720
                        $msg = "Error! You did not set the primary key(s) for the new object of type '$tableName'. The primary key is not set to 'autoincrement' so you must either set the primary key in the object or modify the DB model to create an primary key with auto-increment.";
721
                        throw new TDBMException($msg);
722
                    }
723
724
                }*/
725
726
                $types = [];
727
728
                foreach ($dbRowData as $columnName => $value) {
729
                    $columnDescriptor = $tableDescriptor->getColumn($columnName);
730
                    $types[] = $columnDescriptor->getType();
731
                }
732
733
                $this->connection->insert($tableName, $dbRowData, $types);
734
735
                if (!$isPkSet && count($primaryKeyColumns) == 1) {
736
                    $id = $this->connection->lastInsertId();
737
                    $primaryKeys[$primaryKeyColumns[0]] = $id;
738
                }
739
740
                // TODO: change this to some private magic accessor in future
741
                $dbRow->_setPrimaryKeys($primaryKeys);
742
                $unindexedPrimaryKeys = array_values($primaryKeys);
743
744
                /*
745
                 * When attached, on "save", we check if the column updated is part of a primary key
746
                 * If this is part of a primary key, we call the _update_id method that updates the id in the list of known objects.
747
                 * This method should first verify that the id is not already used (and is not auto-incremented)
748
                 *
749
                 * In the object, the key is stored in an array of  (column => value), that can be directly used to update the record.
750
                 *
751
                 *
752
                 */
753
754
                /*try {
0 ignored issues
show
Unused Code Comprehensibility introduced by
42% of this comment could be valid code. Did you maybe forget this after debugging?

Sometimes obsolete code just ends up commented out instead of removed. In this case it is better to remove the code once you have checked you do not need it.

The code might also have been commented out for debugging purposes. In this case it is vital that someone uncomments it again or your project may behave in very unexpected ways in production.

This check looks for comments that seem to be mostly valid code and reports them.

Loading history...
755
                    $this->db_connection->exec($sql);
756
                } catch (TDBMException $e) {
757
                    $this->db_onerror = true;
758
759
                    // Strange..... if we do not have the line below, bad inserts are not catched.
760
                    // It seems that destructors are called before the registered shutdown function (PHP >=5.0.5)
761
                    //if ($this->tdbmService->isProgramExiting())
762
                    //	trigger_error("program exiting");
763
                    trigger_error($e->getMessage(), E_USER_ERROR);
764
765
                    if (!$this->tdbmService->isProgramExiting())
766
                        throw $e;
767
                    else
768
                    {
769
                        trigger_error($e->getMessage(), E_USER_ERROR);
770
                    }
771
                }*/
772
773
                // Let's remove this object from the $new_objects static table.
774
                $this->removeFromToSaveObjectList($dbRow);
775
776
                // TODO: change this behaviour to something more sensible performance-wise
777
                // Maybe a setting to trigger this globally?
778
                //$this->status = TDBMObjectStateEnum::STATE_NOT_LOADED;
0 ignored issues
show
Unused Code Comprehensibility introduced by
40% of this comment could be valid code. Did you maybe forget this after debugging?

Sometimes obsolete code just ends up commented out instead of removed. In this case it is better to remove the code once you have checked you do not need it.

The code might also have been commented out for debugging purposes. In this case it is vital that someone uncomments it again or your project may behave in very unexpected ways in production.

This check looks for comments that seem to be mostly valid code and reports them.

Loading history...
779
                //$this->db_modified_state = false;
0 ignored issues
show
Unused Code Comprehensibility introduced by
50% of this comment could be valid code. Did you maybe forget this after debugging?

Sometimes obsolete code just ends up commented out instead of removed. In this case it is better to remove the code once you have checked you do not need it.

The code might also have been commented out for debugging purposes. In this case it is vital that someone uncomments it again or your project may behave in very unexpected ways in production.

This check looks for comments that seem to be mostly valid code and reports them.

Loading history...
780
                //$dbRow = array();
0 ignored issues
show
Unused Code Comprehensibility introduced by
63% of this comment could be valid code. Did you maybe forget this after debugging?

Sometimes obsolete code just ends up commented out instead of removed. In this case it is better to remove the code once you have checked you do not need it.

The code might also have been commented out for debugging purposes. In this case it is vital that someone uncomments it again or your project may behave in very unexpected ways in production.

This check looks for comments that seem to be mostly valid code and reports them.

Loading history...
781
782
                // Let's add this object to the list of objects in cache.
783
                $this->_addToCache($dbRow);
784
            }
785
786
            $object->_setStatus(TDBMObjectStateEnum::STATE_LOADED);
787
        } elseif ($status === TDBMObjectStateEnum::STATE_DIRTY) {
788
            $dbRows = $object->_getDbRows();
789
790
            foreach ($dbRows as $dbRow) {
791
                $references = $dbRow->_getReferences();
792
793
                // Let's save all references in NEW state (we need their primary key)
794
                foreach ($references as $fkName => $reference) {
795
                    if ($reference->_getStatus() === TDBMObjectStateEnum::STATE_NEW) {
796
                        $this->save($reference);
797
                    }
798
                }
799
800
                // Let's first get the primary keys
801
                $tableName = $dbRow->_getDbTableName();
802
                $dbRowData = $dbRow->_getDbRow();
803
804
                $schema = $this->tdbmSchemaAnalyzer->getSchema();
805
                $tableDescriptor = $schema->getTable($tableName);
806
807
                $primaryKeys = $dbRow->_getPrimaryKeys();
808
809
                $types = [];
810
811
                foreach ($dbRowData as $columnName => $value) {
812
                    $columnDescriptor = $tableDescriptor->getColumn($columnName);
813
                    $types[] = $columnDescriptor->getType();
814
                }
815
                foreach ($primaryKeys as $columnName => $value) {
816
                    $columnDescriptor = $tableDescriptor->getColumn($columnName);
817
                    $types[] = $columnDescriptor->getType();
818
                }
819
820
                $this->connection->update($tableName, $dbRowData, $primaryKeys, $types);
821
822
                // Let's check if the primary key has been updated...
823
                $needsUpdatePk = false;
824
                foreach ($primaryKeys as $column => $value) {
825
                    if (!isset($dbRowData[$column]) || $dbRowData[$column] != $value) {
826
                        $needsUpdatePk = true;
827
                        break;
828
                    }
829
                }
830
                if ($needsUpdatePk) {
831
                    $this->objectStorage->remove($tableName, $this->getObjectHash($primaryKeys));
832
                    $newPrimaryKeys = $this->getPrimaryKeysForObjectFromDbRow($dbRow);
833
                    $dbRow->_setPrimaryKeys($newPrimaryKeys);
834
                    $this->objectStorage->set($tableName, $this->getObjectHash($primaryKeys), $dbRow);
835
                }
836
837
                // Let's remove this object from the list of objects to save.
838
                $this->removeFromToSaveObjectList($dbRow);
839
            }
840
841
            $object->_setStatus(TDBMObjectStateEnum::STATE_LOADED);
842
        } elseif ($status === TDBMObjectStateEnum::STATE_DELETED) {
843
            throw new TDBMInvalidOperationException('This object has been deleted. It cannot be saved.');
844
        }
845
846
        // Finally, let's save all the many to many relationships to this bean.
847
        $this->persistManyToManyRelationships($object);
848
    }
849
850
    private function persistManyToManyRelationships(AbstractTDBMObject $object)
851
    {
852
        foreach ($object->_getCachedRelationships() as $pivotTableName => $storage) {
853
            $tableDescriptor = $this->tdbmSchemaAnalyzer->getSchema()->getTable($pivotTableName);
854
            list($localFk, $remoteFk) = $this->getPivotTableForeignKeys($pivotTableName, $object);
855
856
            foreach ($storage as $remoteBean) {
857
                /* @var $remoteBean AbstractTDBMObject */
858
                $statusArr = $storage[$remoteBean];
859
                $status = $statusArr['status'];
860
                $reverse = $statusArr['reverse'];
861
                if ($reverse) {
862
                    continue;
863
                }
864
865
                if ($status === 'new') {
866
                    $remoteBeanStatus = $remoteBean->_getStatus();
867
                    if ($remoteBeanStatus === TDBMObjectStateEnum::STATE_NEW || $remoteBeanStatus === TDBMObjectStateEnum::STATE_DETACHED) {
868
                        // Let's save remote bean if needed.
869
                        $this->save($remoteBean);
870
                    }
871
872
                    $filters = $this->getPivotFilters($object, $remoteBean, $localFk, $remoteFk);
873
874
                    $types = [];
875
876
                    foreach ($filters as $columnName => $value) {
877
                        $columnDescriptor = $tableDescriptor->getColumn($columnName);
878
                        $types[] = $columnDescriptor->getType();
879
                    }
880
881
                    $this->connection->insert($pivotTableName, $filters, $types);
882
883
                    // Finally, let's mark relationships as saved.
884
                    $statusArr['status'] = 'loaded';
885
                    $storage[$remoteBean] = $statusArr;
886
                    $remoteStorage = $remoteBean->_getCachedRelationships()[$pivotTableName];
887
                    $remoteStatusArr = $remoteStorage[$object];
888
                    $remoteStatusArr['status'] = 'loaded';
889
                    $remoteStorage[$object] = $remoteStatusArr;
890
                } elseif ($status === 'delete') {
891
                    $filters = $this->getPivotFilters($object, $remoteBean, $localFk, $remoteFk);
892
893
                    $types = [];
894
895
                    foreach ($filters as $columnName => $value) {
896
                        $columnDescriptor = $tableDescriptor->getColumn($columnName);
897
                        $types[] = $columnDescriptor->getType();
898
                    }
899
900
                    $this->connection->delete($pivotTableName, $filters, $types);
901
902
                    // Finally, let's remove relationships completely from bean.
903
                    $storage->detach($remoteBean);
904
                    $remoteBean->_getCachedRelationships()[$pivotTableName]->detach($object);
905
                }
906
            }
907
        }
908
    }
909
910
    private function getPivotFilters(AbstractTDBMObject $localBean, AbstractTDBMObject $remoteBean, ForeignKeyConstraint $localFk, ForeignKeyConstraint $remoteFk)
911
    {
912
        $localBeanPk = $this->getPrimaryKeyValues($localBean);
913
        $remoteBeanPk = $this->getPrimaryKeyValues($remoteBean);
914
        $localColumns = $localFk->getLocalColumns();
915
        $remoteColumns = $remoteFk->getLocalColumns();
916
917
        $localFilters = array_combine($localColumns, $localBeanPk);
918
        $remoteFilters = array_combine($remoteColumns, $remoteBeanPk);
919
920
        return array_merge($localFilters, $remoteFilters);
921
    }
922
923
    /**
924
     * Returns the "values" of the primary key.
925
     * This returns the primary key from the $primaryKey attribute, not the one stored in the columns.
926
     *
927
     * @param AbstractTDBMObject $bean
928
     *
929
     * @return array numerically indexed array of values.
930
     */
931
    private function getPrimaryKeyValues(AbstractTDBMObject $bean)
932
    {
933
        $dbRows = $bean->_getDbRows();
934
        $dbRow = reset($dbRows);
935
936
        return array_values($dbRow->_getPrimaryKeys());
937
    }
938
939
    /**
940
     * Returns a unique hash used to store the object based on its primary key.
941
     * If the array contains only one value, then the value is returned.
942
     * Otherwise, a hash representing the array is returned.
943
     *
944
     * @param array $primaryKeys An array of columns => values forming the primary key
945
     *
946
     * @return string
947
     */
948
    public function getObjectHash(array $primaryKeys)
949
    {
950
        if (count($primaryKeys) === 1) {
951
            return reset($primaryKeys);
952
        } else {
953
            ksort($primaryKeys);
954
955
            return md5(json_encode($primaryKeys));
956
        }
957
    }
958
959
    /**
960
     * Returns an array of primary keys from the object.
961
     * The primary keys are extracted from the object columns and not from the primary keys stored in the
962
     * $primaryKeys variable of the object.
963
     *
964
     * @param DbRow $dbRow
965
     *
966
     * @return array Returns an array of column => value
967
     */
968
    public function getPrimaryKeysForObjectFromDbRow(DbRow $dbRow)
969
    {
970
        $table = $dbRow->_getDbTableName();
971
        $dbRowData = $dbRow->_getDbRow();
972
973
        return $this->_getPrimaryKeysFromObjectData($table, $dbRowData);
974
    }
975
976
    /**
977
     * Returns an array of primary keys for the given row.
978
     * The primary keys are extracted from the object columns.
979
     *
980
     * @param $table
981
     * @param array $columns
982
     *
983
     * @return array
984
     */
985
    public function _getPrimaryKeysFromObjectData($table, array $columns)
986
    {
987
        $primaryKeyColumns = $this->getPrimaryKeyColumns($table);
988
        $values = array();
989
        foreach ($primaryKeyColumns as $column) {
990
            if (isset($columns[$column])) {
991
                $values[$column] = $columns[$column];
992
            }
993
        }
994
995
        return $values;
996
    }
997
998
    /**
999
     * Attaches $object to this TDBMService.
1000
     * The $object must be in DETACHED state and will pass in NEW state.
1001
     *
1002
     * @param AbstractTDBMObject $object
1003
     *
1004
     * @throws TDBMInvalidOperationException
1005
     */
1006
    public function attach(AbstractTDBMObject $object)
1007
    {
1008
        $object->_attach($this);
1009
    }
1010
1011
    /**
1012
     * Returns an associative array (column => value) for the primary keys from the table name and an
1013
     * indexed array of primary key values.
1014
     *
1015
     * @param string $tableName
1016
     * @param array  $indexedPrimaryKeys
1017
     */
1018
    public function _getPrimaryKeysFromIndexedPrimaryKeys($tableName, array $indexedPrimaryKeys)
1019
    {
1020
        $primaryKeyColumns = $this->tdbmSchemaAnalyzer->getSchema()->getTable($tableName)->getPrimaryKeyColumns();
1021
1022
        if (count($primaryKeyColumns) !== count($indexedPrimaryKeys)) {
1023
            throw new TDBMException(sprintf('Wrong number of columns passed for primary key. Expected %s columns for table "%s",
1024
			got %s instead.', count($primaryKeyColumns), $tableName, count($indexedPrimaryKeys)));
1025
        }
1026
1027
        return array_combine($primaryKeyColumns, $indexedPrimaryKeys);
1028
    }
1029
1030
    /**
1031
     * Return the list of tables (from child to parent) joining the tables passed in parameter.
1032
     * Tables must be in a single line of inheritance. The method will find missing tables.
1033
     *
1034
     * Algorithm: one of those tables is the ultimate child. From this child, by recursively getting the parent,
1035
     * we must be able to find all other tables.
1036
     *
1037
     * @param string[] $tables
1038
     *
1039
     * @return string[]
1040
     */
1041
    public function _getLinkBetweenInheritedTables(array $tables)
1042
    {
1043
        sort($tables);
1044
1045
        return $this->fromCache($this->cachePrefix.'_linkbetweeninheritedtables_'.implode('__split__', $tables),
1046
            function () use ($tables) {
1047
                return $this->_getLinkBetweenInheritedTablesWithoutCache($tables);
1048
            });
1049
    }
1050
1051
    /**
1052
     * Return the list of tables (from child to parent) joining the tables passed in parameter.
1053
     * Tables must be in a single line of inheritance. The method will find missing tables.
1054
     *
1055
     * Algorithm: one of those tables is the ultimate child. From this child, by recursively getting the parent,
1056
     * we must be able to find all other tables.
1057
     *
1058
     * @param string[] $tables
1059
     *
1060
     * @return string[]
1061
     */
1062
    private function _getLinkBetweenInheritedTablesWithoutCache(array $tables)
1063
    {
1064
        $schemaAnalyzer = $this->schemaAnalyzer;
1065
1066
        foreach ($tables as $currentTable) {
1067
            $allParents = [$currentTable];
1068
            while ($currentFk = $schemaAnalyzer->getParentRelationship($currentTable)) {
1069
                $currentTable = $currentFk->getForeignTableName();
1070
                $allParents[] = $currentTable;
1071
            };
1072
1073
            // Now, does the $allParents contain all the tables we want?
1074
            $notFoundTables = array_diff($tables, $allParents);
1075
            if (empty($notFoundTables)) {
1076
                // We have a winner!
1077
                return $allParents;
1078
            }
1079
        }
1080
1081
        throw new TDBMException(sprintf('The tables (%s) cannot be linked by an inheritance relationship.', implode(', ', $tables)));
1082
    }
1083
1084
    /**
1085
     * Returns the list of tables related to this table (via a parent or child inheritance relationship).
1086
     *
1087
     * @param string $table
1088
     *
1089
     * @return string[]
1090
     */
1091
    public function _getRelatedTablesByInheritance($table)
1092
    {
1093
        return $this->fromCache($this->cachePrefix.'_relatedtables_'.$table, function () use ($table) {
1094
            return $this->_getRelatedTablesByInheritanceWithoutCache($table);
1095
        });
1096
    }
1097
1098
    /**
1099
     * Returns the list of tables related to this table (via a parent or child inheritance relationship).
1100
     *
1101
     * @param string $table
1102
     *
1103
     * @return string[]
1104
     */
1105
    private function _getRelatedTablesByInheritanceWithoutCache($table)
1106
    {
1107
        $schemaAnalyzer = $this->schemaAnalyzer;
1108
1109
        // Let's scan the parent tables
1110
        $currentTable = $table;
1111
1112
        $parentTables = [];
1113
1114
        // Get parent relationship
1115
        while ($currentFk = $schemaAnalyzer->getParentRelationship($currentTable)) {
1116
            $currentTable = $currentFk->getForeignTableName();
1117
            $parentTables[] = $currentTable;
1118
        };
1119
1120
        // Let's recurse in children
1121
        $childrenTables = $this->exploreChildrenTablesRelationships($schemaAnalyzer, $table);
1122
1123
        return array_merge(array_reverse($parentTables), $childrenTables);
1124
    }
1125
1126
    /**
1127
     * Explore all the children and descendant of $table and returns ForeignKeyConstraints on those.
1128
     *
1129
     * @param string $table
1130
     *
1131
     * @return string[]
1132
     */
1133
    private function exploreChildrenTablesRelationships(SchemaAnalyzer $schemaAnalyzer, $table)
1134
    {
1135
        $tables = [$table];
1136
        $keys = $schemaAnalyzer->getChildrenRelationships($table);
1137
1138
        foreach ($keys as $key) {
1139
            $tables = array_merge($tables, $this->exploreChildrenTablesRelationships($schemaAnalyzer, $key->getLocalTableName()));
1140
        }
1141
1142
        return $tables;
1143
    }
1144
1145
    /**
1146
     * Casts a foreign key into SQL, assuming table name is used with no alias.
1147
     * The returned value does contain only one table. For instance:.
1148
     *
1149
     * " LEFT JOIN table2 ON table1.id = table2.table1_id"
1150
     *
1151
     * @param ForeignKeyConstraint $fk
1152
     * @param bool                 $leftTableIsLocal
1153
     *
1154
     * @return string
1155
     */
1156
    /*private function foreignKeyToSql(ForeignKeyConstraint $fk, $leftTableIsLocal) {
0 ignored issues
show
Unused Code Comprehensibility introduced by
59% of this comment could be valid code. Did you maybe forget this after debugging?

Sometimes obsolete code just ends up commented out instead of removed. In this case it is better to remove the code once you have checked you do not need it.

The code might also have been commented out for debugging purposes. In this case it is vital that someone uncomments it again or your project may behave in very unexpected ways in production.

This check looks for comments that seem to be mostly valid code and reports them.

Loading history...
1157
        $onClauses = [];
1158
        $foreignTableName = $this->connection->quoteIdentifier($fk->getForeignTableName());
1159
        $foreignColumns = $fk->getForeignColumns();
1160
        $localTableName = $this->connection->quoteIdentifier($fk->getLocalTableName());
1161
        $localColumns = $fk->getLocalColumns();
1162
        $columnCount = count($localTableName);
1163
1164
        for ($i = 0; $i < $columnCount; $i++) {
1165
            $onClauses[] = sprintf("%s.%s = %s.%s",
1166
                $localTableName,
1167
                $this->connection->quoteIdentifier($localColumns[$i]),
1168
                $foreignColumns,
1169
                $this->connection->quoteIdentifier($foreignColumns[$i])
1170
                );
1171
        }
1172
1173
        $onClause = implode(' AND ', $onClauses);
1174
1175
        if ($leftTableIsLocal) {
1176
            return sprintf(" LEFT JOIN %s ON (%s)", $foreignTableName, $onClause);
1177
        } else {
1178
            return sprintf(" LEFT JOIN %s ON (%s)", $localTableName, $onClause);
1179
        }
1180
    }*/
1181
1182
    /**
1183
     * Returns an identifier for the group of tables passed in parameter.
1184
     *
1185
     * @param string[] $relatedTables
1186
     *
1187
     * @return string
1188
     */
1189
    private function getTableGroupName(array $relatedTables)
1190
    {
1191
        sort($relatedTables);
1192
1193
        return implode('_``_', $relatedTables);
1194
    }
1195
1196
    /**
1197
     * Returns a `ResultIterator` object representing filtered records of "$mainTable" .
1198
     *
1199
     * The findObjects method should be the most used query method in TDBM if you want to query the database for objects.
1200
     * (Note: if you want to query the database for an object by its primary key, use the findObjectByPk method).
1201
     *
1202
     * The findObjects method takes in parameter:
1203
     * 	- mainTable: the kind of bean you want to retrieve. In TDBM, a bean matches a database row, so the
1204
     * 			`$mainTable` parameter should be the name of an existing table in database.
1205
     *  - filter: The filter is a filter bag. It is what you use to filter your request (the WHERE part in SQL).
1206
     *          It can be a string (SQL Where clause), or even a bean or an associative array (key = column to filter, value = value to find)
1207
     *  - parameters: The parameters used in the filter. If you pass a SQL string as a filter, be sure to avoid
1208
     *          concatenating parameters in the string (this leads to SQL injection and also to poor caching performance).
1209
     *          Instead, please consider passing parameters (see documentation for more details).
1210
     *  - additionalTablesFetch: An array of SQL tables names. The beans related to those tables will be fetched along
1211
     *          the main table. This is useful to avoid hitting the database with numerous subqueries.
1212
     *  - mode: The fetch mode of the result. See `setFetchMode()` method for more details.
1213
     *
1214
     * The `findObjects` method will return a `ResultIterator`. A `ResultIterator` is an object that behaves as an array
1215
     * (in ARRAY mode) at least. It can be iterated using a `foreach` loop.
1216
     *
1217
     * Finally, if filter_bag is null, the whole table is returned.
1218
     *
1219
     * @param string            $mainTable             The name of the table queried
1220
     * @param string|array|null $filter                The SQL filters to apply to the query (the WHERE part). All columns must be prefixed by the table name (in the form: table.column)
1221
     * @param array             $parameters
1222
     * @param string|null       $orderString           The ORDER BY part of the query. All columns must be prefixed by the table name (in the form: table.column)
1223
     * @param array             $additionalTablesFetch
1224
     * @param int               $mode
1225
     * @param string            $className             Optional: The name of the class to instantiate. This class must extend the TDBMObject class. If none is specified, a TDBMObject instance will be returned.
1226
     *
1227
     * @return ResultIterator An object representing an array of results.
1228
     *
1229
     * @throws TDBMException
1230
     */
1231
    public function findObjects($mainTable, $filter = null, array $parameters = array(), $orderString = null, array $additionalTablesFetch = array(), $mode = null, $className = null)
1232
    {
1233
        // $mainTable is not secured in MagicJoin, let's add a bit of security to avoid SQL injection.
1234
        if (!preg_match('/^[a-zA-Z_][a-zA-Z0-9_]*$/', $mainTable)) {
1235
            throw new TDBMException(sprintf("Invalid table name: '%s'", $mainTable));
1236
        }
1237
1238
        list($filterString, $additionalParameters) = $this->buildFilterFromFilterBag($filter);
1239
1240
        $parameters = array_merge($parameters, $additionalParameters);
1241
1242
        // From the table name and the additional tables we want to fetch, let's build a list of all tables
1243
        // that must be part of the select columns.
1244
1245
        $tableGroups = [];
1246
        $allFetchedTables = $this->_getRelatedTablesByInheritance($mainTable);
1247
        $tableGroupName = $this->getTableGroupName($allFetchedTables);
1248
        foreach ($allFetchedTables as $table) {
1249
            $tableGroups[$table] = $tableGroupName;
1250
        }
1251
1252
        foreach ($additionalTablesFetch as $additionalTable) {
1253
            $relatedTables = $this->_getRelatedTablesByInheritance($additionalTable);
1254
            $tableGroupName = $this->getTableGroupName($relatedTables);
1255
            foreach ($relatedTables as $table) {
1256
                $tableGroups[$table] = $tableGroupName;
1257
            }
1258
            $allFetchedTables = array_merge($allFetchedTables, $relatedTables);
1259
        }
1260
1261
        // Let's remove any duplicate
1262
        $allFetchedTables = array_flip(array_flip($allFetchedTables));
1263
1264
        $columnsList = [];
1265
        $columnDescList = [];
1266
        $schema = $this->tdbmSchemaAnalyzer->getSchema();
1267
1268
        // Now, let's build the column list
1269
        foreach ($allFetchedTables as $table) {
1270
            foreach ($schema->getTable($table)->getColumns() as $column) {
1271
                $columnName = $column->getName();
1272
                $columnDescList[] = [
1273
                    'as' => $table.'____'.$columnName,
1274
                    'table' => $table,
1275
                    'column' => $columnName,
1276
                    'type' => $column->getType(),
1277
                    'tableGroup' => $tableGroups[$table],
1278
                ];
1279
                $columnsList[] = $this->connection->quoteIdentifier($table).'.'.$this->connection->quoteIdentifier($columnName).' as '.
1280
                    $this->connection->quoteIdentifier($table.'____'.$columnName);
1281
            }
1282
        }
1283
1284
        $sql = 'SELECT DISTINCT '.implode(', ', $columnsList).' FROM MAGICJOIN('.$mainTable.')';
1285
        $countSql = 'SELECT COUNT(1) FROM MAGICJOIN('.$mainTable.')';
1286
1287
        if (!empty($filterString)) {
1288
            $sql .= ' WHERE '.$filterString;
1289
            $countSql .= ' WHERE '.$filterString;
1290
        }
1291
1292
        if (!empty($orderString)) {
1293
            $sql .= ' ORDER BY '.$orderString;
1294
            $countSql .= ' ORDER BY '.$orderString;
1295
        }
1296
1297 View Code Duplication
        if ($mode !== null && $mode !== self::MODE_CURSOR && $mode !== self::MODE_ARRAY) {
0 ignored issues
show
Duplication introduced by
This code seems to be duplicated across your project.

Duplicated code is one of the most pungent code smells. If you need to duplicate the same code in three or more different places, we strongly encourage you to look into extracting the code into a single class or operation.

You can also find more detailed suggestions in the “Code” section of your repository.

Loading history...
1298
            throw new TDBMException("Unknown fetch mode: '".$this->mode."'");
1299
        }
1300
1301
        $mode = $mode ?: $this->mode;
1302
1303
        return new ResultIterator($sql, $countSql, $parameters, $columnDescList, $this->objectStorage, $className, $this, $this->magicQuery, $mode);
1304
    }
1305
1306
    /**
1307
     * @param $table
1308
     * @param array  $primaryKeys
1309
     * @param array  $additionalTablesFetch
1310
     * @param bool   $lazy                  Whether to perform lazy loading on this object or not.
1311
     * @param string $className
1312
     *
1313
     * @return AbstractTDBMObject
1314
     *
1315
     * @throws TDBMException
1316
     */
1317
    public function findObjectByPk($table, array $primaryKeys, array $additionalTablesFetch = array(), $lazy = false, $className = null)
1318
    {
1319
        $primaryKeys = $this->_getPrimaryKeysFromObjectData($table, $primaryKeys);
1320
        $hash = $this->getObjectHash($primaryKeys);
1321
1322
        if ($this->objectStorage->has($table, $hash)) {
1323
            $dbRow = $this->objectStorage->get($table, $hash);
1324
            $bean = $dbRow->getTDBMObject();
1325
            if ($className !== null && !is_a($bean, $className)) {
1326
                throw new TDBMException("TDBM cannot create a bean of class '".$className."'. The requested object was already loaded and its class is '".get_class($bean)."'");
1327
            }
1328
1329
            return $bean;
1330
        }
1331
1332
        // Are we performing lazy fetching?
1333
        if ($lazy === true) {
1334
            // Can we perform lazy fetching?
1335
            $tables = $this->_getRelatedTablesByInheritance($table);
1336
            // Only allowed if no inheritance.
1337
            if (count($tables) === 1) {
1338
                if ($className === null) {
1339
                    $className = isset($this->tableToBeanMap[$table]) ? $this->tableToBeanMap[$table] : 'Mouf\\Database\\TDBM\\TDBMObject';
1340
                }
1341
1342
                // Let's construct the bean
1343
                if (!isset($this->reflectionClassCache[$className])) {
1344
                    $this->reflectionClassCache[$className] = new \ReflectionClass($className);
1345
                }
1346
                // Let's bypass the constructor when creating the bean!
1347
                $bean = $this->reflectionClassCache[$className]->newInstanceWithoutConstructor();
1348
                /* @var $bean AbstractTDBMObject */
1349
                $bean->_constructLazy($table, $primaryKeys, $this);
1350
            }
1351
        }
1352
1353
        // Did not find the object in cache? Let's query it!
1354
        return $this->findObjectOrFail($table, $primaryKeys, [], $additionalTablesFetch, $className);
1355
    }
1356
1357
    /**
1358
     * Returns a unique bean (or null) according to the filters passed in parameter.
1359
     *
1360
     * @param string            $mainTable             The name of the table queried
1361
     * @param string|array|null $filter                The SQL filters to apply to the query (the WHERE part). All columns must be prefixed by the table name (in the form: table.column)
1362
     * @param array             $parameters
1363
     * @param array             $additionalTablesFetch
1364
     * @param string            $className             Optional: The name of the class to instantiate. This class must extend the TDBMObject class. If none is specified, a TDBMObject instance will be returned.
1365
     *
1366
     * @return AbstractTDBMObject|null The object we want, or null if no object matches the filters.
1367
     *
1368
     * @throws TDBMException
1369
     */
1370
    public function findObject($mainTable, $filter = null, array $parameters = array(), array $additionalTablesFetch = array(), $className = null)
1371
    {
1372
        $objects = $this->findObjects($mainTable, $filter, $parameters, null, $additionalTablesFetch, self::MODE_ARRAY, $className);
1373
        $page = $objects->take(0, 2);
1374
        $count = $page->count();
1375
        if ($count > 1) {
1376
            throw new DuplicateRowException("Error while querying an object for table '$mainTable': More than 1 row have been returned, but we should have received at most one.");
1377
        } elseif ($count === 0) {
1378
            return;
1379
        }
1380
1381
        return $objects[0];
1382
    }
1383
1384
    /**
1385
     * Returns a unique bean according to the filters passed in parameter.
1386
     * Throws a NoBeanFoundException if no bean was found for the filter passed in parameter.
1387
     *
1388
     * @param string            $mainTable             The name of the table queried
1389
     * @param string|array|null $filter                The SQL filters to apply to the query (the WHERE part). All columns must be prefixed by the table name (in the form: table.column)
1390
     * @param array             $parameters
1391
     * @param array             $additionalTablesFetch
1392
     * @param string            $className             Optional: The name of the class to instantiate. This class must extend the TDBMObject class. If none is specified, a TDBMObject instance will be returned.
1393
     *
1394
     * @return AbstractTDBMObject The object we want
1395
     *
1396
     * @throws TDBMException
1397
     */
1398
    public function findObjectOrFail($mainTable, $filter = null, array $parameters = array(), array $additionalTablesFetch = array(), $className = null)
1399
    {
1400
        $bean = $this->findObject($mainTable, $filter, $parameters, $additionalTablesFetch, $className);
1401
        if ($bean === null) {
1402
            throw new NoBeanFoundException("No result found for query on table '".$mainTable."'");
1403
        }
1404
1405
        return $bean;
1406
    }
1407
1408
    /**
1409
     * @param array $beanData An array of data: array<table, array<column, value>>
1410
     *
1411
     * @return array an array with first item = class name and second item = table name
1412
     */
1413
    public function _getClassNameFromBeanData(array $beanData)
1414
    {
1415
        if (count($beanData) === 1) {
1416
            $tableName = array_keys($beanData)[0];
1417
        } else {
1418
            $tables = [];
1419
            foreach ($beanData as $table => $row) {
1420
                $primaryKeyColumns = $this->getPrimaryKeyColumns($table);
1421
                $pkSet = false;
1422
                foreach ($primaryKeyColumns as $columnName) {
1423
                    if ($row[$columnName] !== null) {
1424
                        $pkSet = true;
1425
                        break;
1426
                    }
1427
                }
1428
                if ($pkSet) {
1429
                    $tables[] = $table;
1430
                }
1431
            }
1432
1433
            // $tables contains the tables for this bean. Let's view the top most part of the hierarchy
1434
            $allTables = $this->_getLinkBetweenInheritedTables($tables);
1435
            $tableName = $allTables[0];
1436
        }
1437
1438
        // Only one table in this bean. Life is sweat, let's look at its type:
1439
        if (isset($this->tableToBeanMap[$tableName])) {
1440
            return [$this->tableToBeanMap[$tableName], $tableName];
1441
        } else {
1442
            return ['Mouf\\Database\\TDBM\\TDBMObject', $tableName];
1443
        }
1444
    }
1445
1446
    /**
1447
     * Returns an item from cache or computes it using $closure and puts it in cache.
1448
     *
1449
     * @param string   $key
1450
     * @param callable $closure
1451
     *
1452
     * @return mixed
1453
     */
1454
    private function fromCache($key, callable $closure)
1455
    {
1456
        $item = $this->cache->fetch($key);
1457
        if ($item === false) {
1458
            $item = $closure();
1459
            $this->cache->save($key, $item);
1460
        }
1461
1462
        return $item;
1463
    }
1464
1465
    /**
1466
     * Returns the foreign key object.
1467
     *
1468
     * @param string $table
1469
     * @param string $fkName
1470
     *
1471
     * @return ForeignKeyConstraint
1472
     */
1473
    public function _getForeignKeyByName($table, $fkName)
1474
    {
1475
        return $this->tdbmSchemaAnalyzer->getSchema()->getTable($table)->getForeignKey($fkName);
1476
    }
1477
1478
    /**
1479
     * @param $pivotTableName
1480
     * @param AbstractTDBMObject $bean
1481
     *
1482
     * @return AbstractTDBMObject[]
1483
     */
1484
    public function _getRelatedBeans($pivotTableName, AbstractTDBMObject $bean)
1485
    {
1486
        list($localFk, $remoteFk) = $this->getPivotTableForeignKeys($pivotTableName, $bean);
1487
        /* @var $localFk ForeignKeyConstraint */
1488
        /* @var $remoteFk ForeignKeyConstraint */
1489
        $remoteTable = $remoteFk->getForeignTableName();
1490
1491
        $primaryKeys = $this->getPrimaryKeyValues($bean);
1492
        $columnNames = array_map(function ($name) use ($pivotTableName) { return $pivotTableName.'.'.$name; }, $localFk->getLocalColumns());
1493
1494
        $filter = array_combine($columnNames, $primaryKeys);
1495
1496
        return $this->findObjects($remoteTable, $filter);
1497
    }
1498
1499
    /**
1500
     * @param $pivotTableName
1501
     * @param AbstractTDBMObject $bean The LOCAL bean
1502
     *
1503
     * @return ForeignKeyConstraint[] First item: the LOCAL bean, second item: the REMOTE bean.
1504
     *
1505
     * @throws TDBMException
1506
     */
1507
    private function getPivotTableForeignKeys($pivotTableName, AbstractTDBMObject $bean)
1508
    {
1509
        $fks = array_values($this->tdbmSchemaAnalyzer->getSchema()->getTable($pivotTableName)->getForeignKeys());
1510
        $table1 = $fks[0]->getForeignTableName();
1511
        $table2 = $fks[1]->getForeignTableName();
1512
1513
        $beanTables = array_map(function (DbRow $dbRow) { return $dbRow->_getDbTableName(); }, $bean->_getDbRows());
1514
1515
        if (in_array($table1, $beanTables)) {
1516
            return [$fks[0], $fks[1]];
1517
        } elseif (in_array($table2, $beanTables)) {
1518
            return [$fks[1], $fks[0]];
1519
        } else {
1520
            throw new TDBMException("Unexpected bean type in getPivotTableForeignKeys. Awaiting beans from table {$table1} and {$table2} for pivot table {$pivotTableName}");
1521
        }
1522
    }
1523
1524
    /**
1525
     * Returns a list of pivot tables linked to $bean.
1526
     *
1527
     * @param AbstractTDBMObject $bean
1528
     *
1529
     * @return string[]
1530
     */
1531
    public function _getPivotTablesLinkedToBean(AbstractTDBMObject $bean)
1532
    {
1533
        $junctionTables = [];
1534
        $allJunctionTables = $this->schemaAnalyzer->detectJunctionTables();
1535
        foreach ($bean->_getDbRows() as $dbRow) {
1536
            foreach ($allJunctionTables as $table) {
1537
                // There are exactly 2 FKs since this is a pivot table.
1538
                $fks = array_values($table->getForeignKeys());
1539
1540
                if ($fks[0]->getForeignTableName() === $dbRow->_getDbTableName() || $fks[1]->getForeignTableName() === $dbRow->_getDbTableName()) {
1541
                    $junctionTables[] = $table->getName();
1542
                }
1543
            }
1544
        }
1545
1546
        return $junctionTables;
1547
    }
1548
}
1549