开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 数据库系统 > MySQL数据库 > 查看源码
Exec_expr_func.cpp
资源名称：mysql-4.1.16-win-src.zip　[点击查看]
上传用户：romrleung
上传日期：2022-05-23
资源大小：18897k
文件大小：8k
源码类别：

MySQL数据库

开发平台：

Visual C++

Exec_expr_func.cpp：源码内容
  1. /* Copyright (C) 2003 MySQL AB
  3.    This program is free software; you can redistribute it and/or modify
  4.    it under the terms of the GNU General Public License as published by
  5.    the Free Software Foundation; either version 2 of the License, or
  6.    (at your option) any later version.
  8.    This program is distributed in the hope that it will be useful,
  9.    but WITHOUT ANY WARRANTY; without even the implied warranty of
  10.    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11.    GNU General Public License for more details.
  13.    You should have received a copy of the GNU General Public License
  14.    along with this program; if not, write to the Free Software
  15.    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
  17. #include <time.h>
  18. #include <NdbTick.h>
  19. #include <codegen/Code_expr_func.hpp>
  21. void
  22. Exec_expr_func::init(Ctx& ctx, Ctl& ctl)
  23. {
  24.     const Code& code = getCode();
  25.     Data& data = getData();
  26.     const SqlType& t = code.sqlSpec().sqlType();
  27.     SqlField& f = ctl.m_groupIndex == 0 ? data.m_sqlField : data.groupField(code.sqlSpec().sqlType(), ctl.m_groupIndex, ctl.m_groupInit);
  28.     if (ctl.m_groupIndex >= data.m_groupAcc.size())
  29. data.m_groupAcc.resize(1 + ctl.m_groupIndex);
  30.     Data::Acc& acc = data.m_groupAcc[ctl.m_groupIndex];
  31.     acc.m_count = 0;
  32.     Expr_func::Code fc = code.m_func.m_code;
  33.     if (fc == Expr_func::Substr || fc == Expr_func::Left || fc == Expr_func::Right) {
  34.     } else if (fc == Expr_func::Count) {
  35. f.sqlBigint(0);
  36.     } else if (fc == Expr_func::Min || fc == Expr_func::Max) {
  37. f.sqlNull(true);
  38.     } else if (fc == Expr_func::Sum) {
  39. f.sqlNull(true);
  40. switch (t.type()) {
  41. case SqlType::Bigint:
  42.     acc.m_bigint = 0;
  43.     break;
  44. case SqlType::Double:
  45.     acc.m_double = 0;
  46.     break;
  47. default:
  48.     ctx_assert(false);
  49.     break;
  50. }
  51.     } else if (fc == Expr_func::Avg) {
  52. f.sqlNull(true);
  53. switch (t.type()) {
  54. case SqlType::Double:
  55.     acc.m_double = 0.0;
  56.     break;
  57. default:
  58.     ctx_assert(false);
  59.     break;
  60. }
  61.     } else if (fc == Expr_func::Rownum) {
  62. // uses only m_count
  63.     } else if (fc == Expr_func::Sysdate) {
  64. // set time once
  65. NDB_TICKS secs = 0;
  66. Uint32 micros = 0;
  67. NdbTick_CurrentMicrosecond(&secs, &micros);
  68. time_t clock = secs;
  69. struct tm* t = gmtime(&clock);
  70. SqlDatetime& d = acc.m_sysdate;
  71. d.cc((1900 + t->tm_year) / 100);
  72. d.yy((1900 + t->tm_year) % 100);
  73. d.mm(1 + t->tm_mon);
  74. d.dd(t->tm_mday);
  75. d.HH(t->tm_hour);
  76. d.MM(t->tm_min);
  77. d.SS(t->tm_sec);
  78. d.ff(1000 * micros);
  79.     } else {
  80. ctx_assert(false);
  81.     }
  82. }
  84. void
  85. Exec_expr_func::evaluate(Ctx& ctx, Ctl& ctl)
  86. {
  87.     const Code& code = getCode();
  88.     Data& data = getData();
  89.     const SqlType& t = code.sqlSpec().sqlType();
  90.     if (ctl.m_groupInit)
  91. init(ctx, ctl);
  92.     SqlField& f = ctl.m_groupIndex == 0 ? data.m_sqlField : data.groupField(code.sqlSpec().sqlType(), ctl.m_groupIndex, false);
  93.     Data::Acc& acc = data.m_groupAcc[ctl.m_groupIndex];
  94.     Expr_func::Code fc = code.m_func.m_code;
  95.     const unsigned narg = code.m_narg;
  96.     Exec_expr** args = code.m_args;
  97.     ctx_assert(args != 0);
  98.     // evaluate arguments
  99.     for (unsigned i = 1; i <= narg; i++) {
  100. ctx_assert(args[i] != 0);
  101. unsigned save = ctl.m_groupIndex;
  102. if (code.m_func.m_aggr)
  103.     ctl.m_groupIndex = 0;
  104. args[i]->evaluate(ctx, ctl);
  105. if (! ctx.ok())
  106.     return;
  107. ctl.m_groupIndex = save;
  108.     }
  109.     if (fc == Expr_func::Substr || fc == Expr_func::Left || fc == Expr_func::Right) {
  110. ctx_assert((narg == (unsigned)2) || (narg == (unsigned)(fc == Expr_func::Substr ? 3 : 2)));
  111. const SqlType& t1 = args[1]->getCode().sqlSpec().sqlType();
  112. const SqlField& f1 = args[1]->getData().sqlField();
  113. int pos, len;
  114. for (unsigned i = 2; i <= narg; i++) {
  115.     int& num = (fc == Expr_func::Substr ? (i == 2 ? pos : len) : len);
  116.     const SqlType& t2 = args[i]->getCode().sqlSpec().sqlType();
  117.     const SqlField& f2 = args[i]->getData().sqlField();
  118.     switch (t2.type()) {
  119.     case SqlType::Smallint:
  120. num = static_cast<int>(f2.sqlSmallint());
  121. break;
  122.     case SqlType::Integer:
  123. num = static_cast<int>(f2.sqlInteger());
  124. break;
  125.     case SqlType::Bigint:
  126. num = static_cast<int>(f2.sqlBigint());
  127. break;
  128.     default:
  129. ctx_assert(false);
  130. break;
  131.     }
  132. }
  133. int length = 0;
  134. const SqlChar* data = 0;
  135. switch (t1.type()) {
  136. case SqlType::Char:
  137.     length = t1.length();
  138.     data = f1.sqlChar();
  139.     break;
  140. case SqlType::Varchar:
  141.     unsigned ulength;
  142.     data = f1.sqlVarchar(&ulength);
  143.     length = ulength;
  144.     break;
  145. default:
  146.     ctx_assert(false);
  147.     break;
  148. }
  149. if (fc == Expr_func::Left)
  150.     pos = 1;
  151. else if (fc == Expr_func::Right)
  152.     pos = len > length ? 1 : length - len + 1;
  153. else if (pos < 0)
  154.     pos += length + 1;
  155. if (pos <= 0 || pos > length || len <= 0) {
  156.     f.sqlNull(true); // oracle-ish
  157.     return;
  158. }
  159. if (len > length - pos + 1)
  160.     len = length - pos + 1;
  161. switch (t1.type()) {
  162. case SqlType::Char:
  163.     f.sqlChar(data + (pos - 1), len);
  164.     break;
  165. case SqlType::Varchar:
  166.     f.sqlVarchar(data + (pos - 1), len);
  167.     break;
  168. default:
  169.     ctx_assert(false);
  170.     break;
  171. }
  172.     } else if (fc == Expr_func::Count) {
  173. ctx_assert(narg == 0 || narg == 1);
  174. if (ctl.m_postEval)
  175.     return;
  176. if (narg == 1) {
  177.     const SqlField& f1 = args[1]->getData().sqlField();
  178.     if (f1.sqlNull())
  179. return;
  180. }
  181. f.sqlBigint(++acc.m_count);
  182.     } else if (fc == Expr_func::Min) {
  183. ctx_assert(narg == 1);
  184. if (ctl.m_postEval)
  185.     return;
  186. const SqlField& f1 = args[1]->getData().sqlField();
  187. if (f1.sqlNull())
  188.     return;
  189. if (f.sqlNull() || f1.less(f))
  190.     f1.copy(ctx, f);
  191.     } else if (fc == Expr_func::Max) {
  192. ctx_assert(narg == 1);
  193. if (ctl.m_postEval)
  194.     return;
  195. const SqlField& f1 = args[1]->getData().sqlField();
  196. if (f1.sqlNull())
  197.     return;
  198. if (f.sqlNull() || f.less(f1))
  199.     f1.copy(ctx, f);
  200.     } else if (fc == Expr_func::Sum) {
  201. ctx_assert(narg == 1);
  202. if (ctl.m_postEval)
  203.     return;
  204. const SqlType& t1 = args[1]->getCode().sqlSpec().sqlType();
  205. const SqlField& f1 = args[1]->getData().sqlField();
  206. if (f1.sqlNull())
  207.     return;
  208. switch (t.type()) {
  209. case SqlType::Bigint:
  210.     switch (t1.type()) {
  211.     case SqlType::Integer:
  212. acc.m_bigint += f1.sqlInteger();
  213. break;
  214.     case SqlType::Bigint:
  215. acc.m_bigint += f1.sqlBigint();
  216. break;
  217.     default:
  218. ctx_assert(false);
  219. break;
  220.     }
  221.     f.sqlBigint(acc.m_bigint);
  222.     break;
  223. case SqlType::Double:
  224.     switch (t1.type()) {
  225.     case SqlType::Real:
  226. acc.m_double += f1.sqlReal();
  227. break;
  228.     case SqlType::Double:
  229. acc.m_double += f1.sqlDouble();
  230. break;
  231.     default:
  232. ctx_assert(false);
  233. break;
  234.     }
  235.     f.sqlDouble(acc.m_double);
  236.     break;
  237. default:
  238.     ctx_assert(false);
  239.     break;
  240. }
  241.     } else if (fc == Expr_func::Avg) {
  242. ctx_assert(narg == 1);
  243. if (ctl.m_postEval)
  244.     return;
  245. const SqlType& t1 = args[1]->getCode().sqlSpec().sqlType();
  246. const SqlField& f1 = args[1]->getData().sqlField();
  247. if (f1.sqlNull())
  248.     return;
  249. switch (t1.type()) {
  250. case SqlType::Smallint:
  251.     acc.m_bigint += f1.sqlSmallint();
  252.     break;
  253. case SqlType::Integer:
  254.     acc.m_bigint += f1.sqlInteger();
  255.     break;
  256. case SqlType::Bigint:
  257.     acc.m_bigint += f1.sqlBigint();
  258.     break;
  259. case SqlType::Real:
  260.     acc.m_double += f1.sqlReal();
  261.     break;
  262. case SqlType::Double:
  263.     acc.m_double += f1.sqlDouble();
  264.     break;
  265. default:
  266.     ctx_assert(false);
  267.     break;
  268. }
  269. f.sqlDouble(acc.m_double / (SqlDouble)++acc.m_count);
  270.     } else if (fc == Expr_func::Rownum) {
  271. ctx_assert(narg == 0);
  272. if (! ctl.m_postEval)
  273.     f.sqlBigint(1 + acc.m_count);
  274. else
  275.     acc.m_count++;
  276.     } else if (fc == Expr_func::Sysdate) {
  277. ctx_assert(narg == 0);
  278. if (ctl.m_postEval)
  279.     return;
  280. f.sqlDatetime(acc.m_sysdate);
  281.     } else {
  282. ctx_assert(false);
  283.     }
  284. }