← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to plugin/pbxt/src/backup_xt.cc

  • Committer: Brian Aker
  • Date: 2009-02-21 00:18:15 UTC
  • Revision ID: brian@tangent.org-20090221001815-x20e8h71e984lvs1
Completion (?) of uint conversion.

Show diffs side-by-side

added added

removed removed

Lines of Context:
plugin/pbxt/src/backup_xt.cc
1
 
/* Copyright (C) 2009 PrimeBase Technologies GmbH
2
 
 *
3
 
 * PrimeBase XT
4
 
 *
5
 
 * This program is free software; you can redistribute it and/or modify
6
 
 * it under the terms of the GNU General Public License as published by
7
 
 * the Free Software Foundation; either version 2 of the License, or
8
 
 * (at your option) any later version.
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
 
 * 2009-09-07   Paul McCullagh
20
 
 *
21
 
 * H&G2JCtL
22
 
 */
23
 
 
24
 
#include "xt_config.h"
25
 
 
26
 
#ifdef MYSQL_SUPPORTS_BACKUP
27
 
 
28
 
#include <string.h>
29
 
#include <stdio.h>
30
 
#include <stdlib.h>
31
 
#include <time.h>
32
 
#include <ctype.h>
33
 
 
34
 
#include "mysql_priv.h"
35
 
#include <backup/api_types.h>
36
 
#include <backup/backup_engine.h>
37
 
#include <backup/backup_aux.h>         // for build_table_list()
38
 
#include <hash.h>
39
 
 
40
 
#include "ha_pbxt.h"
41
 
 
42
 
#include "backup_xt.h"
43
 
#include "pthread_xt.h"
44
 
#include "filesys_xt.h"
45
 
#include "database_xt.h"
46
 
#include "strutil_xt.h"
47
 
#include "memory_xt.h"
48
 
#include "trace_xt.h"
49
 
#include "myxt_xt.h"
50
 
 
51
 
#ifdef OK
52
 
#undef OK
53
 
#endif
54
 
 
55
 
#ifdef byte
56
 
#undef byte
57
 
#endif
58
 
 
59
 
#ifdef DEBUG
60
 
//#define TRACE_BACKUP_CALLS
61
 
//#define TEST_SMALL_BLOCK                      100000
62
 
#endif
63
 
 
64
 
using backup::byte;
65
 
using backup::result_t;
66
 
using backup::version_t;
67
 
using backup::Table_list;
68
 
using backup::Table_ref;
69
 
using backup::Buffer;
70
 
 
71
 
#ifdef TRACE_BACKUP_CALLS
72
 
#define XT_TRACE_CALL()                         ha_trace_function(__FUNC__, NULL)
73
 
#else
74
 
#define XT_TRACE_CALL()
75
 
#endif
76
 
 
77
 
#define XT_RESTORE_BATCH_SIZE           10000
78
 
 
79
 
#define BUP_STATE_BEFORE_LOCK           0
80
 
#define BUP_STATE_AFTER_LOCK            1
81
 
 
82
 
#define BUP_STANDARD_VAR_RECORD         1
83
 
#define BUP_RECORD_BLOCK_4_START        2                       // Part of a record, with a 4 byte total length, and 4 byte data length
84
 
#define BUP_RECORD_BLOCK_4                      3                       // Part of a record, with a 4 byte length
85
 
#define BUP_RECORD_BLOCK_4_END          4                       // Last part of a record with a 4 byte length
86
 
 
87
 
/*
88
 
 * -----------------------------------------------------------------------
89
 
 * UTILITIES
90
 
 */
91
 
 
92
 
#ifdef TRACE_BACKUP_CALLS
93
 
static void ha_trace_function(const char *function, char *table)
94
 
{
95
 
        char            func_buf[50], *ptr;
96
 
        XTThreadPtr     thread = xt_get_self(); 
97
 
        
98
 
        if ((ptr = strchr(function, '('))) {
99
 
                ptr--;
100
 
                while (ptr > function) {
101
 
                        if (!(isalnum(*ptr) || *ptr == '_'))
102
 
                                break;
103
 
                        ptr--;
104
 
                }
105
 
                ptr++;
106
 
                xt_strcpy(50, func_buf, ptr);
107
 
                if ((ptr = strchr(func_buf, '(')))
108
 
                        *ptr = 0;
109
 
        }
110
 
        else
111
 
                xt_strcpy(50, func_buf, function);
112
 
        if (table)
113
 
                printf("%s %s (%s)\n", thread ? thread->t_name : "-unknown-", func_buf, table);
114
 
        else
115
 
                printf("%s %s\n", thread ? thread->t_name : "-unknown-", func_buf);
116
 
}
117
 
