Doctrine

Timestamp:

01/22/08 21:42:17 (18 months ago)

Author:

jwage

Message:

Connection refactoring to allow create/drop database functionality for sqlite. fixes #480

Location:

trunk

Files:

• lib/Doctrine.php (modified) (2 diffs)
• lib/Doctrine/Connection.php (modified) (4 diffs)
• lib/Doctrine/Connection/Sqlite.php (modified) (1 diff)
• lib/Doctrine/Export.php (modified) (4 diffs)
• lib/Doctrine/Export/Sqlite.php (modified) (1 diff)
• lib/Doctrine/Manager.php (modified) (5 diffs)
• lib/Doctrine/Query/Tokenizer.php (modified) (1 diff)
• lib/Doctrine/Task/CreateDb.php (modified) (1 diff)
• lib/Doctrine/Task/DropDb.php (modified) (1 diff)
• models/TreeLeaf.php (modified) (1 diff)
• tests/Export/SqliteTestCase.php (modified) (3 diffs)
• tests/ManagerTestCase.php (modified) (1 diff)
• tests/run.php (modified) (1 diff)
• tests/Ticket/480TestCase.php (modified) (1 diff)
• tests/TreeStructureTestCase.php (modified) (1 diff)
• tools/sandbox/config.php.dist (modified) (2 diffs)
• tools/sandbox/sandbox.db (deleted)

trunk/lib/Doctrine.php

@@ -895 +895 @@
     public static function createDatabases($specifiedConnections = array())
     {
-        if ( ! is_array($specifiedConnections)) {
-            $specifiedConnections = (array) $specifiedConnections;
-        }
-
-        $manager = Doctrine_Manager::getInstance();
-        $connections = $manager->getConnections();
-
-        $results = array();
-
-        foreach ($connections as $name => $connection) {
-            if ( ! empty($specifiedConnections) && !in_array($name, $specifiedConnections)) {
-                continue;
-            }
-
-            $info = $manager->parsePdoDsn($connection->getOption('dsn'));
-            $username = $connection->getOption('username');
-            $password = $connection->getOption('password');
-
-            // Make connection without database specified so we can create it
-            $connect = $manager->openConnection(new PDO($info['scheme'] . ':host=' . $info['host'], $username, $password), 'tmp_connection', false);
-
-            try {
-                // Create database
-                $connect->export->createDatabase($name);
-
-                // Close the tmp connection with no database
-                $manager->closeConnection($connect);
-
-                // Close original connection
-                $manager->closeConnection($connection);
-
-                // Reopen original connection with newly created database
-                $manager->openConnection(new PDO($info['dsn'], $username, $password), $name, true);
-
-                $results[$name] = true;
-            } catch (Exception $e) {
-                $results[$name] = false;
-            }
-        }
-
-        return $results;
+        return Doctrine_Manager::getInstance()->createDatabases($specifiedConnections);
     }
 
@@ -948 +908 @@
     public static function dropDatabases($specifiedConnections = array())
     {
-        if ( ! is_array($specifiedConnections)) {
-            $specifiedConnections = (array) $specifiedConnections;
-        }
-
-        $manager = Doctrine_Manager::getInstance();
-
-        $connections = $manager->getConnections();
-
-        $results = array();
-
-        foreach ($connections as $name => $connection) {
-            if ( ! empty($specifiedConnections) && !in_array($name, $specifiedConnections)) {
-                continue;
-            }
-
-            try {
-                $connection->export->dropDatabase($connection->getDatabaseName());
-
-                $results[$name] = true;
-            } catch (Exception $e) {
-                $results[$name] = false;
-            }
-        }
-
-        return $results;
+        return Doctrine_Manager::getInstance()->dropDatabases($specifiedConnections);
     }
 

trunk/lib/Doctrine/Connection.php

@@ -76 +76 @@
      */
     protected $_mappers = array();
