← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to plugin/console/console.cc

  • Committer: Eric Day
  • Date: 2009-08-27 19:21:43 UTC
  • mto: This revision was merged to the branch mainline in revision 1131.
  • Revision ID: eday@oddments.org-20090827192143-t750y6bepooi0dt8
Added console plugin.

Show diffs side-by-side

added added

removed removed

Lines of Context:
plugin/console/console.cc
 
1
/* Copyright (C) 2009 Sun Microsystems
 
2
 
 
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; version 2 of the License.
 
6
 
 
7
   This program is distributed in the hope that it will be useful,
 
8
   but WITHOUT ANY WARRANTY; without even the implied warranty of
 
9
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
10
   GNU General Public License for more details.
 
11
 
 
12
   You should have received a copy of the GNU General Public License
 
13
   along with this program; if not, write to the Free Software
 
14
   Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA */
 
15
 
 
16
#include <drizzled/server_includes.h>
 
17
#include <drizzled/gettext.h>
 
18
#include <drizzled/plugin/listen_tcp.h>
 
19
#include <drizzled/plugin/client.h>
 
20
 
 
21
#include <iostream>
 
22
 
 
23
using namespace std;
 
24
using namespace drizzled;
 
25
 
 
26
static bool enable= false;
 
27
static bool debug= false;
 
28
 
 
29
class ClientConsole: public plugin::Client
 