#endif
118
 
 
119
 
/*
120
 
 * -----------------------------------------------------------------------
121
 
 * BACKUP DRIVER
122
 
 */
123
 
 
124
 
class PBXTBackupDriver: public Backup_driver
125
 
{
126
 
        public:
127
 
        PBXTBackupDriver(const Table_list &);
128
 
        virtual ~PBXTBackupDriver();
129
 
 
130
 
        virtual size_t          size();
131
 
        virtual size_t          init_size();
132
 
        virtual result_t        begin(const size_t);
133
 
        virtual result_t        end();
134
 
        virtual result_t        get_data(Buffer &);
135
 
        virtual result_t        prelock();
136
 
        virtual result_t        lock();
137
 
        virtual result_t        unlock();
138
 
        virtual result_t        cancel();
139
 
        virtual void            free();
140
 
        void                            lock_tables_TL_READ_NO_INSERT();
141
 
 
142
 
        private:
143
 
        XTThreadPtr             bd_thread;
144
 
        int                             bd_state;
145
 
        u_int                   bd_table_no;
146
 
        XTOpenTablePtr  bd_ot;
147
 
        xtWord1                 *bd_row_buf;
148
 
 
149
 
        /* Non-zero if we last returned only part of
150
 
         * a row.
151
 
         */
152
 
        xtWord1                 *db_write_block(xtWord1 *buffer, xtWord1 bup_type, size_t *size, xtWord4 row_len);
153
 
        xtWord1                 *db_write_block(xtWord1 *buffer, xtWord1 bup_type, size_t *size, xtWord4 total_len, xtWord4 row_len);
154
 
 
155
 
        xtWord4                 bd_row_offset;
156
 
        xtWord4                 bd_row_size;
157
 
};
158
 
 
159
 
 
160
 
PBXTBackupDriver::PBXTBackupDriver(const Table_list &tables):
161
 
Backup_driver(tables),
162
 
bd_state(BUP_STATE_BEFORE_LOCK),
163
 
bd_table_no(0),
164
 
bd_ot(NULL),
165
 
bd_row_buf(NULL),
166
 
bd_row_offset(0),
167
 
bd_row_size(0)
168
 
{
169
 
}
170
 
 
171
 
PBXTBackupDriver::~PBXTBackupDriver()
172
 
{
173
 
}
174
 
 
175
 
/** Estimates total size of backup. @todo improve it */
176
 
size_t PBXTBackupDriver::size()
177
 
{
178
 
        XT_TRACE_CALL();
179
 
        return UNKNOWN_SIZE;
180
 
}
181
 
 
182
 
/** Estimates size of backup before lock. @todo improve it */
183
 
size_t PBXTBackupDriver::init_size()
184
 
{
185
 
        XT_TRACE_CALL();
186
 
        return 0;
187
 
}
188
 
 
189
 
result_t PBXTBackupDriver::begin(const size_t)
190
 
{
191
 
        THD                             *thd = current_thd;
192
 
        XTExceptionRec  e;
193
 
 
194
 
        XT_TRACE_CALL();
195
 
        
196
 
        if (!(bd_thread = xt_ha_set_current_thread(thd, &e))) {
197
 
                xt_log_exception(NULL, &e, XT_LOG_DEFAULT);
198
 
                return backup::ERROR;
199
 
        }
200
 
        
201
 
        return backup::OK;
202
 
}
203
 
 
204
 
result_t PBXTBackupDriver::end()
205
 
{
206
 
        XT_TRACE_CALL();
207
 
        if (bd_ot) {
208
 
                xt_tab_seq_exit(bd_ot);
209
 
                xt_db_return_table_to_pool_ns(bd_ot);
210
 
                bd_ot = NULL;
211
 
        }
212
 
        if (bd_thread->st_xact_data) {
213
 
                if (!xt_xn_commit(bd_thread))
214
 
                        return backup::ERROR;
215
 
        }
216
 
        return backup::OK;
217
 
}
218
 
 
219
 
