Skip to content

Commit

Permalink
Improved speed of optimizer trace
Browse files Browse the repository at this point in the history
- Added unlikely() to optimize for not having optimizer trace enabled
- Made THD::trace_started() inline
- Added 'if (trace_enabled())' around some potentially expensive code
  (not many found)
- Added ASSERT's to ensure we don't call expensive optimizer trace calls
  if optimizer trace is not enabled
- Added length to Json_writer functions to speed up buffer writes
  when optimizer trace is enabled.
- Changed LEX_CSTRING argument handling to not send full struct to writer
  function on_add_str() functions now trusts length arguments
  • Loading branch information
montywi committed Mar 9, 2020
1 parent 1ad8693 commit 940fcbe
Show file tree
Hide file tree
Showing 9 changed files with 176 additions and 102 deletions.
89 changes: 60 additions & 29 deletions sql/my_json_writer.cc
Original file line number Diff line number Diff line change
Expand Up @@ -79,15 +79,31 @@ void Json_writer::end_array()

Json_writer& Json_writer::add_member(const char *name)
{
if (fmt_helper.on_add_member(name))
size_t len= strlen(name);
if (fmt_helper.on_add_member(name, len))
return *this; // handled

// assert that we are in an object
DBUG_ASSERT(!element_started);
start_element();

output.append('"');
output.append(name);
output.append(name, len);
output.append("\": ", 3);
return *this;
}

Json_writer& Json_writer::add_member(const char *name, size_t len)
{
if (fmt_helper.on_add_member(name, len))
return *this; // handled

// assert that we are in an object
DBUG_ASSERT(!element_started);
start_element();

output.append('"');
output.append(name, len);
output.append("\": ");
return *this;
}
Expand Down Expand Up @@ -141,28 +157,31 @@ void Json_writer::add_ull(ulonglong val)
void Json_writer::add_size(longlong val)
{
char buf[64];
size_t len;
if (val < 1024)
my_snprintf(buf, sizeof(buf), "%lld", val);
len= my_snprintf(buf, sizeof(buf), "%lld", val);
else if (val < 1024*1024*16)
{
/* Values less than 16MB are specified in KB for precision */
size_t len= my_snprintf(buf, sizeof(buf), "%lld", val/1024);
len= my_snprintf(buf, sizeof(buf), "%lld", val/1024);
strcpy(buf + len, "Kb");
len+= 2;
}
else
{
size_t len= my_snprintf(buf, sizeof(buf), "%lld", val/(1024*1024));
len= my_snprintf(buf, sizeof(buf), "%lld", val/(1024*1024));
strcpy(buf + len, "Mb");
len+= 2;
}
add_str(buf);
add_str(buf, len);
}


void Json_writer::add_double(double val)
{
char buf[64];
my_snprintf(buf, sizeof(buf), "%lg", val);
add_unquoted_str(buf);
size_t len= my_snprintf(buf, sizeof(buf), "%lg", val);
add_unquoted_str(buf, len);
}


Expand All @@ -174,32 +193,46 @@ void Json_writer::add_bool(bool val)

void Json_writer::add_null()
{
add_unquoted_str("null");
add_unquoted_str("null", (size_t) 4);
}


void Json_writer::add_unquoted_str(const char* str)
{
if (fmt_helper.on_add_str(str, 0))
size_t len= strlen(str);
if (fmt_helper.on_add_str(str, len))
return;

if (!element_started)
start_element();

output.append(str, len);
element_started= false;
}

void Json_writer::add_unquoted_str(const char* str, size_t len)
{
if (fmt_helper.on_add_str(str, len))
return;

if (!element_started)
start_element();

output.append(str);
output.append(str, len);
element_started= false;
}

void Json_writer::add_str(const char *str)
{
if (fmt_helper.on_add_str(str, 0))
size_t len= strlen(str);
if (fmt_helper.on_add_str(str, len))
return;

if (!element_started)
start_element();

output.append('"');
output.append(str);
output.append(str, len);
output.append('"');
element_started= false;
}
Expand Down Expand Up @@ -230,41 +263,41 @@ void Json_writer::add_str(const String &str)
Json_writer_object::Json_writer_object(THD *thd) :
Json_writer_struct(thd)
{
if (my_writer)
if (unlikely(my_writer))
my_writer->start_object();
}

Json_writer_object::Json_writer_object(THD* thd, const char *str) :
Json_writer_struct(thd)
{
if (my_writer)
if (unlikely(my_writer))
my_writer->add_member(str).start_object();
}

Json_writer_object::~Json_writer_object()
{
if (!closed && my_writer)
if (my_writer && !closed)
my_writer->end_object();
closed= TRUE;
}

Json_writer_array::Json_writer_array(THD *thd) :
Json_writer_struct(thd)
{
if (my_writer)
if (unlikely(my_writer))
my_writer->start_array();
}

Json_writer_array::Json_writer_array(THD *thd, const char *str) :
Json_writer_struct(thd)
{
if (my_writer)
if (unlikely(my_writer))
my_writer->add_member(str).start_array();

}
Json_writer_array::~Json_writer_array()
{
if (!closed && my_writer)
if (unlikely(my_writer && !closed))
{
my_writer->end_array();
closed= TRUE;
Expand All @@ -281,7 +314,8 @@ Json_writer_temp_disable::~Json_writer_temp_disable()
thd->opt_trace.enable_tracing_if_required();
}

bool Single_line_formatting_helper::on_add_member(const char *name)
bool Single_line_formatting_helper::on_add_member(const char *name,
size_t len)
{
DBUG_ASSERT(state== INACTIVE || state == DISABLED);
if (state != DISABLED)
Expand All @@ -290,7 +324,6 @@ bool Single_line_formatting_helper::on_add_member(const char *name)
buf_ptr= buffer;

//append member name to the array
size_t len= strlen(name);
if (len < MAX_LINE_LEN)
{
memcpy(buf_ptr, name, len);
Expand Down Expand Up @@ -344,12 +377,10 @@ void Single_line_formatting_helper::on_start_object()


bool Single_line_formatting_helper::on_add_str(const char *str,
size_t num_bytes)
size_t len)
{
if (state == IN_ARRAY)
{
size_t len= num_bytes ? num_bytes : strlen(str);

// New length will be:
// "$string",
// quote + quote + comma + space = 4
Expand Down Expand Up @@ -425,23 +456,23 @@ void Single_line_formatting_helper::disable_and_flush()
while (ptr < buf_ptr)
{
char *str= ptr;
size_t len= strlen(str);

if (nr == 0)
{
owner->add_member(str);
owner->add_member(str, len);
if (start_array)
owner->start_array();
}
else
{
//if (nr == 1)
// owner->start_array();
owner->add_str(str);
owner->add_str(str, len);
}

nr++;
while (*ptr!=0)
ptr++;
ptr++;
ptr+= len+1;
}
buf_ptr= buffer;
state= INACTIVE;
Expand Down
Loading

0 comments on commit 940fcbe

Please sign in to comment.