eiendb_base.hpp

浏览该文件的文档.

#ifndef __EIENDB_BASE_HPP__
#define __EIENDB_BASE_HPP__

#include "winux.hpp"

namespace eiendb
{

#ifdef EIENDB_DLL_USE
    #if defined(_MSC_VER) || defined(WIN32)
        #pragma warning( disable: 4251 )
        #ifdef  EIENDB_DLL_EXPORTS
            #define EIENDB_DLL  __declspec(dllexport)
        #else
            #define EIENDB_DLL  __declspec(dllimport)
        #endif

        #define EIENDB_API __stdcall
    #else
        #define EIENDB_DLL
        #define EIENDB_API
    #endif
#else
    #define EIENDB_DLL
    #define EIENDB_API
#endif

#define EIENDB_FUNC_DECL(ret) EIENDB_DLL ret EIENDB_API
#define EIENDB_FUNC_IMPL(ret) ret EIENDB_API


#ifndef interface
#define interface struct
#endif

typedef void * DbHandle;
typedef void * DbResHandle;

class EIENDB_DLL DbError : public winux::Error
{
public:
    enum
    {
        dbeNone, 
        dbeDbSelfError, 
        dbeNotImplemented, 
        dbeConfigParamsError, 
        dbeInvalidDbConn,  
        dbeMysqlError, 
        dbeSqliteError, 
        dbePgsqlError, 
    };

    DbError( int errType, winux::AnsiString const & s ) throw() : winux::Error( errType, s ) { }
};

interface IDbConnection;
interface IDbStatement;
interface IDbResult;

interface EIENDB_DLL IDbConnection
{
    virtual ~IDbConnection() { }

    virtual bool connect();

    virtual bool close();

    virtual void alive();

    virtual bool selectDb( winux::String const & database );

    virtual bool setLinkCharset( winux::String const & charset );

    virtual bool createDb( winux::String const & database );

    virtual bool dropDb( winux::String const & database );

    virtual size_t affectedRows();

    virtual size_t insertId();

    virtual int errNo();

    virtual winux::String error();

    virtual winux::SharedPointer<IDbResult> query( winux::String const & sql );

    virtual winux::SharedPointer<IDbResult> query( winux::SharedPointer<IDbStatement> stmt );

    virtual bool exec( winux::String const & sql );

    virtual bool exec( winux::SharedPointer<IDbStatement> stmt );

    virtual winux::String escape( winux::String const & str, winux::String const & addQuote = "\'" );

    virtual winux::String escape( void const * buf, size_t size, winux::String const & addQuote = "\'" );

    virtual winux::SharedPointer<IDbResult> listDbs();

    virtual winux::SharedPointer<IDbResult> listFields( winux::String const & tableName );

    virtual winux::SharedPointer<IDbResult> listTables();

    virtual winux::String tableDdl( winux::String const & tableName );

    virtual winux::String symbolQuotes( winux::String const & str, bool periodAsSeparator = true );

    virtual size_t loadSqlFile( winux::IFile * sqlFile, winux::StringArray * arrSql );
    virtual size_t getPrimaryKey( winux::String const & tableName, winux::StringArray * arrKeyColumn );

    virtual winux::SharedPointer<IDbStatement> buildStmt( winux::String const & sql, winux::Mixed const & params = winux::mxNull );

    virtual operator bool() const = 0;
};

interface EIENDB_DLL IDbStatement
{
    virtual ~IDbStatement() { }
    virtual bool build( winux::String const & sql, winux::Mixed const & params = winux::mxNull );
    virtual bool bind( size_t paramIndex, winux::Mixed const & val );
    virtual bool bind( winux::String const & paramName, winux::Mixed const & val );
};

interface EIENDB_DLL IDbResult
{
    virtual ~IDbResult() { }

    virtual bool free();

    virtual bool dataSeek( size_t index );

    virtual bool fetchRow( winux::Mixed * fields, int type = 0 );

    virtual bool fetchArray( winux::MixedArray * fields );

    virtual bool fetchMap( winux::StringMixedMap * fields );

    virtual winux::String fieldName( size_t fieldIndex );

    virtual size_t fieldsCount();

    virtual size_t rowsCount();

    virtual winux::String fieldType( size_t fieldIndex );

};

interface EIENDB_DLL IDbModifier
{
    virtual ~IDbModifier() { }

    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 bool modifyEx( winux::StringArray const & fieldNames, winux::MixedArray const & values, winux::String const & where );

    virtual bool modifyEx( winux::Mixed const & fields, winux::String const & where );

    virtual bool deleteOne( winux::Mixed const & prkValue );

    virtual bool deleteEx( winux::String const & where );
};

EIENDB_FUNC_DECL(winux::String) MixedToEscapedStr( winux::Mixed const & m, IDbConnection * cnn, winux::String const & addQuote = "\'" );

class EIENDB_DLL DbStatement : public IDbStatement
{
public:
    DbStatement( IDbConnection * cnn );
    DbStatement( IDbConnection * cnn, winux::String const & sql, winux::Mixed const & params = winux::mxNull );

    virtual bool build( winux::String const & sql, winux::Mixed const & params = winux::mxNull ) 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 bool() const { return !_orgSql.empty(); }
    IDbConnection * getCnn() const { return _cnn; }
    winux::String const & getSql();

    // paramIndex base on 1 
    winux::Mixed const & getParam( size_t paramIndex ) const;
    // paramName prefix with ':'
    winux::Mixed const & getParam( winux::String const & paramName ) const;

private:
    // 生成SQL
    void _generateSql();

    IDbConnection * _cnn;
    winux::String _orgSql; // 原SQL
    winux::String _sql; // 生成的SQL
    bool _isGeneratedSql; // 是否已经生成SQL
    bool _isNameBinding; // 是否采用参数名称绑定
    std::vector< std::pair< size_t, winux::Mixed > > _bindingParams; // 已绑定的参数
    std::vector< std::pair< winux::String, winux::Mixed > > _bindingNameParams; // 已绑定的名称参数

    DISABLE_OBJECT_COPY(DbStatement)
};

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

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

protected:
    // 获取表信息
    virtual void _getTableInfo();
    // 构建`INSERT INTO`语句（含待绑定变量占位符，语句末尾不带';'）
    virtual winux::String _buildInsertInto( winux::StringArray const * fieldNames, size_t valuesCount );
    // 执行`INSERT INTO`语句
    virtual bool _execInsertInto( winux::SharedPointer<IDbStatement> stmt );

    IDbConnection * _cnn;
    winux::String _tableName;
    winux::StringArray _prkColumn; // 主键引用列
    winux::StringArray _fieldNames; // 原始字段名

    DISABLE_OBJECT_COPY(DbModifier)
};


} // namespace eiendb

#endif // __EIENDB_BASE_HPP__