xtWord1 *PBXTBackupDriver::db_write_block(xtWord1 *buffer, xtWord1 bup_type, size_t *ret_size, xtWord4 row_len)
220
 
{
221
 
        register size_t size = *ret_size;
222
 
 
223
 
        *buffer = bup_type;     // Record type identifier.
224
 
        buffer++;
225
 
        size--;
226
 
        memcpy(buffer, bd_ot->ot_row_wbuffer, row_len);
227
 
        buffer += row_len;
228
 
        size -= row_len;
229
 
        *ret_size = size;
230
 
        return buffer;
231
 
}
232
 
 
233
 
xtWord1 *PBXTBackupDriver::db_write_block(xtWord1 *buffer, xtWord1 bup_type, size_t *ret_size, xtWord4 total_len, xtWord4 row_len)
234
 
{
235
 
        register size_t size = *ret_size;
236
 
 
237
 
        *buffer = bup_type;     // Record type identifier.
238
 
        buffer++;
239
 
        size--;
240
 
        if (bup_type == BUP_RECORD_BLOCK_4_START) {
241
 
                XT_SET_DISK_4(buffer, total_len);
242
 
                buffer += 4;
243
 
                size -= 4;
244
 
        }
245
 
        XT_SET_DISK_4(buffer, row_len);
246
 
        buffer += 4;
247
 
        size -= 4;
248
 
        memcpy(buffer, bd_ot->ot_row_wbuffer+bd_row_offset, row_len);
249
 
        buffer += row_len;
250
 
        size -= row_len;
251
 
        bd_row_size -= row_len;
252
 
        bd_row_offset += row_len;
253
 
        *ret_size = size;
254
 
        return buffer;
255
 
}
256
 
 
257
 
result_t PBXTBackupDriver::get_data(Buffer &buf)
258
 
