Make setter fluent. (#783)

ongr-io · Apr 13, 2017 · c3c53f1 · c3c53f1
1 parent aa75c61
commit c3c53f1
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/Generator/DocumentGenerator.php b/Generator/DocumentGenerator.php
@@ -59,10 +59,14 @@ public function is<methodName>()
  * Sets <fieldName>
  *
  * @param string $<fieldName>
+ *
+ * @return self
  */
 public function set<methodName>($<fieldName>)
 {
 <spaces>$this-><fieldName> = $<fieldName>;
+
+<spaces>return $this;
 }';
 
     /**
@@ -160,7 +164,7 @@ private function generateDocumentMethods(array $metadata)
             if (isset($property['property_type']) && $property['property_type'] === 'boolean') {
                 $lines[] = $this->generateDocumentMethod($property, $this->isMethodTemplate) . "\n";
             }
-            
+
             $lines[] = $this->generateDocumentMethod($property, $this->getMethodTemplate) . "\n";
         }