Code_query_count.hpp
上传用户:romrleung
上传日期:2022-05-23
资源大小:18897k
文件大小:4k
源码类别:

MySQL数据库

开发平台:

Visual C++

  1. /* Copyright (C) 2003 MySQL AB
  2.    This program is free software; you can redistribute it and/or modify
  3.    it under the terms of the GNU General Public License as published by
  4.    the Free Software Foundation; either version 2 of the License, or
  5.    (at your option) any later version.
  6.    This program is distributed in the hope that it will be useful,
  7.    but WITHOUT ANY WARRANTY; without even the implied warranty of
  8.    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  9.    GNU General Public License for more details.
  10.    You should have received a copy of the GNU General Public License
  11.    along with this program; if not, write to the Free Software
  12.    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
  13. #ifndef ODBC_CODEGEN_Code_query_count_hpp
  14. #define ODBC_CODEGEN_Code_query_count_hpp
  15. #include <common/common.hpp>
  16. #include "Code_query.hpp"
  17. #include "Code_expr_row.hpp"
  18. #include "Code_root.hpp"
  19. class Ctx;
  20. /**
  21.  * @class Plan_query_count
  22.  * @brief Select count and other aggregates (no group by)
  23.  */
  24. class Plan_query_count : public Plan_query {
  25. public:
  26.     Plan_query_count(Plan_root* root);
  27.     virtual ~Plan_query_count();
  28.     Plan_base* analyze(Ctx& ctx, Ctl& ctl);
  29.     Exec_base* codegen(Ctx& ctx, Ctl& ctl);
  30.     void print(Ctx& ctx);
  31.     // children
  32.     void setQuery(Plan_query* query);
  33.     void setRow(Plan_expr_row* exprRow);
  34. protected:
  35.     Plan_expr_row* getRow();
  36.     Plan_query* m_query;
  37.     Plan_expr_row* m_exprRow;
  38. };
  39. inline
  40. Plan_query_count::Plan_query_count(Plan_root* root) :
  41.     Plan_query(root),
  42.     m_query(0),
  43.     m_exprRow(0)
  44. {
  45. }
  46. // children
  47. inline void
  48. Plan_query_count::setQuery(Plan_query* query)
  49. {
  50.     ctx_assert(query != 0);
  51.     m_query = query;
  52. }
  53. inline void
  54. Plan_query_count::setRow(Plan_expr_row* exprRow)
  55. {
  56.     ctx_assert(exprRow != 0);
  57.     m_exprRow = exprRow;
  58. }
  59. /**
  60.  * @class Exec_query_count
  61.  * @brief Select count and other aggregates (no group by)
  62.  */
  63. class Exec_query_count : public Exec_query {
  64. public:
  65.     class Code : public Exec_query::Code {
  66.     public:
  67. Code(const SqlSpecs& sqlSpecs);
  68. virtual ~Code();
  69.     protected:
  70. friend class Exec_query_count;
  71. // sets reference to Sqlspecs from the row
  72.     };
  73.     class Data : public Exec_query::Data {
  74.     public:
  75. Data(Exec_query_count* node, const SqlRow& sqlRow);
  76. virtual ~Data();
  77.     protected:
  78. friend class Exec_query_count;
  79. // sets reference to SqlRow from the row
  80. bool m_done; // returns one row
  81.     };
  82.     Exec_query_count(Exec_root* root);
  83.     virtual ~Exec_query_count();
  84.     void alloc(Ctx& ctx, Ctl& ctl);
  85.     void execImpl(Ctx& ctx, Ctl& ctl);
  86.     bool fetchImpl(Ctx& ctx, Ctl& ctl);
  87.     void close(Ctx& ctx);
  88.     void print(Ctx& ctx);
  89.     // children
  90.     const Code& getCode() const;
  91.     Data& getData() const;
  92.     void setQuery(Exec_query* query);
  93.     void setRow(Exec_expr_row* exprRow);
  94.     const Exec_query* getRawQuery() const;
  95. protected:
  96.     Exec_query* m_query;
  97.     Exec_expr_row* m_exprRow;
  98. };
  99. inline
  100. Exec_query_count::Code::Code(const SqlSpecs& sqlSpecs) :
  101.     Exec_query::Code(sqlSpecs)
  102. {
  103. }
  104. inline
  105. Exec_query_count::Data::Data(Exec_query_count* node, const SqlRow& sqlRow) :
  106.     Exec_query::Data(node, sqlRow)
  107. {
  108. }
  109. inline
  110. Exec_query_count::Exec_query_count(Exec_root* root) :
  111.     Exec_query(root),
  112.     m_query(0),
  113.     m_exprRow(0)
  114. {
  115. }
  116. // children
  117. inline const Exec_query_count::Code&
  118. Exec_query_count::getCode() const
  119. {
  120.     const Code* code = static_cast<const Code*>(m_code);
  121.     return *code;
  122. }
  123. inline Exec_query_count::Data&
  124. Exec_query_count::getData() const
  125. {
  126.     Data* data = static_cast<Data*>(m_data);
  127.     return *data;
  128. }
  129. inline void
  130. Exec_query_count::setQuery(Exec_query* query)
  131. {
  132.     ctx_assert(query != 0);
  133.     m_query = query;
  134. }
  135. inline void
  136. Exec_query_count::setRow(Exec_expr_row* exprRow)
  137. {
  138.     ctx_assert(m_exprRow == 0 && exprRow != 0);
  139.     m_exprRow = exprRow;
  140. }
  141. #endif