Code_query_distinct.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_distinct_hpp
  14. #define ODBC_CODEGEN_Code_query_distinct_hpp
  15. #include <functional>
  16. #include <common/common.hpp>
  17. #include "Code_query.hpp"
  18. #include "Code_expr_row.hpp"
  19. #include "Code_pred.hpp"
  20. /**
  21.  * @class Plan_query_distinct
  22.  * @brief Group-by node in PlanTree
  23.  */
  24. class Plan_query_distinct : public Plan_query {
  25. public:
  26.     Plan_query_distinct(Plan_root* root);
  27.     virtual ~Plan_query_distinct();
  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.     Plan_expr_row* getRow();
  34. protected:
  35.     Plan_query* m_query;
  36. };
  37. inline
  38. Plan_query_distinct::Plan_query_distinct(Plan_root* root) :
  39.     Plan_query(root),
  40.     m_query(0)
  41. {
  42. }
  43. // children
  44. inline void
  45. Plan_query_distinct::setQuery(Plan_query* query)
  46. {
  47.     ctx_assert(query != 0);
  48.     m_query = query;
  49. }
  50. /**
  51.  * Distinct preserves order of input rows so we use 2 data structures:
  52.  * map<row> = index and vector<index> = row (index >= 0).
  53.  */
  54. class Exec_query_distinct;
  55. struct DistinctLess : std::binary_function<const SqlRow*, const SqlRow*, bool> {
  56.     bool operator()(const SqlRow* s1, const SqlRow* s2) const;
  57. };
  58. typedef std::map<const SqlRow*, unsigned, DistinctLess> DistinctList;
  59. typedef std::vector<const SqlRow*> DistinctVector;
  60. /**
  61.  * @class Exec_query_distinct
  62.  * @brief Group-by node in ExecTree
  63.  */
  64. class Exec_query_distinct : public Exec_query {
  65. public:
  66.     class Code : public Exec_query::Code {
  67.     public:
  68. Code(const SqlSpecs& sqlSpecs);
  69. virtual ~Code();
  70.     protected:
  71. friend class Exec_query_distinct;
  72. // sets reference to Sqlspecs from subquery
  73.     };
  74.     class Data : public Exec_query::Data {
  75.     public:
  76. Data(Exec_query_distinct* node, const SqlSpecs& sqlSpecs);
  77. virtual ~Data();
  78.     protected:
  79. friend class Exec_query_distinct;
  80. SqlRow m_sqlRow; // current row
  81. bool m_grouped; // fetch and group-by done
  82. unsigned m_count;
  83. DistinctList m_groupList;
  84. DistinctVector m_groupVector;
  85. unsigned m_index;
  86.     };
  87.     Exec_query_distinct(Exec_root* root);
  88.     virtual ~Exec_query_distinct();
  89.     void alloc(Ctx& ctx, Ctl& ctl);
  90.     void execImpl(Ctx& ctx, Ctl& ctl);
  91.     bool fetchImpl(Ctx& ctx, Ctl& ctl);
  92.     void close(Ctx& ctx);
  93.     void print(Ctx& ctx);
  94.     // children
  95.     const Code& getCode() const;
  96.     Data& getData() const;
  97.     void setQuery(Exec_query* query);
  98.     const Exec_query* getRawQuery() const;
  99. protected:
  100.     friend class Exec_query;
  101.     Exec_query* m_query;
  102. };
  103. inline
  104. Exec_query_distinct::Code::Code(const SqlSpecs& sqlSpecs) :
  105.     Exec_query::Code(sqlSpecs)
  106. {
  107. }
  108. inline
  109. Exec_query_distinct::Data::Data(Exec_query_distinct* node, const SqlSpecs& sqlSpecs) :
  110.     Exec_query::Data(node, m_sqlRow),
  111.     m_sqlRow(sqlSpecs),
  112.     m_grouped(false),
  113.     m_count(0),
  114.     m_index(0)
  115. {
  116. }
  117. inline
  118. Exec_query_distinct::Exec_query_distinct(Exec_root* root) :
  119.     Exec_query(root),
  120.     m_query(0)
  121. {
  122. }
  123. // children
  124. inline const Exec_query_distinct::Code&
  125. Exec_query_distinct::getCode() const
  126. {
  127.     const Code* code = static_cast<const Code*>(m_code);
  128.     return *code;
  129. }
  130. inline Exec_query_distinct::Data&
  131. Exec_query_distinct::getData() const
  132. {
  133.     Data* data = static_cast<Data*>(m_data);
  134.     return *data;
  135. }
  136. inline void
  137. Exec_query_distinct::setQuery(Exec_query* query)
  138. {
  139.     ctx_assert(m_query == 0 && query != 0);
  140.     m_query = query;
  141. }
  142. #endif