hatoku_hton.cc 62.8 KB
Newer Older
1
/* -*- mode: C; c-basic-offset: 4 -*- */
Zardosht Kasheff's avatar
Zardosht Kasheff committed
2
#define MYSQL_SERVER 1
3
#include "hatoku_defines.h"
Zardosht Kasheff's avatar
Zardosht Kasheff committed
4
#include <db.h>
Zardosht Kasheff's avatar
Zardosht Kasheff committed
5 6 7 8 9

#include "stdint.h"
#if defined(_WIN32)
#include "misc.h"
#endif
Barry Perlman's avatar
Barry Perlman committed
10
#define __STDC_FORMAT_MACROS
Zardosht Kasheff's avatar
Zardosht Kasheff committed
11
#include <inttypes.h>
Zardosht Kasheff's avatar
Zardosht Kasheff committed
12
#include "toku_os.h"
13
#include "toku_time.h"
14
#include "partitioned_counter.h"
Zardosht Kasheff's avatar
Zardosht Kasheff committed
15

Zardosht Kasheff's avatar
Zardosht Kasheff committed
16 17 18 19 20 21 22 23 24 25 26 27 28 29 30
/* We define DTRACE after mysql_priv.h in case it disabled dtrace in the main server */
#ifdef HAVE_DTRACE
#define _DTRACE_VERSION 1
#else
#endif

#include <mysql/plugin.h>
#include "hatoku_hton.h"
#include "ha_tokudb.h"

#undef PACKAGE
#undef VERSION
#undef HAVE_DTRACE
#undef _DTRACE_VERSION

31
#define TOKU_METADB_NAME "tokudb_meta"
Zardosht Kasheff's avatar
Zardosht Kasheff committed
32

33 34 35 36 37 38
typedef struct savepoint_info {
    DB_TXN* txn;
    tokudb_trx_data* trx;
    bool in_sub_stmt;
} *SP_INFO, SP_INFO_T;

Zardosht Kasheff's avatar
Zardosht Kasheff committed
39 40 41 42 43 44
static uchar *tokudb_get_key(TOKUDB_SHARE * share, size_t * length, my_bool not_used __attribute__ ((unused))) {
    *length = share->table_name_length;
    return (uchar *) share->table_name;
}

static handler *tokudb_create_handler(handlerton * hton, TABLE_SHARE * table, MEM_ROOT * mem_root);
45

46 47 48 49 50
static MYSQL_THDVAR_BOOL(commit_sync, 
    PLUGIN_VAR_THDLOCAL, 
    "sync on txn commit",
    /* check */ NULL, 
    /* update */ NULL,
Yoni Fogel's avatar
Yoni Fogel committed
51
    /* default*/ true
52
);
53
static MYSQL_THDVAR_UINT(pk_insert_mode,
54 55 56 57 58 59 60 61 62
    0,
    "set the primary key insert mode",
    NULL, 
    NULL, 
    1, // default
    0, // min?
    2, // max
    1 // blocksize
);
63
static MYSQL_THDVAR_BOOL(load_save_space,
64 65 66 67 68 69
    0,
    "if on, intial loads are slower but take less space",
    NULL, 
    NULL, 
    false
);
70
static MYSQL_THDVAR_BOOL(disable_slow_alter,
71 72 73 74 75 76
    0,
    "if on, alter tables that require copy are disabled",
    NULL, 
    NULL, 
    false
);
77
static MYSQL_THDVAR_BOOL(disable_hot_alter,
78 79 80 81 82 83
    0,
    "if on, hot alter table is disabled",
    NULL, 
    NULL, 
    false
);
84
static MYSQL_THDVAR_BOOL(create_index_online,
85 86 87 88 89 90
    0,
    "if on, create index done online",
    NULL, 
    NULL, 
    true
);
91
static MYSQL_THDVAR_BOOL(disable_prefetching,
92 93 94 95 96 97
    0,
    "if on, prefetching disabled",
    NULL, 
    NULL, 
   false
);
98
static MYSQL_THDVAR_BOOL(prelock_empty,
99 100 101 102 103 104
    0,
    "Tokudb Prelock Empty Table",
    NULL, 
    NULL, 
    true
);
105
static MYSQL_THDVAR_BOOL(log_client_errors,
106 107 108 109 110 111
    0,
    "Tokudb Log Client Errors",
    NULL, 
    NULL, 
    false
);
112
static MYSQL_THDVAR_UINT(block_size,
113 114 115 116 117 118
    0,
    "fractal tree block size",
    NULL, 
    NULL, 
    4<<20, // default
    4096,  // min
119
    ~0U,   // max
120 121
    1      // blocksize???
);
122
static MYSQL_THDVAR_UINT(read_block_size,
123 124 125 126 127 128
    0,
    "fractal tree read block size",
    NULL, 
    NULL, 
    128*1024, // default
    4096,  // min
129
    ~0U,   // max
130 131
    1      // blocksize???
);
132
static MYSQL_THDVAR_UINT(read_buf_size,
133 134 135 136 137 138 139 140 141
    0,
    "fractal tree read block size", //TODO: Is this a typo?
    NULL, 
    NULL, 
    128*1024, // default
    0,  // min
    1*1024*1024,   // max
    1      // blocksize???
);
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
#if TOKU_INCLUDE_UPSERT
static MYSQL_THDVAR_BOOL(enable_fast_update, 
    PLUGIN_VAR_THDLOCAL, 
    "enable fast update",
    NULL, // check
    NULL, // update
    false // default
);
static MYSQL_THDVAR_BOOL(disable_slow_update, 
    PLUGIN_VAR_THDLOCAL, 
    "disable slow update",
    NULL, // check
    NULL, // update
    false // default
);
static MYSQL_THDVAR_BOOL(enable_fast_upsert, 
    PLUGIN_VAR_THDLOCAL, 
    "enable fast upsert",
    NULL, // check
    NULL, // update
    false // default
);
static MYSQL_THDVAR_BOOL(disable_slow_upsert, 
    PLUGIN_VAR_THDLOCAL, 
    "disable slow upsert",
    NULL, // check
    NULL, // update
    false // default
);
#endif
172

173 174
static void tokudb_checkpoint_lock(THD * thd);
static void tokudb_checkpoint_unlock(THD * thd);
175

176
static void
177 178 179 180 181 182 183
tokudb_checkpoint_lock_update(
    THD* thd,
    struct st_mysql_sys_var* var,
    void* var_ptr,
    const void* save) 
{
    my_bool* val = (my_bool *) var_ptr;
Yoni Fogel's avatar
Yoni Fogel committed
184
    *val= *(my_bool *) save ? true : false;
185 186 187 188 189 190 191 192 193
    if (*val) {
        tokudb_checkpoint_lock(thd);
    }
    else {
        tokudb_checkpoint_unlock(thd);
    }
}
  
static MYSQL_THDVAR_BOOL(checkpoint_lock,
194 195 196 197 198 199
    0,
    "Tokudb Checkpoint Lock",
    NULL, 
    tokudb_checkpoint_lock_update, 
    false
);
200

201 202 203 204 205 206 207
static const char *tokudb_row_format_names[] = {
    "tokudb_uncompressed",
    "tokudb_zlib",
    "tokudb_quicklz",
    "tokudb_lzma",
    "tokudb_fast",
    "tokudb_small",
208
    "tokudb_default",
209 210 211 212 213 214 215 216 217 218
    NullS
};

static TYPELIB tokudb_row_format_typelib = {
    array_elements(tokudb_row_format_names) - 1,
    "tokudb_row_format_typelib",
    tokudb_row_format_names,
    NULL
};

219
static MYSQL_THDVAR_ENUM(row_format, PLUGIN_VAR_OPCMDARG,
220 221
                         "Specifies the compression method for a table during this session. "
                         "Possible values are TOKUDB_UNCOMPRESSED, TOKUDB_ZLIB, TOKUDB_QUICKLZ, "
222 223
                         "TOKUDB_LZMA, TOKUDB_FAST, TOKUDB_SMALL and TOKUDB_DEFAULT",
                         NULL, NULL, SRV_ROW_FORMAT_DEFAULT, &tokudb_row_format_typelib);
224 225 226 227 228 229

srv_row_format_t get_row_format(THD *thd)
{
    return (srv_row_format_t) THDVAR(thd, row_format);
}

Zardosht Kasheff's avatar
Zardosht Kasheff committed
230 231 232 233 234 235 236 237
static void tokudb_print_error(const DB_ENV * db_env, const char *db_errpfx, const char *buffer);
static void tokudb_cleanup_log_files(void);
static int tokudb_end(handlerton * hton, ha_panic_function type);
static bool tokudb_flush_logs(handlerton * hton);
static bool tokudb_show_status(handlerton * hton, THD * thd, stat_print_fn * print, enum ha_stat_type);
static int tokudb_close_connection(handlerton * hton, THD * thd);
static int tokudb_commit(handlerton * hton, THD * thd, bool all);
static int tokudb_rollback(handlerton * hton, THD * thd, bool all);
238
#if TOKU_INCLUDE_XA
239 240 241 242
static int tokudb_xa_prepare(handlerton* hton, THD* thd, bool all);
static int tokudb_xa_recover(handlerton* hton, XID*  xid_list, uint  len);
static int tokudb_commit_by_xid(handlerton* hton, XID* xid);
static int tokudb_rollback_by_xid(handlerton* hton, XID*  xid);
243
#endif
244

245
#if defined(HA_GENERAL_ONLINE) || defined(HA_INPLACE_ADD_INDEX_NO_READ_WRITE)
Zardosht Kasheff's avatar
Zardosht Kasheff committed
246
static uint tokudb_alter_table_flags(uint flags);
247
#endif
Zardosht Kasheff's avatar
Zardosht Kasheff committed
248 249 250
static int tokudb_rollback_to_savepoint(handlerton * hton, THD * thd, void *savepoint);
static int tokudb_savepoint(handlerton * hton, THD * thd, void *savepoint);
static int tokudb_release_savepoint(handlerton * hton, THD * thd, void *savepoint);
251 252
static int tokudb_discover(handlerton *hton, THD* thd, const char *db, const char *name, uchar **frmblob, size_t *frmlen);
static int tokudb_discover2(handlerton *hton, THD* thd, const char *db, const char *name, bool translate_name,uchar **frmblob, size_t *frmlen);
Zardosht Kasheff's avatar
Zardosht Kasheff committed
253 254 255 256 257 258
handlerton *tokudb_hton;

const char *ha_tokudb_ext = ".tokudb";
char *tokudb_data_dir;
ulong tokudb_debug;
DB_ENV *db_env;
259
DB* metadata_db;
Zardosht Kasheff's avatar
Zardosht Kasheff committed
260 261
HASH tokudb_open_tables;
pthread_mutex_t tokudb_mutex;
262
pthread_mutex_t tokudb_meta_mutex;
Zardosht Kasheff's avatar
Zardosht Kasheff committed
263

264
static ulonglong tokudb_lock_timeout;
265 266
static ulong tokudb_cleaner_period;
static ulong tokudb_cleaner_iterations;
Zardosht Kasheff's avatar
Zardosht Kasheff committed
267

268 269 270 271 272
#define ASSERT_MSGLEN 1024

void toku_hton_assert_fail(const char* expr_as_string, const char * fun, const char * file, int line, int caller_errno) {
    char msg[ASSERT_MSGLEN];
    if (db_env) {
273 274
        snprintf(msg, ASSERT_MSGLEN, "Handlerton: %s ", expr_as_string);
        db_env->crash(db_env, msg, fun, file, line,caller_errno);
275 276
    }
    else {
277 278 279
        snprintf(msg, ASSERT_MSGLEN, "Handlerton assertion failed, no env, %s, %d, %s, %s (errno=%d)\n", file, line, fun, expr_as_string, caller_errno);
        perror(msg);
        fflush(stderr);
280 281 282 283
    }
    abort();
}