{
259
 
        xtBool  eof = FALSE;
260
 
        size_t  size;
261
 
        xtWord4 row_len;
262
 
        xtWord1 *buffer;
263
 
 
264
 
        XT_TRACE_CALL();
265
 
 
266
 
        if (bd_state == BUP_STATE_BEFORE_LOCK) {
267
 
                buf.table_num = 0;
268
 
                buf.size = 0;
269
 
                buf.last = FALSE;
270
 
                return backup::READY;
271
 
        }
272
 
 
273
 
        /* Open the backup table: */
274
 
        if (!bd_ot) {
275
 
                XTThreadPtr             self = bd_thread;
276
 
                XTTableHPtr             tab;
277
 
                char                    path[PATH_MAX];
278
 
        
279
 
                if (bd_table_no == m_tables.count()) {
280
 
                        buf.size = 0;
281
 
                        buf.table_num = 0;
282
 
                        buf.last = TRUE;
283
 
                        return backup::DONE;
284
 
                }
285
 
                
286
 
                m_tables[bd_table_no].internal_name(path, sizeof(path));
287
 
                bd_table_no++;
288
 
                try_(a) {
289
 
                        xt_ha_open_database_of_table(self, (XTPathStrPtr) path);
290
 
                        tab = xt_use_table(self, (XTPathStrPtr) path, FALSE, FALSE);
291
 
                        pushr_(xt_heap_release, tab);
292
 
                        if (!(bd_ot = xt_db_open_table_using_tab(tab, bd_thread)))
293
 
                                xt_throw(self);
294
 
                        freer_(); // xt_heap_release(tab)
295
 
 
296
 
                        /* Prepare the seqential scan: */
297
 
                        xt_tab_seq_exit(bd_ot);
298
 
                        if (!xt_tab_seq_init(bd_ot))
299
 
                                xt_throw(self);
300
 
                        
301
 
                        if (bd_row_buf) {
302
 
                                xt_free(self, bd_row_buf);
303
 
                                bd_row_buf = NULL;
304
 
                        }
305
 
                        bd_row_buf = (xtWord1 *) xt_malloc(self, bd_ot->ot_table->tab_dic.dic_mysql_buf_size);
306
 
                        bd_ot->ot_cols_req = bd_ot->ot_table->tab_dic.dic_no_of_cols;
307
 
                }
308
 
                catch_(a) {
309
 
                        ;
310
 
                }
311
 
                cont_(a);
312
 
 
313
 
                if (!bd_ot)
314
 
                        goto failed;
315
 
        }
316
 
 
317
 
        buf.table_num = bd_table_no;
318
 
#ifdef TEST_SMALL_BLOCK
319
 
        buf.size = TEST_SMALL_BLOCK;
320
 
#endif
321
 
        size = buf.size;
322
 
        buffer = (xtWord1 *) buf.data;
323
 
        ASSERT_NS(size > 9);
324
 
 
325
 
        /* First check of a record was partically written
326
 
         * last time.
327
 
         */
328
 
        write_row:
329
 
        if (bd_row_size > 0) {
330
 
                row_len = bd_row_size;
331
 
                if (bd_row_offset == 0) {
332
 
                        if (row_len+1 > size) {
333
 
                                ASSERT_NS(size > 9);
334
 
                                row_len = size - 9;
335
 
                                buffer = db_write_block(buffer, BUP_RECORD_BLOCK_4_START, &size, bd_row_size, row_len);
336
 
                                goto done;
337
 
                        }
338
 
                        buffer = db_write_block(buffer, BUP_STANDARD_VAR_RECORD, &size, row_len);
339
 
                        bd_row_size = 0;
340
 
                }
341
 
                else {
342
 
                        if (row_len+5 > size) {
343
 
                                row_len = size - 5;
344
 
                                buffer = db_write_block(buffer, BUP_RECORD_BLOCK_4, &size, 0, row_len);
345
 
                                goto done;
346
 
                        }
347
 
                        buffer = db_write_block(buffer, BUP_RECORD_BLOCK_4_END, &size, 0, row_len);
348
 
                }
349
 
        }
350
 
 
351
 
        /* Now continue with the sequential scan. */
352
 
        while (size > 1) {
353
 
                if (!xt_tab_seq_next(bd_ot, bd_row_buf, &eof))
354
 
                        goto failed;
355
 
                if (eof) {
356
 
                        /* We will go the next table, on the next call. */
357
 
                        xt_tab_seq_exit(bd_ot);
358
 
                        xt_db_return_table_to_pool_ns(bd_ot);
359
 
                        bd_ot = NULL;
360
 
                        break;
361
 
                }
362
 
                if (!(row_len = myxt_store_row_data(bd_ot, 0, (char *) bd_row_buf)))
363
 
                        goto failed;
364
 
                if (row_len+1 > size) {
365
 
                        /* Does not fit: */
366
 
                        bd_row_offset = 0;
367
 
                        bd_row_size = row_len;
368
 
                        /* Only add part of the row, if there is still
369
 
                         * quite a bit of space left:
370
 
                         */
371
 
                        if (size >= (32 * 1024))
372
 
                                goto write_row;
373
 
                        break;
374
 
                }
375
 
                buffer = db_write_block(buffer, BUP_STANDARD_VAR_RECORD, &size, row_len);
376
 
        }
377
 
 
378
 
        done:
379
 
        buf.size = buf.size - size;
380
 
        /* This indicates wnd of data for a table! */
381
 
    buf.last = eof;
382
 
 
383
 
        return backup::OK;
384
 
 
385
 
        failed:
386
 
        xt_log_and_clear_exception(bd_thread);
387
 
        return backup::ERROR;
388
 
}
389
 
 
390
 
result_t PBXTBackupDriver::prelock()
391
 
{
392
 
        XT_TRACE_CALL();
393
 
        return backup::READY;
394
 
}
395
 
 
396
 
result_t PBXTBackupDriver::lock()
397
 
{
398
 
        XT_TRACE_CALL();
399
 
        bd_thread->st_xact_mode = XT_XACT_COMMITTED_READ;
400
 
        bd_thread->st_ignore_fkeys = FALSE;
401
 
        bd_thread->st_auto_commit = FALSE;
402
 
        bd_thread->st_table_trans = FALSE;
403
 
        bd_thread->st_abort_trans = FALSE;
404
 
        bd_thread->st_stat_ended = FALSE;
405
 
        bd_thread->st_stat_trans = FALSE;
406
 
        bd_thread->st_is_update = NULL;
407
 
        if (!xt_xn_begin(bd_thread))
408
 
                return backup::ERROR;
409
 
        bd_state = BUP_STATE_AFTER_LOCK;
410
 
        return backup::OK;
411
 
}
412
 
 
413
 