30
{
 
31
  bool is_dead;
 
32
  uint32_t column;
 
33
  uint32_t max_column;
 
34
 
 
35
public:
 
36
  ClientConsole():
 
37
    is_dead(false),
 
38
    column(0),
 
39
    max_column(0)
 
40
  {}
 
41
 
 
42
  virtual void printDebug(const char *message)
 
43
  {
 
44
    if (debug)
 
45
      cout << "CONSOLE: " << message << endl;
 
46
  }
 
47
 
 
48
  virtual int getFileDescriptor(void)
 
49
  {
 
50
    printDebug("getFileDescriptor");
 
51
    return 0;
 
52
  }
 
53
 
 
54
  virtual bool isConnected(void)
 
55
  {
 
56
    printDebug("isConnected");
 
57
    return true;
 
58
  }
 
59
 
 
60
  virtual bool isReading(void)
 
61
  {
 
62
    printDebug("isReading");
 
63
    return false;
 
64
  }
 
65
 
 
66
  virtual bool isWriting(void)
 
67
  {
 
68
    printDebug("isWriting");
 
69
    return false;
 
70
  }
 
71
 
 
72
  virtual bool flush(void)
 
73
  {
 
74
    printDebug("flush");
 
75
    return false;
 
76
  }
 
77
 
 
78
  virtual void close(void)
 
79
  {
 
80
    printDebug("close");
 
81
    is_dead= true;
 
82
  }
 
83
 
 
84
  virtual bool authenticate(void)
 
85
  {
 
86
    printDebug("authenticate");
 
87
    return true;
 
88
  }
 
89
 
 
90
  virtual bool readCommand(char **packet, uint32_t *packet_length)
 
91
  {
 
92
    char buffer[8192];
 
93
    printf("drizzled> ");
 
94
    fflush(stdout);
 
95
 
 
96
    if (is_dead)
 
97
      return false;
 
98
 
 
99
    if (fgets(buffer, 8192, stdin) == NULL ||!strcasecmp(buffer, "quit\n") ||
 
100
        !strcasecmp(buffer, "exit\n"))
 
101
    {
 
102
      is_dead= true;
 
103
      *packet_length= 1;
 
104
      *packet= (char *)malloc(*packet_length);
 
105
      (*packet)[0]= COM_SHUTDOWN;
 
106
      return true;
 
107
    }
 
108
 
 
109
    *packet_length= strlen(buffer);
 
110
    *packet= (char *)malloc(*packet_length);
 
111
    (*packet)[0]= COM_QUERY;
 
112
    memcpy(*packet + 1, buffer, *packet_length - 1);
 
113
 
 
114
    return true;
 
115
  }
 
116
 
 
117
  virtual void sendOK(void)
 
118
  {
 
119
    cout << "OK" << endl;
 
120
  }
 
121
 
 
122
  virtual void sendEOF(void)
 
123
  {
 
124
    printDebug("sendEOF");
 
125
  }
 
126
 
 
127
  virtual void sendError(uint32_t sql_errno, const char *err)
 
128
  {
 
129
    cout << "Error: " << sql_errno << " " << err << endl;
 
130
  }
 
131
 
 
132
  virtual bool sendFields(List<Item> *list)
 
133
  {
 
134
    List_iterator_fast<Item> it(*list);
 
135
    Item *item;
 
136
 
 
137
    column= 0;
 
138
    max_column= 0;
 
139
 
 
140
    while ((item=it++))
 
141
    {
 
142
      SendField field;
 
143
      item->make_field(&field);
 
144
      cout << field.col_name << "\t";
 
145
      max_column++;
 
146
    }
 
147
 
 
148
    cout << endl;
 
149
 
 
150
    return false;
 
151
  }
 
152
 
 
153
  virtual void checkRowEnd(void)
 
154
  {
 
155
    if (++column % max_column == 0)
 
156
      cout << endl;
 
157
  }
 
158
 
 
159
  using Client::store;
 
160
 
 
161
  virtual bool store(Field *from)
 
162
  {
 
163
    if (from->is_null())
 
164
      return store();
 
165
 
 
166
    char buff[MAX_FIELD_WIDTH];
 
167
    String str(buff, sizeof(buff), &my_charset_bin);
 
168
    from->val_str(&str);
 
169
    return store(str.ptr(), str.length());
 
170
  }
 
171
 
 
172
  virtual bool store(void)
 
173
  {
 
174
    cout << "NULL" << "\t";
 
175
    checkRowEnd();
 
176
    return false;
 
177
  }
 
178
 
 
179
  virtual bool store(int32_t from)
 
180
  {
 
181
    cout << from << "\t";
 
182
    checkRowEnd();
 
183
    return false;
 
184
  }
 
185
 
 
186
  virtual bool store(uint32_t from)
 
187
  {
 
188
    cout << from << "\t";
 
189
    checkRowEnd();
 
190
    return false;
 
191
  }
 
192
 
 
193
  virtual bool store(int64_t from)
 
194
  {
 
195
    cout << from << "\t";
 
196
    checkRowEnd();
 
197
    return false;
 
198
  }
 
199
 
 
200
  virtual bool store(uint64_t from)
 
201
  {
 
202
    cout << from << "\t";
 
203
    checkRowEnd();
 
204
    return false;
 
205
  }
 
206
 
 
207
  virtual bool store(double from, uint32_t decimals, String *buffer)
 
208
  {
 
209
    buffer->set_real(from, decimals, &my_charset_bin);
 
210
    return store(buffer->ptr(), buffer->length());
 
211
  }
 
212
 
 
213
  virtual bool store(const DRIZZLE_TIME *tm)
 
214
  {
 
215
    char buff[40];
 
216
    uint32_t length;
 
217
    uint32_t day;
 
218
 
 
219
    switch (tm->time_type)
 
220
    {
 
221
    case DRIZZLE_TIMESTAMP_DATETIME:
 
222
      length= sprintf(buff, "%04d-%02d-%02d %02d:%02d:%02d",
 
223
                      (int) tm->year,
 
224
                      (int) tm->month,
 
225
                      (int) tm->day,
 
226
                      (int) tm->hour,
 
227
                      (int) tm->minute,
 
228
                      (int) tm->second);
 
229
      if (tm->second_part)
 
230
        length+= sprintf(buff+length, ".%06d", (int)tm->second_part);
 
231
      break;
 
232
 
 
233
    case DRIZZLE_TIMESTAMP_DATE:
 
234
      length= sprintf(buff, "%04d-%02d-%02d",
 
235
                      (int) tm->year,
 
236
                      (int) tm->month,
 
237
                      (int) tm->day);
 
238
      break;
 
239
 
 
240
    case DRIZZLE_TIMESTAMP_TIME:
 
241
      day= (tm->year || tm->month) ? 0 : tm->day;
 
242
      length= sprintf(buff, "%s%02ld:%02d:%02d", tm->neg ? "-" : "",
 
243
                      (long) day*24L+(long) tm->hour, (int) tm->minute,
 
244
                      (int) tm->second);
 
245
      if (tm->second_part)
 
246
        length+= sprintf(buff+length, ".%06d", (int)tm->second_part);
 
247
      break;
 
248
 
 
249
    case DRIZZLE_TIMESTAMP_NONE:
 
250
    case DRIZZLE_TIMESTAMP_ERROR:
 
251
    default:
 
252
      assert(0);
 
253
      return false;
 
254
    }
 
255
 
 
256
    return store(buff);
 
257
  }
 
258
 
 
259
  virtual bool store(const char *from, size_t length)
 
260
  {
 
261
    printf("%.*s\t", (uint32_t)length, from);
 
262
    checkRowEnd();
 
263
    return false;
 
264
  }
 
265
 
 
266
  virtual bool haveMoreData(void)
 
267
  {
 
268
    printDebug("haveMoreData");
 
269
    return false;
 
270
  }
 
271
 
 
272
  virtual bool haveError(void)
 
273
  {
 
274
    printDebug("haveError");
 
275
    return false;
 
276
  }
 
277
 
 
278
  virtual bool wasAborted(void)
 
279
  {
 
280
    printDebug("wasAborted");
 
281
    return false;
 
282
  }
 
283
};
 