+    
+    /**
+     * $_name
+     *
+     * Name of the connection
+     *
+     * @var string $_name
+     */
+    protected $_name;
 
     /**
@@ -307 +316 @@
     public function getName()
     {
+        return $this->_name;
+    }
+    
+    public function setName($name)
+    {
+        $this->_name = $name;
+    }
+    
+    public function getDriverName()
+    {
         return $this->driverName;
     }
@@ -341 +360 @@
                     break;
                 default:
-                    $class = 'Doctrine_' . ucwords($name) . '_' . $this->getName();
+                    $class = 'Doctrine_' . ucwords($name) . '_' . $this->getDriverName();
                     $this->modules[$name] = new $class($this);
                 }
@@ -1382 +1401 @@
 
     /**
+     * createDatabase
+     *
+     * @return void
+     */
+    public function createDatabase()
+    {
+      $manager = $this->getManager();
+
+      $info = $manager->parsePdoDsn($this->getOption('dsn'));
+      $username = $this->getOption('username');
+      $password = $this->getOption('password');
+
+      // Make connection without database specified so we can create it
+      $connect = $manager->openConnection(new PDO($info['scheme'] . ':host=' . $info['host'], $username, $password), 'tmp_connection', false);
+      
+      try {
+          // Create database
+          $connect->export->createDatabase($info['dbname']);
+
+          // Close the tmp connection with no database
+          $manager->closeConnection($connect);
+
+          // Close original connection
+          $manager->closeConnection($this);
+
+          // Reopen original connection with newly created database
+          $manager->openConnection(new PDO($info['dsn'], $username, $password), $this->getName(), true);
+
+          return 'Successfully created database for connection "' . $this->getName() . '" named "' . $info['dbname'] . '"';
+      } catch (Exception $e) {
+          return $e;
+      }
+    }
+
+    /**
+     * dropDatabase
+     *
+     * @return void
+     */
+    public function dropDatabase()
+    {
+      try {
+          $info = $this->getManager()->parsePdoDsn($this->getOption('dsn'));
+          
+          $this->export->dropDatabase($info['dbname']);
+
+          return 'Successfully dropped database for connection "' . $this->getName() . '" named "' . $info['dbname'] . '"';
+      } catch (Exception $e) {
+          return $e;
+      }
+    }
+
+    /**
      * returns a string representation of this object
      * @return string

trunk/lib/Doctrine/Connection/Sqlite.php

@@ -97 +97 @@
 
     /**
-     * getDatabaseFile
+     * createDatabase
      *
-     * @param string $name      the name of the database
-     * @return string
+     * @return void
      */
-    public function getDatabaseFile($name)
+    public function createDatabase()
     {
-        return $name . '.db';
+      try {
+          $manager = $this->getManager();
+
+          $info = $manager->parseDsn($this->getOption('dsn'));
+
+          $this->export->createDatabase($info['database']);
+
+          return 'Successfully created database for connection "' . $this->getName() . '" at path "' . $info['database'] . '"';
+      } catch (Exception $e) {
+          return $e;
+      }
+    }
+
+    /**
+     * dropDatabase
+     *
+     * @return void
+     */
+    public function dropDatabase()
+    {
+      try {
+          $info = $this->getManager()->parseDsn($this->getOption('dsn'));
+
+          $this->export->dropDatabase($info['database']);
+
+          return 'Successfully dropped database for connection "' . $this->getName() . '" at path "' . $info['database'] . '"';
+      } catch (Exception $e) {
+          return $e;
+      }
     }
 }

trunk/lib/Doctrine/Export.php