result_t PBXTBackupDriver::unlock()
414
 
{
415
 
        XT_TRACE_CALL();
416
 
        return backup::OK;
417
 
}
418
 
 
419
 
result_t PBXTBackupDriver::cancel()
420
 
{
421
 
        XT_TRACE_CALL();
422
 
        return backup::OK; // free() will be called and suffice
423
 
}
424
 
 
425
 
void PBXTBackupDriver::free()
426
 
{
427
 
        XT_TRACE_CALL();
428
 
        if (bd_ot) {
429
 
                xt_tab_seq_exit(bd_ot);
430
 
                xt_db_return_table_to_pool_ns(bd_ot);
431
 
                bd_ot = NULL;
432
 
        }
433
 
        if (bd_row_buf) {
434
 
                xt_free_ns(bd_row_buf);
435
 
                bd_row_buf = NULL;
436
 
        }
437
 
        if (bd_thread->st_xact_data)
438
 
                xt_xn_rollback(bd_thread);
439
 
        delete this;
440
 
}
441
 
 
442
 
void PBXTBackupDriver::lock_tables_TL_READ_NO_INSERT()
443
 
{
444
 
        XT_TRACE_CALL();
445
 
}
446
 
 
447
 
/*
448
 
 * -----------------------------------------------------------------------
449
 
 * BACKUP DRIVER
450
 
 */
451
 
 
452
 
class PBXTRestoreDriver: public Restore_driver
453
 
{
454
 
        public:
455
 
        PBXTRestoreDriver(const Table_list &tables);
456
 
        virtual ~PBXTRestoreDriver();
457
 
 
458
 
        virtual result_t  begin(const size_t);
459
 
        virtual result_t  end();
460
 
        virtual result_t  send_data(Buffer &buf);
461
 
        virtual result_t  cancel();
462
 
        virtual void      free();
463
 
        
464
 
        private:
465
 
        XTThreadPtr             rd_thread;
466
 
        u_int                   rd_table_no;
467
 
        XTOpenTablePtr  rd_ot;
468
 
        STRUCT_TABLE    *rd_my_table;
469
 
        xtWord1                 *rb_row_buf;
470
 
        u_int                   rb_col_cnt;
471
 
        u_int                   rb_insert_count;
472
 
 
473
 
        /* Long rows are accumulated here: */
474
 
        xtWord4                 rb_row_len;
475
 
        xtWord4                 rb_data_size;
476
 
        xtWord1                 *rb_row_data;
477
 
};
478
 
 
479
 
PBXTRestoreDriver::PBXTRestoreDriver(const Table_list &tables):
480
 
Restore_driver(tables),
481
 
rd_thread(NULL),
482
 
rd_table_no(0),
483
 
rd_ot(NULL),
484
 
rb_row_buf(NULL),
485
 
rb_row_len(0),
486
 
rb_data_size(0),
487
 
rb_row_data(NULL)
488
 
{
489
 
}
490
 
 
491
 
PBXTRestoreDriver::~PBXTRestoreDriver()
492
 
{
493
 
}
494
 
 
495
 
result_t PBXTRestoreDriver::begin(const size_t)
496
 
{
497
 
        THD                             *thd = current_thd;
498
 
        XTExceptionRec  e;
499
 
        
500
 
        XT_TRACE_CALL();
501
 
        
502
 
        if (!(rd_thread = xt_ha_set_current_thread(thd, &e))) {
503
 
                xt_log_exception(NULL, &e, XT_LOG_DEFAULT);
504
 
                return backup::ERROR;
505
 
        }
506
 
        
507
 
        return backup::OK;
508
 
}
509
 
 
510
 
result_t PBXTRestoreDriver::end()
511
 
