added more flexibility to configure the factories

oscarotero · Dec 3, 2015 · f31af29 · f31af29
1 parent b39b015
commit f31af29
Show file tree

Hide file tree

Showing 2 changed files with 44 additions and 7 deletions.
diff --git a/src/EntityFactory.php b/src/EntityFactory.php
@@ -53,7 +53,7 @@ public function setNamespace($namespace)
     }
 
     /**
-     * Set the queryFactory instance used by the entities.
+     * Set the QueryFactory instance used by the entities.
      *
      * @param QueryFactory $queryFactory
      *
@@ -67,7 +67,17 @@ public function setQueryFactory(QueryFactory $queryFactory)
     }
 
     /**
-     * Set the fieldFactory instance used by the entities.
+     * Returns the QueryFactory instance used by the entities.
+     *
+     * @return QueryFactory
+     */
+    public function getQueryFactory()
+    {
+        return $this->queryFactory;
+    }
+
+    /**
+     * Set the FieldFactory instance used by the entities.
      *
      * @param FieldFactory $fieldFactory
      *
@@ -80,6 +90,16 @@ public function setFieldFactory(FieldFactory $fieldFactory)
         return $this;
     }
 
+    /**
+     * Returns the FieldFactory instance used by the entities.
+     *
+     * @return FieldFactory
+     */
+    public function getFieldFactory(FieldFactory $fieldFactory)
+    {
+        return $this->fieldFactory;
+    }
+
     /**
      * Set whether the entities are autocreated or not.
      *

diff --git a/src/SimpleCrud.php b/src/SimpleCrud.php
@@ -21,15 +21,32 @@ class SimpleCrud
      */
     public function __construct(PDO $connection, EntityFactory $entityFactory = null)
     {
-        if ($entityFactory === null) {
-            $entityFactory = (new EntityFactory())->setAutocreate();
-        }
+        $this->connection = $connection;
+        $this->connection->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);
+
+        $this->setEntityFactory($entityFactory ?: (new EntityFactory())->setAutocreate());
+    }
 
+    /**
+     * Set the EntityFactory instance used to create all entities.
+     *
+     * @param EntityFactory $entityFactory
+     */
+    public function setEntityFactory(EntityFactory $entityFactory)
+    {
         $entityFactory->setDb($this);
+
         $this->entityFactory = $entityFactory;
+    }
 
-        $this->connection = $connection;
-        $this->connection->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);
+    /**
+     * Returns the EntityFactory instance used.
+     *
+     * @return EntityFactory
+     */
+    public function getEntityFactory()
+    {
+        return $this->entityFactory;
     }
 
     /**