add opt for logging slow query

Qihoo360 · Oct 9, 2014 · 768c7ce · 768c7ce
1 parent 102f9fb
commit 768c7ce
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/plugins/proxy/proxy-plugin.c b/plugins/proxy/proxy-plugin.c
@@ -229,6 +229,7 @@ struct chassis_plugin_config {
 
 	FILE *sql_log;
 	gchar *sql_log_type;
+	gint sql_log_slow_ms;
 
 	gchar *charset;
 };
@@ -1681,18 +1682,20 @@ void merge_rows(network_mysqld_con* con, injection* inj) {
 
 void log_sql(network_mysqld_con* con, injection* inj) {
 	if (sql_log_type == OFF) return;
+
+	float latency_ms = (inj->ts_read_query_result_last - inj->ts_read_query)/1000.0;
+	if ((gint)latency_ms < config->sql_log_slow_ms) return;
 
 	GString* message = g_string_new(NULL);
 
 	time_t t = time(NULL);
 	struct tm* tm = localtime(&t);
 	g_string_printf(message, "[%02d/%02d/%d %02d:%02d:%02d] C:%s S:", tm->tm_mon+1, tm->tm_mday, tm->tm_year+1900, tm->tm_hour, tm->tm_min, tm->tm_sec, con->client->src->name->str);
-	gint latency = inj->ts_read_query_result_last - inj->ts_read_query;
 
 	if (inj->qstat.query_status == MYSQLD_PACKET_OK) {
-		g_string_append_printf(message, "%s OK %.3f \"%s\"\n", con->server->dst->name->str, latency/1000.0, inj->query->str+1);
+		g_string_append_printf(message, "%s OK %.3f \"%s\"\n", con->server->dst->name->str, latency_ms, inj->query->str+1);
 	} else {
-		g_string_append_printf(message, "%s ERR %.3f \"%s\"\n", con->server->dst->name->str, latency/1000.0, inj->query->str+1);
+		g_string_append_printf(message, "%s ERR %.3f \"%s\"\n", con->server->dst->name->str, latency_ms, inj->query->str+1);
 	}
 
 	fwrite(message->str, message->len, 1, config->sql_log);
@@ -2224,6 +2227,7 @@ chassis_plugin_config * network_mysqld_proxy_plugin_new(void) {
 	config->sql_log = NULL;
 	config->sql_log_type = NULL;
 	config->charset = NULL;
+	config->sql_log_slow_ms = 0;
 
 	return config;
 }
@@ -2330,6 +2334,7 @@ static GOptionEntry * network_mysqld_proxy_plugin_get_options(chassis_plugin_con
 		{ "charset", 0, 0, G_OPTION_ARG_STRING, NULL, "original charset(default: LATIN1)", NULL },
 
 		{ "sql-log", 0, 0, G_OPTION_ARG_STRING, NULL, "sql log type(default: OFF)", NULL },
+		{ "sql-log-slow", 0, 0, G_OPTION_ARG_INT, NULL, "only log sql which takes longer than this milliseconds (default: 0)", NULL },
 
 		{ NULL,                       0, 0, G_OPTION_ARG_NONE,   NULL, NULL, NULL }
 	};
@@ -2351,6 +2356,7 @@ static GOptionEntry * network_mysqld_proxy_plugin_get_options(chassis_plugin_con
 	config_entries[i++].arg_data = &(config->pwds);
 	config_entries[i++].arg_data = &(config->charset);
 	config_entries[i++].arg_data = &(config->sql_log_type);
+	config_entries[i++].arg_data = &(config->sql_log_slow_ms);
 
 	return config_entries;
 }