{
512
 
        XT_TRACE_CALL();
513
 
        if (rd_ot) {
514
 
                xt_db_return_table_to_pool_ns(rd_ot);
515
 
                rd_ot = NULL;
516
 
        }
517
 
        //if (rb_row_buf) {
518
 
        //      xt_free_ns(rb_row_buf);
519
 
        //      rb_row_buf = NULL;
520
 
        //}
521
 
        if (rb_row_data) {
522
 
                xt_free_ns(rb_row_data);
523
 
                rb_row_data = NULL;
524
 
        }
525
 
        if (rd_thread->st_xact_data) {
526
 
                if (!xt_xn_commit(rd_thread))
527
 
                        return backup::ERROR;
528
 
        }
529
 
        return backup::OK;
530
 
}
531
 
 
532
 
 
533
 
result_t PBXTRestoreDriver::send_data(Buffer &buf)
534
 
{
535
 
        size_t  size;
536
 
        xtWord1 type;
537
 
        xtWord1 *buffer;
538
 
        xtWord4 row_len;
539
 
        xtWord1 *rec_data;
540
 
 
541
 
        XT_TRACE_CALL();
542
 
 
543
 
        if (buf.table_num != rd_table_no) {
544
 
                XTThreadPtr             self = rd_thread;
545
 
                XTTableHPtr             tab;
546
 
                char                    path[PATH_MAX];
547
 
                
548
 
                if (rd_ot) {
549
 
                        xt_db_return_table_to_pool_ns(rd_ot);
550
 
                        rd_ot = NULL;
551
 
                }
552
 
 
553
 
                if (rd_thread->st_xact_data) {
554
 
                        if (!xt_xn_commit(rd_thread))
555
 
                                goto failed;
556
 
                }
557
 
                if (!xt_xn_begin(rd_thread))
558
 
                        goto failed;
559
 
                rb_insert_count = 0;
560
 
                
561
 
                rd_table_no = buf.table_num;
562
 
                m_tables[rd_table_no-1].internal_name(path, sizeof(path));
563
 
                try_(a) {
564
 
                        xt_ha_open_database_of_table(self, (XTPathStrPtr) path);
565
 
                        tab = xt_use_table(self, (XTPathStrPtr) path, FALSE, FALSE);
566
 
                        pushr_(xt_heap_release, tab);
567
 
                        if (!(rd_ot = xt_db_open_table_using_tab(tab, rd_thread)))
568
 
                                xt_throw(self);
569
 
                        freer_(); // xt_heap_release(tab)
570
 
 
571
 
                        rd_my_table = rd_ot->ot_table->tab_dic.dic_my_table;
572
 
                        if (rd_my_table->found_next_number_field) {
573
 
                                rd_my_table->in_use = current_thd;
574
 
                                rd_my_table->next_number_field = rd_my_table->found_next_number_field;
575
 
                                rd_my_table->mark_columns_used_by_index_no_reset(rd_my_table->s->next_number_index, rd_my_table->read_set);
576
 
                        }
577
 
 
578
 
                        /* This is safe because only one thread can restore a table at 
579
 
                         * a time!
580
 
                         */
581
 
                        rb_row_buf = (xtWord1 *) rd_my_table->RECORD_0;
582
 
                        //if (rb_row_buf) {
583
 
                        //      xt_free(self, rb_row_buf);
584
 
                        //      rb_row_buf = NULL;
585
 
                        //}
586
 
                        //rb_row_buf = (xtWord1 *) xt_malloc(self, rd_ot->ot_table->tab_dic.dic_mysql_buf_size);
587
 
        
588
 
                        rb_col_cnt = rd_ot->ot_table->tab_dic.dic_no_of_cols;
589
 
 
590
 
                }
591
 
                catch_(a) {
592
 
                        ;
593
 
                }
594
 
                cont_(a);
595
 
                
596
 
                if (!rd_ot)
597
 
                        goto failed;
598
 
        }
599
 
 
600
 
        buffer = (xtWord1 *) buf.data;
601
 
        size = buf.size;
602
 
 
603
 
        while (size > 0) {
604
 
                type = *buffer;
605
 
                switch (type) {
606
 
                        case BUP_STANDARD_VAR_RECORD:
607
 
                                rec_data = buffer + 1;
608
 
                                break;
609
 
                        case BUP_RECORD_BLOCK_4_START:
610
 
                                buffer++;
611
 
                                row_len = XT_GET_DISK_4(buffer);
612
 
                                buffer += 4;
613
 
                                if (rb_data_size < row_len) {
614
 
                                        if (!xt_realloc_ns((void **) &rb_row_data, row_len))
615
 
                                                goto failed;
616
 
                                        rb_data_size = row_len;
617
 
                                }
618
 
                                row_len = XT_GET_DISK_4(buffer);
619
 
                                buffer += 4;
620
 
                                ASSERT_NS(row_len <= rb_data_size);
621
 
                                if (row_len > rb_data_size) {
622
 
                                        xt_register_xterr(XT_REG_CONTEXT, XT_ERR_BAD_BACKUP_FORMAT);
623
 
                                        goto failed;
624
 
                                }
625
 
                                memcpy(rb_row_data, buffer, row_len);
626
 
                                rb_row_len = row_len;
627
 
                                buffer += row_len;
628
 
                                if (row_len + 9 > size) {
629
 
                                        xt_register_xterr(XT_REG_CONTEXT, XT_ERR_BAD_BACKUP_FORMAT);
630
 
                                        goto failed;
631
 
                                }
632
 
                                size -= row_len + 9;
633
 
                                continue;
634
 
                        case BUP_RECORD_BLOCK_4:
635
 
                                buffer++;
636
 
                                row_len = XT_GET_DISK_4(buffer);
637
 
                                buffer += 4;
638
 
                                ASSERT_NS(rb_row_len + row_len <= rb_data_size);
639
 
                                if (rb_row_len + row_len > rb_data_size) {
640
 
                                        xt_register_xterr(XT_REG_CONTEXT, XT_ERR_BAD_BACKUP_FORMAT);
641
 
                                        goto failed;
642
 
                                }
643
 
                                memcpy(rb_row_data + rb_row_len, buffer, row_len);
644
 
                                rb_row_len += row_len;
645
 
                                buffer += row_len;
646
 
                                if (row_len + 5 > size) {
647
 
                                        xt_register_xterr(XT_REG_CONTEXT, XT_ERR_BAD_BACKUP_FORMAT);
648
 
                                        goto failed;
649
 
                                }
650
 
                                size -= row_len + 5;
651
 
                                continue;
652
 
                        case BUP_RECORD_BLOCK_4_END:
653
 
                                buffer++;
654
 
                                row_len = XT_GET_DISK_4(buffer);
655
 
                                buffer += 4;
656
 
                                ASSERT_NS(rb_row_len + row_len <= rb_data_size);
657
 
                                if (rb_row_len + row_len > rb_data_size) {
658
 
                                        xt_register_xterr(XT_REG_CONTEXT, XT_ERR_BAD_BACKUP_FORMAT);
659
 
                                        goto failed;
660
 
                                }
661
 
                                memcpy(rb_row_data + rb_row_len, buffer, row_len);
662
 
                                buffer += row_len;
663
 
                                if (row_len + 5 > size) {
664
 
                                        xt_register_xterr(XT_REG_CONTEXT, XT_ERR_BAD_BACKUP_FORMAT);
665
 
                                        goto failed;
666
 
                                }
667
 
                                size -= row_len + 5;
668
 
                                rec_data = rb_row_data;
669
 
                                break;
670
 
                        default:
671
 
                                xt_register_xterr(XT_REG_CONTEXT, XT_ERR_BAD_BACKUP_FORMAT);
672
 
                                goto failed;
673
 
                }
674
 
                
675
 
                if (!(row_len = myxt_load_row_data(rd_ot, rec_data, rb_row_buf, rb_col_cnt)))
676
 
                        goto failed;
677
 
 
678
 
                if (rd_ot->ot_table->tab_dic.dic_my_table->found_next_number_field)
679
 
                        ha_set_auto_increment(rd_ot, rd_ot->ot_table->tab_dic.dic_my_table->found_next_number_field);
680
 
 
681
 
                if (!xt_tab_new_record(rd_ot, rb_row_buf))
682
 
                        goto failed;
683
 
 
684
 
                if (type == BUP_STANDARD_VAR_RECORD) {
685
 
                        buffer += row_len+1;
686
 
                        if (row_len + 1 > size) {
687
 
                                xt_register_xterr(XT_REG_CONTEXT, XT_ERR_BAD_BACKUP_FORMAT);
688
 
                                goto failed;
689
 
                        }
690
 
                        size -= row_len + 1;
691
 
                }
692
 
 
693
 
                rb_insert_count++;
694
 
                if (rb_insert_count == XT_RESTORE_BATCH_SIZE) {
695
 
                        if (!xt_xn_commit(rd_thread))
696
 
                                goto failed;
697
 
                        if (!xt_xn_begin(rd_thread))
698
 
                                goto failed;
699
 
                        rb_insert_count = 0;
700
 
                }
701
 
        }
702
 
 
703
 
        return backup::OK;
704
 
        
705
 
        failed:
706
 
        xt_log_and_clear_exception(rd_thread);
707
 
        return backup::ERROR;
708
 
}
709
 
 
710
 
 
711
 
