KVSQLiteStatement.h

 
/*************************************************************************
 * Copyright (C) 1995-2013, Rene Brun and Fons Rademakers.               *
 * All rights reserved.                                                  *
 *                                                                       *
 * For the licensing terms see $ROOTSYS/LICENSE.                         *
 * For the list of contributors see $ROOTSYS/README/CREDITS.             *
 *************************************************************************/
 
#ifndef ROOT_TSQLiteStatement
#define ROOT_TSQLiteStatement
 
#include "TSQLStatement.h"
 
class sqlite3;
class sqlite3_stmt;
 
struct SQLite3_Stmt_t {
   sqlite3*      fConn;
   sqlite3_stmt* fRes;
};
 
class KVSQLiteStatement : public TSQLStatement {
 
private:
   SQLite3_Stmt_t*       fStmt{nullptr};     
   Int_t                 fWorkingMode{0};    
   Int_t                 fNumPars{0};        
   Int_t                 fIterationCount{0}; 
 
   Bool_t      IsSetParsMode() const
   {
      return fWorkingMode == 1;
   }
   Bool_t      IsResultSetMode() const
   {
      return fWorkingMode == 2;
   }
 
   Bool_t      SetSQLParamType(Int_t npar, int sqltype, bool sig, int sqlsize = 0);
 
   long double ConvertToNumeric(Int_t npar);
   const char* ConvertToString(Int_t npar);
 
   Bool_t CheckBindError(const char* method, int res);
 
public:
   KVSQLiteStatement(SQLite3_Stmt_t* stmt, Bool_t errout = kTRUE);
   virtual ~KVSQLiteStatement();
 
   void        Close(Option_t* = "") final;
 
   Int_t       GetBufferLength() const final
   {
      return 1;
   }
   Int_t       GetNumParameters() final;
 
   Bool_t      SetNull(Int_t npar) final;
   Bool_t      SetInt(Int_t npar, Int_t value) final;
   Bool_t      SetUInt(Int_t npar, UInt_t value) final;
   Bool_t      SetLong(Int_t npar, Long_t value) final;
   Bool_t      SetLong64(Int_t npar, Long64_t value) final;
   Bool_t      SetULong64(Int_t npar, ULong64_t value) final;
   Bool_t      SetDouble(Int_t npar, Double_t value) final;
   Bool_t      SetString(Int_t npar, const char* value, Int_t maxsize = 256) final;
   Bool_t      SetBinary(Int_t npar, void* mem, Long_t size, Long_t maxsize = 0x1000) final;
   Bool_t      SetDate(Int_t npar, Int_t year, Int_t month, Int_t day) final;
   Bool_t      SetTime(Int_t npar, Int_t hour, Int_t min, Int_t sec) final;
   Bool_t      SetDatime(Int_t npar, Int_t year, Int_t month, Int_t day, Int_t hour, Int_t min, Int_t sec) final;
   using TSQLStatement::SetTimestamp;
   Bool_t      SetTimestamp(Int_t npar, Int_t year, Int_t month, Int_t day, Int_t hour, Int_t min, Int_t sec, Int_t frac = 0) final;
 
   Bool_t      NextIteration() final;
 
   Bool_t      Process() final;
   Int_t       GetNumAffectedRows() final;
 
   Bool_t      StoreResult() final;
   Int_t       GetNumFields() final;
   const char* GetFieldName(Int_t nfield) final;
   Bool_t      NextResultRow() final;
 
   Bool_t      IsNull(Int_t npar) final;
   Int_t       GetInt(Int_t npar) final;
   UInt_t      GetUInt(Int_t npar) final;
   Long_t      GetLong(Int_t npar) final;
   Long64_t    GetLong64(Int_t npar) final;
   ULong64_t   GetULong64(Int_t npar) final;
   Double_t    GetDouble(Int_t npar) final;
   const char* GetString(Int_t npar) final;
   Bool_t      GetBinary(Int_t npar, void*& mem, Long_t& size) final;
   Bool_t      GetDate(Int_t npar, Int_t& year, Int_t& month, Int_t& day) final;
   Bool_t      GetTime(Int_t npar, Int_t& hour, Int_t& min, Int_t& sec) final;
   Bool_t      GetDatime(Int_t npar, Int_t& year, Int_t& month, Int_t& day, Int_t& hour, Int_t& min, Int_t& sec) final;
   using TSQLStatement::GetTimestamp;
   Bool_t      GetTimestamp(Int_t npar, Int_t& year, Int_t& month, Int_t& day, Int_t& hour, Int_t& min, Int_t& sec, Int_t&) final;
 
   ClassDefOverride(KVSQLiteStatement, 0);  // SQL statement class for SQLite DB
};
 
#endif