1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120
|
#include "sqlitecreatevirtualtable.h"
#include "sqlitequerytype.h"
#include <parser/lexer.h>
#include <parser/statementtokenbuilder.h>
SqliteCreateVirtualTable::SqliteCreateVirtualTable()
{
queryType = SqliteQueryType::CreateVirtualTable;
}
SqliteCreateVirtualTable::SqliteCreateVirtualTable(const SqliteCreateVirtualTable& other) :
SqliteQuery(other), ifNotExistsKw(other.ifNotExistsKw), database(other.database), table(other.table), module(other.module), args(other.args)
{
}
SqliteCreateVirtualTable::SqliteCreateVirtualTable(bool ifNotExists, const QString &name1, const QString &name2, const QString &name3) :
SqliteCreateVirtualTable()
{
initName(name1, name2);
this->ifNotExistsKw = ifNotExists;
module = name3;
}
SqliteCreateVirtualTable::SqliteCreateVirtualTable(bool ifNotExists, const QString &name1, const QString &name2, const QString &name3, const QList<QString> &args) :
SqliteCreateVirtualTable()
{
initName(name1, name2);
this->ifNotExistsKw = ifNotExists;
module = name3;
this->args = args;
}
SqliteStatement*SqliteCreateVirtualTable::clone()
{
return new SqliteCreateVirtualTable(*this);
}
QStringList SqliteCreateVirtualTable::getTablesInStatement()
{
return getStrListFromValue(table);
}
QStringList SqliteCreateVirtualTable::getDatabasesInStatement()
{
return getStrListFromValue(database);
}
TokenList SqliteCreateVirtualTable::getTableTokensInStatement()
{
return getObjectTokenListFromFullname();
}
TokenList SqliteCreateVirtualTable::getDatabaseTokensInStatement()
{
return getDbTokenListFromFullname();
}
QList<SqliteStatement::FullObject> SqliteCreateVirtualTable::getFullObjectsInStatement()
{
QList<FullObject> result;
// Table object
FullObject fullObj = getFullObjectFromFullname(FullObject::TABLE);
if (fullObj.isValid())
result << fullObj;
// Db object
fullObj = getFirstDbFullObject();
if (fullObj.isValid())
{
result << fullObj;
dbTokenForFullObjects = fullObj.database;
}
return result;
}
void SqliteCreateVirtualTable::initName(const QString &name1, const QString &name2)
{
if (!name2.isNull())
{
database = name1;
table = name2;
}
else
table = name1;
}
TokenList SqliteCreateVirtualTable::rebuildTokensFromContents()
{
StatementTokenBuilder builder;
builder.withTokens(SqliteQuery::rebuildTokensFromContents());
builder.withKeyword("CREATE").withSpace().withKeyword("VIRTUAL").withSpace().withKeyword("TABLE");
if (ifNotExistsKw)
builder.withKeyword("IF").withSpace().withKeyword("NOT").withSpace().withKeyword("EXISTS").withSpace();
if (!database.isNull())
builder.withOther(database).withOperator(".");
builder.withKeyword("USING").withSpace().withOther(module);
if (!args.isEmpty())
{
builder.withSpace();
int i = 0;
for (const QString& arg : args)
{
if (i > 0)
builder.withOperator(",").withSpace();
builder.withTokens(Lexer::tokenize(arg));
i++;
}
}
builder.withOperator(";");
return builder.build();
}
|