Yoni Fogel's avatar
Yoni Fogel committed
284 285
//my_bool tokudb_shared_data = false;
static uint32_t tokudb_init_flags = 
Zardosht Kasheff's avatar
Zardosht Kasheff committed
286 287 288 289
    DB_CREATE | DB_THREAD | DB_PRIVATE | 
    DB_INIT_LOCK | 
    DB_INIT_MPOOL |
    DB_INIT_TXN | 
290
    DB_INIT_LOG |
Zardosht Kasheff's avatar
Zardosht Kasheff committed
291
    DB_RECOVER;
Yoni Fogel's avatar
Yoni Fogel committed
292 293
static uint32_t tokudb_env_flags = 0;
// static uint32_t tokudb_lock_type = DB_LOCK_DEFAULT;
Zardosht Kasheff's avatar
Zardosht Kasheff committed
294 295 296
// static ulong tokudb_log_buffer_size = 0;
// static ulong tokudb_log_file_size = 0;
static ulonglong tokudb_cache_size = 0;
297
static ulonglong tokudb_max_lock_memory = 0;
Zardosht Kasheff's avatar
Zardosht Kasheff committed
298
static char *tokudb_home;
299
static char *tokudb_tmp_dir;
Zardosht Kasheff's avatar
Zardosht Kasheff committed
300 301 302 303
static char *tokudb_log_dir;
// static long tokudb_lock_scan_time = 0;
// static ulong tokudb_region_size = 0;
// static ulong tokudb_cache_parts = 1;
304
const char *tokudb_hton_name = "TokuDB";
Yoni Fogel's avatar
Yoni Fogel committed
305 306 307
static uint32_t tokudb_checkpointing_period;
uint32_t tokudb_write_status_frequency;
uint32_t tokudb_read_status_frequency;
Zardosht Kasheff's avatar
Zardosht Kasheff committed
308
#ifdef TOKUDB_VERSION
309
char *tokudb_version = (char*) TOKUDB_VERSION;
Zardosht Kasheff's avatar
Zardosht Kasheff committed
310
#else
311
char *tokudb_version;
Zardosht Kasheff's avatar
Zardosht Kasheff committed
312
#endif
313
static int tokudb_fs_reserve_percent;  // file system reserve as a percentage of total disk space
Zardosht Kasheff's avatar
Zardosht Kasheff committed
314

315
#if defined(_WIN32)
Zardosht Kasheff's avatar
Zardosht Kasheff committed
316 317 318
extern "C" {
#include "ydb.h"
}
319
#endif
Zardosht Kasheff's avatar
Zardosht Kasheff committed
320

321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340
// A flag set if the handlerton is in an initialized, usable state,
// plus a reader-write lock to protect it without serializing reads.
// Since we don't have static initializers for the opaque rwlock type,
// use constructor and destructor functions to create and destroy
// the lock before and after main(), respectively.
static int tokudb_hton_initialized;
static rw_lock_t tokudb_hton_initialized_lock;

static void create_tokudb_hton_intialized_lock(void)  __attribute__((constructor));
static void destroy_tokudb_hton_initialized_lock(void) __attribute__((destructor));

static void create_tokudb_hton_intialized_lock(void)
{
    my_rwlock_init(&tokudb_hton_initialized_lock, 0);
}

static void destroy_tokudb_hton_initialized_lock(void)
{
    rwlock_destroy(&tokudb_hton_initialized_lock);
}
341

Zardosht Kasheff's avatar
Zardosht Kasheff committed
342 343
static int tokudb_init_func(void *p) {
    TOKUDB_DBUG_ENTER("tokudb_init_func");
Zardosht Kasheff's avatar
Zardosht Kasheff committed
344
    int r;
345 346 347 348 349 350 351
#if defined(_WIN64)
        r = toku_ydb_init();
        if (r) {
            printf("got error %d\n", r);
            goto error;
        }
#endif
352 353 354 355 356

    // 3938: lock the handlerton's initialized status flag for writing
    r = rw_wrlock(&tokudb_hton_initialized_lock);
    assert(r == 0);

357 358
    db_env = NULL;
    metadata_db = NULL;
Zardosht Kasheff's avatar
Zardosht Kasheff committed
359 360 361

    tokudb_hton = (handlerton *) p;

362 363
    pthread_mutex_init(&tokudb_mutex, MY_MUTEX_INIT_FAST);
    pthread_mutex_init(&tokudb_meta_mutex, MY_MUTEX_INIT_FAST);
364
    (void) my_hash_init(&tokudb_open_tables, table_alias_charset, 32, 0, 0, (my_hash_get_key) tokudb_get_key, 0, 0);
Zardosht Kasheff's avatar
Zardosht Kasheff committed
365 366 367

    tokudb_hton->state = SHOW_OPTION_YES;
    // tokudb_hton->flags= HTON_CAN_RECREATE;  // QQQ this came from skeleton
368
    tokudb_hton->flags = HTON_CLOSE_CURSORS_AT_COMMIT;
Zardosht Kasheff's avatar
Zardosht Kasheff committed
369 370 371 372 373 374 375 376
#ifdef DB_TYPE_TOKUDB
    tokudb_hton->db_type = DB_TYPE_TOKUDB;
#else
    tokudb_hton->db_type = DB_TYPE_UNKNOWN;
#endif

    tokudb_hton->create = tokudb_create_handler;
    tokudb_hton->close_connection = tokudb_close_connection;
377 378

    tokudb_hton->savepoint_offset = sizeof(SP_INFO_T);
Zardosht Kasheff's avatar
Zardosht Kasheff committed
379 380 381
    tokudb_hton->savepoint_set = tokudb_savepoint;
    tokudb_hton->savepoint_rollback = tokudb_rollback_to_savepoint;
    tokudb_hton->savepoint_release = tokudb_release_savepoint;
382

383
    tokudb_hton->discover = tokudb_discover;
384 385 386
#if defined(MYSQL_HANDLERTON_INCLUDE_DISCOVER2)
    tokudb_hton->discover2 = tokudb_discover2;
#endif
Zardosht Kasheff's avatar
Zardosht Kasheff committed
387 388
    tokudb_hton->commit = tokudb_commit;
    tokudb_hton->rollback = tokudb_rollback;
389
#if TOKU_INCLUDE_XA
390 391 392 393
    tokudb_hton->prepare=tokudb_xa_prepare;
    tokudb_hton->recover=tokudb_xa_recover;
    tokudb_hton->commit_by_xid=tokudb_commit_by_xid;
    tokudb_hton->rollback_by_xid=tokudb_rollback_by_xid;
394
#endif
395

Zardosht Kasheff's avatar
Zardosht Kasheff committed
396 397 398
    tokudb_hton->panic = tokudb_end;
    tokudb_hton->flush_logs = tokudb_flush_logs;
    tokudb_hton->show_status = tokudb_show_status;
399
#if defined(HA_GENERAL_ONLINE) || defined(HA_INPLACE_ADD_INDEX_NO_READ_WRITE)
Zardosht Kasheff's avatar
Zardosht Kasheff committed
400
    tokudb_hton->alter_table_flags = tokudb_alter_table_flags;
401
#endif
Zardosht Kasheff's avatar
Zardosht Kasheff committed
402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431
    if (!tokudb_home)
        tokudb_home = mysql_real_data_home;
    DBUG_PRINT("info", ("tokudb_home: %s", tokudb_home));
#if 0
    if (!tokudb_log_buffer_size) { // QQQ
        tokudb_log_buffer_size = max(table_cache_size * 512, 32 * 1024);
        DBUG_PRINT("info", ("computing tokudb_log_buffer_size %ld\n", tokudb_log_buffer_size));
    }
    tokudb_log_file_size = tokudb_log_buffer_size * 4;
    tokudb_log_file_size = MY_ALIGN(tokudb_log_file_size, 1024 * 1024L);
    tokudb_log_file_size = max(tokudb_log_file_size, 10 * 1024 * 1024L);
    DBUG_PRINT("info", ("computing tokudb_log_file_size: %ld\n", tokudb_log_file_size));
#endif
    if ((r = db_env_create(&db_env, 0))) {
        DBUG_PRINT("info", ("db_env_create %d\n", r));
        goto error;
    }

    DBUG_PRINT("info", ("tokudb_env_flags: 0x%x\n", tokudb_env_flags));
    r = db_env->set_flags(db_env, tokudb_env_flags, 1);
    if (r) { // QQQ
        if (tokudb_debug & TOKUDB_DEBUG_INIT) 
            TOKUDB_TRACE("%s:WARNING: flags=%x r=%d\n", __FUNCTION__, tokudb_env_flags, r); 
        // goto error;
    }

    // config error handling
    db_env->set_errcall(db_env, tokudb_print_error);
    db_env->set_errpfx(db_env, "TokuDB");

432 433 434 435 436 437 438 439 440
    //
    // set default comparison functions
    //
    r = db_env->set_default_bt_compare(db_env, tokudb_cmp_dbt_key);
    if (r) {
        DBUG_PRINT("info", ("set_default_bt_compare%d\n", r));
        goto error; 
    }

Zardosht Kasheff's avatar
Zardosht Kasheff committed
441
    {
442
    char *tmp_dir = tokudb_tmp_dir;
Zardosht Kasheff's avatar
Zardosht Kasheff committed
443
    char *data_dir = tokudb_data_dir;
444
    if (data_dir == 0) {
Zardosht Kasheff's avatar
Zardosht Kasheff committed
445
        data_dir = mysql_data_home;
446
    }
447 448
    if (tmp_dir == 0) {
        tmp_dir = data_dir;
449
    }
Zardosht Kasheff's avatar
Zardosht Kasheff committed
450 451
    DBUG_PRINT("info", ("tokudb_data_dir: %s\n", data_dir));
    db_env->set_data_dir(db_env, data_dir);
452

453 454
    DBUG_PRINT("info", ("tokudb_tmp_dir: %s\n", tmp_dir));
    db_env->set_tmp_dir(db_env, tmp_dir);
Zardosht Kasheff's avatar
Zardosht Kasheff committed
455 456 457 458 459 460 461
    }

    if (tokudb_log_dir) {
        DBUG_PRINT("info", ("tokudb_log_dir: %s\n", tokudb_log_dir));
        db_env->set_lg_dir(db_env, tokudb_log_dir);
    }

462
    // config the cache table size to min(1/2 of physical memory, 1/8 of the process address space)
Zardosht Kasheff's avatar
Zardosht Kasheff committed
463
    if (tokudb_cache_size == 0) {
464 465 466 467 468 469 470 471
        uint64_t physmem, maxdata;
        physmem = toku_os_get_phys_memory_size();
        tokudb_cache_size = physmem / 2;
        r = toku_os_get_max_process_data_size(&maxdata);
        if (r == 0) {
            if (tokudb_cache_size > maxdata / 8)
                tokudb_cache_size = maxdata / 8;
        }
Zardosht Kasheff's avatar
Zardosht Kasheff committed
472 473 474
    }
    if (tokudb_cache_size) {
        DBUG_PRINT("info", ("tokudb_cache_size: %lld\n", tokudb_cache_size));
Yoni Fogel's avatar
Yoni Fogel committed
475
        r = db_env->set_cachesize(db_env, (uint32_t)(tokudb_cache_size >> 30), (uint32_t)(tokudb_cache_size % (1024L * 1024L * 1024L)), 1);
Zardosht Kasheff's avatar
Zardosht Kasheff committed
476 477 478 479 480
        if (r) {
            DBUG_PRINT("info", ("set_cachesize %d\n", r));
            goto error; 
        }
    }
481 482 483 484 485 486 487 488 489 490 491 492
    if (tokudb_max_lock_memory == 0) {
        tokudb_max_lock_memory = tokudb_cache_size/8;
    }
    if (tokudb_max_lock_memory) {
        DBUG_PRINT("info", ("tokudb_max_lock_memory: %lld\n", tokudb_max_lock_memory));
        r = db_env->set_lk_max_memory(db_env, tokudb_max_lock_memory);
        if (r) {
            DBUG_PRINT("info", ("set_lk_max_memory %d\n", r));
            goto error; 
        }
    }
    
Yoni Fogel's avatar
Yoni Fogel committed
493
    uint32_t gbytes, bytes; int parts;
Zardosht Kasheff's avatar
Zardosht Kasheff committed
494 495 496 497 498 499 500 501 502 503 504 505 506 507 508
    r = db_env->get_cachesize(db_env, &gbytes, &bytes, &parts);
    if (r == 0) 
        if (tokudb_debug & TOKUDB_DEBUG_INIT) 
            TOKUDB_TRACE("%s:tokudb_cache_size=%lld\n", __FUNCTION__, ((unsigned long long) gbytes << 30) + bytes);

#if 0
    // QQQ config the logs
    DBUG_PRINT("info", ("tokudb_log_file_size: %ld\n", tokudb_log_file_size));
    db_env->set_lg_max(db_env, tokudb_log_file_size);
    DBUG_PRINT("info", ("tokudb_log_buffer_size: %ld\n", tokudb_log_buffer_size));
    db_env->set_lg_bsize(db_env, tokudb_log_buffer_size);
    // DBUG_PRINT("info",("tokudb_region_size: %ld\n", tokudb_region_size));
    // db_env->set_lg_regionmax(db_env, tokudb_region_size);
#endif

509 510 511 512 513 514
    if (db_env->set_redzone) {
        r = db_env->set_redzone(db_env, tokudb_fs_reserve_percent);
        if (r && (tokudb_debug & TOKUDB_DEBUG_INIT))
            TOKUDB_TRACE("%s:%d r=%d\n", __FUNCTION__, __LINE__, r);
    }

Zardosht Kasheff's avatar
Zardosht Kasheff committed
515 516
    if (tokudb_debug & TOKUDB_DEBUG_INIT) TOKUDB_TRACE("%s:env open:flags=%x\n", __FUNCTION__, tokudb_init_flags);

Zardosht Kasheff's avatar
Zardosht Kasheff committed
517
    r = db_env->set_generate_row_callback_for_put(db_env,generate_row_for_put);
518
    assert(!r);
519 520
    r = db_env->set_generate_row_callback_for_del(db_env,generate_row_for_del);
    assert(!r);
521
    db_env->set_update(db_env, tokudb_update_fun);
Zardosht Kasheff's avatar
Zardosht Kasheff committed
522
    r = db_env->open(db_env, tokudb_home, tokudb_init_flags, S_IRUSR|S_IWUSR|S_IRGRP|S_IWGRP|S_IROTH|S_IWOTH);
Zardosht Kasheff's avatar
Zardosht Kasheff committed
523 524 525 526 527 528 529 530

    if (tokudb_debug & TOKUDB_DEBUG_INIT) TOKUDB_TRACE("%s:env opened:return=%d\n", __FUNCTION__, r);

    if (r) {
        DBUG_PRINT("info", ("env->open %d\n", r));
        goto error;
    }

Zardosht Kasheff's avatar
Zardosht Kasheff committed
531 532
    r = db_env->checkpointing_set_period(db_env, tokudb_checkpointing_period);
    assert(!r);
533 534 535 536
    r = db_env->cleaner_set_period(db_env, tokudb_cleaner_period);
    assert(r == 0);
    r = db_env->cleaner_set_iterations(db_env, tokudb_cleaner_iterations);
    assert(r == 0);
Zardosht Kasheff's avatar
Zardosht Kasheff committed
537

538
    r = db_env->set_lock_timeout(db_env, tokudb_lock_timeout);
539 540
    assert(r == 0);

541 542 543 544 545 546 547
    r = db_create(&metadata_db, db_env, 0);
    if (r) {
        DBUG_PRINT("info", ("failed to create metadata db %d\n", r));
        goto error;
    }
    

548
    r= metadata_db->open(metadata_db, NULL, TOKU_METADB_NAME, NULL, DB_BTREE, DB_THREAD, 0);
549
    if (r) {
550 551 552 553 554 555
        if (r != ENOENT) {
            sql_print_error("Got error %d when trying to open metadata_db", r);
            goto error;
        }
        r = metadata_db->close(metadata_db,0);
        assert(r == 0);
556 557 558 559 560
        r = db_create(&metadata_db, db_env, 0);
        if (r) {
            DBUG_PRINT("info", ("failed to create metadata db %d\n", r));
            goto error;
        }
561 562

        r= metadata_db->open(metadata_db, NULL, TOKU_METADB_NAME, NULL, DB_BTREE, DB_THREAD | DB_CREATE | DB_EXCL, my_umask);
563 564 565 566 567
        if (r) {
            goto error;
        }
    }

568 569 570
    //3938: succeeded, set the init status flag and unlock
    tokudb_hton_initialized = 1;
    rw_unlock(&tokudb_hton_initialized_lock);
Yoni Fogel's avatar
Yoni Fogel committed
571
    DBUG_RETURN(false);
Zardosht Kasheff's avatar
Zardosht Kasheff committed
572 573

error:
574
    if (metadata_db) {
575 576
        int rr = metadata_db->close(metadata_db, 0);
        assert(rr==0);
577
    }
Zardosht Kasheff's avatar
Zardosht Kasheff committed
578
    if (db_env) {
579 580
        int rr= db_env->close(db_env, 0);
        assert(rr==0);
Zardosht Kasheff's avatar
Zardosht Kasheff committed
581 582
        db_env = 0;
    }
583 584 585 586

    // 3938: failed to initialized, drop the flag and lock
    tokudb_hton_initialized = 0;
    rw_unlock(&tokudb_hton_initialized_lock);
Yoni Fogel's avatar
Yoni Fogel committed
587
    DBUG_RETURN(true);
Zardosht Kasheff's avatar
Zardosht Kasheff committed
588 589 590 591
}