result_t PBXTRestoreDriver::cancel()
712
 
{
713
 
        XT_TRACE_CALL();
714
 
        /* Nothing to do in cancel(); free() will suffice */
715
 
        return backup::OK;
716
 
}
717
 
 
718
 
void PBXTRestoreDriver::free()
719
 
{
720
 
        XT_TRACE_CALL();
721
 
        if (rd_ot) {
722
 
                xt_db_return_table_to_pool_ns(rd_ot);
723
 
                rd_ot = NULL;
724
 
        }
725
 
        //if (rb_row_buf) {
726
 
        //      xt_free_ns(rb_row_buf);
727
 
        //      rb_row_buf = NULL;
728
 
        //}
729
 
        if (rb_row_data) {
730
 
                xt_free_ns(rb_row_data);
731
 
                rb_row_data = NULL;
732
 
        }
733
 
        if (rd_thread->st_xact_data)
734
 
                xt_xn_rollback(rd_thread);
735
 
        delete this;
736
 
}
737
 
 
738
 
/*
739
 
 * -----------------------------------------------------------------------
740
 
 * BACKUP ENGINE FACTORY
741
 
 */
742
 
 
743
 
#define PBXT_BACKUP_VERSION 1
744
 
 
745
 
 
746
 
class PBXTBackupEngine: public Backup_engine
747
 
