summaryrefslogtreecommitdiff
path: root/Userland/Libraries/LibSQL/SQLResult.h
diff options
context:
space:
mode:
authorJan de Visser <jan@de-visser.net>2021-06-27 21:00:08 -0400
committerAli Mohammad Pur <Ali.mpfard@gmail.com>2021-07-08 17:55:59 +0430
commit30691549fdc0d6f703735fb0c30ab54cf0cecb65 (patch)
treeefb328f7a7b42e5d647cbe8085d94920c57d6932 /Userland/Libraries/LibSQL/SQLResult.h
parent633dc74606a556cdea4606f93fa50f01f46e2cd6 (diff)
downloadserenity-30691549fdc0d6f703735fb0c30ab54cf0cecb65.zip
LibSQL: Move Order and Nulls enums from SQL::AST to SQL namespace
The Order enum is used in the Meta component of LibSQL. Using this enum meant having to include the monster AST/AST.h include file. Furthermore, they are sort of basic and therefore can live in the general SQL namespace. Moved to LibSQL/Type.h. Also introduced a new class, SQLResult, which is needed in future patches.
Diffstat (limited to 'Userland/Libraries/LibSQL/SQLResult.h')
-rw-r--r--Userland/Libraries/LibSQL/SQLResult.h141
1 files changed, 141 insertions, 0 deletions
diff --git a/Userland/Libraries/LibSQL/SQLResult.h b/Userland/Libraries/LibSQL/SQLResult.h
new file mode 100644
index 0000000000..0c83b19b40
--- /dev/null
+++ b/Userland/Libraries/LibSQL/SQLResult.h
@@ -0,0 +1,141 @@
+/*
+ * Copyright (c) 2021, Jan de Visser <jan@de-visser.net>
+ *
+ * SPDX-License-Identifier: BSD-2-Clause
+ */
+
+#pragma once
+
+#include <AK/NonnullOwnPtrVector.h>
+#include <AK/Vector.h>
+#include <LibCore/Object.h>
+#include <LibSQL/Tuple.h>
+#include <LibSQL/Type.h>
+
+namespace SQL {
+
+#define ENUMERATE_SQL_COMMANDS(S) \
+ S(Create) \
+ S(Delete) \
+ S(Insert) \
+ S(Select) \
+ S(Update)
+
+enum class SQLCommand {
+#undef __ENUMERATE_SQL_COMMAND
+#define __ENUMERATE_SQL_COMMAND(command) command,
+ ENUMERATE_SQL_COMMANDS(__ENUMERATE_SQL_COMMAND)
+#undef __ENUMERATE_SQL_COMMAND
+};
+
+constexpr char const* command_tag(SQLCommand command)
+{
+ switch (command) {
+#undef __ENUMERATE_SQL_COMMAND
+#define __ENUMERATE_SQL_COMMAND(command) \
+ case SQLCommand::command: \
+ return #command;
+ ENUMERATE_SQL_COMMANDS(__ENUMERATE_SQL_COMMAND)
+#undef __ENUMERATE_SQL_COMMAND
+ }
+}
+
+#define ENUMERATE_SQL_ERRORS(S) \
+ S(NoError, "No error") \
+ S(DatabaseUnavailable, "Database Unavailable") \
+ S(StatementUnavailable, "Statement with id {} Unavailable") \
+ S(SyntaxError, "Syntax Error") \
+ S(DatabaseDoesNotExist, "Database {} does not exist") \
+ S(SchemaDoesNotExist, "Schema {} does not exist") \
+ S(SchemaExists, "Schema {} already exist") \
+ S(TableDoesNotExist, "Table {} does not exist") \
+ S(TableExists, "Table {} already exist") \
+ S(InvalidType, "Invalid type {}")
+
+enum class SQLErrorCode {
+#undef __ENUMERATE_SQL_ERROR
+#define __ENUMERATE_SQL_ERROR(error, description) error,
+ ENUMERATE_SQL_ERRORS(__ENUMERATE_SQL_ERROR)
+#undef __ENUMERATE_SQL_ERROR
+};
+
+struct SQLError {
+ SQLErrorCode code { SQLErrorCode::NoError };
+ String error_argument { "" };
+
+ String to_string() const
+ {
+ String code_string;
+ String message;
+ switch (code) {
+#undef __ENUMERATE_SQL_ERROR
+#define __ENUMERATE_SQL_ERROR(error, description) \
+ case SQLErrorCode::error: \
+ code_string = #error; \
+ message = description; \
+ break;
+ ENUMERATE_SQL_ERRORS(__ENUMERATE_SQL_ERROR)
+#undef __ENUMERATE_SQL_ERROR
+ default:
+ VERIFY_NOT_REACHED();
+ }
+ if (!error_argument.is_null() && !error_argument.is_empty()) {
+ if (message.find("{}").has_value()) {
+ message = String::formatted(message, error_argument);
+ } else {
+ message = String::formatted("{}: {}", message, error_argument);
+ }
+ }
+ if (message.is_null() || (message.is_empty())) {
+ return code_string;
+ } else {
+ return String::formatted("{}: {}", code_string, message);
+ }
+ }
+};
+
+class SQLResult : public Core::Object {
+ C_OBJECT(SQLResult)
+
+public:
+ void append(Tuple const& tuple)
+ {
+ m_has_results = true;
+ m_result_set.append(tuple);
+ }
+
+ SQLCommand command() const { return m_command; }
+ int updated() const { return m_update_count; }
+ int inserted() const { return m_insert_count; }
+ int deleted() const { return m_delete_count; }
+ SQLError const& error() const { return m_error; }
+ bool has_results() const { return m_has_results; }
+ Vector<Tuple> const& results() const { return m_result_set; }
+
+private:
+ SQLResult() = default;
+
+ explicit SQLResult(SQLCommand command, int update_count = 0, int insert_count = 0, int delete_count = 0)
+ : m_command(command)
+ , m_update_count(update_count)
+ , m_insert_count(insert_count)
+ , m_delete_count(delete_count)
+ {
+ }
+
+ SQLResult(SQLCommand command, SQLErrorCode error_code, String error_argument)
+ : m_command(command)
+ , m_error({ error_code, move(error_argument) })
+ {
+ }
+
+ SQLCommand m_command { SQLCommand::Select };
+ SQLError m_error { SQLErrorCode::NoError, "" };
+ int m_update_count { 0 };
+ int m_insert_count { 0 };
+ int m_delete_count { 0 };
+ bool m_has_results { false };
+ Vector<Tuple> m_result_set;
+};
+
+}