← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to examples/server.c

  • Committer: Stewart Smith
  • Date: 2008-12-18 23:54:47 UTC
  • mto: This revision was merged to the branch mainline in revision 719.
  • Revision ID: stewart@flamingspork.com-20081218235447-qf6dofgd6guwefo2
fix RENAME TABLE

(problem was missing hton to mysql_rename_table)

If engine implements table_exists_in_engine we get the hton from there.

Else, we reintroduce the mysql_frm_type hackery (temporarily, it *will* go away again with FRM removal).

also fix rename test: no views, plus current FLUSH LOCKS bug.

Show diffs side-by-side

added added

removed removed

Lines of Context:
examples/server.c
1
 
/*
2
 
 * Drizzle Client & Protocol Library
3
 
 *
4
 
 * Copyright (C) 2008 Eric Day (eday@oddments.org)
5
 
 * All rights reserved.
6
 
 *
7
 
 * Redistribution and use in source and binary forms, with or without
8
 
 * modification, are permitted provided that the following conditions are
9
 
 * met:
10
 
 *
11
 
 *     * Redistributions of source code must retain the above copyright
12
 
 * notice, this list of conditions and the following disclaimer.
13
 
 *
14
 
 *     * Redistributions in binary form must reproduce the above
15
 
 * copyright notice, this list of conditions and the following disclaimer
16
 
 * in the documentation and/or other materials provided with the
17
 
 * distribution.
18
 
 *
19
 
 *     * The names of its contributors may not be used to endorse or
20
 
 * promote products derived from this software without specific prior
21
 
 * written permission.
22
 
 *
23
 
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
24
 
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
25
 
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
26
 
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
27
 
 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
28
 
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
29
 
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
30
 
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
31
 
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
32
 
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
33
 
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
34
 
 *
35
 
 */
36
 
 
37
 
 
38
 
#include "config.h"
39
 
 
40
 
#include <errno.h>
41
 
#include <stdio.h>
42
 
#include <stdlib.h>
43
 
#include <unistd.h>
44
 
 
45
 
#include <libdrizzle/drizzle_server.h>
46
 
 
47
 
#define DRIZZLE_FIELD_MAX 32
48
 
#define DRIZZLE_RESULT_ROWS 20
49
 
 
50
 
#define DRIZZLE_RETURN_CHECK(__ret, __function, __drizzle) \
51
 
{ \
52
 
  if ((__ret) != DRIZZLE_RETURN_OK) \
53
 
    DRIZZLE_RETURN_ERROR(__function, __drizzle) \
54
 
}
55
 
 
56
 
#define DRIZZLE_RETURN_ERROR(__function, __drizzle) \
57
 
{ \
58
 
  printf(__function ":%s\n", drizzle_error(__drizzle)); \
59
 
  return; \
60
 
}
61
 
 
62
 
static void server(drizzle_st *drizzle, drizzle_con_st *con,
63
 
                   drizzle_result_st *result, drizzle_column_st *column);
64
 
 
65
 
int main(int argc, char *argv[])
66
 
