a22cde620d
Statistics for these are a bit different than for a data source or a connection: 1) The statistics are simple (currently only 3), but range over sets of indices, so traversing the stats is a two level operation. Added a private callback API to advance to the next index when the stats have been exhausted. 2) cursors are used as single threaded operations, so arrays of stats is to avoid thread conflicts is not needed. Minor change: the stats_desc callback changed to return an int, and added an arg to provide some context. In the join cursor case, the returned desc is not a static string, it needs to be created and managed.
259 lines
7.7 KiB
Python
259 lines
7.7 KiB
Python
# Read the source files and output the statistics #defines plus the
|
|
# initialize and refresh code.
|
|
|
|
import re, string, sys, textwrap
|
|
from dist import compare_srcfile
|
|
|
|
# Read the source files.
|
|
from stat_data import groups, dsrc_stats, connection_stats, join_stats
|
|
|
|
def print_struct(title, name, base, stats):
|
|
'''Print the structures for the stat.h file.'''
|
|
f.write('/*\n')
|
|
f.write(' * Statistics entries for ' + title + '.\n')
|
|
f.write(' */\n')
|
|
f.write('#define\tWT_' + name.upper() + '_STATS_BASE\t' + str(base) + '\n')
|
|
f.write('struct __wt_' + name + '_stats {\n')
|
|
|
|
for l in stats:
|
|
f.write('\tint64_t ' + l.name + ';\n')
|
|
f.write('};\n\n')
|
|
|
|
# Update the #defines in the stat.h file.
|
|
tmp_file = '__tmp'
|
|
f = open(tmp_file, 'w')
|
|
skip = 0
|
|
for line in open('../src/include/stat.h', 'r'):
|
|
if not skip:
|
|
f.write(line)
|
|
if line.count('Statistics section: END'):
|
|
f.write(line)
|
|
skip = 0
|
|
elif line.count('Statistics section: BEGIN'):
|
|
f.write('\n')
|
|
skip = 1
|
|
print_struct(
|
|
'connections', 'connection', 1000, connection_stats)
|
|
print_struct('data sources', 'dsrc', 2000, dsrc_stats)
|
|
print_struct('join cursors', 'join', 3000, join_stats)
|
|
f.close()
|
|
compare_srcfile(tmp_file, '../src/include/stat.h')
|
|
|
|
def print_defines_one(capname, base, stats):
|
|
for v, l in enumerate(stats, base):
|
|
f.write('/*! %s */\n' % '\n * '.join(textwrap.wrap(l.desc, 70)))
|
|
f.write('#define\tWT_STAT_' + capname + '_' + l.name.upper() + "\t" *
|
|
max(1, 6 - int((len('WT_STAT_' + capname + '_' + l.name)) / 8)) +
|
|
str(v) + '\n')
|
|
|
|
def print_defines():
|
|
'''Print the #defines for the wiredtiger.in file.'''
|
|
f.write('''
|
|
/*!
|
|
* @name Connection statistics
|
|
* @anchor statistics_keys
|
|
* @anchor statistics_conn
|
|
* Statistics are accessed through cursors with \c "statistics:" URIs.
|
|
* Individual statistics can be queried through the cursor using the following
|
|
* keys. See @ref data_statistics for more information.
|
|
* @{
|
|
*/
|
|
''')
|
|
print_defines_one('CONN', 1000, connection_stats)
|
|
f.write('''
|
|
/*!
|
|
* @}
|
|
* @name Statistics for data sources
|
|
* @anchor statistics_dsrc
|
|
* @{
|
|
*/
|
|
''')
|
|
print_defines_one('DSRC', 2000, dsrc_stats)
|
|
f.write('''
|
|
/*!
|
|
* @}
|
|
* @name Statistics for join cursors
|
|
* @anchor statistics_join
|
|
* @{
|
|
*/
|
|
''')
|
|
print_defines_one('JOIN', 3000, join_stats)
|
|
f.write('/*! @} */\n')
|
|
|
|
# Update the #defines in the wiredtiger.in file.
|
|
tmp_file = '__tmp'
|
|
f = open(tmp_file, 'w')
|
|
skip = 0
|
|
for line in open('../src/include/wiredtiger.in', 'r'):
|
|
if not skip:
|
|
f.write(line)
|
|
if line.count('Statistics section: END'):
|
|
f.write(line)
|
|
skip = 0
|
|
elif line.count('Statistics section: BEGIN'):
|
|
f.write(' */\n')
|
|
skip = 1
|
|
print_defines()
|
|
f.write('/*\n')
|
|
f.close()
|
|
compare_srcfile(tmp_file, '../src/include/wiredtiger.in')
|
|
|
|
def print_func(name, handle, list):
|
|
'''Print the structures/functions for the stat.c file.'''
|
|
f.write('\n')
|
|
f.write('static const char * const __stats_' + name + '_desc[] = {\n')
|
|
for l in list:
|
|
f.write('\t"' + l.desc + '",\n')
|
|
f.write('};\n')
|
|
|
|
f.write('''
|
|
int
|
|
__wt_stat_''' + name + '''_desc(WT_CURSOR_STAT *cst, int slot, const char **p)
|
|
{
|
|
\tWT_UNUSED(cst);
|
|
\t*p = __stats_''' + name + '''_desc[slot];
|
|
\treturn (0);
|
|
}
|
|
''')
|
|
|
|
f.write('''
|
|
void
|
|
__wt_stat_''' + name + '_init_single(WT_' + name.upper() + '''_STATS *stats)
|
|
{
|
|
\tmemset(stats, 0, sizeof(*stats));
|
|
}
|
|
''')
|
|
|
|
if handle != None:
|
|
f.write('''
|
|
void
|
|
__wt_stat_''' + name + '_init(' + handle + ''' *handle)
|
|
{
|
|
\tint i;
|
|
|
|
\tfor (i = 0; i < WT_COUNTER_SLOTS; ++i) {
|
|
\t\thandle->stats[i] = &handle->stat_array[i];
|
|
\t\t__wt_stat_''' + name + '''_init_single(handle->stats[i]);
|
|
\t}
|
|
}
|
|
''')
|
|
|
|
f.write('''
|
|
void
|
|
__wt_stat_''' + name + '_clear_single(WT_' + name.upper() + '''_STATS *stats)
|
|
{
|
|
''')
|
|
for l in sorted(list):
|
|
# no_clear: don't clear the value.
|
|
if 'no_clear' in l.flags:
|
|
f.write('\t\t/* not clearing ' + l.name + ' */\n')
|
|
else:
|
|
f.write('\tstats->' + l.name + ' = 0;\n')
|
|
f.write('}\n')
|
|
|
|
f.write('''
|
|
void
|
|
__wt_stat_''' + name + '_clear_all(WT_' + name.upper() + '''_STATS **stats)
|
|
{
|
|
\tu_int i;
|
|
|
|
\tfor (i = 0; i < WT_COUNTER_SLOTS; ++i)
|
|
\t\t__wt_stat_''' + name + '''_clear_single(stats[i]);
|
|
}
|
|
''')
|
|
|
|
# Single structure aggregation is currently only used by data sources.
|
|
if name == 'dsrc':
|
|
f.write('''
|
|
void
|
|
__wt_stat_''' + name + '''_aggregate_single(
|
|
WT_''' + name.upper() + '_STATS *from, WT_' + name.upper() + '''_STATS *to)
|
|
{
|
|
''')
|
|
for l in sorted(list):
|
|
if 'no_aggregate' in l.flags:
|
|
o = '\tto->' + l.name + ' = from->' + l.name + ';\n'
|
|
elif 'max_aggregate' in l.flags:
|
|
o = '\tif (from->' + l.name + ' > to->' + l.name + ')\n' +\
|
|
'\t\tto->' + l.name + ' = from->' + l.name + ';\n'
|
|
else:
|
|
o = '\tto->' + l.name + ' += from->' + l.name + ';\n'
|
|
if len(o) > 72: # Account for the leading tab.
|
|
o = o.replace(' += ', ' +=\n\t ')
|
|
f.write(o)
|
|
f.write('}\n')
|
|
|
|
f.write('''
|
|
void
|
|
__wt_stat_''' + name + '''_aggregate(
|
|
WT_''' + name.upper() + '_STATS **from, WT_' + name.upper() + '''_STATS *to)
|
|
{
|
|
''')
|
|
# Connection level aggregation does not currently have any computation
|
|
# of a maximum value; I'm leaving in support for it, but don't declare
|
|
# a temporary variable until it's needed.
|
|
for l in sorted(list):
|
|
if 'max_aggregate' in l.flags:
|
|
f.write('\tint64_t v;\n\n')
|
|
break;
|
|
for l in sorted(list):
|
|
if 'no_aggregate' in l.flags:
|
|
o = '\tto->' + l.name + ' = from[0]->' + l.name + ';\n'
|
|
elif 'max_aggregate' in l.flags:
|
|
o = '\tif ((v = WT_STAT_READ(from, ' + l.name + ')) >\n' +\
|
|
'\t to->' + l.name + ')\n' +\
|
|
'\t\tto->' + l.name + ' = v;\n'
|
|
else:
|
|
o = '\tto->' + l.name + ' += WT_STAT_READ(from, ' + l.name + ');\n'
|
|
if len(o) > 72: # Account for the leading tab.
|
|
o = o.replace(' += ', ' +=\n\t ')
|
|
f.write(o)
|
|
f.write('}\n')
|
|
|
|
# Write the stat initialization and refresh routines to the stat.c file.
|
|
f = open(tmp_file, 'w')
|
|
f.write('/* DO NOT EDIT: automatically built by dist/stat.py. */\n\n')
|
|
f.write('#include "wt_internal.h"\n')
|
|
|
|
print_func('dsrc', 'WT_DATA_HANDLE', dsrc_stats)
|
|
print_func('connection', 'WT_CONNECTION_IMPL', connection_stats)
|
|
print_func('join', None, join_stats)
|
|
f.close()
|
|
compare_srcfile(tmp_file, '../src/support/stat.c')
|
|
|
|
# Update the statlog file with the entries we can scale per second.
|
|
scale_info = 'no_scale_per_second_list = [\n'
|
|
clear_info = 'no_clear_list = [\n'
|
|
prefix_list = []
|
|
for l in sorted(connection_stats):
|
|
prefix_list.append(l.prefix)
|
|
if 'no_scale' in l.flags:
|
|
scale_info += ' \'' + l.desc + '\',\n'
|
|
if 'no_clear' in l.flags:
|
|
clear_info += ' \'' + l.desc + '\',\n'
|
|
for l in sorted(dsrc_stats):
|
|
prefix_list.append(l.prefix)
|
|
if 'no_scale' in l.flags:
|
|
scale_info += ' \'' + l.desc + '\',\n'
|
|
if 'no_clear' in l.flags:
|
|
clear_info += ' \'' + l.desc + '\',\n'
|
|
# No join statistics can be captured in wtstats
|
|
scale_info += ']\n'
|
|
clear_info += ']\n'
|
|
prefix_info = 'prefix_list = [\n'
|
|
# Remove the duplicates and print out the list
|
|
for l in list(set(prefix_list)):
|
|
prefix_info += ' \'' + l + '\',\n'
|
|
prefix_info += ']\n'
|
|
group_info = 'groups = ' + str(groups)
|
|
|
|
tmp_file = '__tmp'
|
|
f = open(tmp_file, 'w')
|
|
f.write('# DO NOT EDIT: automatically built by dist/stat.py. */\n\n')
|
|
f.write(scale_info)
|
|
f.write(clear_info)
|
|
f.write(prefix_info)
|
|
f.write(group_info)
|
|
f.close()
|
|
compare_srcfile(tmp_file, '../tools/wtstats/stat_data.py')
|