eiendb_sqlite.hpp

浏览该文件的文档.

#ifndef __EIENDB_SQLITE_HPP__
#define __EIENDB_SQLITE_HPP__

#include "eiendb_base.hpp"

struct sqlite3;
struct sqlite3_stmt;

namespace eiendb
{

class EIENDB_DLL SqliteDbError: public DbError
{
    int _sqliteErrNo;
public:
    SqliteDbError( int sqliteErrNo, winux::AnsiString const & s ) throw() : DbError( DbError::dbeSqliteError, s ), _sqliteErrNo(sqliteErrNo) { }
    int getSqliteErrCode() const throw() { return _sqliteErrNo; }
};

class EIENDB_DLL SqliteConnection : public IDbConnection
{
public:
    SqliteConnection( winux::String const & path = "", winux::AnsiString const & dbKey = "", winux::String linkCharset = "", bool doConnect = true );
    virtual ~SqliteConnection();

    virtual bool connect() override;
    virtual bool close() override;
    virtual void alive() override;
    virtual bool setLinkCharset( winux::String const & charset ) override;
    virtual bool selectDb( winux::String const & database ) override;
    virtual bool createDb( winux::String const & database ) override;
    virtual bool dropDb( winux::String const & database ) override;
    virtual size_t affectedRows() override;
    virtual int errNo() override;
    virtual winux::String error() override;
    // 只是简单的引号double。内部不会进行utf8换码
    virtual winux::String escape( winux::String const & str, winux::String const & addQuote = "\'" ) override;
    // 将二进制数据转换成X'******'的形式
    virtual winux::String escape( void const * buf, size_t size, winux::String const & addQuote = "\'" ) override;
    // 执行SQL语句查询并返回结果集
    virtual winux::SharedPointer<IDbResult> query( winux::String const & sql ) override;
    virtual winux::SharedPointer<IDbResult> query( winux::SharedPointer<IDbStatement> stmt ) override;
    // 执行SQL语句,返回bool代表是否成功
    virtual bool exec( winux::String const & sql ) override;
    virtual bool exec( winux::SharedPointer<IDbStatement> stmt ) override;
    virtual size_t insertId() override;
    virtual winux::SharedPointer<IDbResult> listDbs() override;
    // 执行 PRAGMA table_info(tablename)
    virtual winux::SharedPointer<IDbResult> listFields( winux::String const & tableName ) override;
    // 执行 SELECT name FROM sqlite_master WHERE type = 'table' AND substr( name, 1, 7 ) <> 'sqlite_' ORDER BY name;
    virtual winux::SharedPointer<IDbResult> listTables() override;
    virtual winux::String tableDdl( winux::String const & tableName ) override;
    virtual winux::String symbolQuotes( winux::String const & str ) override;

    virtual size_t loadSql( winux::String const & sqlText, winux::StringArray * arrSql ) override;
    virtual size_t loadSqlFile( winux::IFile * sqlScriptFile, winux::StringArray * arrSql ) override;
    virtual size_t getPrimaryKey( winux::String const & tableName, winux::StringArray * arrKeyColumn ) override;

    virtual winux::SharedPointer<IDbStatement> buildStmt( winux::String const & sql ) override;
    virtual winux::SharedPointer<IDbStatement> buildStmt( winux::String const & sql, winux::Mixed const & params ) override;
    virtual winux::SharedPointer<IDbStatement> buildStmt( winux::String const & sql, winux::MixedArray const & params ) override;
    virtual winux::SharedPointer<IDbStatement> buildStmt( winux::String const & sql, winux::StringMixedMap const & params ) override;

    virtual operator bool() const override { return _db != NULL; }
    operator sqlite3 * () const { return _db; }
    // SQLite字符串转到winux::String，NULL也能正确处理
    winux::String sqliteStrToString( char const * s );
    // winux::String转到SQLite字符串
    winux::AnsiString stringToSqliteStr( winux::String const & s );
private:
    bool _open( winux::String const & database, winux::uint flags );

    sqlite3 * _db;
    winux::String _path; // 输入的路径参数
    winux::String _dirPath; // 解析到的目录路径
    winux::String _database; // 解析到的数据库名
    winux::String _linkCharset; // 连接校验字符集
    winux::AnsiString _dbKey; // 数据库密码,可以为二进制数据