static int tokudb_done_func(void *p) {
    TOKUDB_DBUG_ENTER("tokudb_done_func");
592
    my_hash_free(&tokudb_open_tables);
Zardosht Kasheff's avatar
Zardosht Kasheff committed
593
    pthread_mutex_destroy(&tokudb_mutex);
594
    pthread_mutex_destroy(&tokudb_meta_mutex);
595
#if defined(_WIN64)
596
    toku_ydb_destroy();
597
#endif
Zardosht Kasheff's avatar
Zardosht Kasheff committed
598 599 600 601 602 603 604 605 606 607
    TOKUDB_DBUG_RETURN(0);
}

static handler *tokudb_create_handler(handlerton * hton, TABLE_SHARE * table, MEM_ROOT * mem_root) {
    return new(mem_root) ha_tokudb(hton, table);
}

int tokudb_end(handlerton * hton, ha_panic_function type) {
    TOKUDB_DBUG_ENTER("tokudb_end");
    int error = 0;
608 609 610 611 612 613 614 615
    
    // 3938: if we finalize the storage engine plugin, it is no longer
    // initialized. grab a writer lock for the duration of the
    // call, so we can drop the flag and destroy the mutexes
    // in isolation.
    rw_wrlock(&tokudb_hton_initialized_lock);
    assert(tokudb_hton_initialized);

616
    if (metadata_db) {
617 618
        int r = metadata_db->close(metadata_db, 0);
        assert(r==0);
619
    }
Zardosht Kasheff's avatar
Zardosht Kasheff committed
620 621 622 623
    if (db_env) {
        if (tokudb_init_flags & DB_INIT_LOG)
            tokudb_cleanup_log_files();
        error = db_env->close(db_env, 0);       // Error is logged
624
        assert(error==0);
Zardosht Kasheff's avatar
Zardosht Kasheff committed
625 626
        db_env = NULL;
    }
627 628 629 630 631

    // 3938: drop the initialized flag and unlock
    tokudb_hton_initialized = 0;
    rw_unlock(&tokudb_hton_initialized_lock);

Zardosht Kasheff's avatar
Zardosht Kasheff committed
632 633 634 635
    TOKUDB_DBUG_RETURN(error);
}

static int tokudb_close_connection(handlerton * hton, THD * thd) {
636 637 638 639 640 641 642 643
    int error = 0;
    tokudb_trx_data* trx = NULL;
    trx = (tokudb_trx_data *) thd_data_get(thd, tokudb_hton->slot);
    if (trx && trx->checkpoint_lock_taken) {
        error = db_env->checkpointing_resume(db_env);
    }
    my_free(trx, MYF(0));
    return error;
Zardosht Kasheff's avatar
Zardosht Kasheff committed
644 645 646 647 648 649
}

bool tokudb_flush_logs(handlerton * hton) {
    TOKUDB_DBUG_ENTER("tokudb_flush_logs");
    int error;
    bool result = 0;
Yoni Fogel's avatar
Yoni Fogel committed
650
    uint32_t curr_tokudb_checkpointing_period = 0;
Zardosht Kasheff's avatar
Zardosht Kasheff committed
651 652 653 654 655 656 657 658 659 660 661 662 663

    //
    // get the current checkpointing period
    //
    error = db_env->checkpointing_get_period(
        db_env, 
        &curr_tokudb_checkpointing_period
        );
    if (error) {
        my_error(ER_ERROR_DURING_CHECKPOINT, MYF(0), error);
        result = 1;
        goto exit;
    }
Zardosht Kasheff's avatar
Zardosht Kasheff committed
664

Zardosht Kasheff's avatar
Zardosht Kasheff committed
665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683
    //
    // if the current period is not the same as the variable, that means
    // the user has changed the period and now we need to update it
    //
    if (tokudb_checkpointing_period != curr_tokudb_checkpointing_period) {
        error = db_env->checkpointing_set_period(
            db_env, 
            tokudb_checkpointing_period
            );
        if (error) {
            my_error(ER_ERROR_DURING_CHECKPOINT, MYF(0), error);
            result = 1;
            goto exit;
        }
    }
    
    //
    // take the checkpoint
    //
Zardosht Kasheff's avatar
Zardosht Kasheff committed
684 685 686 687
    error = db_env->txn_checkpoint(db_env, 0, 0, 0);
    if (error) {
        my_error(ER_ERROR_DURING_CHECKPOINT, MYF(0), error);
        result = 1;
Zardosht Kasheff's avatar
Zardosht Kasheff committed
688
        goto exit;
Zardosht Kasheff's avatar
Zardosht Kasheff committed
689
    }
Zardosht Kasheff's avatar
Zardosht Kasheff committed
690 691 692

    result = 0;
exit:
Zardosht Kasheff's avatar
Zardosht Kasheff committed
693 694 695
    TOKUDB_DBUG_RETURN(result);
}

696 697 698 699
uint get_pk_insert_mode(THD* thd) {
    return THDVAR(thd, pk_insert_mode);
}

700 701 702 703
bool get_load_save_space(THD* thd) {
    return (THDVAR(thd, load_save_space) != 0);
}

704 705 706 707
bool get_disable_slow_alter(THD* thd) {
    return (THDVAR(thd, disable_slow_alter) != 0);
}

708 709 710 711
bool get_disable_hot_alter(THD* thd) {
    return THDVAR(thd, disable_hot_alter) != 0;
}

712 713 714 715
bool get_create_index_online(THD* thd) {
    return (THDVAR(thd, create_index_online) != 0);
}

716 717 718 719
bool get_disable_prefetching(THD* thd) {
    return (THDVAR(thd, disable_prefetching) != 0);
}

720 721 722
bool get_prelock_empty(THD* thd) {
    return (THDVAR(thd, prelock_empty) != 0);
}
723

724 725 726 727
bool get_log_client_errors(THD* thd) {
    return (THDVAR(thd, log_client_errors) != 0);
}

728 729 730 731
uint get_tokudb_block_size(THD* thd) {
    return THDVAR(thd, block_size);
}

