Query Unit Of Work

QueryUnitOfWorkMixin.php

<?php

namespace App\Library\Mixin;

trait QueryUnitOfWorkMixin
{
	public function queryUow($className, array $properties)
    {
        foreach($this->getEntityManager()->getUnitOfWork()->getScheduledEntityInsertions() as $entity){
             if(is_a($entity, $className)){
                $matches = true;
                foreach($properties as $propertName => $value){
                    $getMethodName = 'get'.ucfirst($propertName);
                    if(is_a($value, \DateTime::class)){
                        if($entity->$getMethodName() != $value){
                            $matches = false;
                        }
                    }else{
                        if($entity->$getMethodName() !== $value){
                            $matches = false;
                        }
                    }
                }
                if($matches){
                    return $entity;
                }
            }
        }
        return null;
    }

    public function getUowInsertions($className)
    {
        $entities = [];
        foreach($this->getEntityManager()->getUnitOfWork()->getScheduledEntityInsertions() as $entity){
             if(is_a($entity, $className)){
                $entities[] = $entity;
            }
        }
        return $entities;
    }
}