1#ifndef ENGINE_SERVER_DATABASES_CONNECTION_H
2#define ENGINE_SERVER_DATABASES_CONNECTION_H
3
4#include "connection_pool.h"
5
6#include <engine/shared/protocol.h>
7
8#include <memory>
9#include <optional>
10
11enum
12{
13 // MAX_NAME_LENGTH includes the size with \0, which is not necessary in SQL
14 MAX_NAME_LENGTH_SQL = MAX_NAME_LENGTH - 1,
15};
16
17// can hold one PreparedStatement with Results
18class IDbConnection
19{
20public:
21 IDbConnection(const char *pPrefix);
22 virtual ~IDbConnection() = default;
23 IDbConnection &operator=(const IDbConnection &) = delete;
24 virtual void Print(const char *pMode) = 0;
25
26 // returns the database prefix
27 const char *GetPrefix() const { return m_aPrefix; }
28 virtual const char *BinaryCollate() const = 0;
29 // can be inserted into queries to convert a timestamp variable to the unix timestamp
30 virtual void ToUnixTimestamp(const char *pTimestamp, char *aBuf, unsigned int BufferSize) = 0;
31 // since MySQL automatically converts timestamps to utc, meanwhile sqlite code has to
32 // explicitly convert before inserting timestamps, NOTE: CURRENT_TIMESTAMP in SQLite is UTC by
33 // default and doesn't have to be converted
34 virtual const char *InsertTimestampAsUtc() const = 0;
35 // can be used in the context of `LIKE Map`, adds `? COLLATE`
36 virtual const char *CollateNocase() const = 0;
37 // syntax to insert a row into table or ignore if it already exists
38 virtual const char *InsertIgnore() const = 0;
39 // ORDER BY RANDOM()/RAND()
40 virtual const char *Random() const = 0;
41 // Get Median Map Time from l.Map
42 virtual const char *MedianMapTime(char *pBuffer, int BufferSize) const = 0;
43 virtual const char *False() const = 0;
44 virtual const char *True() const = 0;
45
46 // tries to allocate the connection from the pool established
47 //
48 // returns true on success
49 virtual bool Connect(char *pError, int ErrorSize) = 0;
50 // has to be called to return the connection back to the pool
51 virtual void Disconnect() = 0;
52
53 // ? for Placeholders, connection has to be established, can overwrite previous prepared statements
54 //
55 // returns true on success
56 virtual bool PrepareStatement(const char *pStmt, char *pError, int ErrorSize) = 0;
57
58 // PrepareStatement has to be called beforehand,
59 virtual void BindString(int Idx, const char *pString) = 0;
60 virtual void BindBlob(int Idx, unsigned char *pBlob, int Size) = 0;
61 virtual void BindInt(int Idx, int Value) = 0;
62 virtual void BindInt64(int Idx, int64_t Value) = 0;
63 virtual void BindFloat(int Idx, float Value) = 0;
64 virtual void BindNull(int Idx) = 0;
65
66 // Print expanded sql statement
67 virtual void Print() = 0;
68
69 // executes the query and returns if a result row exists and selects it
70 // when called multiple times the next row is selected
71 //
72 // returns true on success
73 virtual bool Step(bool *pEnd, char *pError, int ErrorSize) = 0;
74 // executes the query and returns the number of rows affected by the update/insert/delete
75 //
76 // returns true on success
77 virtual bool ExecuteUpdate(int *pNumUpdated, char *pError, int ErrorSize) = 0;
78
79 virtual bool IsNull(int Col) = 0;
80 virtual float GetFloat(int Col) = 0;
81 virtual int GetInt(int Col) = 0;
82 virtual int64_t GetInt64(int Col) = 0;
83 // ensures that the string is null terminated
84 virtual void GetString(int Col, char *pBuffer, int BufferSize) = 0;
85 // returns number of bytes read into the buffer
86 virtual int GetBlob(int Col, unsigned char *pBuffer, int BufferSize) = 0;
87 std::optional<float> GetOptionalFloat(int Col);
88 std::optional<int> GetOptionalInt(int Col);
89 std::optional<int64_t> GetOptionalInt64(int Col);
90
91 // SQL statements, that can't be abstracted, has side effects to the result
92 virtual bool AddPoints(const char *pPlayer, int Points, char *pError, int ErrorSize) = 0;
93
94private:
95 char m_aPrefix[64];
96
97protected:
98 void FormatCreateRace(char *aBuf, unsigned int BufferSize, bool Backup) const;
99 void FormatCreateTeamrace(char *aBuf, unsigned int BufferSize, const char *pIdType, bool Backup) const;
100 void FormatCreateMaps(char *aBuf, unsigned int BufferSize) const;
101 void FormatCreateSaves(char *aBuf, unsigned int BufferSize, bool Backup) const;
102 void FormatCreatePoints(char *aBuf, unsigned int BufferSize) const;
103};
104
105bool MysqlAvailable();
106int MysqlInit();
107void MysqlUninit();
108
109std::unique_ptr<IDbConnection> CreateSqliteConnection(const char *pFilename, bool Setup);
110// Returns nullptr if MySQL support is not compiled in.
111std::unique_ptr<IDbConnection> CreateMysqlConnection(CMysqlConfig Config);
112
113#endif // ENGINE_SERVER_DATABASES_CONNECTION_H
114