Completed
Push — 4.0 ( e1def4...8f8e98 )
by David
15:31
created

TDBMService::buildFilterFromFilterBag()   C

Complexity

Conditions 10
Paths 10

Size

Total Lines 48
Code Lines 38

Duplication

Lines 0
Ratio 0 %

Importance

Changes 9
Bugs 2 Features 0
Metric Value
c 9
b 2
f 0
dl 0
loc 48
rs 5.3454
cc 10
eloc 38
nc 10
nop 2

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);
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));
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
     *
633
     * @return \string[] the list of tables
634
     */
635
    public function generateAllDaosAndBeans($daoFactoryClassName, $daonamespace, $beannamespace, $storeInUtc)
636
    {
637
        $tdbmDaoGenerator = new TDBMDaoGenerator($this->schemaAnalyzer, $this->tdbmSchemaAnalyzer->getSchema(), $this->tdbmSchemaAnalyzer);
638
639
        return $tdbmDaoGenerator->generateAllDaosAndBeans($daoFactoryClassName, $daonamespace, $beannamespace, $storeInUtc);
640
    }
641
642
    /**
643
     * @param array<string, string> $tableToBeanMap
644
     */
645
    public function setTableToBeanMap(array $tableToBeanMap)
646
    {
647
        $this->tableToBeanMap = $tableToBeanMap;
648
    }
649
650
    /**
651
     * Saves $object by INSERTing or UPDAT(E)ing it in the database.
652
     *
653
     * @param AbstractTDBMObject $object
654
     *
655
     * @throws TDBMException
656
     */
657
    public function save(AbstractTDBMObject $object)
