SQLite 中如果不存在则更改表添加列

Question

我们最近需要向一些现有的 SQLite 数据库表添加列。这可以通过 ALTER TABLE ADD COLUMN 来完成。当然，如果表已经被更改，我们不想理会它。不幸的是，SQLite 不支持 ALTER TABLE 上的 IF NOT EXISTS 子句。

我们当前的解决方法是执行 ALTER TABLE 语句并忽略任何“重复列名”错误，就像 此 Python 示例（但使用 C++）。

但是，我们设置数据库模式的常用方法是使用包含 CREATE TABLE IF NOT EXISTS 和 CREATE INDEX IF NOT EXISTS 语句的 .sql 脚本，可以使用以下命令执行sqlite3_exec 或 sqlite3 命令行工具。我们不能将 ALTER TABLE 放入这些脚本文件中，因为如果该语句失败，则该语句后面的任何内容都不会被执行。

我希望将表定义放在一处，而不是在 .sql 和 .cpp 文件之间拆分。有没有办法在纯 SQLite SQL 中编写 ALTER TABLE ADD COLUMN IF NOT EXISTS 的解决方法？

Answer 1

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

• 使用 'user_version' pragma 命令 (PRAGMA user_version）来存储数据库架构版本的增量数字。

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

这样，当软件启动时，它可以检查数据库架构，并在需要时运行 ALTER TABLE 查询，然后增加存储的版本。这比“盲目”尝试各种更新要好得多，特别是如果您的数据库多年来增长和更改了几次。

Answer 2

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

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

Answer 3

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

PRAGMA table_info(foo_table_name)

示例输出：

cid	name	type	notnull	dflt_value	pk
0	id	integer	0	null	1
1	type	text	0	null	0
2	data	json	0	null	0

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

Answer 4

一种解决方法是仅创建列并捕获该列已存在时出现的异常/错误。添加多列时，请将它们添加到单独的 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));
    }
}

Answer 5

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

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

关键部分是使用 pragma_table_info('') 而不是 pragma table_info('')。

---

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

Answer 6

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);

Answer 7

如果您在 flex/adobe air 中遇到此问题并首先发现自己在这里，我已经找到了解决方案，并将其发布在相关问题上：如果不存在，则将列添加到 sqlite db - flex/air sqlite？

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

Answer 8

您也可以将 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') 

Answer 9

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%'

Answer 10

我在 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;
}

Answer 11

这是我的解决方案，但是在 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 命令添加列。

Answer 12

如果您一次执行一行，所有这些答案都很好。然而，最初的问题是输入一个将由单个数据库执行执行的 sql 脚本，并且所有解决方案（例如提前检查列是否存在）都需要执行程序了解哪些表和正在更改/添加列或对输入脚本进行预处理和解析以确定此信息。通常，您不会实时或经常运行它。所以捕获异常然后继续前进的想法是可以接受的。问题就在这里……如何继续前进。幸运的是，错误消息为我们提供了执行此操作所需的所有信息。我们的想法是，如果alter table调用出现异常，则执行sql，我们可以在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 ) ) 
        )
-------------

Answer 13

如果“your_column”尚未在表中的列中，则使用简单的 python 代码将“your_column”添加到“your_table”。如果您在 for 循环中找到“your_column”，您也可以中断，对于我这个问题的特殊情况，我不能这样做。您可能还想通过在 col[1] 和“your_column”上使用 .upper() 或 .lower() 来强制比较区分大小写。如果您要添加多于一列，那么我建议更改 for 循环以从您要添加的列表中删除这些列。

cols = []
for col in DB.cursor.execute(f"PRAGMA table_info({your_table})").fetchall():
    cols.append(col[1])
if "your_column" not in cols:
      DB.cursor.execute(f"ALTER TABLE {your_table} ADD COLUMN {your_column} {your_datatype}")

Answer 14

我想出了这个查询，

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。
• 根据结果​​，更改列

Answer 15

我用两个查询解决了它。这是我使用 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; }
            }

Answer 16

显然...在 SQLite 中...如果列已经存在，“alter table”语句不会生成异常。

在支持论坛中找到这篇帖子并进行了测试。