1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
/*-
* See the file LICENSE for redistribution information.
*
* Copyright (c) 2008-2011 WiredTiger, Inc.
* All rights reserved.
*/
#include "wt_internal.h"
/*
* __wt_session_add_btree --
* Add a btree handle to the session's cache.
*/
int
__wt_session_add_btree(SESSION *session, BTREE_SESSION **btree_sessionp)
{
const char *config;
char *format;
BTREE_SESSION *btree_session;
CONNECTION *conn;
WT_CONFIG_ITEM cval;
WT_RET(__wt_calloc_def(session, 1, &btree_session));
btree_session->btree = session->btree;
conn = S2C(session);
/*
* Make a copy of the key and value format, it's easier for everyone
* if they are NUL-terminated. They live in the BTREE_SESSION to save
* allocating memory on every cursor open.
*/
config = session->btree->config;
WT_RET(__wt_config_getones(config, "key_format", &cval));
WT_RET(__wt_calloc_def(session, cval.len + 1, &format));
memcpy(format, cval.str, cval.len);
btree_session->key_format = format;
WT_RET(__wt_config_getones(config, "value_format", &cval));
WT_RET(__wt_calloc_def(session, cval.len + 1, &format));
memcpy(format, cval.str, cval.len);
btree_session->value_format = format;
TAILQ_INSERT_HEAD(&session->btrees, btree_session, q);
__wt_lock(session, conn->mtx);
++session->btree->refcnt;
__wt_unlock(session, conn->mtx);
if (btree_sessionp != NULL)
*btree_sessionp = btree_session;
return (0);
}
/*
* __wt_session_get_btree --
* Get the btree handle for the named table.
*/
int
__wt_session_get_btree(SESSION *session,
const char *name, size_t namelen, BTREE_SESSION **btree_sessionp)
{
BTREE *btree;
BTREE_SESSION *btree_session;
TAILQ_FOREACH(btree_session, &session->btrees, q) {
btree = btree_session->btree;
if (strncmp(name, btree->name, namelen) == 0 &&
btree->name[namelen] == '\0') {
*btree_sessionp = btree_session;
return (0);
}
}
return (WT_NOTFOUND);
}
/*
* __wt_session_remove_btree --
* Remove the btree handle from the session, closing if necessary.
*/
int
__wt_session_remove_btree(SESSION *session, BTREE_SESSION *btree_session)
{
CONNECTION *conn;
int need_close;
conn = S2C(session);
TAILQ_REMOVE(&session->btrees, btree_session, q);
session->btree = btree_session->btree;
__wt_free(session, btree_session->key_format);
__wt_free(session, btree_session->value_format);
__wt_free(session, btree_session);
__wt_lock(session, conn->mtx);
WT_ASSERT(session, session->btree->refcnt > 0);
need_close = (--session->btree->refcnt == 0);
__wt_unlock(session, conn->mtx);
return (need_close ? __wt_btree_close(session) : 0);
}
|