ALTER TABLE添加列,如果不是SQLite中的EXISTS

时间:2010-08-30 21:21:00

标签: sqlite alter-table

我们最近需要在现有的几个SQLite数据库表中添加列。这可以使用ALTER TABLE ADD COLUMN完成。当然,如果表已被更改,我们希望不管它。不幸的是,SQLite不支持IF NOT EXISTS上的ALTER TABLE子句。

我们当前的解决方法是执行ALTER TABLE语句并忽略任何“重复列名”错误,就像this Python example一样(但在C ++中)。

但是,我们设置数据库模式的常用方法是使用包含CREATE TABLE IF NOT EXISTSCREATE INDEX IF NOT EXISTS语句的.sql脚本,这些语句可以使用sqlite3_exec或{{1}执行命令行工具。我们不能将sqlite3放在这些脚本文件中,因为如果该语句失败,则不会执行任何后续操作。

我希望将表定义放在一个地方,而不是在.sql和.cpp文件之间拆分。有没有办法在纯SQLite SQL中为ALTER TABLE写一个变通方法?

15 个答案:

答案 0 :(得分:50)

我有一个99%的纯SQL方法。我们的想法是对架构进行版本控制。您可以通过两种方式执行此操作:

  • 使用'user_version'pragma命令(PRAGMA user_version)存储数据库架构版本的增量编号。

  • 将您的版本号存储在您自己定义的表格中。

通过这种方式,当软件启动时,它可以检查数据库模式,并在需要时运行ALTER TABLE查询,然后增加存储的版本。这比尝试各种更新“盲目”要好得多,特别是如果您的数据库在过去几年中增长和变化几次。

答案 1 :(得分:28)

一种解决方法是只创建列并捕获列已存在时出现的异常/错误。添加多个列时,请将它们添加到单独的ALTER TABLE语句中,以便一个副本不会阻止创建其他列。

使用sqlite-net,我们做了类似的事情。它并不完美,因为我们无法区分重复的sqlite错误和其他sqlite错误。

Dictionary<string, string> columnNameToAddColumnSql = new Dictionary<string, string>
{
    {
        "Column1",
        "ALTER TABLE MyTable ADD COLUMN Column1 INTEGER"
    },
    {
        "Column2",
        "ALTER TABLE MyTable ADD COLUMN Column2 TEXT"
    }
};

foreach (var pair in columnNameToAddColumnSql)
{
    string columnName = pair.Key;
    string sql = pair.Value;

    try
    {
        this.DB.ExecuteNonQuery(sql);
    }
    catch (System.Data.SQLite.SQLiteException e)
    {
        _log.Warn(e, string.Format("Failed to create column [{0}]. Most likely it already exists, which is fine.", columnName));
    }
}

答案 2 :(得分:25)

SQLite还支持名为“table_info”的pragma语句,该语句在表中每列返回一行,其中包含列的名称(以及有关该列的其他信息)。您可以在查询中使用它来检查缺少的列,如果不存在则更改表。

PRAGMA table_info(foo_table_name)

http://www.sqlite.org/pragma.html#pragma_table_info

答案 3 :(得分:15)

如果您在数据库升级语句中执行此操作,可能最简单的方法是在尝试添加可能已存在的字段时捕获引发的异常。

try {
   db.execSQL("ALTER TABLE " + TABLE_NAME + " ADD COLUMN foo TEXT default null");
} catch (SQLiteException ex) {
   Log.w(TAG, "Altering " + TABLE_NAME + ": " + ex.getMessage());
}

答案 4 :(得分:10)

threre是一个PRAGMA的方法是table_info(table_name),它返回表的所有信息。

以下是如何使用它来检查列是否存在,

    public boolean isColumnExists (String table, String column) {
         boolean isExists = false
         Cursor cursor;
         try {           
            cursor = db.rawQuery("PRAGMA table_info("+ table +")", null);
            if (cursor != null) {
                while (cursor.moveToNext()) {
                    String name = cursor.getString(cursor.getColumnIndex("name"));
                    if (column.equalsIgnoreCase(name)) {
                        isExists = true;
                        break;
                    }
                }
            }

         } finally {
            if (cursor != null && !cursor.isClose()) 
               cursor.close();
         }
         return isExists;
    }

您也可以在不使用循环的情况下使用此查询,

cursor = db.rawQuery("PRAGMA table_info("+ table +") where name = " + column, null);

答案 5 :(得分:3)

对于那些想将pragma table_info()的结果用作较大SQL的一部分的人。

select count(*) from
pragma_table_info('<table_name>')
where name='<column_name>';

关键部分是使用pragma_table_info('<table_name>')而不是pragma table_info('<table_name>')


此答案的灵感来自@Robert Hawkey的回复。我将其发布为新答案的原因是我没有足够的声誉将其发布为评论。

答案 6 :(得分:1)

我提出了这个查询

SELECT CASE (SELECT count(*) FROM pragma_table_info(''product'') c WHERE c.name = ''purchaseCopy'') WHEN 0 THEN ALTER TABLE product ADD purchaseCopy BLOB END
  • 如果列存在,内部查询将返回0或1。
  • 根据结果,更改列

答案 7 :(得分:0)

如果您在弹性/ adobe air中遇到此问题并首先找到自己,我找到了解决方案,并将其发布在相关问题上:ADD COLUMN to sqlite db IF NOT EXISTS - flex/air sqlite?

我的评论:https://stackoverflow.com/a/24928437/2678219

答案 8 :(得分:0)

我在C#/。Net中接受了上面的答案,并将其改写为Qt / C ++,并没有太大的改变,但是我想把它留在这里供将来寻找C ++“ ish”答案的任何人使用。

    bool MainWindow::isColumnExisting(QString &table, QString &columnName){

    QSqlQuery q;

    try {
        if(q.exec("PRAGMA table_info("+ table +")"))
            while (q.next()) {
                QString name = q.value("name").toString();     
                if (columnName.toLower() == name.toLower())
                    return true;
            }

    } catch(exception){
        return false;
    }
    return false;
}

答案 9 :(得分:0)

您也可以将CASE-WHEN TSQL语句与pragma_table_info结合使用,以了解是否存在列:

select case(CNT) 
    WHEN 0 then printf('not found')
    WHEN 1 then printf('found')
    END
FROM (SELECT COUNT(*) AS CNT FROM pragma_table_info('myTableName') WHERE name='columnToCheck') 

答案 10 :(得分:0)

这是我的解决方案,但是在python中(我尝试并未能找到与python相关的主题的任何帖子):

# modify table for legacy version which did not have leave type and leave time columns of rings3 table.
sql = 'PRAGMA table_info(rings3)' # get table info. returns an array of columns.
result = inquire (sql) # call homemade function to execute the inquiry
if len(result)<= 6: # if there are not enough columns add the leave type and leave time columns
    sql = 'ALTER table rings3 ADD COLUMN leave_type varchar'
    commit(sql) # call homemade function to execute sql
    sql = 'ALTER table rings3 ADD COLUMN leave_time varchar'
    commit(sql)

我使用PRAGMA来获取表格信息。它返回一个充满有关列信息的多维数组-每列一个数组。我计算数组的数量以获得列数。如果没有足够的列,那么我使用ALTER TABLE命令添加列。

答案 11 :(得分:0)

如果您一次执行一行,则所有这些答案都可以。但是,最初的问题是输入将由单个db execute执行的sql脚本,并且所有解决方案(例如提前检查列是否存在)都将要求执行程序知道哪些表和列正在被更改/添加或对输入脚本进行预处理和解析,以确定此信息。通常,您不会实时或经常运行此程序。因此,捕获异常的想法是可以接受的,然后继续进行下去。问题就在这里...如何继续前进。幸运的是,错误消息为我们提供了执行此操作所需的所有信息。这个想法是执行SQL,如果它在alter table调用上异常,我们可以在sql中找到alter table行并返回剩余的行并执行,直到成功或找不到匹配的alter table行为止。这是一些示例代码,其中我们在数组中包含sql脚本。我们迭代执行每个脚本的数组。我们两次调用它来使alter table命令失败,但是程序成功,因为我们从sql中删除了alter table命令并重新执行更新的代码。

#!/bin/sh
# the next line restarts using wish \

exec /opt/usr8.6.3/bin/tclsh8.6  "$0" ${1+"$@"}
foreach pkg {sqlite3 } {
    if { [ catch {package require {*}$pkg } err ] != 0 } {
    puts stderr "Unable to find package $pkg\n$err\n ... adjust your auto_path!";
    }
}
array set sqlArray {
    1 {
    CREATE TABLE IF NOT EXISTS Notes (
                      id INTEGER PRIMARY KEY AUTOINCREMENT,
                      name text,
                      note text,
                      createdDate integer(4) DEFAULT ( cast( strftime('%s', 'now') as int ) ) ,
                      updatedDate integer(4) DEFAULT ( cast( strftime('%s', 'now') as int ) ) 
                      );
    CREATE TABLE IF NOT EXISTS Version (
                        id INTEGER PRIMARY KEY AUTOINCREMENT,
                        version text,
                        createdDate integer(4) DEFAULT ( cast( strftime('%s', 'now') as int ) ) ,
                        updatedDate integer(4) DEFAULT ( cast( strftime('%s', 'now') as int ) )
                        );
    INSERT INTO Version(version) values('1.0');
    }
    2 {
    CREATE TABLE IF NOT EXISTS Tags (
        id INTEGER PRIMARY KEY AUTOINCREMENT,
        name text,
        tag text,
        createdDate integer(4) DEFAULT ( cast( strftime('%s', 'now') as int ) ) ,
        updatedDate integer(4) DEFAULT ( cast( strftime('%s', 'now') as int ) ) 
        );
    ALTER TABLE Notes ADD COLUMN dump text;
    INSERT INTO Version(version) values('2.0');
    }
    3 {
    ALTER TABLE Version ADD COLUMN sql text;
    INSERT INTO Version(version) values('3.0');
    }
}

# create db command , use in memory database for demonstration purposes
sqlite3 db :memory:

proc createSchema { sqlArray } {
    upvar $sqlArray sql
    # execute each sql script in order 
    foreach version [lsort -integer [array names sql ] ] {
    set cmd $sql($version)
    set ok 0
    while { !$ok && [string length $cmd ] } {  
        try {
        db eval $cmd
        set ok 1  ;   # it succeeded if we get here
        } on error { err backtrace } {
        if { [regexp {duplicate column name: ([a-zA-Z0-9])} [string trim $err ] match columnname ] } {
            puts "Error:  $err ... trying again" 
            set cmd [removeAlterTable $cmd $columnname ]
        } else {
            throw DBERROR "$err\n$backtrace"
        }
        }
    }
    }
}
# return sqltext with alter table command with column name removed
# if no matching alter table line found or result is no lines then
# returns ""
proc removeAlterTable { sqltext columnname } {
    set mode skip
    set result [list]
    foreach line [split $sqltext \n ] {
    if { [string first "alter table" [string tolower [string trim $line] ] ] >= 0 } {
        if { [string first $columnname $line ] } {
        set mode add
        continue;
        }
    }
    if { $mode eq "add" } {
        lappend result $line
    }
    }
    if { $mode eq "skip" } {
    puts stderr "Unable to find matching alter table line"
    return ""
    } elseif { [llength $result ] }  { 
    return [ join $result \n ]
    } else {
    return ""
    }
}
               
proc printSchema { } {
    db eval { select * from sqlite_master } x {
    puts "Table: $x(tbl_name)"
    puts "$x(sql)"
    puts "-------------"
    }
}
createSchema sqlArray
printSchema
# run again to see if we get alter table errors 
createSchema sqlArray
printSchema

预期产量

Table: Notes
CREATE TABLE Notes (
                      id INTEGER PRIMARY KEY AUTOINCREMENT,
                      name text,
                      note text,
                      createdDate integer(4) DEFAULT ( cast( strftime('%s', 'now') as int ) ) ,
                      updatedDate integer(4) DEFAULT ( cast( strftime('%s', 'now') as int ) ) 
                      , dump text)
-------------
Table: sqlite_sequence
CREATE TABLE sqlite_sequence(name,seq)
-------------
Table: Version
CREATE TABLE Version (
                        id INTEGER PRIMARY KEY AUTOINCREMENT,
                        version text,
                        createdDate integer(4) DEFAULT ( cast( strftime('%s', 'now') as int ) ) ,
                        updatedDate integer(4) DEFAULT ( cast( strftime('%s', 'now') as int ) )
                        , sql text)
-------------
Table: Tags
CREATE TABLE Tags (
        id INTEGER PRIMARY KEY AUTOINCREMENT,
        name text,
        tag text,
        createdDate integer(4) DEFAULT ( cast( strftime('%s', 'now') as int ) ) ,
        updatedDate integer(4) DEFAULT ( cast( strftime('%s', 'now') as int ) ) 
        )
-------------
Error:  duplicate column name: dump ... trying again
Error:  duplicate column name: sql ... trying again
Table: Notes
CREATE TABLE Notes (
                      id INTEGER PRIMARY KEY AUTOINCREMENT,
                      name text,
                      note text,
                      createdDate integer(4) DEFAULT ( cast( strftime('%s', 'now') as int ) ) ,
                      updatedDate integer(4) DEFAULT ( cast( strftime('%s', 'now') as int ) ) 
                      , dump text)
-------------
Table: sqlite_sequence
CREATE TABLE sqlite_sequence(name,seq)
-------------
Table: Version
CREATE TABLE Version (
                        id INTEGER PRIMARY KEY AUTOINCREMENT,
                        version text,
                        createdDate integer(4) DEFAULT ( cast( strftime('%s', 'now') as int ) ) ,
                        updatedDate integer(4) DEFAULT ( cast( strftime('%s', 'now') as int ) )
                        , sql text)
-------------
Table: Tags
CREATE TABLE Tags (
        id INTEGER PRIMARY KEY AUTOINCREMENT,
        name text,
        tag text,
        createdDate integer(4) DEFAULT ( cast( strftime('%s', 'now') as int ) ) ,
        updatedDate integer(4) DEFAULT ( cast( strftime('%s', 'now') as int ) ) 
        )
-------------

答案 12 :(得分:0)

select * from sqlite_master where type = 'table' and tbl_name = 'TableName' and sql like '%ColumnName%'

逻辑:sqlite_master中的sql列包含表定义,因此它肯定包含带有列名的字符串。

在搜索子字符串时,它有明显的局限性。因此,我建议在ColumnName中使用更具限制性的子字符串,例如这样的东西(需要测试为“`”字符并不总是存在):

select * from sqlite_master where type = 'table' and tbl_name = 'MyTable' and sql like '%`MyColumn` TEXT%'

答案 13 :(得分:0)

我用2个查询解决了这个问题。这是我使用System.Data.SQLite的Unity3D脚本。

IDbCommand command = dbConnection.CreateCommand();
            command.CommandText = @"SELECT count(*) FROM pragma_table_info('Candidat') c WHERE c.name = 'BirthPlace'";
            IDataReader reader = command.ExecuteReader();
            while (reader.Read())
            {
                try
                {
                    if (int.TryParse(reader[0].ToString(), out int result))
                    {
                        if (result == 0)
                        {
                            command = dbConnection.CreateCommand();
                            command.CommandText = @"ALTER TABLE Candidat ADD COLUMN BirthPlace VARCHAR";
                            command.ExecuteNonQuery();
                            command.Dispose();
                        }
                    }
                }
                catch { throw; }
            }

答案 14 :(得分:0)

显然……在 SQLite 中……如果列已经存在,“alter table”语句不会产生异常。

在支持论坛中找到了 this 个帖子并对其进行了测试。