123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343 |
- // Copyright (c) Athena Dev Teams - Licensed under GNU GPL
- // For more information, see LICENCE in the main folder
- #ifndef _COMMON_SQL_H_
- #define _COMMON_SQL_H_
- #ifndef _CBASETYPES_H_
- #include "../common/cbasetypes.h"
- #endif
- #include <stdarg.h>// va_list
- // Return codes
- #define SQL_ERROR -1
- #define SQL_SUCCESS 0
- #define SQL_NO_DATA 100
- /// Data type identifier.
- /// String, enum and blob data types need the buffer length specified.
- enum SqlDataType
- {
- SQLDT_NULL,
- // fixed size
- SQLDT_INT8,
- SQLDT_INT16,
- SQLDT_INT32,
- SQLDT_INT64,
- SQLDT_UINT8,
- SQLDT_UINT16,
- SQLDT_UINT32,
- SQLDT_UINT64,
- // platform dependent size
- SQLDT_CHAR,
- SQLDT_SHORT,
- SQLDT_INT,
- SQLDT_LONG,
- SQLDT_LONGLONG,
- SQLDT_UCHAR,
- SQLDT_USHORT,
- SQLDT_UINT,
- SQLDT_ULONG,
- SQLDT_ULONGLONG,
- // floating point
- SQLDT_FLOAT,
- SQLDT_DOUBLE,
- // other
- SQLDT_STRING,
- SQLDT_ENUM,
- // Note: An ENUM is a string with restricted values. When an invalid value
- // is inserted, it is saved as an empty string (numerical value 0).
- SQLDT_BLOB,
- SQLDT_LASTID
- };
- struct Sql;// Sql handle (private access)
- struct SqlStmt;// Sql statement (private access)
- typedef enum SqlDataType SqlDataType;
- typedef struct Sql Sql;
- typedef struct SqlStmt SqlStmt;
- /// Allocates and initializes a new Sql handle.
- struct Sql* Sql_Malloc(void);
- /// Establishes a connection.
- ///
- /// @return SQL_SUCCESS or SQL_ERROR
- int Sql_Connect(Sql* self, const char* user, const char* passwd, const char* host, uint16 port, const char* db);
- /// Retrieves the timeout of the connection.
- ///
- /// @return SQL_SUCCESS or SQL_ERROR
- int Sql_GetTimeout(Sql* self, uint32* out_timeout);
- /// Retrieves the name of the columns of a table into out_buf, with the separator after each name.
- ///
- /// @return SQL_SUCCESS or SQL_ERROR
- int Sql_GetColumnNames(Sql* self, const char* table, char* out_buf, size_t buf_len, char sep);
- /// Changes the encoding of the connection.
- ///
- /// @return SQL_SUCCESS or SQL_ERROR
- int Sql_SetEncoding(Sql* self, const char* encoding);
- /// Pings the connection.
- ///
- /// @return SQL_SUCCESS or SQL_ERROR
- int Sql_Ping(Sql* self);
- /// Escapes a string.
- /// The output buffer must be at least strlen(from)*2+1 in size.
- ///
- /// @return The size of the escaped string
- size_t Sql_EscapeString(Sql* self, char* out_to, const char* from);
- /// Escapes a string.
- /// The output buffer must be at least from_len*2+1 in size.
- ///
- /// @return The size of the escaped string
- size_t Sql_EscapeStringLen(Sql* self, char* out_to, const char* from, size_t from_len);
- /// Executes a query.
- /// Any previous result is freed.
- /// The query is constructed as if it was sprintf.
- ///
- /// @return SQL_SUCCESS or SQL_ERROR
- int Sql_Query(Sql* self, const char* query, ...);
- /// Executes a query.
- /// Any previous result is freed.
- /// The query is constructed as if it was svprintf.
- ///
- /// @return SQL_SUCCESS or SQL_ERROR
- int Sql_QueryV(Sql* self, const char* query, va_list args);
- /// Executes a query.
- /// Any previous result is freed.
- /// The query is used directly.
- ///
- /// @return SQL_SUCCESS or SQL_ERROR
- int Sql_QueryStr(Sql* self, const char* query);
- /// Returns the number of the AUTO_INCREMENT column of the last INSERT/UPDATE query.
- ///
- /// @return Value of the auto-increment column
- uint64 Sql_LastInsertId(Sql* self);
- /// Returns the number of columns in each row of the result.
- ///
- /// @return Number of columns
- uint32 Sql_NumColumns(Sql* self);
- /// Returns the number of rows in the result.
- ///
- /// @return Number of rows
- uint64 Sql_NumRows(Sql* self);
- /// Fetches the next row.
- /// The data of the previous row is no longer valid.
- ///
- /// @return SQL_SUCCESS, SQL_ERROR or SQL_NO_DATA
- int Sql_NextRow(Sql* self);
- /// Gets the data of a column.
- /// The data remains valid until the next row is fetched or the result is freed.
- ///
- /// @return SQL_SUCCESS or SQL_ERROR
- int Sql_GetData(Sql* self, size_t col, char** out_buf, size_t* out_len);
- /// Frees the result of the query.
- void Sql_FreeResult(Sql* self);
- #if defined(SQL_REMOVE_SHOWDEBUG)
- #define Sql_ShowDebug(self) (void)0
- #else
- #define Sql_ShowDebug(self) Sql_ShowDebug_(self, __FILE__, __LINE__)
- #endif
- /// Shows debug information (last query).
- void Sql_ShowDebug_(Sql* self, const char* debug_file, const unsigned long debug_line);
- /// Frees a Sql handle returned by Sql_Malloc.
- void Sql_Free(Sql* self);
- ///////////////////////////////////////////////////////////////////////////////
- // Prepared Statements
- ///////////////////////////////////////////////////////////////////////////////
- // Parameters are placed in the statement by embedding question mark ('?')
- // characters into the query at the appropriate positions.
- // The markers are legal only in places where they represent data.
- // The markers cannot be inside quotes. Quotes will be added automatically
- // when they are required.
- //
- // example queries with parameters:
- // 1) SELECT col FROM table WHERE id=?
- // 2) INSERT INTO table(col1,col2) VALUES(?,?)
- /// Allocates and initializes a new SqlStmt handle.
- /// It uses the connection of the parent Sql handle.
- /// Queries in Sql and SqlStmt are independent and don't affect each other.
- ///
- /// @return SqlStmt handle or NULL if an error occured
- struct SqlStmt* SqlStmt_Malloc(Sql* sql);
- /// Prepares the statement.
- /// Any previous result is freed and all parameter bindings are removed.
- /// The query is constructed as if it was sprintf.
- ///
- /// @return SQL_SUCCESS or SQL_ERROR
- int SqlStmt_Prepare(SqlStmt* self, const char* query, ...);
- /// Prepares the statement.
- /// Any previous result is freed and all parameter bindings are removed.
- /// The query is constructed as if it was svprintf.
- ///
- /// @return SQL_SUCCESS or SQL_ERROR
- int SqlStmt_PrepareV(SqlStmt* self, const char* query, va_list args);
- /// Prepares the statement.
- /// Any previous result is freed and all parameter bindings are removed.
- /// The query is used directly.
- ///
- /// @return SQL_SUCCESS or SQL_ERROR
- int SqlStmt_PrepareStr(SqlStmt* self, const char* query);
- /// Returns the number of parameters in the prepared statement.
- ///
- /// @return Number or paramenters
- size_t SqlStmt_NumParams(SqlStmt* self);
- /// Binds a parameter to a buffer.
- /// The buffer data will be used when the statement is executed.
- /// All parameters should have bindings.
- ///
- /// @return SQL_SUCCESS or SQL_ERROR
- int SqlStmt_BindParam(SqlStmt* self, size_t idx, SqlDataType buffer_type, void* buffer, size_t buffer_len);
- /// Executes the prepared statement.
- /// Any previous result is freed and all column bindings are removed.
- ///
- /// @return SQL_SUCCESS or SQL_ERROR
- int SqlStmt_Execute(SqlStmt* self);
- /// Returns the number of the AUTO_INCREMENT column of the last INSERT/UPDATE statement.
- ///
- /// @return Value of the auto-increment column
- uint64 SqlStmt_LastInsertId(SqlStmt* self);
- /// Returns the number of columns in each row of the result.
- ///
- /// @return Number of columns
- size_t SqlStmt_NumColumns(SqlStmt* self);
- /// Binds the result of a column to a buffer.
- /// The buffer will be filled with data when the next row is fetched.
- /// For string/enum buffer types there has to be enough space for the data
- /// and the nul-terminator (an extra byte).
- ///
- /// @return SQL_SUCCESS or SQL_ERROR
- int SqlStmt_BindColumn(SqlStmt* self, size_t idx, SqlDataType buffer_type, void* buffer, size_t buffer_len, uint32* out_length, int8* out_is_null);
- /// Returns the number of rows in the result.
- ///
- /// @return Number of rows
- uint64 SqlStmt_NumRows(SqlStmt* self);
- /// Fetches the next row.
- /// All column bindings will be filled with data.
- ///
- /// @return SQL_SUCCESS, SQL_ERROR or SQL_NO_DATA
- int SqlStmt_NextRow(SqlStmt* self);
- /// Frees the result of the statement execution.
- void SqlStmt_FreeResult(SqlStmt* self);
- #if defined(SQL_REMOVE_SHOWDEBUG)
- #define SqlStmt_ShowDebug(self) (void)0
- #else
- #define SqlStmt_ShowDebug(self) SqlStmt_ShowDebug_(self, __FILE__, __LINE__)
- #endif
- /// Shows debug information (with statement).
- void SqlStmt_ShowDebug_(SqlStmt* self, const char* debug_file, const unsigned long debug_line);
- /// Frees a SqlStmt returned by SqlStmt_Malloc.
- void SqlStmt_Free(SqlStmt* self);
- #endif /* _COMMON_SQL_H_ */
|