Fix mysql create_table method
This commit is contained in:
parent
0567c3443f
commit
1178b5704d
@ -57,7 +57,7 @@ class MySQL extends DB_PDO {
|
||||
public function get_dbs()
|
||||
{
|
||||
$res = $this->query("SHOW DATABASES");
|
||||
return $this->fetchAll(PDO::FETCH_ASSOC);
|
||||
return array_values($this->fetchAll(PDO::FETCH_ASSOC));
|
||||
}
|
||||
|
||||
// --------------------------------------------------------------------------
|
||||
@ -70,7 +70,16 @@ class MySQL extends DB_PDO {
|
||||
public function get_tables()
|
||||
{
|
||||
$res = $this->query("SHOW TABLES");
|
||||
return $res->fetchAll(PDO::FETCH_ASSOC);
|
||||
|
||||
$tables = array();
|
||||
$rows = $res->fetchAll(PDO::FETCH_NUM);
|
||||
|
||||
foreach($rows as $r)
|
||||
{
|
||||
$tables[] = $r[0];
|
||||
}
|
||||
|
||||
return $tables;
|
||||
}
|
||||
|
||||
// --------------------------------------------------------------------------
|
||||
|
@ -29,7 +29,60 @@
|
||||
*/
|
||||
public function create_table($name, $columns, array $constraints=array(), array $indexes=array())
|
||||
{
|
||||
//TODO: implement
|
||||
$column_array = array();
|
||||
|
||||
// Reorganize into an array indexed with column information
|
||||
// Eg $column_array[$colname] = array(
|
||||
// 'type' => ...,
|
||||
// 'constraint' => ...,
|
||||
// 'index' => ...,
|
||||
// )
|
||||
foreach($columns as $colname => $type)
|
||||
{
|
||||
if(is_numeric($colname))
|
||||
{
|
||||
$colname = $type;
|
||||
}
|
||||
|
||||
$column_array[$colname] = array();
|
||||
$column_array[$colname]['type'] = ($type !== $colname) ? $type : '';
|
||||
}
|
||||
|
||||
if( ! empty($constraints))
|
||||
{
|
||||
foreach($constraints as $col => $const)
|
||||
{
|
||||
$column_array[$col]['constraint'] = "{$const} ({$col})";
|
||||
}
|
||||
}
|
||||
|
||||
// Join column definitons together
|
||||
$columns = array();
|
||||
foreach($column_array as $n => $props)
|
||||
{
|
||||
$n = trim($n, '`');
|
||||
|
||||
$str = "`{$n}` ";
|
||||
$str .= (isset($props['type'])) ? "{$props['type']} " : "";
|
||||
|
||||
$columns[] = $str;
|
||||
}
|
||||
|
||||
// Add constraints
|
||||
foreach($column_array as $n => $props)
|
||||
{
|
||||
if (isset($props['constraint']))
|
||||
{
|
||||
$columns[] = $props['constraint'];
|
||||
}
|
||||
}
|
||||
|
||||
// Generate the sql for the creation of the table
|
||||
$sql = "CREATE TABLE IF NOT EXISTS `{$name}` (";
|
||||
$sql .= implode(", ", $columns);
|
||||
$sql .= ")";
|
||||
|
||||
return $sql;
|
||||
}
|
||||
|
||||
// --------------------------------------------------------------------------
|
||||
|
@ -21,7 +21,7 @@ class MySQLTest extends UnitTestCase {
|
||||
|
||||
function __construct()
|
||||
{
|
||||
|
||||
parent::__construct();
|
||||
}
|
||||
|
||||
function setUp()
|
||||
@ -58,6 +58,7 @@ class MySQLTest extends UnitTestCase {
|
||||
if (empty($this->db)) return;
|
||||
|
||||
$tables = $this->db->get_tables();
|
||||
|
||||
$this->assertTrue(is_array($tables));
|
||||
}
|
||||
|
||||
@ -85,7 +86,7 @@ class MySQLTest extends UnitTestCase {
|
||||
//Attempt to create the table
|
||||
$sql = $this->db->sql->create_table('create_test',
|
||||
array(
|
||||
'id' => 'INTEGER',
|
||||
'id' => 'int(10)',
|
||||
'key' => 'TEXT',
|
||||
'val' => 'TEXT',
|
||||
),
|
||||
@ -93,12 +94,13 @@ class MySQLTest extends UnitTestCase {
|
||||
'id' => 'PRIMARY KEY'
|
||||
)
|
||||
);
|
||||
|
||||
$this->db->query($sql);
|
||||
|
||||
//Attempt to create the table
|
||||
$sql = $this->db->sql->create_table('create_join',
|
||||
array(
|
||||
'id' => 'INTEGER',
|
||||
'id' => 'int(10)',
|
||||
'key' => 'TEXT',
|
||||
'val' => 'TEXT',
|
||||
),
|
||||
@ -110,7 +112,9 @@ class MySQLTest extends UnitTestCase {
|
||||
|
||||
//Check
|
||||
$dbs = $this->db->get_tables();
|
||||
//$this->assertEqual($dbs['create_test'], 'CREATE TABLE "create_test" (id INTEGER PRIMARY KEY, key TEXT , val TEXT )');
|
||||
|
||||
$this->assertTrue(in_array('create_test', $dbs));
|
||||
|
||||
}
|
||||
|
||||
/*function TestTruncate()
|
||||
|
Binary file not shown.
Binary file not shown.
Loading…
Reference in New Issue
Block a user