732 733 734 735
uint get_tokudb_read_block_size(THD* thd) {
    return THDVAR(thd, read_block_size);
}

736 737 738 739
uint get_tokudb_read_buf_size(THD* thd) {
    return THDVAR(thd, read_buf_size);
}

740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757
#if TOKU_INCLUDE_UPSERT
bool get_enable_fast_update(THD* thd) {
    return (THDVAR(thd, enable_fast_update) != 0);
}

bool get_disable_slow_update(THD* thd) {
    return (THDVAR(thd, disable_slow_update) != 0);
}

bool get_enable_fast_upsert(THD* thd) {
    return (THDVAR(thd, enable_fast_upsert) != 0);
}

bool get_disable_slow_upsert(THD* thd) {
    return (THDVAR(thd, disable_slow_upsert) != 0);
}
#endif

758 759 760 761 762 763 764 765 766 767
typedef struct txn_progress_info {
    char status[200];
    THD* thd;
} *TXN_PROGRESS_INFO;


void txn_progress_func(TOKU_TXN_PROGRESS progress, void* extra) {
    TXN_PROGRESS_INFO progress_info = (TXN_PROGRESS_INFO)extra;
    int r;
    if (progress->stalled_on_checkpoint) {
768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785
        if (progress->is_commit) {
            r = sprintf(
                progress_info->status, 
                "Writing committed changes to disk, processing commit of transaction, %"PRId64" out of %"PRId64, 
                progress->entries_processed, 
                progress->entries_total
                ); 
            assert(r >= 0);
        }
        else {
            r = sprintf(
                progress_info->status, 
                "Writing committed changes to disk, processing abort of transaction, %"PRId64" out of %"PRId64, 
                progress->entries_processed, 
                progress->entries_total
                ); 
            assert(r >= 0);
        }
786 787
    }
    else {
788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803 804 805
        if (progress->is_commit) {
            r = sprintf(
                progress_info->status, 
                "processing commit of transaction, %"PRId64" out of %"PRId64, 
                progress->entries_processed, 
                progress->entries_total
                ); 
            assert(r >= 0);
        }
        else {
            r = sprintf(
                progress_info->status, 
                "processing abort of transaction, %"PRId64" out of %"PRId64, 
                progress->entries_processed, 
                progress->entries_total
                ); 
            assert(r >= 0);
        }
806 807 808 809 810
    }
    thd_proc_info(progress_info->thd, progress_info->status);
}


Yoni Fogel's avatar
Yoni Fogel committed
811
static void commit_txn_with_progress(DB_TXN* txn, uint32_t flags, THD* thd) {
812 813 814 815 816 817 818 819 820 821 822 823 824 825 826 827 828 829 830 831 832
    int r;
    struct txn_progress_info info;
    info.thd = thd;
    r = txn->commit_with_progress(txn, flags, txn_progress_func, &info);
    if (r != 0) {
        sql_print_error("tried committing transaction %p and got error code %d", txn, r);
    }
    assert(r == 0);
}

static void abort_txn_with_progress(DB_TXN* txn, THD* thd) {
    int r;
    struct txn_progress_info info;
    info.thd = thd;
    r = txn->abort_with_progress(txn, txn_progress_func, &info);
    if (r != 0) {
        sql_print_error("tried aborting transaction %p and got error code %d", txn, r);
    }
    assert(r == 0);
}

Zardosht Kasheff's avatar
Zardosht Kasheff committed
833 834 835
static int tokudb_commit(handlerton * hton, THD * thd, bool all) {
    TOKUDB_DBUG_ENTER("tokudb_commit");
    DBUG_PRINT("trans", ("ending transaction %s", all ? "all" : "stmt"));
Yoni Fogel's avatar
Yoni Fogel committed
836
    uint32_t syncflag = THDVAR(thd, commit_sync) ? 0 : DB_TXN_NOSYNC;
Zardosht Kasheff's avatar
Zardosht Kasheff committed
837 838 839
    tokudb_trx_data *trx = (tokudb_trx_data *) thd_data_get(thd, hton->slot);
    DB_TXN **txn = all ? &trx->all : &trx->stmt;
    if (*txn) {
840
        if (tokudb_debug & TOKUDB_DEBUG_TXN) {
841
            TOKUDB_TRACE("doing txn commit:%d:%p\n", all, *txn);
842
        }
843 844
        // test hook to induce a crash on a debug build
        DBUG_EXECUTE_IF("tokudb_crash_commit_before", DBUG_SUICIDE(););
845
        commit_txn_with_progress(*txn, syncflag, thd);
846 847
        // test hook to induce a crash on a debug build
        DBUG_EXECUTE_IF("tokudb_crash_commit_after", DBUG_SUICIDE(););
848
        if (*txn == trx->sp_level) {
Zardosht Kasheff's avatar
Zardosht Kasheff committed
849
            trx->sp_level = 0;
850
        }
Zardosht Kasheff's avatar
Zardosht Kasheff committed
851
        *txn = 0;
852
        trx->sub_sp_level = NULL;
Zardosht Kasheff's avatar
Zardosht Kasheff committed
853 854
    } 
    else if (tokudb_debug & TOKUDB_DEBUG_TXN) {
855
        TOKUDB_TRACE("nothing to commit %d\n", all);
Zardosht Kasheff's avatar
Zardosht Kasheff committed
856
    }
857
    reset_stmt_progress(&trx->stmt_progress);
858
    TOKUDB_DBUG_RETURN(0);
Zardosht Kasheff's avatar
Zardosht Kasheff committed
859 860 861 862 863 864 865 866
}

static int tokudb_rollback(handlerton * hton, THD * thd, bool all) {
    TOKUDB_DBUG_ENTER("tokudb_rollback");
    DBUG_PRINT("trans", ("aborting transaction %s", all ? "all" : "stmt"));
    tokudb_trx_data *trx = (tokudb_trx_data *) thd_data_get(thd, hton->slot);
    DB_TXN **txn = all ? &trx->all : &trx->stmt;
    if (*txn) {
867
        if (tokudb_debug & TOKUDB_DEBUG_TXN) {
Zardosht Kasheff's avatar
Zardosht Kasheff committed
868
            TOKUDB_TRACE("rollback:%p\n", *txn);
869
        }
870
        abort_txn_with_progress(*txn, thd);
871 872 873 874
        if (*txn == trx->sp_level) {
            trx->sp_level = 0;
        }
        *txn = 0;
875
        trx->sub_sp_level = NULL;
Zardosht Kasheff's avatar
Zardosht Kasheff committed
876 877 878
    } 
    else {
        if (tokudb_debug & TOKUDB_DEBUG_TXN) {
Zardosht Kasheff's avatar
Zardosht Kasheff committed
879
            TOKUDB_TRACE("abort0\n");
Zardosht Kasheff's avatar
Zardosht Kasheff committed
880 881
        }
    }
882
    reset_stmt_progress(&trx->stmt_progress);
883
    TOKUDB_DBUG_RETURN(0);
Zardosht Kasheff's avatar
Zardosht Kasheff committed
884 885
}

886 887
#if TOKU_INCLUDE_XA

888 889 890 891 892 893 894 895 896 897 898 899 900
static int tokudb_xa_prepare(handlerton* hton, THD* thd, bool all) {
    TOKUDB_DBUG_ENTER("tokudb_xa_prepare");
    int r = 0;
    DBUG_PRINT("trans", ("preparing transaction %s", all ? "all" : "stmt"));
    tokudb_trx_data *trx = (tokudb_trx_data *) thd_data_get(thd, hton->slot);
    DB_TXN* txn = all ? trx->all : trx->stmt;
    if (txn) {
        if (tokudb_debug & TOKUDB_DEBUG_TXN) {
            TOKUDB_TRACE("doing txn prepare:%d:%p\n", all, txn);
        }
        // a TOKU_XA_XID is identical to a MYSQL_XID
        TOKU_XA_XID thd_xid;
        thd_get_xid(thd, (MYSQL_XID*) &thd_xid);
901 902
        // test hook to induce a crash on a debug build
        DBUG_EXECUTE_IF("tokudb_crash_prepare_before", DBUG_SUICIDE(););
903
        r = txn->xa_prepare(txn, &thd_xid);
904 905
        // test hook to induce a crash on a debug build
        DBUG_EXECUTE_IF("tokudb_crash_prepare_after", DBUG_SUICIDE(););
906 907 908 909 910 911
    } 
    else if (tokudb_debug & TOKUDB_DEBUG_TXN) {
        TOKUDB_TRACE("nothing to prepare %d\n", all);
    }
    TOKUDB_DBUG_RETURN(r);
}
912

913 914 915 916
static int tokudb_xa_recover(handlerton* hton, XID*  xid_list, uint  len) {
    TOKUDB_DBUG_ENTER("tokudb_xa_recover");
    int r = 0;
    if (len == 0 || xid_list == NULL) {
917
        TOKUDB_DBUG_RETURN(0);
918 919 920 921 922 923 924 925 926
    }
    long num_returned = 0;
    r = db_env->txn_xa_recover(
        db_env,
        (TOKU_XA_XID*)xid_list,
        len,
        &num_returned,
        DB_NEXT
        );
927
    assert(r == 0);
928 929
    TOKUDB_DBUG_RETURN((int)num_returned);
}
930

931 932 933 934 935 936 937 938 939 940 941 942 943 944 945 946
static int tokudb_commit_by_xid(handlerton* hton, XID* xid) {
    TOKUDB_DBUG_ENTER("tokudb_commit_by_xid");
    int r = 0;
    DB_TXN* txn = NULL;
    TOKU_XA_XID* toku_xid = (TOKU_XA_XID*)xid;

    r = db_env->get_txn_from_xid(db_env, toku_xid, &txn);
    if (r) { goto cleanup; }

    r = txn->commit(txn, 0);
    if (r) { goto cleanup; }

    r = 0;
cleanup:
    TOKUDB_DBUG_RETURN(r);
}
947

948 949 950 951 952 953 954 955 956 957 958 959 960 961 962 963
static int tokudb_rollback_by_xid(handlerton* hton, XID*  xid) {
    TOKUDB_DBUG_ENTER("tokudb_rollback_by_xid");
    int r = 0;
    DB_TXN* txn = NULL;
    TOKU_XA_XID* toku_xid = (TOKU_XA_XID*)xid;

    r = db_env->get_txn_from_xid(db_env, toku_xid, &txn);
    if (r) { goto cleanup; }

    r = txn->abort(txn);
    if (r) { goto cleanup; }

    r = 0;
cleanup:
    TOKUDB_DBUG_RETURN(r);
}
Zardosht Kasheff's avatar
Zardosht Kasheff committed
964

965 966
#endif

Zardosht Kasheff's avatar
Zardosht Kasheff committed
967 968 969
static int tokudb_savepoint(handlerton * hton, THD * thd, void *savepoint) {
    TOKUDB_DBUG_ENTER("tokudb_savepoint");
    int error;
970
    SP_INFO save_info = (SP_INFO)savepoint;
Zardosht Kasheff's avatar
Zardosht Kasheff committed
971
    tokudb_trx_data *trx = (tokudb_trx_data *) thd_data_get(thd, hton->slot);
972 973
    if (thd->in_sub_stmt) {
        assert(trx->stmt);
974
        error = db_env->txn_begin(db_env, trx->sub_sp_level, &(save_info->txn), DB_INHERIT_ISOLATION);
975 976 977 978 979
        if (error) {
            goto cleanup;
        }
        trx->sub_sp_level = save_info->txn;
        save_info->in_sub_stmt = true;
Zardosht Kasheff's avatar
Zardosht Kasheff committed
980
    }
981
    else {
982
        error = db_env->txn_begin(db_env, trx->sp_level, &(save_info->txn), DB_INHERIT_ISOLATION);
983 984 985 986 987 988 989 990 991
        if (error) {
            goto cleanup;
        }
        trx->sp_level = save_info->txn;
        save_info->in_sub_stmt = false;
    }
    save_info->trx = trx;
    error = 0;
cleanup:
Zardosht Kasheff's avatar
Zardosht Kasheff committed
992 993 994 995 996 997
    TOKUDB_DBUG_RETURN(error);
}

