// file : odb/sqlite/query.hxx // copyright : Copyright (c) 2009-2012 Code Synthesis Tools CC // license : GNU GPL v2; see accompanying LICENSE file #ifndef ODB_SQLITE_QUERY_HXX #define ODB_SQLITE_QUERY_HXX #include #include #include #include // std::size_t #include #include #include #include #include #include #include #include #include #include namespace odb { namespace sqlite { template class val_bind { public: explicit val_bind (const T& v): val (v) {} const T& val; }; template class ref_bind { public: explicit ref_bind (const T& r): ref (r) {} const T& ref; }; struct LIBODB_SQLITE_EXPORT query_param: details::shared_base { virtual ~query_param (); bool reference () const { return value_ != 0; } virtual bool init () = 0; virtual void bind (sqlite::bind*) = 0; protected: query_param (const void* value) : value_ (value) { } protected: const void* value_; }; class query_base; class LIBODB_SQLITE_EXPORT query_params: public details::shared_base { public: typedef sqlite::binding binding_type; void init (); binding_type& binding () {return binding_;} private: friend class query_base; query_params (): binding_ (0, 0) {} query_params (const query_params&); query_params& operator= (const query_params&); query_params& operator+= (const query_params&); void add (details::shared_ptr); private: typedef std::vector > params; params params_; std::vector bind_; binding_type binding_; }; // // template struct query_column; class LIBODB_SQLITE_EXPORT query_base { public: struct clause_part { enum kind_type { kind_column, kind_param, kind_native, kind_bool }; clause_part (kind_type k): kind (k) {} clause_part (kind_type k, const std::string& p): kind (k), part (p) {} clause_part (bool p): kind (kind_bool), bool_part (p) {} kind_type kind; std::string part; // If kind is param, then part is conversion expr. bool bool_part; }; query_base () : parameters_ (new (details::shared) query_params) { } // True or false literal. // explicit query_base (bool v) : parameters_ (new (details::shared) query_params) { clause_.push_back (clause_part (v)); } explicit query_base (const char* native) : parameters_ (new (details::shared) query_params) { clause_.push_back (clause_part (clause_part::kind_native, native)); } explicit query_base (const std::string& native) : parameters_ (new (details::shared) query_params) { clause_.push_back (clause_part (clause_part::kind_native, native)); } query_base (const char* table, const char* column) : parameters_ (new (details::shared) query_params) { append (table, column); } template explicit query_base (val_bind v) : parameters_ (new (details::shared) query_params) { append::db_type_id> ( v, details::conversion::to ()); } template explicit query_base (ref_bind r) : parameters_ (new (details::shared) query_params) { append::db_type_id> ( r, details::conversion::to ()); } template query_base (const query_column&); query_base (const query_base&); query_base& operator= (const query_base&); public: std::string clause () const; const char* clause_prefix () const; // Initialize the by-reference parameters from bound variables. // void init_parameters () const; binding& parameters_binding () const; const details::shared_ptr& parameters () const; public: bool empty () const { return clause_.empty (); } static const query_base true_expr; bool const_true () const { return clause_.size () == 1 && clause_.front ().kind == clause_part::kind_bool && clause_.front ().bool_part; } void optimize (); public: template static val_bind _val (const T& x) { return val_bind (x); } template static ref_bind _ref (const T& x) { return ref_bind (x); } public: query_base& operator+= (const query_base&); query_base& operator+= (const std::string& q) { append (q); return *this; } template query_base& operator+= (val_bind v) { append::db_type_id> ( v, details::conversion::to ()); return *this; } template query_base& operator+= (ref_bind r) { append::db_type_id> ( r, details::conversion::to ()); return *this; } public: template void append (val_bind, const char* conv); template void append (ref_bind, const char* conv); void append (const std::string& native); void append (const char* table, const char* column); private: void add (details::shared_ptr, const char* conv); private: typedef std::vector clause_type; clause_type clause_; details::shared_ptr parameters_; }; inline query_base operator+ (const query_base& x, const query_base& y) { query_base r (x); r += y; return r; } template inline query_base operator+ (const query_base& q, val_bind b) { query_base r (q); r += b; return r; } template inline query_base operator+ (const query_base& q, ref_bind b) { query_base r (q); r += b; return r; } template inline query_base operator+ (val_bind b, const query_base& q) { query_base r; r += b; r += q; return r; } template inline query_base operator+ (ref_bind b, const query_base& q) { query_base r; r += b; r += q; return r; } inline query_base operator+ (const query_base& q, const std::string& s) { query_base r (q); r += s; return r; } inline query_base operator+ (const std::string& s, const query_base& q) { query_base r (s); r += q; return r; } template inline query_base operator+ (const std::string& s, val_bind b) { query_base r (s); r += b; return r; } template inline query_base operator+ (const std::string& s, ref_bind b) { query_base r (s); r += b; return r; } template inline query_base operator+ (val_bind b, const std::string& s) { query_base r; r += b; r += s; return r; } template inline query_base operator+ (ref_bind b, const std::string& s) { query_base r; r += b; r += s; return r; } LIBODB_SQLITE_EXPORT query_base operator&& (const query_base&, const query_base&); LIBODB_SQLITE_EXPORT query_base operator|| (const query_base&, const query_base&); LIBODB_SQLITE_EXPORT query_base operator! (const query_base&); // query_column // template class copy_bind: public val_bind { public: explicit copy_bind (const T2& v): val_bind (val), val (v) {} const T val; }; template const T& type_instance (); template struct query_column { // Note that we keep shallow copies of the table, column, and conversion // expression. The latter can be NULL. // query_column (const char* table, const char* column, const char* conv) : table_ (table), column_ (column), conversion_ (conv) { } const char* table () const { return table_; } const char* column () const { return column_; } // Can be NULL. // const char* conversion () const { return conversion_; } // is_null, is_not_null // public: query_base is_null () const { query_base q (table_, column_); q += "IS NULL"; return q; } query_base is_not_null () const { query_base q (table_, column_); q += "IS NOT NULL"; return q; } // in // public: query_base in (const T&, const T&) const; query_base in (const T&, const T&, const T&) const; query_base in (const T&, const T&, const T&, const T&) const; query_base in (const T&, const T&, const T&, const T&, const T&) const; template query_base in_range (I begin, I end) const; // = // public: query_base equal (const T& v) const { return equal (val_bind (v)); } query_base equal (val_bind v) const { query_base q (table_, column_); q += "="; q.append (v, conversion_); return q; } template query_base equal (val_bind v) const { copy_bind c (v.val); return equal (c); } query_base equal (ref_bind r) const { query_base q (table_, column_); q += "="; q.append (r, conversion_); return q; } friend query_base operator== (const query_column& c, const T& v) { return c.equal (v); } friend query_base operator== (const T& v, const query_column& c) { return c.equal (v); } friend query_base operator== (const query_column& c, val_bind v) { return c.equal (v); } friend query_base operator== (val_bind v, const query_column& c) { return c.equal (v); } template friend query_base operator== (const query_column& c, val_bind v) { return c.equal (v); } template friend query_base operator== (val_bind v, const query_column& c) { return c.equal (v); } friend query_base operator== (const query_column& c, ref_bind r) { return c.equal (r); } friend query_base operator== (ref_bind r, const query_column& c) { return c.equal (r); } // != // public: query_base unequal (const T& v) const { return unequal (val_bind (v)); } query_base unequal (val_bind v) const { query_base q (table_, column_); q += "!="; q.append (v, conversion_); return q; } template query_base unequal (val_bind v) const { copy_bind c (v.val); return unequal (c); } query_base unequal (ref_bind r) const { query_base q (table_, column_); q += "!="; q.append (r, conversion_); return q; } friend query_base operator!= (const query_column& c, const T& v) { return c.unequal (v); } friend query_base operator!= (const T& v, const query_column& c) { return c.unequal (v); } friend query_base operator!= (const query_column& c, val_bind v) { return c.unequal (v); } friend query_base operator!= (val_bind v, const query_column& c) { return c.unequal (v); } template friend query_base operator!= (const query_column& c, val_bind v) { return c.unequal (v); } template friend query_base operator!= (val_bind v, const query_column& c) { return c.unequal (v); } friend query_base operator!= (const query_column& c, ref_bind r) { return c.unequal (r); } friend query_base operator!= (ref_bind r, const query_column& c) { return c.unequal (r); } // < // public: query_base less (const T& v) const { return less (val_bind (v)); } query_base less (val_bind v) const { query_base q (table_, column_); q += "<"; q.append (v, conversion_); return q; } template query_base less (val_bind v) const { copy_bind c (v.val); return less (c); } query_base less (ref_bind r) const { query_base q (table_, column_); q += "<"; q.append (r, conversion_); return q; } friend query_base operator< (const query_column& c, const T& v) { return c.less (v); } friend query_base operator< (const T& v, const query_column& c) { return c.greater (v); } friend query_base operator< (const query_column& c, val_bind v) { return c.less (v); } friend query_base operator< (val_bind v, const query_column& c) { return c.greater (v); } template friend query_base operator< (const query_column& c, val_bind v) { return c.less (v); } template friend query_base operator< (val_bind v, const query_column& c) { return c.greater (v); } friend query_base operator< (const query_column& c, ref_bind r) { return c.less (r); } friend query_base operator< (ref_bind r, const query_column& c) { return c.greater (r); } // > // public: query_base greater (const T& v) const { return greater (val_bind (v)); } query_base greater (val_bind v) const { query_base q (table_, column_); q += ">"; q.append (v, conversion_); return q; } template query_base greater (val_bind v) const { copy_bind c (v.val); return greater (c); } query_base greater (ref_bind r) const { query_base q (table_, column_); q += ">"; q.append (r, conversion_); return q; } friend query_base operator> (const query_column& c, const T& v) { return c.greater (v); } friend query_base operator> (const T& v, const query_column& c) { return c.less (v); } friend query_base operator> (const query_column& c, val_bind v) { return c.greater (v); } friend query_base operator> (val_bind v, const query_column& c) { return c.less (v); } template friend query_base operator> (const query_column& c, val_bind v) { return c.greater (v); } template friend query_base operator> (val_bind v, const query_column& c) { return c.less (v); } friend query_base operator> (const query_column& c, ref_bind r) { return c.greater (r); } friend query_base operator> (ref_bind r, const query_column& c) { return c.less (r); } // <= // public: query_base less_equal (const T& v) const { return less_equal (val_bind (v)); } query_base less_equal (val_bind v) const { query_base q (table_, column_); q += "<="; q.append (v, conversion_); return q; } template query_base less_equal (val_bind v) const { copy_bind c (v.val); return less_equal (c); } query_base less_equal (ref_bind r) const { query_base q (table_, column_); q += "<="; q.append (r, conversion_); return q; } friend query_base operator<= (const query_column& c, const T& v) { return c.less_equal (v); } friend query_base operator<= (const T& v, const query_column& c) { return c.greater_equal (v); } friend query_base operator<= (const query_column& c, val_bind v) { return c.less_equal (v); } friend query_base operator<= (val_bind v, const query_column& c) { return c.greater_equal (v); } template friend query_base operator<= (const query_column& c, val_bind v) { return c.less_equal (v); } template friend query_base operator<= (val_bind v, const query_column& c) { return c.greater_equal (v); } friend query_base operator<= (const query_column& c, ref_bind r) { return c.less_equal (r); } friend query_base operator<= (ref_bind r, const query_column& c) { return c.greater_equal (r); } // >= // public: query_base greater_equal (const T& v) const { return greater_equal (val_bind (v)); } query_base greater_equal (val_bind v) const { query_base q (table_, column_); q += ">="; q.append (v, conversion_); return q; } template query_base greater_equal (val_bind v) const { copy_bind c (v.val); return greater_equal (c); } query_base greater_equal (ref_bind r) const { query_base q (table_, column_); q += ">="; q.append (r, conversion_); return q; } friend query_base operator>= (const query_column& c, const T& v) { return c.greater_equal (v); } friend query_base operator>= (const T& v, const query_column& c) { return c.less_equal (v); } friend query_base operator>= (const query_column& c, val_bind v) { return c.greater_equal (v); } friend query_base operator>= (val_bind v, const query_column& c) { return c.less_equal (v); } template friend query_base operator>= (const query_column& c, val_bind v) { return c.greater_equal (v); } template friend query_base operator>= (val_bind v, const query_column& c) { return c.less_equal (v); } friend query_base operator>= (const query_column& c, ref_bind r) { return c.greater_equal (r); } friend query_base operator>= (ref_bind r, const query_column& c) { return c.less_equal (r); } // Column comparison. // public: template query_base operator== (const query_column& c) const { // We can compare columns only if we can compare their C++ types. // (void) (sizeof (type_instance () == type_instance ())); query_base q (table_, column_); q += "="; q.append (c.table (), c.column ()); return q; } template query_base operator!= (const query_column& c) const { // We can compare columns only if we can compare their C++ types. // (void) (sizeof (type_instance () != type_instance ())); query_base q (table_, column_); q += "!="; q.append (c.table (), c.column ()); return q; } template query_base operator< (const query_column& c) const { // We can compare columns only if we can compare their C++ types. // (void) (sizeof (type_instance () < type_instance ())); query_base q (table_, column_); q += "<"; q.append (c.table (), c.column ()); return q; } template query_base operator> (const query_column& c) const { // We can compare columns only if we can compare their C++ types. // (void) (sizeof (type_instance () > type_instance ())); query_base q (table_, column_); q += ">"; q.append (c.table (), c.column ()); return q; } template query_base operator<= (const query_column& c) const { // We can compare columns only if we can compare their C++ types. // (void) (sizeof (type_instance () <= type_instance ())); query_base q (table_, column_); q += "<="; q.append (c.table (), c.column ()); return q; } template query_base operator>= (const query_column& c) const { // We can compare columns only if we can compare their C++ types. // (void) (sizeof (type_instance () >= type_instance ())); query_base q (table_, column_); q += ">="; q.append (c.table (), c.column ()); return q; } private: const char* table_; const char* column_; const char* conversion_; }; // Provide operator+() for using columns to construct native // query fragments (e.g., ORDER BY). // template inline query_base operator+ (const query_column& c, const std::string& s) { query_base q (c.table (), c.column ()); q += s; return q; } template inline query_base operator+ (const std::string& s, const query_column& c) { query_base q (s); q.append (c.table (), c.column ()); return q; } template inline query_base operator+ (const query_column& c, const query_base& q) { query_base r (c.table (), c.column ()); r += q; return r; } template inline query_base operator+ (const query_base& q, const query_column& c) { query_base r (q); r.append (c.table (), c.column ()); return r; } // // template struct query_param_impl; // INTEGER // template struct query_param_impl: query_param { query_param_impl (ref_bind r) : query_param (&r.ref) {} query_param_impl (val_bind v) : query_param (0) {init (v.val);} virtual bool init () { init (*static_cast (value_)); return false; } virtual void bind (sqlite::bind* b) { b->type = sqlite::bind::integer; b->buffer = &image_; } private: void init (const T& v) { bool is_null (false); // Can't be NULL. value_traits::set_image (image_, is_null, v); } private: long long image_; }; // REAL // template struct query_param_impl: query_param { query_param_impl (ref_bind r) : query_param (&r.ref) {} query_param_impl (val_bind v) : query_param (0) {init (v.val);} virtual bool init () { init (*static_cast (value_)); return false; } virtual void bind (sqlite::bind* b) { b->type = sqlite::bind::real; b->buffer = &image_; } private: void init (const T& v) { bool is_null (false); // Can't be NULL. value_traits::set_image (image_, is_null, v); } private: double image_; }; // TEXT // template struct query_param_impl: query_param { query_param_impl (ref_bind r) : query_param (&r.ref) {} query_param_impl (val_bind v) : query_param (0) {init (v.val);} virtual bool init () { return init (*static_cast (value_)); } virtual void bind (sqlite::bind* b) { b->type = image_traits::bind_value; b->buffer = buffer_.data (); b->size = &size_; } private: bool init (const T& v) { bool is_null (false); // Can't be NULL. std::size_t cap (buffer_.capacity ()); value_traits::set_image (buffer_, size_, is_null, v); return cap != buffer_.capacity (); } private: details::buffer buffer_; std::size_t size_; }; // BLOB // template struct query_param_impl: query_param { query_param_impl (ref_bind r) : query_param (&r.ref) {} query_param_impl (val_bind v) : query_param (0) {init (v.val);} virtual bool init () { return init (*static_cast (value_)); } virtual void bind (sqlite::bind* b) { b->type = sqlite::bind::blob; b->buffer = buffer_.data (); b->size = &size_; } private: bool init (const T& v) { bool is_null (false); // Can't be NULL. std::size_t cap (buffer_.capacity ()); value_traits::set_image (buffer_, size_, is_null, v); return cap != buffer_.capacity (); } private: details::buffer buffer_; std::size_t size_; }; } } // odb::sqlite::query and odb::query specialization for SQLite. // namespace odb { namespace sqlite { template class query: public query_base, public query_selector::columns_type { public: // We don't define any typedefs here since they may clash with // column names defined by our base type. // query () { } explicit query (bool v) : query_base (v) { } explicit query (const char* q) : query_base (q) { } explicit query (const std::string& q) : query_base (q) { } template explicit query (val_bind v) : query_base (v) { } template explicit query (ref_bind r) : query_base (r) { } query (const query_base& q) : query_base (q) { } template query (const query_column& qc) : query_base (qc) { } }; } // Derive odb::query from odb::sqlite::query so that it can be // implicitly converted in sqlite::database::query() calls. // template class query: public sqlite::query { public: // We don't define any typedefs here since they may clash with // column names defined by our base type. // query () { } explicit query (bool v) : sqlite::query (v) { } explicit query (const char* q) : sqlite::query (q) { } explicit query (const std::string& q) : sqlite::query (q) { } template explicit query (sqlite::val_bind v) : sqlite::query (v) { } template explicit query (sqlite::ref_bind r) : sqlite::query (r) { } query (const sqlite::query_base& q) : sqlite::query (q) { } template query (const sqlite::query_column& qc) : sqlite::query (qc) { } }; } #include #include #include #endif // ODB_SQLITE_QUERY_HXX