658
    {
659
        $status = $object->_getStatus();
660
661
        // Let's attach this object if it is in detached state.
662
        if ($status === TDBMObjectStateEnum::STATE_DETACHED) {
663
            $object->_attach($this);
664
            $status = $object->_getStatus();
665
        }
666
667
        if ($status === TDBMObjectStateEnum::STATE_NEW) {
668
            $dbRows = $object->_getDbRows();
669
670
            $unindexedPrimaryKeys = array();
671
672
            foreach ($dbRows as $dbRow) {
673
                $tableName = $dbRow->_getDbTableName();
674
675
                $schema = $this->tdbmSchemaAnalyzer->getSchema();
676
                $tableDescriptor = $schema->getTable($tableName);
677
678
                $primaryKeyColumns = $this->getPrimaryKeyColumns($tableName);
679
680
                if (empty($unindexedPrimaryKeys)) {
681
                    $primaryKeys = $this->getPrimaryKeysForObjectFromDbRow($dbRow);
682
                } else {
683
                    // First insert, the children must have the same primary key as the parent.
684
                    $primaryKeys = $this->_getPrimaryKeysFromIndexedPrimaryKeys($tableName, $unindexedPrimaryKeys);
685
                    $dbRow->_setPrimaryKeys($primaryKeys);
686
                }
687
688
                $references = $dbRow->_getReferences();
689
690
                // Let's save all references in NEW or DETACHED state (we need their primary key)
691
                foreach ($references as $fkName => $reference) {
692
                    $refStatus = $reference->_getStatus();
693
                    if ($refStatus === TDBMObjectStateEnum::STATE_NEW || $refStatus === TDBMObjectStateEnum::STATE_DETACHED) {
694
                        $this->save($reference);
695
                    }
696
                }
697
698
                $dbRowData = $dbRow->_getDbRow();
699
700
                // Let's see if the columns for primary key have been set before inserting.
701
                // We assume that if one of the value of the PK has been set, the PK is set.
702
                $isPkSet = !empty($primaryKeys);
703
704
                /*if (!$isPkSet) {
705
                    // if there is no autoincrement and no pk set, let's go in error.
706
                    $isAutoIncrement = true;
707
708
                    foreach ($primaryKeyColumns as $pkColumnName) {
709
                        $pkColumn = $tableDescriptor->getColumn($pkColumnName);
710
                        if (!$pkColumn->getAutoincrement()) {
711
                            $isAutoIncrement = false;
712
                        }
713
                    }
714
715
                    if (!$isAutoIncrement) {
716
                        $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.";
717
                        throw new TDBMException($msg);
718
                    }
719
720
                }*/
721
722
                $types = [];
723
724
                foreach ($dbRowData as $columnName => $value) {
725
                    $columnDescriptor = $tableDescriptor->getColumn($columnName);
726
                    $types[] = $columnDescriptor->getType();
727
                }
728
729
                $this->connection->insert($tableName, $dbRowData, $types);
730
731
                if (!$isPkSet && count($primaryKeyColumns) == 1) {
732
                    $id = $this->connection->lastInsertId();
733
                    $primaryKeys[$primaryKeyColumns[0]] = $id;
734
                }
735
736
                // TODO: change this to some private magic accessor in future
737
                $dbRow->_setPrimaryKeys($primaryKeys);
738
                $unindexedPrimaryKeys = array_values($primaryKeys);
739
740
                /*
741
                 * When attached, on "save", we check if the column updated is part of a primary key
742
                 * If this is part of a primary key, we call the _update_id method that updates the id in the list of known objects.
743
                 * This method should first verify that the id is not already used (and is not auto-incremented)
744
                 *
745
                 * In the object, the key is stored in an array of  (column => value), that can be directly used to update the record.
746
                 *
747
                 *
748
                 */
749
750
                /*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...
751
                    $this->db_connection->exec($sql);
752
                } catch (TDBMException $e) {
753
                    $this->db_onerror = true;
754
755
                    // Strange..... if we do not have the line below, bad inserts are not catched.
756
                    // It seems that destructors are called before the registered shutdown function (PHP >=5.0.5)
757
                    //if ($this->tdbmService->isProgramExiting())
758
                    //	trigger_error("program exiting");
759
                    trigger_error($e->getMessage(), E_USER_ERROR);
760
761
                    if (!$this->tdbmService->isProgramExiting())
762
                        throw $e;
763
                    else
764
                    {
765
                        trigger_error($e->getMessage(), E_USER_ERROR);
766
                    }
767
                }*/
768
769
                // Let's remove this object from the $new_objects static table.
770
                $this->removeFromToSaveObjectList($dbRow);
771
772
                // TODO: change this behaviour to something more sensible performance-wise
773
                // Maybe a setting to trigger this globally?
774
                //$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...
775
                //$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...
776
                //$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...
777
778
                // Let's add this object to the list of objects in cache.
779
                $this->_addToCache($dbRow);
780
            }
781
782
            $object->_setStatus(TDBMObjectStateEnum::STATE_LOADED);
783
        } elseif ($status === TDBMObjectStateEnum::STATE_DIRTY) {
784
            $dbRows = $object->_getDbRows();
785
786
            foreach ($dbRows as $dbRow) {
787
                $references = $dbRow->_getReferences();
788
789
                // Let's save all references in NEW state (we need their primary key)
790
                foreach ($references as $fkName => $reference) {
791
                    if ($reference->_getStatus() === TDBMObjectStateEnum::STATE_NEW) {
792
                        $this->save($reference);
793
                    }
794
                }
795
796
                // Let's first get the primary keys
797
                $tableName = $dbRow->_getDbTableName();
798
                $dbRowData = $dbRow->_getDbRow();
799
800
                $schema = $this->tdbmSchemaAnalyzer->getSchema();
801
                $tableDescriptor = $schema->getTable($tableName);
802
803
                $primaryKeys = $dbRow->_getPrimaryKeys();
804
805
                $types = [];
806
807
                foreach ($dbRowData as $columnName => $value) {
808
                    $columnDescriptor = $tableDescriptor->getColumn($columnName);
809
                    $types[] = $columnDescriptor->getType();
810
                }
811
                foreach ($primaryKeys as $columnName => $value) {
812
                    $columnDescriptor = $tableDescriptor->getColumn($columnName);
813
                    $types[] = $columnDescriptor->getType();
814
                }
815
816
                $this->connection->update($tableName, $dbRowData, $primaryKeys, $types);
817
818
                // Let's check if the primary key has been updated...
819
                $needsUpdatePk = false;
820
                foreach ($primaryKeys as $column => $value) {
821
                    if (!isset($dbRowData[$column]) || $dbRowData[$column] != $value) {
822
                        $needsUpdatePk = true;
823
                        break;
824
                    }
825
                }
826
                if ($needsUpdatePk) {
827
                    $this->objectStorage->remove($tableName, $this->getObjectHash($primaryKeys));
828
                    $newPrimaryKeys = $this->getPrimaryKeysForObjectFromDbRow($dbRow);
829
                    $dbRow->_setPrimaryKeys($newPrimaryKeys);
830
                    $this->objectStorage->set($tableName, $this->getObjectHash($primaryKeys), $dbRow);
831
                }
832
833
                // Let's remove this object from the list of objects to save.
834
                $this->removeFromToSaveObjectList($dbRow);
835
            }
836
837
            $object->_setStatus(TDBMObjectStateEnum::STATE_LOADED);
838
        } elseif ($status === TDBMObjectStateEnum::STATE_DELETED) {
839
            throw new TDBMInvalidOperationException('This object has been deleted. It cannot be saved.');
840
        }
841
842
        // Finally, let's save all the many to many relationships to this bean.
843
        $this->persistManyToManyRelationships($object);
844
    }
845
846
    private function persistManyToManyRelationships(AbstractTDBMObject $object)
