| 1 | /* -*- mode: C++; c-basic-offset: 4; indent-tabs-mode: nil -*- */ |
| 2 | // vim: ft=cpp:expandtab:ts=8:sw=4:softtabstop=4: |
| 3 | #ident "$Id$" |
| 4 | /*====== |
| 5 | This file is part of PerconaFT. |
| 6 | |
| 7 | |
| 8 | Copyright (c) 2006, 2015, Percona and/or its affiliates. All rights reserved. |
| 9 | |
| 10 | PerconaFT is free software: you can redistribute it and/or modify |
| 11 | it under the terms of the GNU General Public License, version 2, |
| 12 | as published by the Free Software Foundation. |
| 13 | |
| 14 | PerconaFT is distributed in the hope that it will be useful, |
| 15 | but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 16 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| 17 | GNU General Public License for more details. |
| 18 | |
| 19 | You should have received a copy of the GNU General Public License |
| 20 | along with PerconaFT. If not, see <http://www.gnu.org/licenses/>. |
| 21 | |
| 22 | ---------------------------------------- |
| 23 | |
| 24 | PerconaFT is free software: you can redistribute it and/or modify |
| 25 | it under the terms of the GNU Affero General Public License, version 3, |
| 26 | as published by the Free Software Foundation. |
| 27 | |
| 28 | PerconaFT is distributed in the hope that it will be useful, |
| 29 | but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 30 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| 31 | GNU Affero General Public License for more details. |
| 32 | |
| 33 | You should have received a copy of the GNU Affero General Public License |
| 34 | along with PerconaFT. If not, see <http://www.gnu.org/licenses/>. |
| 35 | ======= */ |
| 36 | |
| 37 | #ident "Copyright (c) 2006, 2015, Percona and/or its affiliates. All rights reserved." |
| 38 | |
| 39 | #pragma once |
| 40 | |
| 41 | #include <ft/ft.h> |
| 42 | |
| 43 | #include "ydb-internal.h" |
| 44 | #include "ydb_txn.h" |
| 45 | |
| 46 | #include <memory> |
| 47 | |
| 48 | typedef enum { |
| 49 | YDB_LAYER_DIRECTORY_WRITE_LOCKS = 0, /* total directory write locks taken */ |
| 50 | YDB_LAYER_DIRECTORY_WRITE_LOCKS_FAIL, /* total directory write locks unable to be taken */ |
| 51 | YDB_LAYER_LOGSUPPRESS, /* number of times logs are suppressed for empty table (2440) */ |
| 52 | YDB_LAYER_LOGSUPPRESS_FAIL, /* number of times unable to suppress logs for empty table (2440) */ |
| 53 | YDB_DB_LAYER_STATUS_NUM_ROWS /* number of rows in this status array */ |
| 54 | } ydb_db_lock_layer_status_entry; |
| 55 | |
| 56 | typedef struct { |
| 57 | bool initialized; |
| 58 | TOKU_ENGINE_STATUS_ROW_S status[YDB_DB_LAYER_STATUS_NUM_ROWS]; |
| 59 | } YDB_DB_LAYER_STATUS_S, *YDB_DB_LAYER_STATUS; |
| 60 | |
| 61 | void ydb_db_layer_get_status(YDB_DB_LAYER_STATUS statp); |
| 62 | |
| 63 | // |
| 64 | // export the following locktree create/destroy callbacks so |
| 65 | // the environment can pass them to the locktree manager. |
| 66 | // |
| 67 | struct { |
| 68 | DB_TXN *; |
| 69 | FT_HANDLE ft_handle; |
| 70 | }; |
| 71 | int toku_db_lt_on_create_callback(toku::locktree *lt, void *); |
| 72 | void toku_db_lt_on_destroy_callback(toku::locktree *lt); |
| 73 | |
| 74 | /* db methods */ |
| 75 | static inline int db_opened(DB *db) { |
| 76 | return db->i->opened != 0; |
| 77 | } |
| 78 | |
| 79 | static inline const toku::comparator &toku_db_get_comparator(DB *db) { |
| 80 | return toku_ft_get_comparator(db->i->ft_handle); |
| 81 | } |
| 82 | |
| 83 | int toku_db_use_builtin_key_cmp(DB *db); |
| 84 | int toku_db_pre_acquire_fileops_lock(DB *db, DB_TXN *txn); |
| 85 | int toku_db_open_iname(DB * db, DB_TXN * txn, const char *iname, uint32_t flags, int mode); |
| 86 | int toku_db_pre_acquire_table_lock(DB *db, DB_TXN *txn); |
| 87 | int toku_db_get (DB * db, DB_TXN * txn, DBT * key, DBT * data, uint32_t flags); |
| 88 | int toku_db_create(DB ** db, DB_ENV * env, uint32_t flags); |
| 89 | int toku_db_close(DB * db); |
| 90 | int toku_setup_db_internal (DB **dbp, DB_ENV *env, uint32_t flags, FT_HANDLE ft_handle, bool is_open); |
| 91 | int db_getf_set(DB *db, DB_TXN *txn, uint32_t flags, DBT *key, YDB_CALLBACK_FUNCTION f, void *); |
| 92 | int autotxn_db_get(DB* db, DB_TXN* txn, DBT* key, DBT* data, uint32_t flags); |
| 93 | |
| 94 | //TODO: DB_AUTO_COMMIT. |
| 95 | //TODO: Nowait only conditionally? |
| 96 | //TODO: NOSYNC change to SYNC if DB_ENV has something in set_flags |
| 97 | static inline int |
| 98 | toku_db_construct_autotxn(DB* db, DB_TXN **txn, bool* changed, bool force_auto_commit) { |
| 99 | assert(db && txn && changed); |
| 100 | DB_ENV* env = db->dbenv; |
| 101 | if (*txn || !(env->i->open_flags & DB_INIT_TXN)) { |
| 102 | *changed = false; |
| 103 | return 0; |
| 104 | } |
| 105 | bool nosync = (bool)(!force_auto_commit && !(env->i->open_flags & DB_AUTO_COMMIT)); |
| 106 | uint32_t txn_flags = DB_TXN_NOWAIT | (nosync ? DB_TXN_NOSYNC : 0); |
| 107 | int r = toku_txn_begin(env, NULL, txn, txn_flags); |
| 108 | if (r!=0) return r; |
| 109 | *changed = true; |
| 110 | return 0; |
| 111 | } |
| 112 | |
| 113 | static inline int |
| 114 | toku_db_destruct_autotxn(DB_TXN *txn, int r, bool changed) { |
| 115 | if (!changed) return r; |
| 116 | if (r==0) { |
| 117 | r = locked_txn_commit(txn, 0); |
| 118 | } |
| 119 | else { |
| 120 | locked_txn_abort(txn); |
| 121 | } |
| 122 | return r; |
| 123 | } |
| 124 | |
| 125 | void create_iname_hint(DB_ENV *env, const char *dname, char *hint); |
| 126 | char *create_iname(DB_ENV *env, |
| 127 | uint64_t id1, |
| 128 | uint64_t id2, |
| 129 | char *hint, |
| 130 | const char *mark, |
| 131 | int n); |
| 132 | std::unique_ptr<char[], decltype(&toku_free)> generate_iname_for_rename_or_open( |
| 133 | DB_ENV *env, |
| 134 | DB_TXN *txn, |
| 135 | const char *dname, |
| 136 | bool is_open); |
| 137 | |