« back to all changes in this revision
Viewing changes to client/drizzledump_data.cc
- browse files at revision 1751.4.19
- compare with another revision
- compare with revision 2241.3.11
- download diff
- download tarball
- view history from revision 1751.4.19
- Committer: Andrew Hutchings
- Date: 2010-09-22 19:58:16 UTC
- mto: (1792.1.1 build)
- mto: This revision was merged to the branch mainline in revision 1793.
- Revision ID: andrew@linuxjedi.co.uk-20100922195816-uc3aud8va2sxponn
Put drizzle and mysql processes in seperate classes/files
added
removed
client/drizzledump_data.cc
1
/* -*- mode: c++; c-basic-offset: 2; indent-tabs-mode: nil; -*-
2
* vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
3
*
4
* Copyright (C) 2010 Andrew Hutchings
5
*
6
* This program is free software; you can redistribute it and/or modify
7
* it under the terms of the GNU General Public License as published by
8
* the Free Software Foundation; version 2 of the License.
9
*
10
* This program is distributed in the hope that it will be useful,
11
* but WITHOUT ANY WARRANTY; without even the implied warranty of
12
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13
* GNU General Public License for more details.
14
*
15
* You should have received a copy of the GNU General Public License
16
* along with this program; if not, write to the Free Software
17
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
18
*/
19
20
#include "drizzledump_data.h"
21
#include "client_priv.h"
22
#include <drizzled/gettext.h>
23
#include <string>
24
#include <iostream>
25
26
extern bool opt_no_create_info;
27
extern bool opt_no_data;
28
extern bool opt_create_db;
29
extern bool opt_disable_keys;
30
extern bool extended_insert;
31
extern bool opt_replace_into;
32
extern bool opt_drop;
33
extern uint32_t show_progress_size;
34
extern int connected_server_type;
35
36
enum server_type {
37
SERVER_MYSQL_FOUND,
38
SERVER_DRIZZLE_FOUND,
39
SERVER_UNKNOWN_FOUND
40
};
41
42
std::ostream& operator <<(std::ostream &os, const DrizzleDumpIndex &obj)
43
{
44
if (obj.isPrimary)
45
{
46
os << " PRIMARY KEY ";
47
}
48
else if (obj.isUnique)
49
{
50
os << " UNIQUE KEY `" << obj.indexName << "` ";
51
}
52
else
53
{
54
os << " KEY `" << obj.indexName << "` ";
55
}
56
57
os << "(";
58
59
std::vector<std::string>::iterator i;
60
std::vector<std::string> fields = obj.columns;
61
for (i= fields.begin(); i != fields.end(); ++i)
62
{
63
if (i != fields.begin())
64
os << ",";
65
std::string field= *i;
66
os << "`" << field << "`";
67
}
68
69
os << ")";
70
71
return os;
72
}
73
74
std::ostream& operator <<(std::ostream &os, const DrizzleDumpField &obj)
75
{
76
os << " `" << obj.fieldName << "` ";
77
os << obj.type;
78
if (((obj.type.compare("VARCHAR") == 0) or
79
(obj.type.compare("VARBINARY") == 0)) and
80
(obj.length > 0))
81
{
82
os << "(" << obj.length << ")";
83
}
84
else if ((obj.type.compare("DECIMAL") == 0) or
85
(obj.type.compare("DOUBLE") == 0))
86
{
87
os << "(" << obj.decimalPrecision << "," << obj.decimalScale << ")";
88
}
89
else if (obj.type.compare("ENUM") == 0)
90
{
91
os << "(" << obj.enumValues << ")";
92
}
93
94
if (not obj.isNull)
95
{
96
os << " NOT NULL";
97
}
98
99
if ((not obj.collation.empty()) and (obj.collation.compare("binary") != 0))
100
{
101
os << " COLLATE " << obj.collation;
102
}
103
104
if (obj.isAutoIncrement)
105
os << " AUTO_INCREMENT";
106
107
if (not obj.defaultValue.empty())
108
{
109
if (obj.defaultValue.compare("CURRENT_TIMESTAMP") != 0)
110
os << " DEFAULT '" << obj.defaultValue << "'";
111
else
112
os << " DEFAULT CURRENT_TIMESTAMP";
113
}
114
else if ((obj.collation.empty()) and (obj.defaultIsNull))
115
{
116
os << " DEFAULT NULL";
117
}
118
119
return os;
120
}
121
122
std::ostream& operator <<(std::ostream &os, const DrizzleDumpDatabase &obj)
123
{
124
os << "--" << std::endl
125
<< "-- Current Database: `" << obj.databaseName << "`" << std::endl
126
<< "--" << std::endl << std::endl;
127
128
/* Love that this variable is the opposite of its name */
129
if (not opt_create_db)
130
{
131
os << "CREATE DATABASE IF NOT EXISTS `" << obj.databaseName
132
<< "` COLLATE = " << obj.collate << ";" << std::endl << std::endl;
133
}
134
135
os << "USE `" << obj.databaseName << "`;" << std::endl << std::endl;
136
137
std::vector<DrizzleDumpTable*>::iterator i;
138
std::vector<DrizzleDumpTable*> output_tables = obj.tables;
139
for (i= output_tables.begin(); i != output_tables.end(); ++i)
140
{
141
DrizzleDumpTable *table= *i;
142
if (not opt_no_create_info)
143
os << *table;
144
if (not opt_no_data)
145
{
146
DrizzleDumpData *data= table->getData();
147
os << *data;
148
delete data;
149
}
150
}
151
152
return os;
153
}
154
155
156
std::ostream& operator <<(std::ostream &os, const DrizzleDumpData &obj)
157
{
158
bool new_insert= true;
159
bool first= true;
160
uint64_t rownr= 0;
161
162
drizzle_row_t row;
163
164
if (drizzle_result_row_count(obj.result) < 1)
165
{
166
os << "--" << std::endl
167
<< "-- No data to dump for table `" << obj.table->tableName << "`" << std::endl
168
<< "--" << std::endl << std::endl;
169
return os;
170
}
171
else
172
{
173
os << "--" << std::endl
174
<< "-- Dumping data for table `" << obj.table->tableName << "`" << std::endl
175
<< "--" << std::endl << std::endl;
176
}
177
if (opt_disable_keys)
178
os << "ALTER TABLE `" << obj.table->tableName << "` DISABLE KEYS;" << std::endl;
179
180
std::streampos out_position= os.tellp();
181
182
while((row= drizzle_row_next(obj.result)))
183
{
184
rownr++;
185
if ((rownr % show_progress_size) == 0)
186
{
187
std::cerr << "-- %" << rownr << _(" rows dumped for table ") << obj.table->tableName << std::endl;
188
}
189
190
size_t* row_sizes= drizzle_row_field_sizes(obj.result);
191
if (not first)
192
{
193
if (extended_insert)
194
os << "),(";
195
else
196
os << ");" << std::endl;
197
}
198
else
199
first= false;
200
201
if (new_insert)
202
{
203
if (opt_replace_into)
204
os << "REPLACE ";
205
else
206
os << "INSERT ";
207
os << "INTO `" << obj.table->tableName << "` VALUES (";
208
if (extended_insert)
209
new_insert= false;
210
}
211
for (uint32_t i= 0; i < drizzle_result_column_count(obj.result); i++)
212
{
213
if (not row[i])
214
{
215
os << "NULL";
216
}
217
/* time/date conversion for MySQL connections */
218
else if (obj.table->fields[i]->convertDateTime)
219
{
220
os << obj.checkDateTime(os, row[i], i);
221
}
222
else
223
{
224
if (obj.table->fields[i]->type.compare("INT") != 0)
225
{
226
/* Hex blob processing or escape text */
227
if (((obj.table->fields[i]->type.compare("BLOB") == 0) or
228
(obj.table->fields[i]->type.compare("VARBINARY") == 0)))
229
os << obj.convertHex(row[i], row_sizes[i]);
230
else
231
os << "'" << obj.escape(row[i], row_sizes[i]) << "'";
232
}
233
else
234
os << row[i];
235
}
236
if (i != obj.table->fields.size() - 1)
237
os << ",";
238
}
239
/* Break insert up if it is too long */
240
if (extended_insert and
241
((os.tellp() - out_position) >= DRIZZLE_MAX_LINE_LENGTH))
242
{
243
os << ");" << std::endl;
244
new_insert= true;
245
out_position= os.tellp();
246
}
247
}
248
os << ");" << std::endl;
249
250
if (opt_disable_keys)
251
os << "ALTER TABLE `" << obj.table->tableName << "` ENABLE KEYS;" << std::endl;
252
253
os << std::endl;
254
255
return os;
256
}
257
258
std::string DrizzleDumpData::convertHex(const char* from, size_t from_size) const
259
{
260
std::ostringstream output;
261
if (from_size > 0)
262
output << "0x";
263
while (from_size > 0)
264
{
265
output << std::uppercase << std::hex << std::setw(2) << std::setfill('0') << int(*from);
266
*from++;
267
from_size--;
268
}
269
270
return output.str();
271
}
272
273
/* Ripped out of libdrizzle, hopefully a little safer */
274
std::string DrizzleDumpData::escape(const char* from, size_t from_size) const
275
{
276
std::string output;
277
278
while (from_size > 0)
279
{
280
if (!(*from & 0x80))
281
{
282
switch (*from)
283
{
284
case 0:
285
case '\n':
286
case '\r':
287
case '\\':
288
case '\'':
289
case '"':
290
case '\032':
291
output.push_back('\\');
292
default:
293
break;
294
}
295
}
296
output.push_back(*from);
297
*from++;
298
from_size--;
299
}
300
301
return output;
302
}
303
304
std::ostream& operator <<(std::ostream &os, const DrizzleDumpTable &obj)
305
{
306
os << "--" << std::endl
307
<< "-- Table structure for table `" << obj.tableName << "`" << std::endl
308
<< "--" << std::endl << std::endl;
309
310
if (opt_drop)
311
os << "DROP TABLE IF EXISTS `" << obj.tableName << "`;" << std::endl;
312
313
os << "CREATE TABLE `" << obj.tableName << "` (" << std::endl;
314
std::vector<DrizzleDumpField*>::iterator i;
315
std::vector<DrizzleDumpField*> output_fields = obj.fields;
316
for (i= output_fields.begin(); i != output_fields.end(); ++i)
317
{
318
if (i != output_fields.begin())
319
os << "," << std::endl;
320
DrizzleDumpField *field= *i;
321
os << *field;
322
}
323
324
std::vector<DrizzleDumpIndex*>::iterator j;
325
std::vector<DrizzleDumpIndex*> output_indexes = obj.indexes;
326
for (j= output_indexes.begin(); j != output_indexes.end(); ++j)
327
{
328
os << "," << std::endl;;
329
DrizzleDumpIndex *index= *j;
330
os << *index;
331
}
332
os << std::endl;
333
os << ") ENGINE=" << obj.engineName << " ";
334
if ((connected_server_type == SERVER_MYSQL_FOUND) and (obj.autoIncrement > 0))
335
os << "AUTO_INCREMENT=" << obj.autoIncrement << " ";
336
337
os << "COLLATE = " << obj.collate << ";" << std::endl << std::endl;
338
339
return os;
340
}
Loggerhead is a web-based interface for Breezy
Version: 2.0.1