static int tokudb_rollback_to_savepoint(handlerton * hton, THD * thd, void *savepoint) {
    TOKUDB_DBUG_ENTER("tokudb_rollback_to_savepoint");
    int error;
998 999 1000 1001
    SP_INFO save_info = (SP_INFO)savepoint;
    DB_TXN* parent = NULL;
    DB_TXN* txn_to_rollback = save_info->txn;

Zardosht Kasheff's avatar
Zardosht Kasheff committed
1002
    tokudb_trx_data *trx = (tokudb_trx_data *) thd_data_get(thd, hton->slot);
1003 1004 1005 1006 1007 1008 1009 1010
    parent = txn_to_rollback->parent;
    if (!(error = txn_to_rollback->abort(txn_to_rollback))) {
        if (save_info->in_sub_stmt) {
            trx->sub_sp_level = parent;
        }
        else {
            trx->sp_level = parent;
        }
Zardosht Kasheff's avatar
Zardosht Kasheff committed
1011 1012 1013 1014 1015 1016 1017 1018
        error = tokudb_savepoint(hton, thd, savepoint);
    }
    TOKUDB_DBUG_RETURN(error);
}

static int tokudb_release_savepoint(handlerton * hton, THD * thd, void *savepoint) {
    TOKUDB_DBUG_ENTER("tokudb_release_savepoint");
    int error;
1019 1020 1021 1022 1023

    SP_INFO save_info = (SP_INFO)savepoint;
    DB_TXN* parent = NULL;
    DB_TXN* txn_to_commit = save_info->txn;

Zardosht Kasheff's avatar
Zardosht Kasheff committed
1024
    tokudb_trx_data *trx = (tokudb_trx_data *) thd_data_get(thd, hton->slot);
1025 1026 1027 1028 1029 1030 1031 1032 1033
    parent = txn_to_commit->parent;
    if (!(error = txn_to_commit->commit(txn_to_commit, 0))) {
        if (save_info->in_sub_stmt) {
            trx->sub_sp_level = parent;
        }
        else {
            trx->sp_level = parent;
        }
        save_info->txn = NULL;
Zardosht Kasheff's avatar
Zardosht Kasheff committed
1034 1035 1036 1037
    }
    TOKUDB_DBUG_RETURN(error);
}

1038 1039 1040 1041 1042 1043
static int tokudb_discover(handlerton *hton, THD* thd, const char *db, const char *name, uchar **frmblob, size_t *frmlen) {
    return tokudb_discover2(hton, thd, db, name, true, frmblob, frmlen);
}

static int tokudb_discover2(handlerton *hton, THD* thd, const char *db, const char *name, bool translate_name,
                            uchar **frmblob, size_t *frmlen) {
1044 1045 1046 1047 1048 1049
    TOKUDB_DBUG_ENTER("tokudb_discover");
    int error;
    DB* status_db = NULL;
    DB_TXN* txn = NULL;
    char path[FN_REFLEN + 1];
    HA_METADATA_KEY curr_key = hatoku_frm_data;
1050
    DBT key, value;    
1051 1052
    memset(&key, 0, sizeof(key));
    memset(&value, 0, sizeof(&value));
1053
    
1054 1055 1056
    error = db_env->txn_begin(db_env, 0, &txn, 0);
    if (error) { goto cleanup; }

1057
    build_table_filename(path, sizeof(path) - 1, db, name, "", translate_name ? 0 : FN_IS_TMP);
1058 1059 1060 1061 1062
    error = open_status_dictionary(&status_db, path, txn);
    if (error) { goto cleanup; }

    key.data = &curr_key;
    key.size = sizeof(curr_key);
1063 1064

    error = status_db->getf_set(
1065
        status_db, 
1066 1067
        txn,
        0,
1068
        &key, 
1069 1070
        smart_dbt_callback_verify_frm,
        &value
1071 1072 1073 1074 1075
        );
    if (error) {
        goto cleanup;
    }

1076
    *frmblob = (uchar *)value.data;
1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087 1088
    *frmlen = value.size;

    error = 0;
cleanup:
    if (status_db) {
        status_db->close(status_db,0);
    }
    if (txn) {
        commit_txn(txn, 0);
    }
    TOKUDB_DBUG_RETURN(error);    
}
Zardosht Kasheff's avatar
Zardosht Kasheff committed
1089

Yoni Fogel's avatar
Yoni Fogel committed
1090
static int store_dbname_tablename_size(TABLE *table, char *name, uint64_t size, THD *thd) {
1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113
    char *tp = strrchr(name, '/');
    assert(tp);
    char *tablename = tp + 1;
    size_t tablename_length = strlen(tablename);

    char *dp = strchr(name, '/');
    char *dbname;
    size_t dbname_length;
    if (dp == tp) {
        dbname = name;
        dbname_length = tp - dbname;
    } else {
        dbname = dp + 1;
        dbname_length = tp - dbname;
    }

    table->field[0]->store(dbname, dbname_length, system_charset_info);
    table->field[1]->store(tablename, tablename_length, system_charset_info);
    table->field[2]->store(size, false);
    int error = schema_table_store_record(thd, table);
    return error;
}

1114 1115 1116 1117 1118 1119 1120 1121 1122 1123 1124 1125 1126 1127 1128 1129 1130 1131 1132 1133 1134 1135 1136 1137 1138 1139 1140 1141 1142 1143 1144 1145 1146 1147 1148 1149 1150 1151 1152 1153 1154 1155 1156 1157 1158 1159 1160 1161 1162 1163 1164
static int tokudb_dictionary_info(TABLE *table, THD *thd) {
    int error;
    DB_TXN* txn = NULL;
    DBC* tmp_cursor = NULL;
    DBT curr_key;
    DBT curr_val;
    memset(&curr_key, 0, sizeof curr_key); 
    memset(&curr_val, 0, sizeof curr_val);
    error = db_env->txn_begin(db_env, 0, &txn, DB_READ_UNCOMMITTED);
    if (error) {
        goto cleanup;
    }
    error = db_env->get_cursor_for_directory(db_env, txn, &tmp_cursor);
    if (error) {
        goto cleanup;
    }
    while (error == 0) {
        error = tmp_cursor->c_get(
            tmp_cursor, 
            &curr_key, 
            &curr_val, 
            DB_NEXT
            );
        if (!error) {
            table->field[0]->store(
                (char *)curr_key.data,
                curr_key.size,
                system_charset_info
                );
            table->field[1]->store(
                (char *)curr_val.data,
                curr_val.size,
                system_charset_info
                );
            error = schema_table_store_record(thd, table);
        }
    }
    if (error == DB_NOTFOUND) {
        error = 0;
    }
cleanup:
    if (tmp_cursor) {
        int r = tmp_cursor->c_close(tmp_cursor);
        assert(r==0);
    }
    if (txn) {
        commit_txn(txn, 0);
    }
    return error;
}

1165
static int tokudb_get_user_data_size(TABLE *table, THD *thd, bool exact) {
1166 1167 1168 1169
    int error;
    DB* curr_db = NULL;
    DB_TXN* txn = NULL;
    DBC* tmp_cursor = NULL;
1170
    DBC* tmp_table_cursor = NULL;
1171 1172
    DBT curr_key;
    DBT curr_val;
1173
    DB_TXN* tmp_txn = NULL;
1174 1175 1176 1177
    memset(&curr_key, 0, sizeof curr_key); 
    memset(&curr_val, 0, sizeof curr_val);
    pthread_mutex_lock(&tokudb_meta_mutex);

1178
    error = db_env->txn_begin(db_env, 0, &txn, DB_READ_UNCOMMITTED);
1179 1180 1181 1182 1183 1184 1185 1186
    if (error) {
        goto cleanup;
    }
    error = metadata_db->cursor(metadata_db, txn, &tmp_cursor, 0);
    if (error) {
        goto cleanup;
    }
    while (error == 0) {
1187
        tmp_txn = NULL;
1188 1189 1190 1191 1192 1193 1194
        //
        // here, and in other places, check if process has been killed
        // if so, get out of function so user is not stalled
        //
        if (thd->killed) {
            break;
        }
1195 1196 1197 1198 1199
        error = db_env->txn_begin(db_env, 0, &tmp_txn, DB_READ_UNCOMMITTED);
        if (error) {
            goto cleanup;
        }

1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 1210
        //
        // do not need this to be super fast, so use old simple API
        //
        error = tmp_cursor->c_get(
            tmp_cursor, 
            &curr_key, 
            &curr_val, 
            DB_NEXT
            );
        if (!error) {
            char* name = (char *)curr_key.data;
1211
            char* newname;
Yoni Fogel's avatar
Yoni Fogel committed
1212
            uint64_t curr_num_bytes = 0;
1213 1214
            DB_BTREE_STAT64 dict_stats;

1215 1216 1217
            error = db_create(&curr_db, db_env, 0);
            if (error) { goto cleanup; }

1218 1219
            newname = (char *)my_malloc(
                get_max_dict_name_path_length(name),
1220
                MYF(MY_WME|MY_ZEROFILL|MY_FAE));
1221 1222 1223

            make_name(newname, name, "main");
            
1224
            error = curr_db->open(curr_db, tmp_txn, newname, NULL, DB_BTREE, DB_THREAD, 0);
1225 1226 1227

            my_free(newname, MYF(0));

1228
            if (error == ENOENT) { error = 0; continue; }
1229 1230
            if (error) { goto cleanup; }

1231 1232 1233 1234 1235
            if (exact) {
                //
                // flatten if exact is required
                //
                uint curr_num_items = 0;                
1236
                error = curr_db->cursor(curr_db, tmp_txn, &tmp_table_cursor, 0);
1237 1238 1239 1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252
                if (error) {
                    tmp_table_cursor = NULL;
                    goto cleanup;
                }
                while (error != DB_NOTFOUND) {
                    error = tmp_table_cursor->c_getf_next(tmp_table_cursor, 0, smart_dbt_do_nothing, NULL);
                    if (error && error != DB_NOTFOUND) {
                        goto cleanup;
                    }
                    curr_num_items++;
                    //
                    // allow early exit if command has been killed
                    //
                    if ( (curr_num_items % 1000) == 0 && thd->killed) {
                        goto cleanup;
                    }
1253 1254 1255
                }
                error = tmp_table_cursor->c_close(tmp_table_cursor);
                assert(error==0);
1256 1257 1258
                tmp_table_cursor = NULL;
            }

1259 1260
            error = curr_db->stat64(
                curr_db, 
1261
                tmp_txn, 
1262 1263 1264 1265 1266 1267 1268 1269 1270 1271
                &dict_stats
                );
            if (error) { goto cleanup; }

            curr_num_bytes = dict_stats.bt_dsize;
            if (*(uchar *)curr_val.data) {
                //
                // in this case, we have a hidden primary key, do not
                // want to report space taken up by the hidden primary key to the user
                //
Yoni Fogel's avatar
Yoni Fogel committed
1272
                uint64_t hpk_space = TOKUDB_HIDDEN_PRIMARY_KEY_LENGTH*dict_stats.bt_ndata;
1273 1274 1275
                curr_num_bytes = (hpk_space > curr_num_bytes) ? 0 : curr_num_bytes - hpk_space;
            }
            else {
1276 1277 1278
                //
                // one infinity byte per key needs to be subtracted
                //
Yoni Fogel's avatar
Yoni Fogel committed
1279
                uint64_t inf_byte_space = dict_stats.bt_ndata;
1280 1281 1282
                curr_num_bytes = (inf_byte_space > curr_num_bytes) ? 0 : curr_num_bytes - inf_byte_space;
            }

1283
            error = store_dbname_tablename_size(table, name, curr_num_bytes, thd);
1284
            if (error) goto cleanup;
1285

1286 1287 1288 1289 1290
            {
                int r = curr_db->close(curr_db, 0);
                assert(r==0);
                curr_db = NULL;
            }
1291
        }
1292 1293 1294 1295 1296

        if (tmp_txn) {
            commit_txn(tmp_txn, 0);
            tmp_txn = NULL;
        }
1297 1298 1299 1300 1301 1302
    }

    error = 0;

cleanup:
    if (tmp_cursor) {
1303 1304
        int r = tmp_cursor->c_close(tmp_cursor);
        assert(r==0);
1305
    }
1306
    if (tmp_table_cursor) {
1307
        int r = tmp_table_cursor->c_close(tmp_table_cursor);
Zardosht Kasheff's avatar
Zardosht Kasheff committed
1308
        assert(r==0);
1309
    }
Zardosht Kasheff's avatar
Zardosht Kasheff committed
1310 1311 1312 1313
    if (curr_db) {
        int r = curr_db->close(curr_db, 0);
        assert(r==0);
    }
1314 1315 1316
    if (tmp_txn) {
        commit_txn(tmp_txn, 0);
    }
1317
    if (txn) {
1318
        commit_txn(txn, 0);
1319 1320
    }
    if (error) {
1321
        sql_print_error("got an error %d in show_data_size\n", error);
1322 1323
    }
    pthread_mutex_unlock(&tokudb_meta_mutex);
1324 1325 1326
    return error;
}