@@ -1057 +1057 @@
     /**
      * exportClasses
+     *
+     * FIXME: This method is a big huge hack. The sql needs to be executed in the correct order. I have some stupid logic to 
+     * make sure they are in the right order.
+     *
      * method for exporting Doctrine_Record classes to a schema
      *
@@ -1074 +1078 @@
             if ( ! isset($connections[$connectionName])) {
                 $connections[$connectionName] = array();
-                $connections[$connectionName]['creates'] = array();
+                $connections[$connectionName]['create_tables'] = array();
+                $connections[$connectionName]['create_sequences'] = array();
                 $connections[$connectionName]['alters'] = array();
             }
@@ -1082 +1087 @@
             // We need these to happen first
             foreach ($sql as $key => $query) {
-                if (strstr($query, 'CREATE')) {
-                    $connections[$connectionName]['creates'][] = $query;
-                    // Unset the create from sql so we can have an array of everything else but creates
+                if (strstr($query, 'CREATE TABLE')) {
+                    $connections[$connectionName]['create_tables'][] = $query;
+
+                    unset($sql[$key]);
+                }
+
+                if (strstr($query, 'CREATE SEQUENCE')) {
+                    $connections[$connectionName]['create_sequences'][] = $query;
+
                     unset($sql[$key]);
                 }
@@ -1095 +1106 @@
         $build = array();
         foreach ($connections as $connectionName => $sql) {
-            $build[$connectionName] = array_merge($sql['creates'], $sql['alters']);
+            $build[$connectionName] = array_merge($sql['create_tables'], $sql['create_sequences'], $sql['alters']);
         }
 

trunk/lib/Doctrine/Export/Sqlite.php

@@ -35 +35 @@
 {
     /**
+     * dropDatabase
+     *
      * drop an existing database
      *
-     * @param string $name                  name of the database that should be dropped
+     * @param string $databaseFile          Path of the database that should be dropped
      * @throws Doctrine_Export_Exception    if the database file does not exist
      * @throws Doctrine_Export_Exception    if something failed during the removal of the database file
      * @return void
      */
-    public function dropDatabase($name)
-    {
-        $databaseFile = $this->conn->getDatabaseFile($name);
+    public function dropDatabase($databaseFile)
+    {
         if ( ! @file_exists($databaseFile)) {
             throw new Doctrine_Export_Exception('database does not exist');
         }
+
         $result = @unlink($databaseFile);
+
         if ( ! $result) {
             throw new Doctrine_Export_Exception('could not remove the database file');
         }
+    }
+
+    /**
+     * createDatabase
+     *
+     * Create sqlite database file
+     *
+     * @param string $databaseFile  Path of the database that should be dropped
+     * @return void
+     */
+    public function createDatabase($databaseFile)
+    {
+        return new PDO('sqlite:' . $databaseFile);
     }
 

trunk/lib/Doctrine/Manager.php

@@ -313 +313 @@
         $className = $drivers[$driverName];
         $conn = new $className($this, $adapter);
+        $conn->setName($name);
 
         $this->_connections[$name] = $conn;
@@ -346 +347 @@
         $e = explode(';', $e[1]);
         foreach ($e as $string) {
-            list($key, $value) = explode('=', $string);
-            $parts[$key] = $value;
-        }
-        
+            if ($string) {
+                $e2 = explode('=', $string);
+
+                if (isset($e2[0]) && isset($e2[1])) {
+                    list($key, $value) = $e2;
+                    $parts[$key] = $value;
+                }
+            }
+        }
+
         return $parts;
     }
@@ -361 +368 @@
     public function parseDsn($dsn)
     {
-        //fix linux sqlite dsn so that it will parse correctly
-        $dsn = str_replace("///", "/", $dsn);
-
+        // fix sqlite dsn so that it will parse correctly
+        $dsn = str_replace("////", "/", $dsn);
+        $dsn = str_replace("///c:/", "//c:/", $dsn);
+        
         // silence any warnings
         $parts = @parse_url($dsn);
@@ -452 +460 @@
         }
 
-
         return $parts;
     }
@@ -694 +701 @@
         }
         return $this->_connections[$i];
+    }
+
+    /**
+     * createDatabases
+     *
+     * Creates databases for connections
+     *
+     * @param string $specifiedConnections Array of connections you wish to create the database for
+     * @return void
+     */
+    public function createDatabases($specifiedConnections = array())
+    {
+        if ( ! is_array($specifiedConnections)) {
+            $specifiedConnections = (array) $specifiedConnections;
+        }
+
+        $results = array();
+
+        foreach ($this as $name => $connection) {
+            if ( ! empty($specifiedConnections) && !in_array($name, $specifiedConnections)) {
+                continue;
+            }
+
+            $results[$name] = $connection->createDatabase();
+        }
+
+        return $results;
+    }
+
+    /**
+     * dropDatabases
+     *
+     * Drops databases for connections
+     *
+     * @param string $specifiedConnections Array of connections you wish to drop the database for
+     * @return void
+     */
+    public function dropDatabases($specifiedConnections = array())
+    {
+        if ( ! is_array($specifiedConnections)) {
+            $specifiedConnections = (array) $specifiedConnections;
+        }
+
+        $results = array();
+
+        foreach ($this as $name => $connection) {
+            if ( ! empty($specifiedConnections) && !in_array($name, $specifiedConnections)) {
+                continue;
+            }
+
+            $results[$name] = $connection->dropDatabase();
+        }
+
+        return $results;
     }
 