847
    {
848
        foreach ($object->_getCachedRelationships() as $pivotTableName => $storage) {
849
            $tableDescriptor = $this->tdbmSchemaAnalyzer->getSchema()->getTable($pivotTableName);
850
            list($localFk, $remoteFk) = $this->getPivotTableForeignKeys($pivotTableName, $object);
851
852
            foreach ($storage as $remoteBean) {
853
                /* @var $remoteBean AbstractTDBMObject */
854
                $statusArr = $storage[$remoteBean];
855
                $status = $statusArr['status'];
856
                $reverse = $statusArr['reverse'];
857
                if ($reverse) {
858
                    continue;
859
                }
860
861
                if ($status === 'new') {
862
                    $remoteBeanStatus = $remoteBean->_getStatus();
863
                    if ($remoteBeanStatus === TDBMObjectStateEnum::STATE_NEW || $remoteBeanStatus === TDBMObjectStateEnum::STATE_DETACHED) {
864
                        // Let's save remote bean if needed.
865
                        $this->save($remoteBean);
866
                    }
867
868
                    $filters = $this->getPivotFilters($object, $remoteBean, $localFk, $remoteFk);
869
870
                    $types = [];
871
872
                    foreach ($filters as $columnName => $value) {
873
                        $columnDescriptor = $tableDescriptor->getColumn($columnName);
874
                        $types[] = $columnDescriptor->getType();
875
                    }
876
877
                    $this->connection->insert($pivotTableName, $filters, $types);
878
879
                    // Finally, let's mark relationships as saved.
880
                    $statusArr['status'] = 'loaded';
881
                    $storage[$remoteBean] = $statusArr;
882
                    $remoteStorage = $remoteBean->_getCachedRelationships()[$pivotTableName];
883
                    $remoteStatusArr = $remoteStorage[$object];
884
                    $remoteStatusArr['status'] = 'loaded';
885
                    $remoteStorage[$object] = $remoteStatusArr;
886
                } elseif ($status === 'delete') {
887
                    $filters = $this->getPivotFilters($object, $remoteBean, $localFk, $remoteFk);
888
889
                    $types = [];
890
891
                    foreach ($filters as $columnName => $value) {
892
                        $columnDescriptor = $tableDescriptor->getColumn($columnName);
893
                        $types[] = $columnDescriptor->getType();
894
                    }
895
896
                    $this->connection->delete($pivotTableName, $filters, $types);
897
898
                    // Finally, let's remove relationships completely from bean.
899
                    $storage->detach($remoteBean);
900
                    $remoteBean->_getCachedRelationships()[$pivotTableName]->detach($object);
901
                }
902
            }
903
        }
904
    }
905
906
    private function getPivotFilters(AbstractTDBMObject $localBean, AbstractTDBMObject $remoteBean, ForeignKeyConstraint $localFk, ForeignKeyConstraint $remoteFk)
907
    {
908
        $localBeanPk = $this->getPrimaryKeyValues($localBean);
909
        $remoteBeanPk = $this->getPrimaryKeyValues($remoteBean);
910
        $localColumns = $localFk->getLocalColumns();
911
        $remoteColumns = $remoteFk->getLocalColumns();
912
913
        $localFilters = array_combine($localColumns, $localBeanPk);
914
        $remoteFilters = array_combine($remoteColumns, $remoteBeanPk);
915
916
        return array_merge($localFilters, $remoteFilters);
917
    }
918
919
    /**
920
     * Returns the "values" of the primary key.
921
     * This returns the primary key from the $primaryKey attribute, not the one stored in the columns.
922
     *
923
     * @param AbstractTDBMObject $bean
924
     *
925
     * @return array numerically indexed array of values.
926
     */
927
    private function getPrimaryKeyValues(AbstractTDBMObject $bean)
928
    {
929
        $dbRows = $bean->_getDbRows();
930
        $dbRow = reset($dbRows);
931
932
        return array_values($dbRow->_getPrimaryKeys());
933
    }
934
935
    /**
936
     * Returns a unique hash used to store the object based on its primary key.
937
     * If the array contains only one value, then the value is returned.
938
     * Otherwise, a hash representing the array is returned.
939
     *
940
     * @param array $primaryKeys An array of columns => values forming the primary key
941
     *
942
     * @return string
943
     */
944
    public function getObjectHash(array $primaryKeys)
945
    {
946
        if (count($primaryKeys) === 1) {
947
            return reset($primaryKeys);
948
        } else {
949
            ksort($primaryKeys);
950
951
            return md5(json_encode($primaryKeys));
952
        }
953
    }
954
955
    /**
956
     * Returns an array of primary keys from the object.
957
     * The primary keys are extracted from the object columns and not from the primary keys stored in the
958
     * $primaryKeys variable of the object.
959
     *
960
     * @param DbRow $dbRow
961
     *
962
     * @return array Returns an array of column => value
963
     */
964
    public function getPrimaryKeysForObjectFromDbRow(DbRow $dbRow)
965
    {
966
        $table = $dbRow->_getDbTableName();
967
        $dbRowData = $dbRow->_getDbRow();
968
969
        return $this->_getPrimaryKeysFromObjectData($table, $dbRowData);
970
    }