1327
#define STATPRINT(legend, val) if (legend != NULL && val != NULL) stat_print(thd,   \
1328
                                          tokudb_hton_name, \
1329
                                          strlen(tokudb_hton_name), \
1330 1331 1332 1333 1334
                                          legend, \
                                          strlen(legend), \
                                          val, \
                                          strlen(val))

1335 1336
extern sys_var *intern_find_sys_var(const char *str, uint length, bool no_error);

1337 1338 1339
static bool tokudb_show_engine_status(THD * thd, stat_print_fn * stat_print) {
    TOKUDB_DBUG_ENTER("tokudb_show_engine_status");
    int error;
1340 1341 1342 1343 1344
    uint64_t panic;
    const int panic_string_len = 1024;
    char panic_string[panic_string_len] = {'\0'};
    uint64_t num_rows;
    fs_redzone_state redzone_state;
1345
    const int bufsiz = 1024;
1346
    char buf[bufsiz];
1347

1348
#if MYSQL_VERSION_ID < 50500
1349
    {
1350 1351 1352
        sys_var * version = intern_find_sys_var("version", 0, false);
        snprintf(buf, bufsiz, "%s", version->value_ptr(thd, (enum_var_type)0, (LEX_STRING*)NULL));
        STATPRINT("Version", buf);
1353
    }
1354
#endif
1355 1356 1357 1358 1359
    error = db_env->get_engine_status_num_rows (db_env, &num_rows);
    TOKU_ENGINE_STATUS_ROW_S mystat[num_rows];
    error = db_env->get_engine_status (db_env, mystat, num_rows, &redzone_state, &panic, panic_string, panic_string_len);

    if (strlen(panic_string)) {
1360
        STATPRINT("Environment panic string", panic_string);
1361
    }
1362
    if (error == 0) {
1363 1364 1365 1366
        if (panic) {
            snprintf(buf, bufsiz, "%" PRIu64, panic);
            STATPRINT("Environment panic", buf);
        }
1367 1368 1369 1370 1371 1372 1373 1374 1375 1376 1377 1378 1379 1380 1381 1382 1383 1384 1385 1386 1387 1388 1389 1390 1391 1392 1393 1394 1395 1396 1397 1398
        
        if(redzone_state == FS_BLOCKED) {
            STATPRINT("*** URGENT WARNING ***", "FILE SYSTEM IS COMPLETELY FULL");
            snprintf(buf, bufsiz, "FILE SYSTEM IS COMPLETELY FULL");
        }
        else if (redzone_state == FS_GREEN) {
            snprintf(buf, bufsiz, "more than %d percent of total file system space", 2*tokudb_fs_reserve_percent);
        }
        else if (redzone_state == FS_YELLOW) {
            snprintf(buf, bufsiz, "*** WARNING *** FILE SYSTEM IS GETTING FULL (less than %d percent free)", 2*tokudb_fs_reserve_percent);
        } 
        else if (redzone_state == FS_RED){
            snprintf(buf, bufsiz, "*** WARNING *** FILE SYSTEM IS GETTING VERY FULL (less than %d percent free): INSERTS ARE PROHIBITED", tokudb_fs_reserve_percent);
        }
        else {
            snprintf(buf, bufsiz, "information unavailable, unknown redzone state %d", redzone_state);
        }
        STATPRINT ("disk free space", buf);

        for (uint64_t row = 0; row < num_rows; row++) {
            switch (mystat[row].type) {
            case FS_STATE:
                snprintf(buf, bufsiz, "%"PRIu64"", mystat[row].value.num);
                break;
            case UINT64:
                snprintf(buf, bufsiz, "%"PRIu64"", mystat[row].value.num);
                break;
            case CHARSTR:
                snprintf(buf, bufsiz, "%s", mystat[row].value.str);
                break;
            case UNIXTIME:
                {
1399
                    time_t t = mystat[row].value.num;
1400
                    char tbuf[26];
1401
                    snprintf(buf, bufsiz, "%.24s", ctime_r(&t, tbuf));
1402 1403 1404 1405 1406
                }
                break;
            case TOKUTIME:
                {
                    double t = tokutime_to_seconds(mystat[row].value.num);
1407
                    snprintf(buf, bufsiz, "%.6f", t);
1408 1409
                }
                break;
1410 1411 1412
            case PARCOUNT:
                {
                    uint64_t v = read_partitioned_counter(mystat[row].value.parcount);
1413
                    snprintf(buf, bufsiz, "%" PRIu64, v);
1414 1415 1416 1417 1418 1419
                }
                break;
#if 0
            case MAXCOUNT:
                {
                    uint64_t v = read_max_partitioned_counter(mystat[row].value.maxcount);
1420
                    snprintf(buf, bufsiz, "%" PRIu64, v);
1421 1422
                }
#endif
1423
            default:
1424
                snprintf(buf, bufsiz, "UNKNOWN STATUS TYPE: %d", mystat[row].type);
1425 1426 1427 1428 1429
                break;                
            }
            STATPRINT(mystat[row].legend, buf);
        }  
    }  
1430 1431 1432 1433
    if (error) { my_errno = error; }
    TOKUDB_DBUG_RETURN(error);
}

1434
static void tokudb_checkpoint_lock(THD * thd) {
1435 1436
    int error;
    tokudb_trx_data* trx = NULL;
1437
    char status_msg[200]; //buffer of 200 should be a good upper bound.
1438 1439
    trx = (tokudb_trx_data *) thd_data_get(thd, tokudb_hton->slot);
    if (!trx) {
1440
        error = create_tokudb_trx_data_instance(&trx);
1441 1442 1443 1444
        //
        // can only fail due to memory allocation, so ok to assert
        //
        assert(!error);
1445 1446 1447 1448 1449 1450
        thd_data_set(thd, tokudb_hton->slot, trx);
    }
    
    if (trx->checkpoint_lock_taken) {
        goto cleanup;
    }
1451 1452 1453 1454
    //
    // This can only fail if environment is not created, which is not possible
    // in handlerton
    //
1455 1456
    sprintf(status_msg, "Trying to grab checkpointing lock.");
    thd_proc_info(thd, status_msg);
1457
    error = db_env->checkpointing_postpone(db_env);
1458
    assert(!error);
1459 1460 1461

    trx->checkpoint_lock_taken = true;
cleanup:
1462
    return;
1463 1464
}

1465
static void tokudb_checkpoint_unlock(THD * thd) {
1466
    int error;
1467
    char status_msg[200]; //buffer of 200 should be a good upper bound.
1468 1469 1470 1471 1472 1473 1474 1475 1476 1477 1478 1479 1480
    tokudb_trx_data* trx = NULL;
    trx = (tokudb_trx_data *) thd_data_get(thd, tokudb_hton->slot);
    if (!trx) {
        error = 0;
        goto  cleanup;
    }
    if (!trx->checkpoint_lock_taken) {
        error = 0;
        goto  cleanup;
    }
    //
    // at this point, we know the checkpoint lock has been taken
    //
1481 1482
    sprintf(status_msg, "Trying to release checkpointing lock.");
    thd_proc_info(thd, status_msg);
1483
    error = db_env->checkpointing_resume(db_env);
1484
    assert(!error);
1485 1486 1487 1488

    trx->checkpoint_lock_taken = false;
    
cleanup:
1489
    return;
1490 1491
}

1492
static bool tokudb_show_status(handlerton * hton, THD * thd, stat_print_fn * stat_print, enum ha_stat_type stat_type) {
Zardosht Kasheff's avatar
Zardosht Kasheff committed
1493
    switch (stat_type) {
1494 1495 1496
    case HA_ENGINE_STATUS:
        return tokudb_show_engine_status(thd, stat_print);
        break;
Zardosht Kasheff's avatar
Zardosht Kasheff committed
1497
    default:
1498
        break;
Zardosht Kasheff's avatar
Zardosht Kasheff committed
1499
    }
Yoni Fogel's avatar
Yoni Fogel committed
1500
    return false;
Zardosht Kasheff's avatar
Zardosht Kasheff committed
1501 1502 1503 1504 1505 1506
}

static void tokudb_print_error(const DB_ENV * db_env, const char *db_errpfx, const char *buffer) {
    sql_print_error("%s:  %s", db_errpfx, buffer);
}

1507
static void tokudb_cleanup_log_files(void) {
Zardosht Kasheff's avatar
Zardosht Kasheff committed
1508 1509 1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521 1522 1523 1524 1525 1526 1527 1528 1529 1530 1531
    TOKUDB_DBUG_ENTER("tokudb_cleanup_log_files");
    char **names;
    int error;

    if ((error = db_env->txn_checkpoint(db_env, 0, 0, 0)))
        my_error(ER_ERROR_DURING_CHECKPOINT, MYF(0), error);

    if ((error = db_env->log_archive(db_env, &names, 0)) != 0) {
        DBUG_PRINT("error", ("log_archive failed (error %d)", error));
        db_env->err(db_env, error, "log_archive");
        DBUG_VOID_RETURN;
    }

    if (names) {
        char **np;
        for (np = names; *np; ++np) {
#if 1
            if (tokudb_debug)
                TOKUDB_TRACE("%s:cleanup:%s\n", __FUNCTION__, *np);
#else
            my_delete(*np, MYF(MY_WME));
#endif
        }

1532
        free(names);
Zardosht Kasheff's avatar
Zardosht Kasheff committed
1533 1534 1535 1536 1537
    }

    DBUG_VOID_RETURN;
}

1538
#if defined(HA_GENERAL_ONLINE)
Zardosht Kasheff's avatar
Zardosht Kasheff committed
1539 1540 1541 1542 1543 1544 1545 1546 1547
//
// *******NOTE*****
// If the flags HA_ONLINE_DROP_INDEX and HA_ONLINE_DROP_UNIQUE_INDEX
// are ever added, prepare_drop_index and final_drop_index will need to be modified
// so that the actual deletion of DB's is done in final_drop_index and not prepare_drop_index
//
static uint tokudb_alter_table_flags(uint flags)
{
    return (HA_ONLINE_ADD_INDEX_NO_WRITES| HA_ONLINE_DROP_INDEX_NO_WRITES |
1548
            HA_ONLINE_ADD_UNIQUE_INDEX_NO_WRITES| HA_ONLINE_DROP_UNIQUE_INDEX_NO_WRITES|HA_GENERAL_ONLINE);
Zardosht Kasheff's avatar
Zardosht Kasheff committed
1549 1550

}
1551 1552 1553 1554
#elif defined(HA_INPLACE_ADD_INDEX_NO_READ_WRITE)
static uint tokudb_alter_table_flags(uint flags) {
    return HA_INPLACE_ADD_INDEX_NO_READ_WRITE
        |  HA_INPLACE_ADD_INDEX_NO_WRITE
1555
        |  HA_INPLACE_DROP_INDEX_NO_READ_WRITE
1556 1557 1558 1559
        |  HA_INPLACE_ADD_UNIQUE_INDEX_NO_READ_WRITE
        |  HA_INPLACE_ADD_UNIQUE_INDEX_NO_WRITE
        |  HA_INPLACE_DROP_UNIQUE_INDEX_NO_READ_WRITE;
}
1560
#endif
Zardosht Kasheff's avatar
Zardosht Kasheff committed
1561 1562 1563 1564 1565 1566 1567 1568 1569 1570 1571 1572 1573 1574 1575


