Retrieve session snippet rate and hat fields

[airbundle] / Entity / Location.php

diff --git a/Entity/Location.php b/Entity/Location.php
index 73fc6b980ff79370d14e6fb6103b4fc0e87cd752..fd619698cbf4ff8b455f70f728bfec446bca5737 100644 (file)
--- a/Entity/Location.php
+++ b/Entity/Location.php
@@ -66,11 +66,23 @@ class Location {
         */
        private $sessions;
 
+       /**
+        * @var \Doctrine\Common\Collections\Collection
+        */
+       private $snippets;
+
+       /**
+        * @var \Doctrine\Common\Collections\Collection
+        */
+       private $users;
+
        /**
         * Constructor
         */
        public function __construct() {
                $this->sessions = new \Doctrine\Common\Collections\ArrayCollection();
+               $this->snippets = new \Doctrine\Common\Collections\ArrayCollection();
+               $this->users = new \Doctrine\Common\Collections\ArrayCollection();
        }
 
        /**
@@ -333,6 +345,68 @@ class Location {
                return $this->sessions;
        }
 
+       /**
+        * Add snippet
+        *
+        * @param \Rapsys\AirBundle\Entity\Snippet $snippet
+        *
+        * @return Location
+        */
+       public function addSnippet(\Rapsys\AirBundle\Entity\Snippet $snippet) {
+               $this->snippets[] = $snippet;
+
+               return $this;
+       }
+
+       /**
+        * Remove snippet
+        *
+        * @param \Rapsys\AirBundle\Entity\Snippet $snippet
+        */
+       public function removeSnippet(\Rapsys\AirBundle\Entity\Snippet $snippet) {
+               $this->snippets->removeElement($snippet);
+       }
+
+       /**
+        * Get snippets
+        *
+        * @return \Doctrine\Common\Collections\Collection
+        */
+       public function getSnippets() {
+               return $this->snippets;
+       }
+
+       /**
+        * Add user
+        *
+        * @param \Rapsys\AirBundle\Entity\User $user
+        *
+        * @return Location
+        */
+       public function addUser(\Rapsys\AirBundle\Entity\User $user) {
+               $this->users[] = $user;
+
+               return $this;
+       }
+
+       /**
+        * Remove user
+        *
+        * @param \Rapsys\AirBundle\Entity\User $user
+        */
+       public function removeUser(\Rapsys\AirBundle\Entity\User $user) {
+               $this->users->removeElement($user);
+       }
+
+       /**
+        * Get users
+        *
+        * @return \Doctrine\Common\Collections\Collection
+        */
+       public function getUsers() {
+               return $this->users;
+       }
+
        /**
         * Returns a string representation of the location
         *