~drizzle-trunk/drizzle/development

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
/* Copyright (C) 1995-2002 MySQL AB

   This program is free software; you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
   the Free Software Foundation; version 2 of the License.

   This program is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU General Public License for more details.

   You should have received a copy of the GNU General Public License
   along with this program; if not, write to the Free Software
   Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA */

/**********************************************************************
This file contains the implementation of error and warnings related

  - Whenever an error or warning occurred, it pushes it to a warning list
    that the user can retrieve with SHOW WARNINGS or SHOW ERRORS.

  - For each statement, we return the number of warnings generated from this
    command.  Note that this can be different from @@warning_count as
    we reset the warning list only for questions that uses a table.
    This is done to allow on to do:
    INSERT ...;
    SELECT @@warning_count;
    SHOW WARNINGS;
    (If we would reset after each command, we could not retrieve the number
     of warnings)

  - When client requests the information using SHOW command, then
    server processes from this list and returns back in the form of
    resultset.

    Supported syntaxes:

    SHOW [COUNT(*)] ERRORS [LIMIT [offset,] rows]
    SHOW [COUNT(*)] WARNINGS [LIMIT [offset,] rows]
    SELECT @@warning_count, @@error_count;

***********************************************************************/

#include "config.h"

#include <cstdio>
#include <stdarg.h>

#include <drizzled/session.h>
#include <drizzled/sql_base.h>
#include <drizzled/item/empty_string.h>
#include <drizzled/item/return_int.h>
#include <drizzled/plugin/client.h>

using namespace std;

namespace drizzled
{

/*
  Store a new message in an error object

  This is used to in group_concat() to register how many warnings we actually
  got after the query has been executed.
*/
void DRIZZLE_ERROR::set_msg(Session *session, const char *msg_arg)
{
  msg= session->warn_root.strdup_root(msg_arg);
}

/*
  Reset all warnings for the thread

  SYNOPSIS
    drizzle_reset_errors()
    session			Thread handle
    force               Reset warnings even if it has been done before

  IMPLEMENTATION
    Don't reset warnings if this has already been called for this query.
    This may happen if one gets a warning during the parsing stage,
    in which case push_warnings() has already called this function.
*/

void drizzle_reset_errors(Session *session, bool force)
{
  if (session->getQueryId() != session->getWarningQueryId() || force)
  {
    session->setWarningQueryId(session->getQueryId());
    session->warn_root.free_root(MYF(0));
    memset(session->warn_count, 0, sizeof(session->warn_count));
    if (force)
      session->total_warn_count= 0;
    session->warn_list.empty();
    session->row_count= 1; // by default point to row 1
  }
  return;
}


/*
  Push the warning/error to error list if there is still room in the list

  SYNOPSIS
    push_warning()
    session			Thread handle
    level		Severity of warning (note, warning, error ...)
    code		Error number
    msg			Clear error message

  RETURN
    pointer on DRIZZLE_ERROR object
*/

DRIZZLE_ERROR *push_warning(Session *session, DRIZZLE_ERROR::enum_warning_level level,
                          uint32_t code, const char *msg)
{
  DRIZZLE_ERROR *err= 0;

  if (level == DRIZZLE_ERROR::WARN_LEVEL_NOTE &&
      !(session->options & OPTION_SQL_NOTES))
    return(0);

  if (session->getQueryId() != session->getWarningQueryId())
    drizzle_reset_errors(session, 0);
  session->got_warning= 1;

  /* Abort if we are using strict mode and we are not using IGNORE */
  if ((int) level >= (int) DRIZZLE_ERROR::WARN_LEVEL_WARN &&
      session->really_abort_on_warning())
  {
    /* Avoid my_message() calling push_warning */
    bool no_warnings_for_error= session->no_warnings_for_error;

    session->no_warnings_for_error= 1;

    session->setKilled(Session::KILL_BAD_DATA);
    my_message(code, msg, MYF(0));

    session->no_warnings_for_error= no_warnings_for_error;
    /* Store error in error list (as my_message() didn't do it) */
    level= DRIZZLE_ERROR::WARN_LEVEL_ERROR;
  }

  if (session->handle_error(code, msg, level))
    return NULL;

  if (session->warn_list.elements < session->variables.max_error_count)
  {
    /* We have to use warn_root, as mem_root is freed after each query */
    if ((err= new (&session->warn_root) DRIZZLE_ERROR(session, code, level, msg)))
      session->warn_list.push_back(err, &session->warn_root);
  }
  session->warn_count[(uint32_t) level]++;
  session->total_warn_count++;
  return(err);
}

/*
  Push the warning/error to error list if there is still room in the list

  SYNOPSIS
    push_warning_printf()
    session			Thread handle
    level		Severity of warning (note, warning, error ...)
    code		Error number
    msg			Clear error message
*/

void push_warning_printf(Session *session, DRIZZLE_ERROR::enum_warning_level level,
			 uint32_t code, const char *format, ...)
{
  va_list args;
  char    warning[ERRMSGSIZE+20];

  va_start(args,format);
  vsnprintf(warning, sizeof(warning), format, args);
  va_end(args);
  push_warning(session, level, code, warning);
  return;
}


/*
  Send all notes, errors or warnings to the client in a result set

  SYNOPSIS
    mysqld_show_warnings()
    session			Thread handler
    levels_to_show	Bitmap for which levels to show

  DESCRIPTION
    Takes into account the current LIMIT

  RETURN VALUES
    false ok
    true  Error sending data to client
*/

const LEX_STRING warning_level_names[]=
{
  { C_STRING_WITH_LEN("Note") },
  { C_STRING_WITH_LEN("Warning") },
  { C_STRING_WITH_LEN("Error") },
  { C_STRING_WITH_LEN("?") }
};

bool mysqld_show_warnings(Session *session,
                          bitset<DRIZZLE_ERROR::NUM_ERRORS> &levels_to_show)
{
  List<Item> field_list;

  field_list.push_back(new Item_empty_string("Level", 7));
  field_list.push_back(new Item_return_int("Code",4, DRIZZLE_TYPE_LONG));
  field_list.push_back(new Item_empty_string("Message",DRIZZLE_ERRMSG_SIZE));

  if (session->client->sendFields(&field_list))
    return true;

  DRIZZLE_ERROR *err;
  Select_Lex *sel= &session->lex->select_lex;
  Select_Lex_Unit *unit= &session->lex->unit;
  ha_rows idx= 0;

  unit->set_limit(sel);

  List_iterator_fast<DRIZZLE_ERROR> it(session->warn_list);
  while ((err= it++))
  {
    /* Skip levels that the user is not interested in */
    if (! levels_to_show.test(err->level))
      continue;
    if (++idx <= unit->offset_limit_cnt)
      continue;
    if (idx > unit->select_limit_cnt)
      break;
    session->client->store(warning_level_names[err->level].str,
		           warning_level_names[err->level].length);
    session->client->store((uint32_t) err->code);
    session->client->store(err->msg, strlen(err->msg));
    if (session->client->flush())
      return(true);
  }
  session->my_eof();
  return(false);
}

} /* namespace drizzled */