// options flags
//   PLUGIN_VAR_THDLOCAL  Variable is per-connection
//   PLUGIN_VAR_READONLY  Server variable is read only
//   PLUGIN_VAR_NOSYSVAR  Not a server variable
//   PLUGIN_VAR_NOCMDOPT  Not a command line option
//   PLUGIN_VAR_NOCMDARG  No argument for cmd line
//   PLUGIN_VAR_RQCMDARG  Argument required for cmd line
//   PLUGIN_VAR_OPCMDARG  Argument optional for cmd line
//   PLUGIN_VAR_MEMALLOC  String needs memory allocated


// system variables

1576 1577 1578 1579 1580 1581 1582
static void tokudb_lock_timeout_update(THD * thd,
        struct st_mysql_sys_var * sys_var, 
        void * var, const void * save)
{
    ulonglong * timeout = (ulonglong *) var;

    *timeout = *(const ulonglong *) save;
1583
    db_env->set_lock_timeout(db_env, *timeout);
1584
}
Zardosht Kasheff's avatar
Zardosht Kasheff committed
1585

1586
#define DEFAULT_LOCK_TIMEOUT_MSEC 4000
1587

1588 1589
static MYSQL_SYSVAR_ULONGLONG(lock_timeout, tokudb_lock_timeout,
        0, "TokuDB lock timeout", 
1590
        NULL, tokudb_lock_timeout_update, DEFAULT_LOCK_TIMEOUT_MSEC,
1591
        0, ~0ULL, 0);
1592 1593 1594 1595 1596 1597 1598 1599 1600 1601 1602 1603 1604 1605 1606 1607 1608

static void tokudb_cleaner_period_update(THD * thd,
        struct st_mysql_sys_var * sys_var, 
        void * var, const void * save)
{
    ulong * cleaner_period = (ulong *) var;

    *cleaner_period = *(const ulong *) save;
    int r = db_env->cleaner_set_period(db_env, *cleaner_period);
    assert(r==0);
}

#define DEFAULT_CLEANER_PERIOD 1

static MYSQL_SYSVAR_ULONG(cleaner_period, tokudb_cleaner_period,
        0, "TokuDB cleaner_period", 
        NULL, tokudb_cleaner_period_update, DEFAULT_CLEANER_PERIOD,
1609
        0, ~0UL, 0);
1610 1611 1612 1613 1614 1615 1616 1617 1618 1619 1620 1621

static void tokudb_cleaner_iterations_update(THD * thd,
        struct st_mysql_sys_var * sys_var, 
        void * var, const void * save)
{
    ulong * cleaner_iterations = (ulong *) var;

    *cleaner_iterations = *(const ulong *) save;
    int r = db_env->cleaner_set_iterations(db_env, *cleaner_iterations);
    assert(r==0);
}

1622
#define DEFAULT_CLEANER_ITERATIONS 5
1623 1624 1625 1626

static MYSQL_SYSVAR_ULONG(cleaner_iterations, tokudb_cleaner_iterations,
        0, "TokuDB cleaner_iterations", 
        NULL, tokudb_cleaner_iterations_update, DEFAULT_CLEANER_ITERATIONS,
1627
        0, ~0UL, 0);
1628

1629 1630
static MYSQL_SYSVAR_ULONGLONG(cache_size, tokudb_cache_size,
        PLUGIN_VAR_READONLY, "TokuDB cache table size", NULL, NULL, 0,
1631 1632 1633
        0, ~0ULL, 0);
static MYSQL_SYSVAR_ULONGLONG(max_lock_memory, tokudb_max_lock_memory, PLUGIN_VAR_READONLY, "TokuDB max memory for locks", NULL, NULL, 0, 0, ~0ULL, 0);
static MYSQL_SYSVAR_ULONG(debug, tokudb_debug, 0, "TokuDB Debug", NULL, NULL, 0, 0, ~0UL, 0);
Zardosht Kasheff's avatar
Zardosht Kasheff committed
1634 1635 1636 1637 1638 1639 1640

static MYSQL_SYSVAR_STR(log_dir, tokudb_log_dir, PLUGIN_VAR_READONLY, "TokuDB Log Directory", NULL, NULL, NULL);

static MYSQL_SYSVAR_STR(data_dir, tokudb_data_dir, PLUGIN_VAR_READONLY, "TokuDB Data Directory", NULL, NULL, NULL);

static MYSQL_SYSVAR_STR(version, tokudb_version, PLUGIN_VAR_READONLY, "TokuDB Version", NULL, NULL, NULL);

1641
static MYSQL_SYSVAR_UINT(init_flags, tokudb_init_flags, PLUGIN_VAR_READONLY, "Sets TokuDB DB_ENV->open flags", NULL, NULL, tokudb_init_flags, 0, ~0U, 0);
Zardosht Kasheff's avatar
Zardosht Kasheff committed
1642

1643 1644 1645
static MYSQL_SYSVAR_UINT(checkpointing_period, tokudb_checkpointing_period, 0, "TokuDB Checkpointing period", NULL, NULL, 60, 0, ~0U, 0);
static MYSQL_SYSVAR_UINT(write_status_frequency, tokudb_write_status_frequency, 0, "TokuDB frequency that show processlist updates status of writes", NULL, NULL, 1000, 0, ~0U, 0);
static MYSQL_SYSVAR_UINT(read_status_frequency, tokudb_read_status_frequency, 0, "TokuDB frequency that show processlist updates status of reads", NULL, NULL, 10000, 0, ~0U, 0);
1646
static MYSQL_SYSVAR_INT(fs_reserve_percent, tokudb_fs_reserve_percent, PLUGIN_VAR_READONLY, "TokuDB file system space reserve (percent free required)", NULL, NULL, 5, 0, 100, 0);
1647
static MYSQL_SYSVAR_STR(tmp_dir, tokudb_tmp_dir, PLUGIN_VAR_READONLY, "Tokudb Tmp Dir", NULL, NULL, NULL);
1648

Zardosht Kasheff's avatar
Zardosht Kasheff committed
1649 1650
static struct st_mysql_sys_var *tokudb_system_variables[] = {
    MYSQL_SYSVAR(cache_size),
1651
    MYSQL_SYSVAR(max_lock_memory),
Zardosht Kasheff's avatar
Zardosht Kasheff committed
1652 1653 1654 1655
    MYSQL_SYSVAR(data_dir),
    MYSQL_SYSVAR(log_dir),
    MYSQL_SYSVAR(debug),
    MYSQL_SYSVAR(commit_sync),
1656
    MYSQL_SYSVAR(lock_timeout),
1657 1658
    MYSQL_SYSVAR(cleaner_period),
    MYSQL_SYSVAR(cleaner_iterations),
1659
    MYSQL_SYSVAR(pk_insert_mode),
1660
    MYSQL_SYSVAR(load_save_space),
1661
    MYSQL_SYSVAR(disable_slow_alter),
1662
    MYSQL_SYSVAR(disable_hot_alter),
1663
    MYSQL_SYSVAR(create_index_online),
1664
    MYSQL_SYSVAR(disable_prefetching),
Zardosht Kasheff's avatar
Zardosht Kasheff committed
1665 1666
    MYSQL_SYSVAR(version),
    MYSQL_SYSVAR(init_flags),
Zardosht Kasheff's avatar
Zardosht Kasheff committed
1667
    MYSQL_SYSVAR(checkpointing_period),
1668
    MYSQL_SYSVAR(prelock_empty),
1669
    MYSQL_SYSVAR(log_client_errors),
1670
    MYSQL_SYSVAR(checkpoint_lock),
1671 1672
    MYSQL_SYSVAR(write_status_frequency),
    MYSQL_SYSVAR(read_status_frequency),
1673
    MYSQL_SYSVAR(fs_reserve_percent),
1674
    MYSQL_SYSVAR(tmp_dir),
1675
    MYSQL_SYSVAR(block_size),
1676
    MYSQL_SYSVAR(read_block_size),
1677
    MYSQL_SYSVAR(read_buf_size),
1678
    MYSQL_SYSVAR(row_format),
1679 1680 1681 1682 1683 1684
#if TOKU_INCLUDE_UPSERT
    MYSQL_SYSVAR(enable_fast_update),
    MYSQL_SYSVAR(disable_slow_update),
    MYSQL_SYSVAR(enable_fast_upsert),
    MYSQL_SYSVAR(disable_slow_upsert),
#endif
Zardosht Kasheff's avatar
Zardosht Kasheff committed
1685 1686 1687
    NULL
};

1688 1689 1690
struct st_mysql_storage_engine tokudb_storage_engine = { MYSQL_HANDLERTON_INTERFACE_VERSION };

static ST_FIELD_INFO tokudb_user_data_field_info[] = {
1691 1692 1693
    {"database_name", 256, MYSQL_TYPE_STRING, 0, 0, NULL, SKIP_OPEN_TABLE },
    {"table_name", 256, MYSQL_TYPE_STRING, 0, 0, NULL, SKIP_OPEN_TABLE },
    {"data_size", 0, MYSQL_TYPE_LONGLONG, 0, 0, NULL, SKIP_OPEN_TABLE },
1694
    {NULL, 0, MYSQL_TYPE_NULL, 0, 0, NULL, SKIP_OPEN_TABLE}
1695 1696
};