971
972
    /**
973
     * Returns an array of primary keys for the given row.
974
     * The primary keys are extracted from the object columns.
975
     *
976
     * @param $table
977
     * @param array $columns
978
     *
979
     * @return array
980
     */
981
    public function _getPrimaryKeysFromObjectData($table, array $columns)
982
    {
983
        $primaryKeyColumns = $this->getPrimaryKeyColumns($table);
984
        $values = array();
985
        foreach ($primaryKeyColumns as $column) {
986
            if (isset($columns[$column])) {
987
                $values[$column] = $columns[$column];
988
            }
989
        }
990
991
        return $values;
992
    }
993
994
    /**
995
     * Attaches $object to this TDBMService.
996
     * The $object must be in DETACHED state and will pass in NEW state.
997
     *
998
     * @param AbstractTDBMObject $object
999
     *
1000
     * @throws TDBMInvalidOperationException
1001
     */
1002
    public function attach(AbstractTDBMObject $object)
1003
    {
1004
        $object->_attach($this);
1005
    }
1006
1007
    /**
1008
     * Returns an associative array (column => value) for the primary keys from the table name and an
1009
     * indexed array of primary key values.
1010
     *
1011
     * @param string $tableName
1012
     * @param array  $indexedPrimaryKeys
1013
     */
1014
    public function _getPrimaryKeysFromIndexedPrimaryKeys($tableName, array $indexedPrimaryKeys)