trunk/lib/Doctrine/Query/Tokenizer.php

@@ -378 +378 @@
             }
         }
-        $term[$i - 1] = array($term[$i - 1], '');
+
+        if (isset($term[$i - 1])) {
+            $term[$i - 1] = array($term[$i - 1], '');
+        }
 
         return $term;

trunk/lib/Doctrine/Task/CreateDb.php

@@ -40 +40 @@
         $results = Doctrine::createDatabases();
         
-        foreach ($results as $dbName => $bool) {
-            $msg = $bool ? 'Successfully created database named: "' . $dbName . '"':'Could not create database named: "' .$dbName . '"';
+        foreach ($results as $name => $result) {
+            $msg = $result instanceof Exception ? 'Could not create database for connection: "' .$name . '." Failed with exception: ' . $result->getMessage():$result;
             
             $this->notify($msg);

trunk/lib/Doctrine/Task/DropDb.php

@@ -49 +49 @@
         $results = Doctrine::dropDatabases();
         
-        foreach ($results as $dbName => $bool) {
-            $msg = $bool ? 'Successfully dropped database named: "' . $dbName . '"':'Could not drop database named: "' .$dbName . '"';
+        foreach ($results as $name => $result) {
+            $msg = $result instanceof Exception ? 'Could not drop database for connection: "' .$name . '." Failed with exception: ' . $result->getMessage():$result;
             
             $this->notify($msg);

trunk/models/TreeLeaf.php

@@ -7 +7 @@
         $this->hasColumn('parent_id', 'integer');
     }
+
     public function setUp() 
     {
-        $this->hasOne('TreeLeaf as Parent', 'TreeLeaf.parent_id');
-        $this->hasMany('TreeLeaf as Children', 'TreeLeaf.parent_id');
+        $this->hasOne('TreeLeaf as Parent', array('local' => 'parent_id', 'foreign' => 'id'));
+        $this->hasMany('TreeLeaf as Children', array('local' => 'id', 'foreign' => 'parent_id'));
     }
 }

trunk/tests/Export/SqliteTestCase.php

@@ -33 +33 @@
 class Doctrine_Export_Sqlite_TestCase extends Doctrine_UnitTestCase
 {
-    public function testCreateDatabaseDoesNotExecuteSql()
+    public function testCreateDatabaseDoesNotExecuteSqlAndCreatesSqliteFile()
     {
-        try {
-            $this->export->createDatabase('db');
-            $this->fail();
-        } catch(Doctrine_Export_Exception $e) {
-            $this->pass();
-        }
+        $this->export->createDatabase('sqlite.db');
+        
+        $this->assertTrue(file_exists('sqlite.db'));
     }
-    public function testDropDatabaseDoesNotExecuteSql()
+    public function testDropDatabaseDoesNotExecuteSqlAndDeletesSqliteFile()
     {
-        try {
-            $this->export->dropDatabase('db');
-            $this->fail();
-        } catch(Doctrine_Export_Exception $e) {
-            $this->pass();
-        }
+        $this->export->dropDatabase('sqlite.db');
+
+        $this->assertFalse(file_exists('sqlite.db'));
     }
     public function testCreateTableSupportsAutoincPks()
@@ -172 +166 @@
         $this->export->createTable('sometable', $fields, $options);
 
-        //removed this assertion and inserted the two below
-//        $this->assertEqual($this->adapter->pop(), 'CREATE TABLE sometable (id INTEGER PRIMARY KEY AUTOINCREMENT, name VARCHAR(4), INDEX myindex (id ASC, name DESC))');
-
         $this->assertEqual($this->adapter->pop(),"CREATE INDEX myindex_idx ON sometable (id ASC, name DESC)");
 
@@ -180 +171 @@
 
     }
-
-    /**
-    public function testExportSupportsEmulationOfCascadingDeletes()
-    {
-        $r = new ForeignKeyTest;
-
-        $this->assertEqual($this->adapter->pop(), 'COMMIT');
-        $this->assertEqual($this->adapter->pop(), 'CREATE TRIGGER doctrine_foreign_key_test_cscd_delete AFTER DELETE ON foreign_key_test BEGIN DELETE FROM foreign_key_test WHERE parent_id = old.id;END;');
-        $this->assertEqual($this->adapter->pop(), 'CREATE TABLE foreign_key_test (id INTEGER PRIMARY KEY AUTOINCREMENT, name VARCHAR(2147483647), code INTEGER, content VARCHAR(4000), parent_id INTEGER)');
-        $this->assertEqual($this->adapter->pop(), 'BEGIN TRANSACTION');
-    }
-    */
 }

trunk/tests/ManagerTestCase.php

@@ -66 +66 @@
         // It expects only // since it thinks it is parsing a url
         // The problem after that is that the dns is not valid when being passed to PDO
-        $sqlite = 'sqlite:///full/unix/path/to/file.db';
-        $sqlitewin = 'sqlite://c:/full/windows/path/to/file.db';
+        $sqlite = 'sqlite:////full/unix/path/to/file.db';
+        $sqlitewin = 'sqlite:///c:/full/windows/path/to/file.db';
         
         $manager = Doctrine_Manager::getInstance();

trunk/tests/run.php

@@ -154 +154 @@
 
 // Utility components
-$plugins = new GroupTest('Plugin tests: View, Validator, Hook','plugins');
-//$utility->addTestCase(new Doctrine_PessimisticLocking_TestCase());
+$plugins = new GroupTest('Plugin tests: View, Validator, Hook', 'plugins');
+//$plugins->addTestCase(new Doctrine_PessimisticLocking_TestCase());
 //$plugins->addTestCase(new Doctrine_Plugin_TestCase());
 $plugins->addTestCase(new Doctrine_View_TestCase());

trunk/tests/Ticket/480TestCase.php

@@ -46 +46 @@
     public function testInit()
     {
-                $this->dbh = new Doctrine_Adapter_Mock('oracle');
-                $this->conn = Doctrine_Manager::getInstance()->openConnection($this->dbh);
+        $this->dbh = new Doctrine_Adapter_Mock('oracle');
+        $this->conn = Doctrine_Manager::getInstance()->openConnection($this->dbh);
     }
 

trunk/tests/TreeStructureTestCase.php

@@ -104 +104 @@
         $o4->save();
 
-        $o1->Children;
         $this->assertFalse(isset($o1->Parent));
+        $this->assertTrue(isset($o2->Parent));
+        $this->assertTrue($o2->Parent === $o1);
+        $this->assertFalse(isset($o4->Parent));
+        
         $this->assertTrue(count($o1->Children) == 2);
         $this->assertTrue(count($o1->get('Children')) == 2);
 
-        $this->assertTrue(isset($o2->Parent));
-        $this->assertTrue($o2->Parent === $o1);
-
         $this->assertTrue(count($o4->Children) == 0);
-        $this->assertFalse(isset($o4->Parent));
     }
     public function testTreeStructureFetchingWorksWithDql()

trunk/tools/sandbox/config.php.dist

@@ -43 +43 @@
 define('YAML_SCHEMA_PATH', SANDBOX_PATH . DIRECTORY_SEPARATOR . 'schema');
 define('DB_PATH', SANDBOX_PATH . DIRECTORY_SEPARATOR . 'sandbox.db');
-define('DSN', 'sqlite:' . DB_PATH);
+define('DSN', 'sqlite:///' . DB_PATH);
 
 require_once(DOCTRINE_PATH . DIRECTORY_SEPARATOR . 'Doctrine.php');
@@ -49 +49 @@
 spl_autoload_register(array('Doctrine', 'autoload'));
 
-$pdo = new PDO(DSN);
-Doctrine_Manager::connection($pdo, 'sandbox');
+Doctrine_Manager::connection(DSN, 'sandbox');
 
 Doctrine_Manager::getInstance()->setAttribute(Doctrine::ATTR_MODEL_LOADING, Doctrine::MODEL_LOADING_CONSERVATIVE);