1697 1698 1699
#if MYSQL_VERSION_ID >= 50600
static int tokudb_user_data_fill_table(THD *thd, TABLE_LIST *tables, Item *cond) {
#else
1700
static int tokudb_user_data_fill_table(THD *thd, TABLE_LIST *tables, COND *cond) {
1701
#endif
1702 1703
    int error;
    TABLE *table = tables->table;
1704 1705 1706 1707 1708 1709
    
    // 3938: Get a read lock on the status flag, since we must
    // read it before safely proceeding
    rw_rdlock(&tokudb_hton_initialized_lock);

    if (!tokudb_hton_initialized) {
1710 1711 1712
        my_error(ER_PLUGIN_IS_NOT_LOADED, MYF(0), "TokuDB");
        error = -1;
    } else {
1713
        error = tokudb_get_user_data_size(table, thd, false);
1714
    }
1715 1716 1717

    // 3938: unlock the status flag lock
    rw_unlock(&tokudb_hton_initialized_lock);
1718 1719 1720 1721 1722 1723 1724 1725 1726 1727 1728 1729 1730 1731
    return error;
}

static int tokudb_user_data_init(void *p) {
    ST_SCHEMA_TABLE *schema = (ST_SCHEMA_TABLE *) p;
    schema->fields_info = tokudb_user_data_field_info;
    schema->fill_table = tokudb_user_data_fill_table;
    return 0;
}

static int tokudb_user_data_done(void *p) {
    return 0;
}

1732
static struct st_mysql_information_schema tokudb_user_data_information_schema = { MYSQL_INFORMATION_SCHEMA_INTERFACE_VERSION };
1733 1734

static ST_FIELD_INFO tokudb_user_data_exact_field_info[] = {
1735 1736 1737
    {"database_name", 256, MYSQL_TYPE_STRING, 0, 0, NULL, SKIP_OPEN_TABLE },
    {"table_name", 256, MYSQL_TYPE_STRING, 0, 0, NULL, SKIP_OPEN_TABLE },
    {"data_size", 0, MYSQL_TYPE_LONGLONG, 0, 0, NULL, SKIP_OPEN_TABLE },
1738
    {NULL, 0, MYSQL_TYPE_NULL, 0, 0, NULL, SKIP_OPEN_TABLE}
1739 1740
};

1741
static ST_FIELD_INFO tokudb_dictionary_field_info[] = {
1742 1743
    {"dictionary_name", 256, MYSQL_TYPE_STRING, 0, 0, NULL, SKIP_OPEN_TABLE },
    {"internal_file_name", 256, MYSQL_TYPE_STRING, 0, 0, NULL, SKIP_OPEN_TABLE },
1744 1745 1746 1747 1748 1749 1750 1751 1752 1753 1754 1755 1756 1757 1758 1759 1760 1761 1762 1763 1764 1765 1766 1767 1768 1769 1770 1771
    {NULL, 0, MYSQL_TYPE_NULL, 0, 0, NULL, SKIP_OPEN_TABLE}
};


#if MYSQL_VERSION_ID >= 50600
static int tokudb_dictionary_info_fill_table(THD *thd, TABLE_LIST *tables, Item *cond) {
#else
static int tokudb_dictionary_info_fill_table(THD *thd, TABLE_LIST *tables, COND *cond) {
#endif
    int error;
    TABLE *table = tables->table;

    // 3938: Get a read lock on the status flag, since we must
    // read it before safely proceeding
    rw_rdlock(&tokudb_hton_initialized_lock);

    if (!tokudb_hton_initialized) {
        my_error(ER_PLUGIN_IS_NOT_LOADED, MYF(0), "TokuDB");
        error = -1;
    } else {
        error = tokudb_dictionary_info(table, thd);
    }

    //3938: unlock the status flag lock
    rw_unlock(&tokudb_hton_initialized_lock);
    return error;
}

1772 1773 1774
#if MYSQL_VERSION_ID >= 50600
static int tokudb_user_data_exact_fill_table(THD *thd, TABLE_LIST *tables, Item *cond) {
#else
1775
static int tokudb_user_data_exact_fill_table(THD *thd, TABLE_LIST *tables, COND *cond) {
1776
#endif
1777 1778
    int error;
    TABLE *table = tables->table;
1779 1780 1781 1782 1783 1784

    // 3938: Get a read lock on the status flag, since we must
    // read it before safely proceeding
    rw_rdlock(&tokudb_hton_initialized_lock);

    if (!tokudb_hton_initialized) {
1785 1786 1787
        my_error(ER_PLUGIN_IS_NOT_LOADED, MYF(0), "TokuDB");
        error = -1;
    } else {
1788
        error = tokudb_get_user_data_size(table, thd, true);
1789
    }
1790 1791 1792

    //3938: unlock the status flag lock
    rw_unlock(&tokudb_hton_initialized_lock);
1793 1794 1795 1796 1797 1798 1799 1800 1801 1802 1803 1804 1805 1806
    return error;
}

static int tokudb_user_data_exact_init(void *p) {
    ST_SCHEMA_TABLE *schema = (ST_SCHEMA_TABLE *) p;
    schema->fields_info = tokudb_user_data_exact_field_info;
    schema->fill_table = tokudb_user_data_exact_fill_table;
    return 0;
}

static int tokudb_user_data_exact_done(void *p) {
    return 0;
}

1807 1808 1809 1810 1811 1812 1813 1814 1815 1816 1817
static int tokudb_dictionary_info_init(void *p) {
    ST_SCHEMA_TABLE *schema = (ST_SCHEMA_TABLE *) p;
    schema->fields_info = tokudb_dictionary_field_info;
    schema->fill_table = tokudb_dictionary_info_fill_table;
    return 0;
}

static int tokudb_dictionary_info_done(void *p) {
    return 0;
}

1818
static struct st_mysql_information_schema tokudb_user_data_exact_information_schema = { MYSQL_INFORMATION_SCHEMA_INTERFACE_VERSION };
1819 1820

enum { TOKUDB_PLUGIN_VERSION = 0x0400 };
1821
#define TOKUDB_PLUGIN_VERSION_STR "1024"
1822 1823 1824

mysql_declare_plugin(tokudb) 
{
Zardosht Kasheff's avatar
Zardosht Kasheff committed
1825
    MYSQL_STORAGE_ENGINE_PLUGIN, 
1826
    &tokudb_storage_engine, 
1827
    tokudb_hton_name, 
Zardosht Kasheff's avatar
Zardosht Kasheff committed
1828
    "Tokutek Inc", 
1829
    "Tokutek TokuDB Storage Engine with Fractal Tree(tm) Technology",
1830
    PLUGIN_LICENSE_PROPRIETARY,
Zardosht Kasheff's avatar
Zardosht Kasheff committed
1831 1832
    tokudb_init_func,          /* plugin init */
    tokudb_done_func,          /* plugin deinit */
1833
    TOKUDB_PLUGIN_VERSION,     /* 4.0.0 */
Zardosht Kasheff's avatar
Zardosht Kasheff committed
1834 1835
    NULL,                      /* status variables */
    tokudb_system_variables,   /* system variables */
1836 1837 1838 1839
    NULL,                      /* config options */
#if MYSQL_VERSION_ID >= 50521
    0,                         /* flags */
#endif
1840 1841 1842 1843 1844 1845 1846
},
{
    MYSQL_INFORMATION_SCHEMA_PLUGIN, 
    &tokudb_user_data_information_schema, 
    "TokuDB_user_data", 
    "Tokutek Inc", 
    "Tokutek TokuDB Storage Engine with Fractal Tree(tm) Technology",
1847
    PLUGIN_LICENSE_PROPRIETARY,
1848 1849 1850 1851 1852
    tokudb_user_data_init,     /* plugin init */
    tokudb_user_data_done,     /* plugin deinit */
    TOKUDB_PLUGIN_VERSION,     /* 4.0.0 */
    NULL,                      /* status variables */
    NULL,                      /* system variables */
1853 1854 1855 1856
    NULL,                      /* config options */
#if MYSQL_VERSION_ID >= 50521
    0,                         /* flags */
#endif
1857 1858 1859 1860 1861 1862 1863
},
{
    MYSQL_INFORMATION_SCHEMA_PLUGIN, 
    &tokudb_user_data_exact_information_schema, 
    "TokuDB_user_data_exact", 
    "Tokutek Inc", 
    "Tokutek TokuDB Storage Engine with Fractal Tree(tm) Technology",
1864
    PLUGIN_LICENSE_PROPRIETARY,
1865 1866 1867 1868 1869
    tokudb_user_data_exact_init,     /* plugin init */
    tokudb_user_data_exact_done,     /* plugin deinit */
    TOKUDB_PLUGIN_VERSION,     /* 4.0.0 */
    NULL,                      /* status variables */
    NULL,                      /* system variables */
1870 1871 1872 1873
    NULL,                      /* config options */
#if MYSQL_VERSION_ID >= 50521
    0,                         /* flags */
#endif
1874 1875 1876 1877
},
{
    MYSQL_INFORMATION_SCHEMA_PLUGIN, 
    &tokudb_user_data_exact_information_schema, 
1878
    "TokuDB_file_map", 
1879 1880 1881 1882 1883 1884 1885 1886 1887 1888 1889 1890
    "Tokutek Inc", 
    "Tokutek TokuDB Storage Engine with Fractal Tree(tm) Technology",
    PLUGIN_LICENSE_PROPRIETARY,
    tokudb_dictionary_info_init,     /* plugin init */
    tokudb_dictionary_info_done,     /* plugin deinit */
    TOKUDB_PLUGIN_VERSION,     /* 4.0.0 */
    NULL,                      /* status variables */
    NULL,                      /* system variables */
    NULL,                      /* config options */
#if MYSQL_VERSION_ID >= 50521
    0,                         /* flags */
#endif
Zardosht Kasheff's avatar
Zardosht Kasheff committed
1891 1892 1893
}
mysql_declare_plugin_end;

1894 1895 1896 1897 1898 1899
#ifdef MARIA_PLUGIN_INTERFACE_VERSION

maria_declare_plugin(tokudb) 
{
    MYSQL_STORAGE_ENGINE_PLUGIN, 
    &tokudb_storage_engine, 
1900
    tokudb_hton_name, 
1901 1902
    "Tokutek Inc", 
    "Tokutek TokuDB Storage Engine with Fractal Tree(tm) Technology",
1903
    PLUGIN_LICENSE_PROPRIETARY,
1904 1905 1906 1907 1908 1909 1910 1911 1912 1913 1914 1915 1916 1917
    tokudb_init_func,          /* plugin init */
    tokudb_done_func,          /* plugin deinit */
    TOKUDB_PLUGIN_VERSION,     /* 4.0.0 */
    NULL,                      /* status variables */
    tokudb_system_variables,   /* system variables */
    TOKUDB_PLUGIN_VERSION_STR, /* string version */
    MariaDB_PLUGIN_MATURITY_STABLE /* maturity */
},
{
    MYSQL_INFORMATION_SCHEMA_PLUGIN, 
    &tokudb_user_data_information_schema, 
    "TokuDB_user_data", 
    "Tokutek Inc", 
    "Tokutek TokuDB Storage Engine with Fractal Tree(tm) Technology",
1918
    PLUGIN_LICENSE_PROPRIETARY,
1919 1920 1921 1922 1923 1924 1925 1926 1927 1928 1929 1930 1931 1932
    tokudb_user_data_init,     /* plugin init */
    tokudb_user_data_done,     /* plugin deinit */
    TOKUDB_PLUGIN_VERSION,     /* 4.0.0 */
    NULL,                      /* status variables */
    NULL,                      /* system variables */
    TOKUDB_PLUGIN_VERSION_STR, /* string version */
    MariaDB_PLUGIN_MATURITY_STABLE /* maturity */
},
{
    MYSQL_INFORMATION_SCHEMA_PLUGIN, 
    &tokudb_user_data_exact_information_schema, 
    "TokuDB_user_data_exact", 
    "Tokutek Inc", 
    "Tokutek TokuDB Storage Engine with Fractal Tree(tm) Technology",
1933
    PLUGIN_LICENSE_PROPRIETARY,
1934 1935 1936 1937 1938 1939 1940
    tokudb_user_data_exact_init,     /* plugin init */
    tokudb_user_data_exact_done,     /* plugin deinit */
    TOKUDB_PLUGIN_VERSION,     /* 4.0.0 */
    NULL,                      /* status variables */
    NULL,                      /* system variables */
    TOKUDB_PLUGIN_VERSION_STR, /* string version */
    MariaDB_PLUGIN_MATURITY_STABLE /* maturity */
1941 1942 1943 1944
},
{
    MYSQL_INFORMATION_SCHEMA_PLUGIN, 
    &tokudb_user_data_exact_information_schema, 
1945
    "TokuDB_file_map", 
1946 1947 1948 1949 1950 1951 1952 1953 1954 1955
    "Tokutek Inc", 
    "Tokutek TokuDB Storage Engine with Fractal Tree(tm) Technology",
    PLUGIN_LICENSE_PROPRIETARY,
    tokudb_dictionary_info_init,     /* plugin init */
    tokudb_dictionary_info_done,     /* plugin deinit */
    TOKUDB_PLUGIN_VERSION,     /* 4.0.0 */
    NULL,                      /* status variables */
    NULL,                      /* system variables */
    TOKUDB_PLUGIN_VERSION_STR, /* string version */
    MariaDB_PLUGIN_MATURITY_STABLE /* maturity */
1956 1957 1958 1959
}
maria_declare_plugin_end;

#endif