PHP Doctrine ORM - Proper truncate table with association/relation

truncate.php

<?php 

use Doctrine\DBAL\Connection;
use Doctrine\DBAL\Platforms\AbstractPlatform;

function truncateTable(
    Connection          $connection,
    AbstractPlatform    $databasePlatform, 
    string              $entityName
): void
{
    $classMetadata      = $this->entityManager->getClassMetadata($entityName);
    $tablesToTruncate   = [$classMetadata->getTableName()];

    foreach ($classMetadata->getAssociationMappings() as $associationMapping) {
        if (array_key_exists('joinTable', $associationMapping)
            && array_key_exists('name', $associationMapping['joinTable'])
            && !in_array($associationMapping['joinTable']['name'], $tablesToTruncate)
        ) {
            $tablesToTruncate[] = $associationMapping['joinTable']['name'];
        }
    }

    $connection->executeQuery('SET FOREIGN_KEY_CHECKS=0');

    foreach ($tablesToTruncate as $tableName) {
        $connection->executeStatement($databasePlatform->getTruncateTableSQL($tableName));
    }

    $connection->executeQuery('SET FOREIGN_KEY_CHECKS=1');
}