Code_query_project.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_project_hpp
  14. #define ODBC_CODEGEN_Code_query_project_hpp
  15. #include <common/common.hpp>
  16. #include "Code_query.hpp"
  17. #include "Code_expr_row.hpp"
  18. /**
  19.  * @class Plan_query_project
  20.  * @brief Project node in PlanTree
  21.  */
  22. class Plan_query_project : public Plan_query {
  23. public:
  24.     Plan_query_project(Plan_root* root);
  25.     virtual ~Plan_query_project();
  26.     Plan_base* analyze(Ctx& ctx, Ctl& ctl);
  27.     Exec_base* codegen(Ctx& ctx, Ctl& ctl);
  28.     void print(Ctx& ctx);
  29.     // children
  30.     void setQuery(Plan_query* query);
  31.     void setRow(Plan_expr_row* exprRow);
  32.     void setLimit(int off, int cnt);
  33. protected:
  34.     Plan_expr_row* getRow();
  35.     Plan_query* m_query;
  36.     Plan_expr_row* m_exprRow;
  37.     int m_limitOff;
  38.     int m_limitCnt;
  39. };
  40. inline
  41. Plan_query_project::Plan_query_project(Plan_root* root) :
  42.     Plan_query(root),
  43.     m_query(0),
  44.     m_exprRow(0),
  45.     m_limitOff(0),
  46.     m_limitCnt(-1)
  47. {
  48. }
  49. // children
  50. inline void
  51. Plan_query_project::setQuery(Plan_query* query)
  52. {
  53.     ctx_assert(query != 0);
  54.     m_query = query;
  55. }
  56. inline void
  57. Plan_query_project::setRow(Plan_expr_row* exprRow)
  58. {
  59.     ctx_assert(exprRow != 0);
  60.     m_exprRow = exprRow;
  61. }
  62. inline void
  63. Plan_query_project::setLimit(int off, int cnt)
  64. {
  65.     m_limitOff = off;
  66.     m_limitCnt = cnt;
  67. }
  68. /**
  69.  * @class Exec_query_project
  70.  * @brief Project node in ExecTree
  71.  */
  72. class Exec_query_project : public Exec_query {
  73. public:
  74.     class Code : public Exec_query::Code {
  75.     public:
  76. Code(const SqlSpecs& sqlSpecs);
  77. virtual ~Code();
  78.     protected:
  79. friend class Plan_query_project;
  80. friend class Exec_query_project;
  81. // sets reference to Sqlspecs from the row
  82. int m_limitOff;
  83. int m_limitCnt;
  84.     };
  85.     class Data : public Exec_query::Data {
  86.     public:
  87. Data(Exec_query_project* node, const SqlRow& sqlRow);
  88. virtual ~Data();
  89.     protected:
  90. friend class Exec_query_project;
  91. // sets reference to SqlRow from the row
  92. unsigned m_cnt;
  93.     };
  94.     Exec_query_project(Exec_root* root);
  95.     virtual ~Exec_query_project();
  96.     void alloc(Ctx& ctx, Ctl& ctl);
  97.     void execImpl(Ctx& ctx, Ctl& ctl);
  98.     bool fetchImpl(Ctx& ctx, Ctl& ctl);
  99.     void close(Ctx& ctx);
  100.     void print(Ctx& ctx);
  101.     // children
  102.     const Code& getCode() const;
  103.     Data& getData() const;
  104.     void setQuery(Exec_query* query);
  105.     void setRow(Exec_expr_row* exprRow);
  106.     const Exec_query* getRawQuery() const;
  107. protected:
  108.     friend class Exec_query;
  109.     Exec_query* m_query;
  110.     Exec_expr_row* m_exprRow;
  111. };
  112. inline
  113. Exec_query_project::Code::Code(const SqlSpecs& sqlSpecs) :
  114.     Exec_query::Code(sqlSpecs),
  115.     m_limitOff(0),
  116.     m_limitCnt(-1)
  117. {
  118. }
  119. inline
  120. Exec_query_project::Data::Data(Exec_query_project* node, const SqlRow& sqlRow) :
  121.     Exec_query::Data(node, sqlRow),
  122.     m_cnt(0)
  123. {
  124. }
  125. inline
  126. Exec_query_project::Exec_query_project(Exec_root* root) :
  127.     Exec_query(root),
  128.     m_query(0),
  129.     m_exprRow(0)
  130. {
  131. }
  132. // children
  133. inline const Exec_query_project::Code&
  134. Exec_query_project::getCode() const
  135. {
  136.     const Code* code = static_cast<const Code*>(m_code);
  137.     return *code;
  138. }
  139. inline Exec_query_project::Data&
  140. Exec_query_project::getData() const
  141. {
  142.     Data* data = static_cast<Data*>(m_data);
  143.     return *data;
  144. }
  145. inline void
  146. Exec_query_project::setQuery(Exec_query* query)
  147. {
  148.     ctx_assert(m_query == 0 && query != 0);
  149.     m_query = query;
  150. }
  151. inline void
  152. Exec_query_project::setRow(Exec_expr_row* exprRow)
  153. {
  154.     ctx_assert(m_exprRow == 0 && exprRow != 0);
  155.     m_exprRow = exprRow;
  156. }
  157. #endif