{
67
 
  int c;
68
 
  uint32_t count= 0;
69
 
  const char *host= NULL;
70
 
  bool mysql= false;
71
 
  in_port_t port= 0;
72
 
  drizzle_verbose_t verbose= DRIZZLE_VERBOSE_NEVER;
73
 
  drizzle_return_t ret;
74
 
  drizzle_st drizzle;
75
 
  drizzle_con_st *con_listen= (drizzle_con_st*)malloc(sizeof(drizzle_con_st));
76
 
  drizzle_con_st *con= (drizzle_con_st*)malloc(sizeof(drizzle_con_st));
77
 
  drizzle_result_st result;
78
 
  drizzle_column_st column;
79
 
 
80
 
  while((c = getopt(argc, argv, "c:h:mp:v")) != -1)
81
 
  {
82
 
    switch(c)
83
 
    {
84
 
    case 'c':
85
 
      count= (uint32_t)atoi(optarg);
86
 
      break;
87
 
 
88
 
    case 'h':
89
 
      host= optarg;
90
 
      break;
91
 
 
92
 
    case 'm':
93
 
      mysql= true;
94
 
      break;
95
 
 
96
 
    case 'p':
97
 
      port= (in_port_t)atoi(optarg);
98
 
      break;
99
 
 
100
 
    case 'v':
101
 
      switch(verbose)
102
 
      {
103
 
      case DRIZZLE_VERBOSE_NEVER:
104
 
        verbose= DRIZZLE_VERBOSE_FATAL;
105
 
        break;
106
 
      case DRIZZLE_VERBOSE_FATAL:
107
 
        verbose= DRIZZLE_VERBOSE_ERROR;
108
 
        break;
109
 
      case DRIZZLE_VERBOSE_ERROR:
110
 
        verbose= DRIZZLE_VERBOSE_INFO;
111
 
        break;
112
 
      case DRIZZLE_VERBOSE_INFO:
113
 
        verbose= DRIZZLE_VERBOSE_DEBUG;
114
 
        break;
115
 
      case DRIZZLE_VERBOSE_DEBUG:
116
 
        verbose= DRIZZLE_VERBOSE_CRAZY;
117
 
        break;
118
 
      case DRIZZLE_VERBOSE_CRAZY:
119
 
      case DRIZZLE_VERBOSE_MAX:
120
 
        break;
121
 
      }
122
 
      break;
123
 
 
124
 
    default:
125
 
      printf("\nusage: %s [-c <count>] [-h <host>] [-m] [-p <port>] [-v]\n",
126
 
             argv[0]);
127
 
      printf("\t-c <count> - Number of connections to accept before exiting\n");
128
 
      printf("\t-h <host>  - Host to listen on\n");
129
 
      printf("\t-m         - Use the MySQL protocol\n");
130
 
      printf("\t-p <port>  - Port to listen on\n");
131
 
      printf("\t-v         - Increase verbosity level\n");
132
 
      return 1;
133
 
    }
134
 
  }
135
 
 
136
 
  if (drizzle_create(&drizzle) == NULL)
137
 
  {
138
 
    printf("drizzle_create:NULL\n");
139
 
    return 1;
140
 
  }
141
 
 
142
 
  drizzle_add_options(&drizzle, DRIZZLE_FREE_OBJECTS);
143
 
  drizzle_set_verbose(&drizzle, verbose);
144
 
 
145
 
  if (drizzle_con_create(&drizzle, con_listen) == NULL)
146
 
  {
147
 
    printf("drizzle_con_create:NULL\n");
148
 
    return 1;
149
 
  }
150
 
 
151
 
  drizzle_con_add_options(con_listen, DRIZZLE_CON_LISTEN);
152
 
  drizzle_con_set_tcp(con_listen, host, port);
153
 
 
154
 
  if (mysql)
155
 
    drizzle_con_add_options(con_listen, DRIZZLE_CON_MYSQL);
156
 
 
157
 
  if (drizzle_con_listen(con_listen) != DRIZZLE_RETURN_OK)
158
 
  {
159
 
    printf("drizzle_con_listen:%s\n", drizzle_error(&drizzle));
160
 
    return 1;
161
 
  }
162
 
 
163
 
  while (1)
164
 
  {
165
 
    (void)drizzle_con_accept(&drizzle, con, &ret);
166
 
    if (ret != DRIZZLE_RETURN_OK)
167
 
    {
168
 
      printf("drizzle_con_accept:%s\n", drizzle_error(&drizzle));
169
 
      return 1;
170
 
    }
171
 
 
172
 
    server(&drizzle, con, &result, &column);
173
 
 
174
 
    drizzle_con_free(con);
175
 
 
176
 
    if (count > 0)
177
 
    {
178
 
      count--;
179
 
 
180
 
      if (count == 0)
181
 
        break;
182
 
    }
183
 
  }
184
 
 
185
 
  drizzle_con_free(con_listen);
186
 
  drizzle_free(&drizzle);
187
 
 
188
 
  free(con);
189
 
  free(con_listen);
190
 
 
191
 
  return 0;
192
 
}
193
 
 
194
 
static void server(drizzle_st *drizzle, drizzle_con_st *con,
195
 
                   drizzle_result_st *result, drizzle_column_st *column)
196
 
