From 1d2f46bda7800fba1c9e85582496b32ec920cb87 Mon Sep 17 00:00:00 2001
From: Benjamin Michotte <bmichotte@gmail.com>
Date: Tue, 24 Jan 2012 19:08:25 +0100
Subject: [PATCH] Add fluent code for relations

---
 lib/Doctrine/ORM/Tools/EntityGenerator.php | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/lib/Doctrine/ORM/Tools/EntityGenerator.php b/lib/Doctrine/ORM/Tools/EntityGenerator.php
index 915675712..0f369c4d0 100644
--- a/lib/Doctrine/ORM/Tools/EntityGenerator.php
+++ b/lib/Doctrine/ORM/Tools/EntityGenerator.php
@@ -128,10 +128,12 @@ public function <methodName>(<methodTypeHint>$<variableName><variableDefault>)
  * <description>
  *
  * @param <variableType>$<variableName>
+ * @return <entity>
  */
 public function <methodName>(<methodTypeHint>$<variableName>)
 {
 <spaces>$this-><fieldName>[] = $<variableName>;
+<spaces>return $this;
 }';
 
     private static $_lifecycleCallbackMethodTemplate =