4#include "../SqlAdvisoryLock.hpp"
5#include "../SqlQueryFormatter.hpp"
7#include <reflection-cpp/reflection.hpp>
14class SQLiteQueryFormatter:
public SqlQueryFormatter
20 [[nodiscard]]
static std::string FormatFromTable(std::string_view table)
22 if (!table.empty() && (table.front() ==
'"' || table.front() ==
'['))
23 return std::string(table);
24 return std::format(R
"("{}")", table);
33 [[nodiscard]]
bool RequiresTableRebuildForForeignKeyChange() const noexcept
override
48 [[nodiscard]]
static std::string BuildColumnExistsQuery(std::string_view tableName, std::string_view columnName)
50 return std::format(R
"(SELECT COUNT(*) FROM pragma_table_info('{}') WHERE name = '{}';)", tableName, columnName);
53 [[nodiscard]] std::string Insert(std::string_view intoTable,
54 std::string_view fields,
55 std::string_view values) const override
57 return std::format(R
"(INSERT INTO "{}" ({}) VALUES ({}))", intoTable, fields, values);
60 [[nodiscard]] std::string Insert(std::string_view ,
61 std::string_view intoTable,
62 std::string_view fields,
63 std::string_view values)
const override
66 return std::format(R
"(INSERT INTO "{}" ({}) VALUES ({}))", intoTable, fields, values);
69 [[nodiscard]] std::string QueryLastInsertId(std::string_view )
const override
72 return "SELECT LAST_INSERT_ROWID()";
75 [[nodiscard]] std::string_view BooleanLiteral(
bool literalValue)
const noexcept override
77 return literalValue ?
"TRUE" :
"FALSE";
80 [[nodiscard]] std::string_view DateFunction() const noexcept
override
85 [[nodiscard]] std::string StringLiteral(std::string_view value)
const noexcept override
91 escaped.reserve(value.size() + 2);
93 for (
char const c: value)
104 [[nodiscard]] std::string StringLiteral(
char value)
const noexcept override
108 return std::format(
"'{}'", value);
111 [[nodiscard]] std::string BinaryLiteral(std::span<uint8_t const> data)
const override
114 result.reserve((data.size() * 2) + 3);
116 for (uint8_t
byte: data)
117 result += std::format(
"{:02X}", byte);
122 [[nodiscard]] std::string QualifiedTableName(std::string_view schema, std::string_view table)
const override
126 return std::format(R
"("{}")", table);
128 return std::format(R
"("{}"."{}")", schema, table);
131 [[nodiscard]] std::string SelectCount(bool distinct,
132 std::string_view fromTable,
133 std::string_view fromTableAlias,
134 std::string_view tableJoins,
135 std::string_view whereCondition)
const override
137 auto const formattedTable = FormatFromTable(fromTable);
138 if (fromTableAlias.empty())
140 "SELECT{} COUNT(*) FROM {}{}{}", distinct ?
" DISTINCT" :
"", formattedTable, tableJoins, whereCondition);
142 return std::format(R
"(SELECT{} COUNT(*) FROM {} AS "{}"{}{})",
143 distinct ? " DISTINCT" :
"",
150 [[nodiscard]] std::string SelectAll(
bool distinct,
152 std::string_view fields,
153 std::string_view fromTable,
154 std::string_view fromTableAlias,
155 std::string_view tableJoins,
156 std::string_view whereCondition,
157 std::string_view orderBy,
158 std::string_view groupBy)
const override
160 std::stringstream sqlQueryString;
161 sqlQueryString <<
"SELECT ";
163 sqlQueryString <<
"DISTINCT ";
164 sqlQueryString << fields;
165 sqlQueryString <<
" FROM " << FormatFromTable(fromTable);
166 if (!fromTableAlias.empty())
167 sqlQueryString <<
" AS \"" << fromTableAlias <<
'"';
168 sqlQueryString << tableJoins;
169 sqlQueryString << whereCondition;
170 sqlQueryString << groupBy;
171 sqlQueryString << orderBy;
173 return sqlQueryString.str();
176 [[nodiscard]] std::string SelectFirst(
bool distinct,
178 std::string_view fields,
179 std::string_view fromTable,
180 std::string_view fromTableAlias,
181 std::string_view tableJoins,
182 std::string_view whereCondition,
183 std::string_view orderBy,
184 size_t count)
const override
186 std::stringstream sqlQueryString;
187 sqlQueryString <<
"SELECT " << fields;
189 sqlQueryString <<
" DISTINCT";
190 sqlQueryString <<
" FROM " << FormatFromTable(fromTable);
191 if (!fromTableAlias.empty())
192 sqlQueryString <<
" AS \"" << fromTableAlias <<
"\"";
193 sqlQueryString << tableJoins;
194 sqlQueryString << whereCondition;
195 sqlQueryString << orderBy;
196 sqlQueryString <<
" LIMIT " << count;
197 return sqlQueryString.str();
200 [[nodiscard]] std::string SelectRange(
bool distinct,
202 std::string_view fields,
203 std::string_view fromTable,
204 std::string_view fromTableAlias,
205 std::string_view tableJoins,
206 std::string_view whereCondition,
207 std::string_view orderBy,
208 std::string_view groupBy,
210 std::size_t limit)
const override
212 std::stringstream sqlQueryString;
213 sqlQueryString <<
"SELECT " << fields;
215 sqlQueryString <<
" DISTINCT";
216 sqlQueryString <<
" FROM " << FormatFromTable(fromTable);
217 if (!fromTableAlias.empty())
218 sqlQueryString <<
" AS \"" << fromTableAlias <<
"\"";
219 sqlQueryString << tableJoins;
220 sqlQueryString << whereCondition;
221 sqlQueryString << groupBy;
222 sqlQueryString << orderBy;
223 sqlQueryString <<
" LIMIT " << limit <<
" OFFSET " << offset;
224 return sqlQueryString.str();
227 [[nodiscard]] std::string Update(std::string_view table,
228 std::string_view tableAlias,
229 std::string_view setFields,
230 std::string_view whereCondition)
const override
232 auto const formattedTable = FormatFromTable(table);
233 if (tableAlias.empty())
234 return std::format(
"UPDATE {} SET {}{}", formattedTable, setFields, whereCondition);
236 return std::format(R
"(UPDATE {} AS "{}" SET {}{})", formattedTable, tableAlias, setFields, whereCondition);
239 [[nodiscard]] std::string Delete(std::string_view fromTable,
240 std::string_view fromTableAlias,
241 std::string_view tableJoins,
242 std::string_view whereCondition) const override
244 auto const formattedTable = FormatFromTable(fromTable);
245 if (fromTableAlias.empty())
246 return std::format(
"DELETE FROM {}{}{}", formattedTable, tableJoins, whereCondition);
248 return std::format(R
"(DELETE FROM {} AS "{}"{}{})", formattedTable, fromTableAlias, tableJoins, whereCondition);
251 [[nodiscard]] virtual std::string BuildColumnDefinition(SqlColumnDeclaration
const& column)
const
253 std::stringstream sqlQueryString;
255 sqlQueryString <<
'"' << column.name <<
"\" ";
257 if (column.primaryKey != SqlPrimaryKeyType::AUTO_INCREMENT)
258 sqlQueryString << ColumnType(column.type);
260 sqlQueryString << ColumnType(SqlColumnTypeDefinitions::Integer {});
263 sqlQueryString <<
" NOT NULL";
265 if (column.primaryKey == SqlPrimaryKeyType::AUTO_INCREMENT)
266 sqlQueryString <<
" PRIMARY KEY AUTOINCREMENT";
267 else if (column.primaryKey == SqlPrimaryKeyType::NONE && !column.index && column.unique)
268 sqlQueryString <<
" UNIQUE";
270 if (!column.defaultValue.empty())
271 sqlQueryString <<
" DEFAULT " << column.defaultValue;
273 return sqlQueryString.str();
276 [[nodiscard]]
static std::string BuildForeignKeyConstraint(std::string_view tableName,
277 std::string_view columnName,
278 SqlForeignKeyReferenceDefinition
const& referencedColumn)
285 return std::format(R
"(CONSTRAINT "{}" FOREIGN KEY ("{}") REFERENCES "{}"("{}"))",
288 referencedColumn.tableName,
289 referencedColumn.columnName);
293 [[nodiscard]]
StringList CreateTable(std::string_view schema,
294 std::string_view tableName,
295 std::vector<SqlColumnDeclaration>
const& columns,
296 std::vector<SqlCompositeForeignKeyConstraint>
const& foreignKeys,
297 bool ifNotExists =
false)
const override
302 sqlQueries.emplace_back([&]() {
303 std::stringstream sqlQueryString;
304 sqlQueryString <<
"CREATE TABLE ";
306 sqlQueryString <<
"IF NOT EXISTS ";
309 sqlQueryString <<
"\"" << tableName <<
"\" (";
310 std::vector<SqlColumnDeclaration const*> pks;
311 size_t currentColumn = 0;
312 std::string foreignKeyConstraints;
313 for (SqlColumnDeclaration
const& column: columns)
315 if (currentColumn > 0)
316 sqlQueryString <<
",";
318 sqlQueryString <<
"\n ";
319 sqlQueryString << BuildColumnDefinition(column);
321 if (column.primaryKey != SqlPrimaryKeyType::NONE)
322 pks.push_back(&column);
324 if (column.foreignKey)
326 foreignKeyConstraints +=
",\n ";
327 foreignKeyConstraints += BuildForeignKeyConstraint(tableName, column.name, *column.foreignKey);
331 for (SqlCompositeForeignKeyConstraint
const& fk: foreignKeys)
337 foreignKeyConstraints +=
",\n CONSTRAINT \"";
339 foreignKeyConstraints +=
"\" FOREIGN KEY (";
340 for (
size_t i = 0; i < fk.columns.size(); ++i)
343 foreignKeyConstraints +=
", ";
344 foreignKeyConstraints +=
'"' + fk.columns[i] +
'"';
346 foreignKeyConstraints +=
") REFERENCES \"";
347 foreignKeyConstraints += fk.referencedTableName;
348 foreignKeyConstraints +=
"\" (";
349 for (
size_t i = 0; i < fk.referencedColumns.size(); ++i)
352 foreignKeyConstraints +=
", ";
353 foreignKeyConstraints +=
'"' + fk.referencedColumns[i] +
'"';
355 foreignKeyConstraints +=
")";
365 pks, [](SqlColumnDeclaration
const* col) {
return col->primaryKey == SqlPrimaryKeyType::AUTO_INCREMENT; });
369 std::ranges::sort(pks, [](SqlColumnDeclaration
const* a, SqlColumnDeclaration
const* b) {
374 if (a->primaryKeyIndex != 0 && b->primaryKeyIndex != 0)
375 return a->primaryKeyIndex < b->primaryKeyIndex;
376 if (a->primaryKeyIndex != 0)
378 if (b->primaryKeyIndex != 0)
383 sqlQueryString <<
",\n PRIMARY KEY (";
384 for (
size_t i = 0; i < pks.size(); ++i)
387 sqlQueryString <<
", ";
388 sqlQueryString <<
'"' << pks[i]->name <<
'"';
390 sqlQueryString <<
")";
393 sqlQueryString << foreignKeyConstraints;
395 sqlQueryString <<
"\n);";
396 return sqlQueryString.str();
399 for (SqlColumnDeclaration
const& column: columns)
401 if (column.index && column.primaryKey == SqlPrimaryKeyType::NONE)
405 sqlQueries.emplace_back(std::format(R
"(CREATE UNIQUE INDEX "{}_{}_index" ON "{}" ("{}");)",
411 sqlQueries.emplace_back(std::format(
412 R
"(CREATE INDEX "{}_{}_index" ON "{}" ("{}");)", tableName, column.name, tableName, column.name));
420 [[nodiscard]] std::string FormatAlterTableCommand(std::string_view tableName,
SqlAlterTableCommand const& command)
const
422 auto const formatTable = [tableName]() {
423 return std::format(R
"("{}")", tableName);
426 using namespace SqlAlterTableCommands;
429 [&formatTable](RenameTable
const& actualCommand) -> std::string {
430 return std::format(R
"(ALTER TABLE {} RENAME TO "{}";)", formatTable(), actualCommand.newTableName);
432 [&formatTable, this](AddColumn
const& actualCommand) -> std::string {
433 return std::format(R
"(ALTER TABLE {} ADD COLUMN "{}" {} {};)",
435 actualCommand.columnName,
436 ColumnType(actualCommand.columnType),
437 actualCommand.nullable == SqlNullable::NotNull ? "NOT NULL" :
"NULL");
439 [&formatTable,
this](AlterColumn
const& actualCommand) -> std::string {
440 return std::format(R
"(ALTER TABLE {} ALTER COLUMN "{}" {} {};)",
442 actualCommand.columnName,
443 ColumnType(actualCommand.columnType),
444 actualCommand.nullable == SqlNullable::NotNull ? "NOT NULL" :
"NULL");
446 [&formatTable](RenameColumn
const& actualCommand) -> std::string {
447 return std::format(R
"(ALTER TABLE {} RENAME COLUMN "{}" TO "{}";)",
449 actualCommand.oldColumnName,
450 actualCommand.newColumnName);
452 [&formatTable](DropColumn const& actualCommand) -> std::string {
453 return std::format(R
"(ALTER TABLE {} DROP COLUMN "{}";)", formatTable(), actualCommand.columnName);
455 [tableName](AddIndex const& actualCommand) -> std::string {
456 using namespace std::string_view_literals;
457 auto const uniqueStr = actualCommand.unique ?
"UNIQUE "sv :
""sv;
458 return std::format(R
"(CREATE {2}INDEX "{0}_{1}_index" ON "{0}" ("{1}");)",
460 actualCommand.columnName,
463 [tableName](DropIndex const& actualCommand) -> std::string {
464 return std::format(R
"(DROP INDEX "{0}_{1}_index";)", tableName, actualCommand.columnName);
466 [tableName](AddForeignKey const& actualCommand) -> std::string {
473 R
"(-- LIGHTWEIGHT_SQLITE_GUARD: ADD_FOREIGN_KEY "{0}" "{1}" "{2}" "{3}"
474-- ALTER TABLE "{0}" ADD {4};)",
476 actualCommand.columnName,
477 actualCommand.referencedColumn.tableName,
478 actualCommand.referencedColumn.columnName,
479 BuildForeignKeyConstraint(tableName, actualCommand.columnName, actualCommand.referencedColumn));
481 [tableName](DropForeignKey const& actualCommand) -> std::string {
485 R
"(-- LIGHTWEIGHT_SQLITE_GUARD: DROP_FOREIGN_KEY "{0}" "{1}"
486-- ALTER TABLE "{0}" DROP CONSTRAINT "{2}";)",
488 actualCommand.columnName,
490 std::array { std::string_view { actualCommand.columnName } }));
492 [tableName](AddCompositeForeignKey const& actualCommand) -> std::string {
499 auto const joinComma = [](std::vector<std::string>
const& v) {
501 for (
size_t i = 0; i < v.size(); ++i)
509 auto const joinQuoted = [](std::vector<std::string>
const& v) {
511 for (
size_t i = 0; i < v.size(); ++i)
523 R
"(-- LIGHTWEIGHT_SQLITE_GUARD: ADD_COMPOSITE_FOREIGN_KEY "{0}" "{1}" "{2}" "{3}"
524-- ALTER TABLE "{0}" ADD CONSTRAINT "{4}" FOREIGN KEY ({5}) REFERENCES "{2}"({6});)",
526 joinComma(actualCommand.columns),
527 actualCommand.referencedTableName,
528 joinComma(actualCommand.referencedColumns),
530 joinQuoted(actualCommand.columns),
531 joinQuoted(actualCommand.referencedColumns));
533 [&formatTable, tableName, this](AddColumnIfNotExists
const& actualCommand) -> std::string {
538 R
"(-- LIGHTWEIGHT_SQLITE_GUARD: ADD_COLUMN_IF_NOT_EXISTS "{0}" "{1}"
539ALTER TABLE {2} ADD COLUMN "{1}" {3} {4};)",
541 actualCommand.columnName,
543 ColumnType(actualCommand.columnType),
544 actualCommand.nullable == SqlNullable::NotNull ? "NOT NULL" :
"NULL");
546 [&formatTable, tableName](DropColumnIfExists
const& actualCommand) -> std::string {
549 R
"(-- LIGHTWEIGHT_SQLITE_GUARD: DROP_COLUMN_IF_EXISTS "{0}" "{1}"
550ALTER TABLE {2} DROP COLUMN "{1}";)",
552 actualCommand.columnName,
555 [tableName](DropIndexIfExists const& actualCommand) -> std::string {
556 return std::format(R
"(DROP INDEX IF EXISTS "{0}_{1}_index";)", tableName, actualCommand.columnName);
563 [[nodiscard]]
StringList AlterTable(std::string_view ,
564 std::string_view tableName,
565 std::vector<SqlAlterTableCommand>
const& commands)
const override
574 auto sql = FormatAlterTableCommand(tableName, command);
576 result.push_back(std::move(sql));
581 [[nodiscard]] std::string ColumnType(SqlColumnTypeDefinition
const& type)
const override
583 using namespace SqlColumnTypeDefinitions;
584 return std::visit(detail::overloaded {
585 [](Bigint
const&) -> std::string {
return "BIGINT"; },
586 [](Binary
const&) -> std::string {
return "BLOB"; },
587 [](Bool
const&) -> std::string {
return "BOOLEAN"; },
588 [](Char
const& type) -> std::string {
return std::format(
"CHAR({})", type.size); },
589 [](Date
const&) -> std::string {
return "DATE"; },
590 [](DateTime
const&) -> std::string {
return "DATETIME"; },
591 [](Decimal
const& type) -> std::string {
592 return std::format(
"DECIMAL({}, {})", type.precision, type.scale);
594 [](Guid
const&) -> std::string {
return "GUID"; },
595 [](Integer
const&) -> std::string {
return "INTEGER"; },
596 [](NChar
const& type) -> std::string {
return std::format(
"NCHAR({})", type.size); },
597 [](NVarchar
const& type) -> std::string {
return std::format(
"NVARCHAR({})", type.size); },
598 [](Real
const&) -> std::string {
return "REAL"; },
599 [](Smallint
const&) -> std::string {
return "SMALLINT"; },
600 [](Text
const&) -> std::string {
return "TEXT"; },
601 [](Time
const&) -> std::string {
return "TIME"; },
602 [](Timestamp
const&) -> std::string {
return "TIMESTAMP"; },
603 [](Tinyint
const&) -> std::string {
return "TINYINT"; },
604 [](VarBinary
const& type) -> std::string {
return std::format(
"VARBINARY({})", type.size); },
605 [](Varchar
const& type) -> std::string {
return std::format(
"VARCHAR({})", type.size); },
610 [[nodiscard]]
StringList DropTable(std::string_view ,
611 std::string_view
const& tableName,
612 bool ifExists =
false,
613 bool cascade =
false)
const override
620 return { std::format(R
"(DROP TABLE IF EXISTS "{}";)", tableName) };
622 return { std::format(R
"(DROP TABLE "{}";)", tableName) };
625 [[nodiscard]] std::string QueryServerVersion() const override
627 return "SELECT sqlite_version()";
638 [[nodiscard]] SqlAdvisoryLockHandler
const& AdvisoryLockOps()
const override
640 return SqliteAdvisoryLockOps();
std::variant< SqlAlterTableCommands::RenameTable, SqlAlterTableCommands::AddColumn, SqlAlterTableCommands::AddColumnIfNotExists, SqlAlterTableCommands::AlterColumn, SqlAlterTableCommands::AddIndex, SqlAlterTableCommands::RenameColumn, SqlAlterTableCommands::DropColumn, SqlAlterTableCommands::DropColumnIfExists, SqlAlterTableCommands::DropIndex, SqlAlterTableCommands::DropIndexIfExists, SqlAlterTableCommands::AddForeignKey, SqlAlterTableCommands::AddCompositeForeignKey, SqlAlterTableCommands::DropForeignKey > SqlAlterTableCommand
Represents a single SQL ALTER TABLE command.