    DISABLE_OBJECT_COPY(SqliteConnection)
};

class EIENDB_DLL SqliteStatement : public IDbStatement
{
public:
    SqliteStatement( SqliteConnection * cnn );
    SqliteStatement( SqliteConnection * cnn, winux::String const & sql );
    SqliteStatement( SqliteConnection * cnn, winux::String const & sql, winux::Mixed const & params );
    SqliteStatement( SqliteConnection * cnn, winux::String const & sql, winux::MixedArray const & params );
    SqliteStatement( SqliteConnection * cnn, winux::String const & sql, winux::StringMixedMap const & params );
    virtual ~SqliteStatement();
    //SqliteStatement( SqliteStatement & other );
    //SqliteStatement & operator = ( SqliteStatement & other );

    virtual bool build( winux::String const & sql ) override;
    virtual bool build( winux::String const & sql, winux::Mixed const & params ) override;
    virtual bool build( winux::String const & sql, winux::MixedArray const & params ) override;
    virtual bool build( winux::String const & sql, winux::StringMixedMap const & params ) override;
    virtual bool bind( size_t paramIndex, winux::Mixed const & val ) override;
    virtual bool bind( winux::String const & paramName, winux::Mixed const & val ) override;

    operator sqlite3_stmt * () const { return _stmt; }
    SqliteConnection * getCnn() const { return _cnn; }
    winux::String getSql() const { return _sql; }

private:
    void _free();

    SqliteConnection * _cnn;
    sqlite3_stmt * _stmt;
    winux::String _sql;
    std::vector< std::pair< size_t, winux::Mixed > > _bindingParams; // 已绑定的参数
    friend class SqliteResult;

    DISABLE_OBJECT_COPY(SqliteStatement)
};

class EIENDB_DLL SqliteResult : public IDbResult
{
public:
    SqliteResult( winux::SharedPointer<SqliteStatement> stmt );
    virtual ~SqliteResult();

    virtual bool fetchRow( winux::Mixed * fields, int type = 0 ) override;
    virtual bool fetchRow( winux::MixedArray * fields ) override;
    virtual bool fetchRow( winux::StringMixedMap * fields ) override;
    virtual winux::String fieldName( size_t fieldIndex ) override;
    virtual size_t fieldsCount() override;
    virtual size_t rowsCount() override;
    virtual bool free() override;
    virtual winux::String fieldType( size_t fieldIndex ) override;

    // 重置结果集，以便再次fetch
    bool reset();
    int retCode() const { return _retCode; }
private:
    winux::SharedPointer<SqliteStatement> _stmt;
    int _retCode;
    bool _first;
    bool _hasRow;
    winux::StringArray _fieldNames; // 非原始字段名

    DISABLE_OBJECT_COPY(SqliteResult)
};

class EIENDB_DLL SqliteModifier : public DbModifier
{
public:
    SqliteModifier( SqliteConnection * cnn, winux::String const & tableName );
    virtual ~SqliteModifier();

    /*virtual bool addNew( winux::StringArray const & fieldNames, winux::MixedArray const & values );
    virtual bool addNew( winux::Mixed const & fields );
    virtual bool modify( winux::StringArray const & fieldNames, winux::MixedArray const & values, winux::Mixed const & prkValue );
    virtual bool modify( winux::Mixed const & fields, winux::Mixed const & prkValue );
    virtual int modifyEx( winux::StringArray const & fieldNames, winux::MixedArray const & values, winux::String const & where );
    virtual int modifyEx( winux::Mixed const & fields, winux::String const & where );
    virtual bool deleteOne( winux::Mixed const & prkValue );
    virtual int deleteEx( winux::String const & where ); */

protected:
    virtual void _getTableInfo() override;
    /*SqliteConnection * _cnn;
    winux::String _tableName;
    winux::StringArray _fieldNames; // 原始字段名
    winux::String _prkName;         // 主键名
    //bool _prkAutoIncrement;  // 自动增长 */

    DISABLE_OBJECT_COPY(SqliteModifier)
};

}

#endif // __EIENDB_SQLITE_HPP__