284
 
 
285
class ListenConsole: public plugin::Listen
 
286
{
 
287
  int pipe_fds[2];
 
288
 
 
289
public:
 
290
  ListenConsole()
 
291
  {
 
292
    pipe_fds[0]= -1;
 
293
  }
 
294
 
 
295
  virtual ~ListenConsole()
 
296
  {
 
297
    if (pipe_fds[0] != -1)
 
298
    {
 
299
      close(pipe_fds[0]);
 
300
      close(pipe_fds[1]);
 
301
    }
 
302
  }
 
303
 
 
304
  virtual bool getFileDescriptors(std::vector<int> &fds)
 
305
  {
 
306
    if (debug)
 
307
      enable= true;
 
308
 
 
309
    if (!enable)
 
310
      return false;
 
311
 
 
312
    if (pipe(pipe_fds) == -1)
 
313
    {
 
314
      errmsg_printf(ERRMSG_LVL_ERROR, _("pipe() failed with errno %d"), errno);
 
315
      return true;
 
316
    }
 
317
 
 
318
    fds.push_back(pipe_fds[0]);
 
319
    assert(write(pipe_fds[1], "\0", 1) == 1);
 
320
    return false;
 
321
  }
 
322
 
 
323
  virtual drizzled::plugin::Client *getClient(int fd)
 
324
  {
 
325
    char buffer[1];
 
326
    assert(read(fd, buffer, 1) == 1);
 
327
    return new ClientConsole;
 
328
  }
 
329
};
 
330
 
 
331
static ListenConsole listen_obj;
 
332
 
 
333
static int init(drizzled::plugin::Registry &registry)
 
334
{
 
335
  registry.listen.add(listen_obj);
 
336
  return 0;
 
337
}
 
338
 
 
339
static int deinit(drizzled::plugin::Registry &registry)
 
340
{
 
341
  registry.listen.remove(listen_obj);
 
342
  return 0;
 
343
}
 
344
 
 
345
static DRIZZLE_SYSVAR_BOOL(enable, enable, PLUGIN_VAR_NOCMDARG,
 
346
                           N_("Enable the console."), NULL, NULL, false);
 
347
 
 
348
static DRIZZLE_SYSVAR_BOOL(debug, debug, PLUGIN_VAR_NOCMDARG,
 
349
                           N_("Turn on extra debugging."), NULL, NULL, false);
 
350
 
 
351
static struct st_mysql_sys_var* vars[]= {
 
352
  DRIZZLE_SYSVAR(enable),
 
353
  DRIZZLE_SYSVAR(debug),
 
354
  NULL
 
355
};
 
356
 
 
357
drizzle_declare_plugin(console)
 
358
{
 
359
  "console",
 
360
  "0.1",
 
361
  "Eric Day",
 
362
  "Console Client",
 
363
  PLUGIN_LICENSE_BSD,
 
364
  init,   /* Plugin Init */
 
365
  deinit, /* Plugin Deinit */
 
366
  NULL,   /* status variables */
 
367
  vars,   /* system variables */
 
368
  NULL    /* config options */
 
369
}
 
370
drizzle_declare_plugin_end;