{
748
 
        public:
749
 
        PBXTBackupEngine() { };
750
 
 
751
 
        virtual version_t version() const {
752
 
                return PBXT_BACKUP_VERSION;
753
 
        };
754
 
 
755
 
        virtual result_t get_backup(const uint32, const Table_list &, Backup_driver* &);
756
 
 
757
 
        virtual result_t get_restore(const version_t, const uint32, const Table_list &,Restore_driver* &);
758
 
 
759
 
        virtual void free()
760
 
        {
761
 
                delete this;
762
 
        }
763
 
};
764
 
 
765
 
result_t PBXTBackupEngine::get_backup(const u_int count, const Table_list &tables, Backup_driver* &drv)
766
 
{
767
 
        PBXTBackupDriver *ptr = new PBXTBackupDriver(tables);
768
 
 
769
 
        if (!ptr)
770
 
                return backup::ERROR;
771
 
        drv = ptr;
772
 
        return backup::OK;
773
 
}
774
 
 
775
 
result_t PBXTBackupEngine::get_restore(const version_t ver, const uint32,
776
 
                             const Table_list &tables, Restore_driver* &drv)
777
 
{
778
 
        if (ver > PBXT_BACKUP_VERSION)
779
 
        {
780
 
                return backup::ERROR;    
781
 
        }
782
 
        
783
 
        PBXTRestoreDriver *ptr = new PBXTRestoreDriver(tables);
784
 
 
785
 
        if (!ptr)
786
 
                return backup::ERROR;
787
 
        drv = (Restore_driver *) ptr;
788
 
        return backup::OK;
789
 
}
790
 
 
791
 
 
792
 
Backup_result_t pbxt_backup_engine(handlerton *self, Backup_engine* &be)
793
 
{
794
 
        be = new PBXTBackupEngine();
795
 
        
796
 
        if (!be)
797
 
                return backup::ERROR;
798
 
        
799
 
        return backup::OK;
800
 
}
801
 
 
802
 
#endif