{
197
 
  drizzle_return_t ret;
198
 
  drizzle_command_t command;
199
 
  uint8_t *data= NULL;
200
 
  size_t total;
201
 
  char *field[2];
202
 
  char field1[DRIZZLE_FIELD_MAX];
203
 
  char field2[DRIZZLE_FIELD_MAX];
204
 
  size_t size[2];
205
 
  uint64_t x;
206
 
 
207
 
  field[0]= field1;
208
 
  field[1]= field2;
209
 
 
210
 
  /* Handshake packets. */
211
 
  drizzle_con_set_protocol_version(con, 10);
212
 
  drizzle_con_set_server_version(con, "libdrizzle example 1.2.3");
213
 
  drizzle_con_set_thread_id(con, 1);
214
 
  drizzle_con_set_scramble(con, (const uint8_t *)"ABCDEFGHIJKLMNOPQRST");
215
 
  drizzle_con_set_capabilities(con, DRIZZLE_CAPABILITIES_NONE);
216
 
  drizzle_con_set_charset(con, 8);
217
 
  drizzle_con_set_status(con, DRIZZLE_CON_STATUS_NONE);
218
 
  drizzle_con_set_max_packet_size(con, DRIZZLE_MAX_PACKET_SIZE);
219
 
 
220
 
  ret= drizzle_handshake_server_write(con);
221
 
  DRIZZLE_RETURN_CHECK(ret, "drizzle_handshake_server_write", drizzle)
222
 
 
223
 
  ret= drizzle_handshake_client_read(con);
224
 
  DRIZZLE_RETURN_CHECK(ret, "drizzle_handshake_client_read", drizzle)
225
 
 
226
 
  if (drizzle_result_create(con, result) == NULL)
227
 
    DRIZZLE_RETURN_ERROR("drizzle_result_create", drizzle)
228
 
 
229
 
  ret= drizzle_result_write(con, result, true);
230
 
  DRIZZLE_RETURN_CHECK(ret, "drizzle_result_write", drizzle)
231
 
 
232
 
  /* Command loop. */
233
 
  while (1)
234
 
  {
235
 
    drizzle_result_free(result);
236
 
    if (data != NULL)
237
 
      free(data);
238
 
 
239
 
    data= (uint8_t *)drizzle_con_command_buffer(con, &command, &total, &ret);
240
 
    if (ret == DRIZZLE_RETURN_LOST_CONNECTION ||
241
 
        (ret == DRIZZLE_RETURN_OK && command == DRIZZLE_COMMAND_QUIT))
242
 
    {
243
 
      if (data != NULL)
244
 
        free(data);
245
 
      return;
246
 
    }
247
 
    DRIZZLE_RETURN_CHECK(ret, "drizzle_con_command_buffer", drizzle)
248
 
 
249
 
    if (drizzle_result_create(con, result) == NULL)
250
 
      DRIZZLE_RETURN_ERROR("drizzle_result_create", drizzle)
251
 
 
252
 
    if (command != DRIZZLE_COMMAND_QUERY)
253
 
    {
254
 
      ret= drizzle_result_write(con, result, true);
255
 
      DRIZZLE_RETURN_CHECK(ret, "drizzle_result_write", drizzle)
256
 
      continue;
257
 
    }
258
 
 
259
 
    drizzle_result_set_column_count(result, 2);
260
 
 
261
 
    ret= drizzle_result_write(con, result, false);
262
 
    DRIZZLE_RETURN_CHECK(ret, "drizzle_result_write", drizzle)
263
 
 
264
 
    /* Columns. */
265
 
    if (drizzle_column_create(result, column) == NULL)
266
 
      DRIZZLE_RETURN_ERROR("drizzle_column_create", drizzle)
267
 
 
268
 
    drizzle_column_set_catalog(column, "default");
269
 
    drizzle_column_set_db(column, "drizzle_test_db");
270
 
    drizzle_column_set_table(column, "drizzle_test_table");
271
 
    drizzle_column_set_orig_table(column, "drizzle_test_table");
272
 
    drizzle_column_set_name(column, "test_column_1");
273
 
    drizzle_column_set_orig_name(column, "test_column_1");
274
 
    drizzle_column_set_charset(column, 8);
275
 
    drizzle_column_set_size(column, DRIZZLE_FIELD_MAX);
276
 
    drizzle_column_set_type(column, DRIZZLE_COLUMN_TYPE_VARCHAR);
277
 
 
278
 
    ret= drizzle_column_write(result, column);
279
 
    DRIZZLE_RETURN_CHECK(ret, "drizzle_column_write", drizzle)
280
 
 
281
 
    drizzle_column_set_name(column, "test_column_2");
282
 
    drizzle_column_set_orig_name(column, "test_column_2");
283
 
 
284
 
    ret= drizzle_column_write(result, column);
285
 
    DRIZZLE_RETURN_CHECK(ret, "drizzle_column_write", drizzle)
286
 
 
287
 
    drizzle_column_free(column);
288
 
 
289
 
    drizzle_result_set_eof(result, true);
290
 
 
291
 
    ret= drizzle_result_write(con, result, false);
292
 
    DRIZZLE_RETURN_CHECK(ret, "drizzle_result_write", drizzle)
293
 
 
294
 
    /* Rows. */
295
 
    for (x= 0; x < DRIZZLE_RESULT_ROWS; x++)
296
 
    {
297
 
      size[0]= (size_t)snprintf(field[0], DRIZZLE_FIELD_MAX,
298
 
                                "field %" PRIu64 "-1", x);
299
 
      if (size[0] >= DRIZZLE_FIELD_MAX)
300
 
        size[0]= DRIZZLE_FIELD_MAX - 1;
301
 
 
302
 
      size[1]= (size_t)snprintf(field[1], DRIZZLE_FIELD_MAX,
303
 
                                "field %" PRIu64 "-2", x);
304
 
      if (size[1] >= DRIZZLE_FIELD_MAX)
305
 
        size[1]= DRIZZLE_FIELD_MAX - 1;
306
 
 
307
 
      /* This is needed for MySQL and old Drizzle protocol. */
308
 
      drizzle_result_calc_row_size(result, (drizzle_field_t *)field, size);
309
 
 
310
 
      ret= drizzle_row_write(result);
311
 
      DRIZZLE_RETURN_CHECK(ret, "drizzle_row_write", drizzle)
312
 
 
313
 
      /* Fields. */
314
 
      ret= drizzle_field_write(result, (drizzle_field_t)field[0], size[0],
315
 
                               size[0]);
316
 
      DRIZZLE_RETURN_CHECK(ret, "drizzle_field_write", drizzle)
317
 
 
318
 
      ret= drizzle_field_write(result, (drizzle_field_t)field[1], size[1],
319
 
                               size[1]);
320
 
      DRIZZLE_RETURN_CHECK(ret, "drizzle_field_write", drizzle)
321
 
    }
322
 
 
323
 
    ret= drizzle_result_write(con, result, true);
324
 
    DRIZZLE_RETURN_CHECK(ret, "drizzle_result_write", drizzle)
325
 
  }
326
 
}