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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
|
/*-
* Copyright (c) 2014-present MongoDB, Inc.
* Copyright (c) 2008-2014 WiredTiger, Inc.
* All rights reserved.
*
* See the file LICENSE for redistribution information.
*/
#include "wt_internal.h"
/*
* __wt_optrack_record_funcid --
* Allocate and record optrack function ID.
*/
void
__wt_optrack_record_funcid(WT_SESSION_IMPL *session, const char *func, uint16_t *func_idp)
{
static uint16_t optrack_uid = 0; /* Unique for the process lifetime. */
WT_CONNECTION_IMPL *conn;
WT_DECL_ITEM(tmp);
WT_DECL_RET;
wt_off_t fsize;
bool locked;
conn = S2C(session);
locked = false;
WT_ERR(__wt_scr_alloc(session, strlen(func) + 32, &tmp));
__wt_spin_lock(session, &conn->optrack_map_spinlock);
locked = true;
if (*func_idp == 0) {
*func_idp = ++optrack_uid;
WT_ERR(__wt_buf_fmt(session, tmp, "%" PRIu16 " %s\n", *func_idp, func));
WT_ERR(__wt_filesize(session, conn->optrack_map_fh, &fsize));
WT_ERR(__wt_write(session, conn->optrack_map_fh, fsize, tmp->size, tmp->data));
}
if (0) {
err:
WT_PANIC_MSG(session, ret, "operation tracking initialization failure");
}
if (locked)
__wt_spin_unlock(session, &conn->optrack_map_spinlock);
__wt_scr_free(session, &tmp);
}
/*
* __optrack_open_file --
* Open the per-session operation-tracking file.
*/
static int
__optrack_open_file(WT_SESSION_IMPL *session)
{
WT_CONNECTION_IMPL *conn;
WT_DECL_ITEM(buf);
WT_DECL_RET;
WT_OPTRACK_HEADER optrack_header = {
WT_OPTRACK_VERSION, 0, (uint32_t)WT_TSC_DEFAULT_RATIO * WT_THOUSAND};
conn = S2C(session);
if (!F_ISSET(conn, WT_CONN_OPTRACK))
WT_RET_MSG(session, WT_ERROR, "WT_CONN_OPTRACK not set");
WT_RET(__wt_scr_alloc(session, 0, &buf));
WT_ERR(__wt_filename_construct(
session, conn->optrack_path, "optrack", conn->optrack_pid, session->id, buf));
WT_ERR(__wt_open(session, (const char *)buf->data, WT_FS_OPEN_FILE_TYPE_REGULAR,
WT_FS_OPEN_CREATE, &session->optrack_fh));
/* Indicate whether this is an internal session */
if (F_ISSET(session, WT_SESSION_INTERNAL))
optrack_header.optrack_session_internal = 1;
/*
* Record the clock ticks to nanoseconds ratio. Multiply it by one thousand, so we can use a
* fixed width integer.
*/
optrack_header.optrack_tsc_nsec_ratio = (uint32_t)(__wt_process.tsc_nsec_ratio * WT_THOUSAND);
/* Write the header into the operation-tracking file. */
WT_ERR(session->optrack_fh->handle->fh_write(session->optrack_fh->handle, (WT_SESSION *)session,
0, sizeof(WT_OPTRACK_HEADER), &optrack_header));
session->optrack_offset = sizeof(WT_OPTRACK_HEADER);
if (0) {
err:
WT_TRET(__wt_close(session, &session->optrack_fh));
}
__wt_scr_free(session, &buf);
return (ret);
}
/*
* __wt_optrack_flush_buffer --
* Flush optrack buffer. Returns the number of bytes flushed to the file.
*/
void
__wt_optrack_flush_buffer(WT_SESSION_IMPL *s)
{
if (s->optrack_fh == NULL && __optrack_open_file(s) != 0)
return;
/*
* We're not using the standard write path deliberately, that's quite a bit of additional code
* (including atomic operations), and this work should be as light-weight as possible.
*/
if (s->optrack_fh->handle->fh_write(s->optrack_fh->handle, (WT_SESSION *)s,
(wt_off_t)s->optrack_offset, s->optrackbuf_ptr * sizeof(WT_OPTRACK_RECORD),
s->optrack_buf) == 0)
s->optrack_offset += s->optrackbuf_ptr * sizeof(WT_OPTRACK_RECORD);
}
|