Symfony2 - Overriding default Doctrine query when working with entities

Question

For my project, I have a workspace (kind of a user) with many projects and I am wondering if there is a way to override the default Doctrine query for when I call $workspace->getProjects() to only fetch the active projects only (not the archived one). This way I won't have to filter my collection and it will reduce the size of the returned data from the database.

/**
 * Acme\DemoBundle\Entity\Workspace
 *
 * @ORM\Table()
 * @ORM\Entity
 */
class Workspace {

    /**
     * @var integer $id
     *
     * @ORM\Column(name="id", type="integer")
     * @ORM\Id
     * @ORM\GeneratedValue(strategy="AUTO")
    private $id;

    /**
     * @var ArrayCollection $projects
     * 
     * @ORM\OneToMany(targetEntity="Project", mappedBy="workspace")
     */
    private $projects;


    /**
     * Add projects
     *
     * @param Project $projects
     * @return Workspace
     */
     public function addProject( Project $projects ) {
         $this->projects[] = $projects;

         return $this;
     }

    /**
     * Remove projects
     *
     * @param Project $projects
     */
    public function removeProject( Project $projects ) {
        $this->projects->removeElement( $projects );
    }

    /**
     * Get projects
     *
     * @return Collection 
     */
    public function getProjects() {
        return $this->projects;
    }

You can achieve this by using ArrayCollections filter method. [stackoverflow.com/questions/8334356/… [1]: stackoverflow.com/questions/8334356/… — Turdaliev Nursultan
– Turdaliev Nursultan, Commented Nov 25, 2013 at 15:08

Eduardo Casas · Accepted Answer · 2013-11-25 15:04:45Z

2

You will have to write your own method in the Entity Repository Class.

http://symfony.com/doc/current/book/doctrine.html#custom-repository-classes

answered Nov 25, 2013 at 15:04

Eduardo Casas

5784 silver badges11 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

Frank6 Over a year ago

And how Doctrine knows which method to use from my custom repository ?

ferdynator Over a year ago

You will have to call it yourself. Instead of using $workflog->getProjects() you use $workflowRepository->findProjectsForWorkflow($workflow);

Niavlys Bobar · Accepted Answer · 2016-08-31 13:20:18Z

2

To do that, you can use criteria.

Here is an exemple from the doctrine doc, to adapt to your needs

use Doctrine\Common\Collections\Criteria;

$criteria = Criteria::create()
->where(Criteria::expr()->eq("birthday", "1982-02-17"))
->orderBy(array("username" => Criteria::ASC))
->setFirstResult(0)
->setMaxResults(20)
;

$birthdayUsers = $object-getUsers()->matching($criteria);

answered Aug 31, 2016 at 13:20

Niavlys Bobar

211 bronze badge

Comments

pomaxa · Accepted Answer · 2013-11-25 15:02:52Z

You can create some listener, and catch doctrine events: http://docs.doctrine-project.org/projects/doctrine1/en/latest/en/manual/event-listeners.html

And modify your query before execute as you wish;

Example from Doctrine docs, for custom hook on delete:

<?php 
class UserListener extends Doctrine_EventListener
{
    /**
     * Skip the normal delete options so we can override it with our own
     *
     * @param Doctrine_Event $event
     * @return void
     */
    public function preDelete( Doctrine_Event $event )
    {
        $event->skipOperation();
    }

    /**
     * Implement postDelete() hook and set the deleted flag to true
     *
     * @param Doctrine_Event $event
     * @return void
     */
    public function postDelete( Doctrine_Event $event )
    {
        $name                       = $this->_options['name'];
        $event->getInvoker()->$name = true;
        $event->getInvoker()->save();
    }

    /**
     * Implement preDqlDelete() hook and modify a dql delete query so it updates the deleted flag
     * instead of deleting the record
     *
     * @param Doctrine_Event $event
     * @return void
     */
    public function preDqlDelete( Doctrine_Event $event )
    {
        $params = $event->getParams();
        $field  = $params['alias'] . '.deleted';
        $q      = $event->getQuery();

        if ( ! $q->contains( $field ) )
        {
            $q->from('')->update( $params['component'] . ' ' . $params['alias'] );
            $q->set( $field, '?', array(false) );
            $q->addWhere( $field . ' = ?', array(true) );
        }
    }

    /**
     * Implement preDqlDelete() hook and add the deleted flag to all queries for which this model
     * is being used in.
     *
     * @param Doctrine_Event $event
     * @return void
     */
    public function preDqlSelect( Doctrine_Event $event )
    {
        $params = $event->getParams();
        $field  = $params['alias'] . '.deleted';
        $q      = $event->getQuery();

        if ( ! $q->contains( $field ) )
        {
            $q->addWhere( $field . ' = ?', array(false) );
        }
    }
}

Rafael Adel · Accepted Answer · 2013-11-25 15:32:17Z

0

I think you can put your logic inside the getter method itself

public function getProjects() {
        $filtered_projects = array()
        foreach($this->projects as $project)
        {
           // Your logic here
           // e.g.
           // if($project->getIsActive()){
           //    $filtered_projects[] = $project;
           // } 
        }
        return $filtered_projects;
    }

answered Nov 25, 2013 at 15:32

Rafael Adel

7,75927 gold badges80 silver badges120 bronze badges

1 Comment

ferdynator Over a year ago

He wanted to filter them in the query. Not afterwards in PHP.

Collectives™ on Stack Overflow

Symfony2 - Overriding default Doctrine query when working with entities

4 Answers 4

2 Comments

Comments

1 Comment

1 Comment

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

4 Answers 4

2 Comments

Comments

1 Comment

1 Comment

Your Answer

Sign up or log in

Post as a guest

Linked

Related