1015
    {
1016
        $primaryKeyColumns = $this->tdbmSchemaAnalyzer->getSchema()->getTable($tableName)->getPrimaryKeyColumns();
1017
1018
        if (count($primaryKeyColumns) !== count($indexedPrimaryKeys)) {
1019
            throw new TDBMException(sprintf('Wrong number of columns passed for primary key. Expected %s columns for table "%s",
1020
			got %s instead.', count($primaryKeyColumns), $tableName, count($indexedPrimaryKeys)));
1021
        }
1022
1023
        return array_combine($primaryKeyColumns, $indexedPrimaryKeys);
1024
    }
1025
1026
    /**
1027
     * Return the list of tables (from child to parent) joining the tables passed in parameter.
1028
     * Tables must be in a single line of inheritance. The method will find missing tables.
1029
     *
1030
     * Algorithm: one of those tables is the ultimate child. From this child, by recursively getting the parent,
1031
     * we must be able to find all other tables.
1032
     *
1033
     * @param string[] $tables
1034
     *
1035
     * @return string[]
1036
     */
1037
    public function _getLinkBetweenInheritedTables(array $tables)
1038
    {
1039
        sort($tables);
1040
1041
        return $this->fromCache($this->cachePrefix.'_linkbetweeninheritedtables_'.implode('__split__', $tables),
1042
            function () use ($tables) {
1043
                return $this->_getLinkBetweenInheritedTablesWithoutCache($tables);
1044
            });
1045
    }
1046
1047
    /**
1048
     * Return the list of tables (from child to parent) joining the tables passed in parameter.
1049
     * Tables must be in a single line of inheritance. The method will find missing tables.
1050
     *
1051
     * Algorithm: one of those tables is the ultimate child. From this child, by recursively getting the parent,
1052
     * we must be able to find all other tables.
1053
     *
1054
     * @param string[] $tables
1055
     *
1056
     * @return string[]
1057
     */
1058
    private function _getLinkBetweenInheritedTablesWithoutCache(array $tables)
1059
    {
1060
        $schemaAnalyzer = $this->schemaAnalyzer;
1061
1062
        foreach ($tables as $currentTable) {
1063
            $allParents = [$currentTable];
1064
            while ($currentFk = $schemaAnalyzer->getParentRelationship($currentTable)) {
1065
                $currentTable = $currentFk->getForeignTableName();
1066
                $allParents[] = $currentTable;
1067
            };
1068
1069
            // Now, does the $allParents contain all the tables we want?
1070
            $notFoundTables = array_diff($tables, $allParents);
1071
            if (empty($notFoundTables)) {
1072
                // We have a winner!
1073
                return $allParents;
1074
            }
1075
        }
1076
1077
        throw new TDBMException(sprintf('The tables (%s) cannot be linked by an inheritance relationship.', implode(', ', $tables)));
1078
    }
1079
1080
    /**
1081
     * Returns the list of tables related to this table (via a parent or child inheritance relationship).
1082
     *
1083
     * @param string $table
1084
     *
1085
     * @return string[]
1086
     */
1087
    public function _getRelatedTablesByInheritance($table)
1088
    {
1089
        return $this->fromCache($this->cachePrefix.'_relatedtables_'.$table, function () use ($table) {
1090
            return $this->_getRelatedTablesByInheritanceWithoutCache($table);
1091
        });
1092
    }
1093
1094
    /**
1095
     * Returns the list of tables related to this table (via a parent or child inheritance relationship).
1096
     *
1097
     * @param string $table
1098
     *
1099
     * @return string[]
1100
     */
1101
    private function _getRelatedTablesByInheritanceWithoutCache($table)
1102
    {
1103
        $schemaAnalyzer = $this->schemaAnalyzer;
1104
1105
        // Let's scan the parent tables
1106
        $currentTable = $table;
1107
1108
        $parentTables = [];
1109
1110
        // Get parent relationship
1111
        while ($currentFk = $schemaAnalyzer->getParentRelationship($currentTable)) {
1112
            $currentTable = $currentFk->getForeignTableName();
1113
            $parentTables[] = $currentTable;
1114
        };
1115
1116
        // Let's recurse in children
1117
        $childrenTables = $this->exploreChildrenTablesRelationships($schemaAnalyzer, $table);
1118
1119
        return array_merge(array_reverse($parentTables), $childrenTables);
1120
    }
1121
1122
    /**
1123
     * Explore all the children and descendant of $table and returns ForeignKeyConstraints on those.
1124
     *
1125
     * @param string $table
1126
     *
1127
     * @return string[]
1128
     */
1129
    private function exploreChildrenTablesRelationships(SchemaAnalyzer $schemaAnalyzer, $table)
1130
    {
1131
        $tables = [$table];
1132
        $keys = $schemaAnalyzer->getChildrenRelationships($table);
1133
1134
        foreach ($keys as $key) {
1135
            $tables = array_merge($tables, $this->exploreChildrenTablesRelationships($schemaAnalyzer, $key->getLocalTableName()));
1136
        }
1137
1138
        return $tables;
1139
    }
1140
1141
    /**
1142
     * Casts a foreign key into SQL, assuming table name is used with no alias.
1143
     * The returned value does contain only one table. For instance:.
1144
     *
1145
     * " LEFT JOIN table2 ON table1.id = table2.table1_id"
1146
     *
1147
     * @param ForeignKeyConstraint $fk
1148
     * @param bool                 $leftTableIsLocal
1149
     *
1150
     * @return string
1151
     */
1152
    /*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...
1153
        $onClauses = [];
1154
        $foreignTableName = $this->connection->quoteIdentifier($fk->getForeignTableName());
1155
        $foreignColumns = $fk->getForeignColumns();
1156
        $localTableName = $this->connection->quoteIdentifier($fk->getLocalTableName());
1157
        $localColumns = $fk->getLocalColumns();
1158
        $columnCount = count($localTableName);
1159
1160
        for ($i = 0; $i < $columnCount; $i++) {
1161
            $onClauses[] = sprintf("%s.%s = %s.%s",
1162
                $localTableName,
1163
                $this->connection->quoteIdentifier($localColumns[$i]),
1164
                $foreignColumns,
1165
                $this->connection->quoteIdentifier($foreignColumns[$i])
1166
                );
1167
        }
1168
1169
        $onClause = implode(' AND ', $onClauses);
1170
1171
        if ($leftTableIsLocal) {
1172
            return sprintf(" LEFT JOIN %s ON (%s)", $foreignTableName, $onClause);
1173
        } else {
1174
            return sprintf(" LEFT JOIN %s ON (%s)", $localTableName, $onClause);
1175
        }
1176
    }*/
1177
1178
    /**
1179
     * Returns an identifier for the group of tables passed in parameter.
1180
     *
1181
     * @param string[] $relatedTables
1182
     *
1183
     * @return string
1184
     */
1185
    private function getTableGroupName(array $relatedTables)
1186
    {
1187
        sort($relatedTables);
1188
1189
        return implode('_``_', $relatedTables);
1190
    }
1191
1192
    /**
1193
     * Returns a `ResultIterator` object representing filtered records of "$mainTable" .
1194
     *
1195
     * The findObjects method should be the most used query method in TDBM if you want to query the database for objects.
1196
     * (Note: if you want to query the database for an object by its primary key, use the findObjectByPk method).
1197
     *
1198
     * The findObjects method takes in parameter:
1199
     * 	- mainTable: the kind of bean you want to retrieve. In TDBM, a bean matches a database row, so the
1200
     * 			`$mainTable` parameter should be the name of an existing table in database.
1201
     *  - filter: The filter is a filter bag. It is what you use to filter your request (the WHERE part in SQL).
1202
     *          It can be a string (SQL Where clause), or even a bean or an associative array (key = column to filter, value = value to find)
1203
     *  - parameters: The parameters used in the filter. If you pass a SQL string as a filter, be sure to avoid
1204
     *          concatenating parameters in the string (this leads to SQL injection and also to poor caching performance).
1205
     *          Instead, please consider passing parameters (see documentation for more details).
1206
     *  - additionalTablesFetch: An array of SQL tables names. The beans related to those tables will be fetched along
1207
     *          the main table. This is useful to avoid hitting the database with numerous subqueries.
1208
     *  - mode: The fetch mode of the result. See `setFetchMode()` method for more details.
1209
     *
1210
     * The `findObjects` method will return a `ResultIterator`. A `ResultIterator` is an object that behaves as an array
1211
     * (in ARRAY mode) at least. It can be iterated using a `foreach` loop.
1212
     *
1213
     * Finally, if filter_bag is null, the whole table is returned.
1214
     *
1215
     * @param string            $mainTable             The name of the table queried
1216
     * @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)
1217
     * @param array             $parameters
1218
     * @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)
1219
     * @param array             $additionalTablesFetch
1220
     * @param int               $mode
1221
     * @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.
1222
     *
1223
     * @return ResultIterator An object representing an array of results.
1224
     *
1225
     * @throws TDBMException
1226
     */
1227
    public function findObjects($mainTable, $filter = null, array $parameters = array(), $orderString = null, array $additionalTablesFetch = array(), $mode = null, $className = null)
1228
    {
1229
        // $mainTable is not secured in MagicJoin, let's add a bit of security to avoid SQL injection.
1230
        if (!preg_match('/^[a-zA-Z_][a-zA-Z0-9_]*$/', $mainTable)) {
1231
            throw new TDBMException(sprintf("Invalid table name: '%s'", $mainTable));
1232
        }
1233
1234
        list($filterString, $additionalParameters) = $this->buildFilterFromFilterBag($filter);
1235
1236
        $parameters = array_merge($parameters, $additionalParameters);
1237
1238
        // From the table name and the additional tables we want to fetch, let's build a list of all tables
1239
        // that must be part of the select columns.
1240
1241
        $tableGroups = [];
1242
        $allFetchedTables = $this->_getRelatedTablesByInheritance($mainTable);
1243
        $tableGroupName = $this->getTableGroupName($allFetchedTables);
1244
        foreach ($allFetchedTables as $table) {
1245
            $tableGroups[$table] = $tableGroupName;
1246
        }
1247
1248
        foreach ($additionalTablesFetch as $additionalTable) {
1249
            $relatedTables = $this->_getRelatedTablesByInheritance($additionalTable);
1250
            $tableGroupName = $this->getTableGroupName($relatedTables);
1251
            foreach ($relatedTables as $table) {
1252
                $tableGroups[$table] = $tableGroupName;
1253
            }
1254
            $allFetchedTables = array_merge($allFetchedTables, $relatedTables);
1255
        }
1256
1257
        // Let's remove any duplicate
1258
        $allFetchedTables = array_flip(array_flip($allFetchedTables));
1259
1260
        $columnsList = [];
1261
        $columnDescList = [];
1262
        $schema = $this->tdbmSchemaAnalyzer->getSchema();
1263
1264
        // Now, let's build the column list
1265
        foreach ($allFetchedTables as $table) {
1266
            foreach ($schema->getTable($table)->getColumns() as $column) {
1267
                $columnName = $column->getName();
1268
                $columnDescList[] = [
1269
                    'as' => $table.'____'.$columnName,
1270
                    'table' => $table,
1271
                    'column' => $columnName,
1272
                    'type' => $column->getType(),
1273
                    'tableGroup' => $tableGroups[$table],
1274
                ];
1275
                $columnsList[] = $this->connection->quoteIdentifier($table).'.'.$this->connection->quoteIdentifier($columnName).' as '.
1276
                    $this->connection->quoteIdentifier($table.'____'.$columnName);
1277
            }
1278
        }
1279
1280
        $sql = 'SELECT DISTINCT '.implode(', ', $columnsList).' FROM MAGICJOIN('.$mainTable.')';
1281
        $countSql = 'SELECT COUNT(1) FROM MAGICJOIN('.$mainTable.')';
1282
1283
        if (!empty($filterString)) {
1284
            $sql .= ' WHERE '.$filterString;
1285
            $countSql .= ' WHERE '.$filterString;
1286
        }
1287
1288
        if (!empty($orderString)) {
1289
            $sql .= ' ORDER BY '.$orderString;
1290
            $countSql .= ' ORDER BY '.$orderString;
1291
        }
1292
1293 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...
1294
            throw new TDBMException("Unknown fetch mode: '".$this->mode."'");
1295
        }
1296
1297
        $mode = $mode ?: $this->mode;
1298
1299
        return new ResultIterator($sql, $countSql, $parameters, $columnDescList, $this->objectStorage, $className, $this, $this->magicQuery, $mode);
1300
    }
1301
1302
    /**
1303
     * @param $table
1304
     * @param array  $primaryKeys
1305
     * @param array  $additionalTablesFetch
1306
     * @param bool   $lazy                  Whether to perform lazy loading on this object or not.
1307
     * @param string $className
1308
     *
1309
     * @return AbstractTDBMObject
1310
     *
1311
     * @throws TDBMException
1312
     */
1313
    public function findObjectByPk($table, array $primaryKeys, array $additionalTablesFetch = array(), $lazy = false, $className = null)
1314
    {
1315
        $primaryKeys = $this->_getPrimaryKeysFromObjectData($table, $primaryKeys);
1316
        $hash = $this->getObjectHash($primaryKeys);
1317
1318
        if ($this->objectStorage->has($table, $hash)) {
1319
            $dbRow = $this->objectStorage->get($table, $hash);
1320
            $bean = $dbRow->getTDBMObject();
1321
            if ($className !== null && !is_a($bean, $className)) {
1322
                throw new TDBMException("TDBM cannot create a bean of class '".$className."'. The requested object was already loaded and its class is '".get_class($bean)."'");
1323
            }
1324
1325
            return $bean;
1326
        }
1327
1328
        // Are we performing lazy fetching?
1329
        if ($lazy === true) {
1330
            // Can we perform lazy fetching?
1331
            $tables = $this->_getRelatedTablesByInheritance($table);
1332
            // Only allowed if no inheritance.
1333
            if (count($tables) === 1) {
1334
                if ($className === null) {
1335
                    $className = isset($this->tableToBeanMap[$table]) ? $this->tableToBeanMap[$table] : 'Mouf\\Database\\TDBM\\TDBMObject';
1336
                }
1337
1338
                // Let's construct the bean
1339
                if (!isset($this->reflectionClassCache[$className])) {
1340
                    $this->reflectionClassCache[$className] = new \ReflectionClass($className);
1341
                }
1342
                // Let's bypass the constructor when creating the bean!
1343
                $bean = $this->reflectionClassCache[$className]->newInstanceWithoutConstructor();
1344
                /* @var $bean AbstractTDBMObject */
1345
                $bean->_constructLazy($table, $primaryKeys, $this);
1346
            }
1347
        }
1348
1349
        // Did not find the object in cache? Let's query it!
1350
        return $this->findObjectOrFail($table, $primaryKeys, [], $additionalTablesFetch, $className);
1351
    }
1352
1353
    /**
1354
     * Returns a unique bean (or null) according to the filters passed in parameter.
1355
     *
1356
     * @param string            $mainTable             The name of the table queried
1357
     * @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)
1358
     * @param array             $parameters
1359
     * @param array             $additionalTablesFetch
1360
     * @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.
1361
     *
1362
     * @return AbstractTDBMObject|null The object we want, or null if no object matches the filters.
1363
     *
1364
     * @throws TDBMException
1365
     */
1366
    public function findObject($mainTable, $filter = null, array $parameters = array(), array $additionalTablesFetch = array(), $className = null)
1367
    {
1368
        $objects = $this->findObjects($mainTable, $filter, $parameters, null, $additionalTablesFetch, self::MODE_ARRAY, $className);
1369
        $page = $objects->take(0, 2);
1370
        $count = $page->count();
1371
        if ($count > 1) {
1372
            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.");
1373
        } elseif ($count === 0) {
1374
            return;
1375
        }
1376
1377
        return $objects[0];
1378
    }
1379
1380
    /**
1381
     * Returns a unique bean according to the filters passed in parameter.
1382
     * Throws a NoBeanFoundException if no bean was found for the filter passed in parameter.
1383
     *
1384
     * @param string            $mainTable             The name of the table queried
1385
     * @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)
1386
     * @param array             $parameters
1387
     * @param array             $additionalTablesFetch
1388
     * @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.
1389
     *
1390
     * @return AbstractTDBMObject The object we want
1391
     *
1392
     * @throws TDBMException
1393
     */
1394
    public function findObjectOrFail($mainTable, $filter = null, array $parameters = array(), array $additionalTablesFetch = array(), $className = null)
1395
    {
1396
        $bean = $this->findObject($mainTable, $filter, $parameters, $additionalTablesFetch, $className);
1397
        if ($bean === null) {
1398
            throw new NoBeanFoundException("No result found for query on table '".$mainTable."'");
1399
        }
1400
1401
        return $bean;
1402
    }
1403
1404
    /**
1405
     * @param array $beanData An array of data: array<table, array<column, value>>
1406
     *
1407
     * @return array an array with first item = class name and second item = table name
1408
     */
1409
    public function _getClassNameFromBeanData(array $beanData)
1410
    {
1411
        if (count($beanData) === 1) {
1412
            $tableName = array_keys($beanData)[0];
1413
        } else {
1414
            $tables = [];
1415
            foreach ($beanData as $table => $row) {
1416
                $primaryKeyColumns = $this->getPrimaryKeyColumns($table);
1417
                $pkSet = false;
1418
                foreach ($primaryKeyColumns as $columnName) {
1419
                    if ($row[$columnName] !== null) {
1420
                        $pkSet = true;
1421
                        break;
1422
                    }
1423
                }
1424
                if ($pkSet) {
1425
                    $tables[] = $table;
1426
                }
1427
            }
1428
1429
            // $tables contains the tables for this bean. Let's view the top most part of the hierarchy
1430
            $allTables = $this->_getLinkBetweenInheritedTables($tables);
1431
            $tableName = $allTables[0];
1432
        }
1433
1434
        // Only one table in this bean. Life is sweat, let's look at its type:
1435
        if (isset($this->tableToBeanMap[$tableName])) {
1436
            return [$this->tableToBeanMap[$tableName], $tableName];
1437
        } else {
1438
            return ['Mouf\\Database\\TDBM\\TDBMObject', $tableName];
1439
        }
1440
    }
1441
1442
    /**
1443
     * Returns an item from cache or computes it using $closure and puts it in cache.
1444
     *
1445
     * @param string   $key
1446
     * @param callable $closure
1447
     *
1448
     * @return mixed
1449
     */
1450
    private function fromCache($key, callable $closure)
1451
    {
1452
        $item = $this->cache->fetch($key);
1453
        if ($item === false) {
1454
            $item = $closure();
1455
            $this->cache->save($key, $item);
1456
        }
1457
1458
        return $item;
1459
    }
1460
1461
    /**
1462
     * Returns the foreign key object.
1463
     *
1464
     * @param string $table
1465
     * @param string $fkName
1466
     *
1467
     * @return ForeignKeyConstraint
1468
     */
1469
    public function _getForeignKeyByName($table, $fkName)
1470
    {
1471
        return $this->tdbmSchemaAnalyzer->getSchema()->getTable($table)->getForeignKey($fkName);
1472
    }
1473
1474
    /**
1475
     * @param $pivotTableName
1476
     * @param AbstractTDBMObject $bean
1477
     *
1478
     * @return AbstractTDBMObject[]
1479
     */
1480
    public function _getRelatedBeans($pivotTableName, AbstractTDBMObject $bean)
1481
    {
1482
        list($localFk, $remoteFk) = $this->getPivotTableForeignKeys($pivotTableName, $bean);
1483
        /* @var $localFk ForeignKeyConstraint */
1484
        /* @var $remoteFk ForeignKeyConstraint */
1485
        $remoteTable = $remoteFk->getForeignTableName();
1486
1487
        $primaryKeys = $this->getPrimaryKeyValues($bean);
1488
        $columnNames = array_map(function ($name) use ($pivotTableName) { return $pivotTableName.'.'.$name; }, $localFk->getLocalColumns());
1489
1490
        $filter = array_combine($columnNames, $primaryKeys);
1491
1492
        return $this->findObjects($remoteTable, $filter);
1493
    }
1494
1495
    /**
1496
     * @param $pivotTableName
1497
     * @param AbstractTDBMObject $bean The LOCAL bean
1498
     *
1499
     * @return ForeignKeyConstraint[] First item: the LOCAL bean, second item: the REMOTE bean.
1500
     *
1501
     * @throws TDBMException
1502
     */
1503
    private function getPivotTableForeignKeys($pivotTableName, AbstractTDBMObject $bean)
1504
    {
1505
        $fks = array_values($this->tdbmSchemaAnalyzer->getSchema()->getTable($pivotTableName)->getForeignKeys());
1506
        $table1 = $fks[0]->getForeignTableName();
1507
        $table2 = $fks[1]->getForeignTableName();
1508
1509
        $beanTables = array_map(function (DbRow $dbRow) { return $dbRow->_getDbTableName(); }, $bean->_getDbRows());
1510
1511
        if (in_array($table1, $beanTables)) {
1512
            return [$fks[0], $fks[1]];
1513
        } elseif (in_array($table2, $beanTables)) {
1514
            return [$fks[1], $fks[0]];
1515
        } else {
1516
            throw new TDBMException("Unexpected bean type in getPivotTableForeignKeys. Awaiting beans from table {$table1} and {$table2} for pivot table {$pivotTableName}");
1517
        }
1518
    }
1519
1520
    /**
1521
     * Returns a list of pivot tables linked to $bean.
1522
     *
1523
     * @param AbstractTDBMObject $bean
1524
     *
1525
     * @return string[]
1526
     */
1527
    public function _getPivotTablesLinkedToBean(AbstractTDBMObject $bean)
1528
    {
1529
        $junctionTables = [];
1530
        $allJunctionTables = $this->schemaAnalyzer->detectJunctionTables();
1531
        foreach ($bean->_getDbRows() as $dbRow) {
1532
            foreach ($allJunctionTables as $table) {
1533
                // There are exactly 2 FKs since this is a pivot table.
1534
                $fks = array_values($table->getForeignKeys());
1535
1536
                if ($fks[0]->getForeignTableName() === $dbRow->_getDbTableName() || $fks[1]->getForeignTableName() === $dbRow->_getDbTableName()) {
1537
                    $junctionTables[] = $table->getName();
1538
                }
1539
            }
1540
        }
1541
1542
        return $junctionTables;
1543
    }
1544
}
1545