← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to plugin/mysql_protocol/mysql_protocol.cc

  • Committer: Brian Aker
  • Date: 2011-01-08 04:23:10 UTC
  • mto: (2064.1.1 clean)
  • mto: This revision was merged to the branch mainline in revision 2065.
  • Revision ID: brian@tangent.org-20110108042310-wrcr0n473h37ljfm
First pass, remove retry.

Show diffs side-by-side

added added

removed removed

Lines of Context:
plugin/mysql_protocol/mysql_protocol.cc
17
17
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
18
18
 */
19
19
 
20
 
#include <config.h>
 
20
#include "config.h"
21
21
#include <drizzled/gettext.h>
22
22
#include <drizzled/error.h>
23
23
#include <drizzled/query_id.h>
24
 
#include <drizzled/error/sql_state.h>
 
24
#include <drizzled/sql_state.h>
25
25
#include <drizzled/session.h>
26
 
#include <drizzled/internal/m_string.h>
 
26
#include "drizzled/internal/m_string.h"
27
27
#include <algorithm>
28
28
#include <boost/program_options.hpp>
29
29
#include <drizzled/module/option_map.h>
30
 
#include <drizzled/util/tokenize.h>
 
30
#include "drizzled/util/tokenize.h"
31
31
#include "errmsg.h"
32
32
#include "mysql_protocol.h"
33
33
#include "mysql_password.h"
34
34
#include "options.h"
35
 
 
36
 
#include <drizzled/identifier.h>
37
 
 
38
 
#include <libdrizzle/constants.h>
 
35
#include "table_function.h"
 
36
 
 
37
#include "drizzled/identifier.h"
39
38
 
40
39
#define PROTOCOL_VERSION 10
41
40
 
67
66
ListenMySQLProtocol::~ListenMySQLProtocol()
68
67
{ }
69
68
 
70
 
void ListenMySQLProtocol::addCountersToTable()
71
 
{
72
 
  counters.push_back(new drizzled::plugin::ListenCounter(new std::string("connection_count"), &getCounters()->connectionCount));
73
 
  counters.push_back(new drizzled::plugin::ListenCounter(new std::string("connected"), &getCounters()->connected));
74
 
  counters.push_back(new drizzled::plugin::ListenCounter(new std::string("failed_connections"), &getCounters()->failedConnections));
75
 
}
76
 
 
77
69
const std::string ListenMySQLProtocol::getHost(void) const
78
70
{
79
71
  return _hostname;
404
396
}
405
397
 
406
398
 
407
 
void ClientMySQLProtocol::sendError(drizzled::error_t sql_errno, const char *err)
 
399
void ClientMySQLProtocol::sendError(uint32_t sql_errno, const char *err)
408
400
{
409
401
  uint32_t length;
410
402
  /*
412
404
  */
413
405
  unsigned char buff[2+1+SQLSTATE_LENGTH+DRIZZLE_ERRMSG_SIZE], *pos;
414
406
 
415
 
  assert(sql_errno != EE_OK);
 
407
  assert(sql_errno);
416
408
  assert(err && err[0]);
417
409
 
418
410
  /*
437
429
    return;
438
430
  }
439
431
 
440
 
  int2store(buff, static_cast<uint16_t>(sql_errno));
 
432
  int2store(buff,sql_errno);
441
433
  pos= buff+2;
442
434
 
443
435
  /* The first # is to make the client backward compatible */
444
436
  buff[2]= '#';
445
 
  pos= (unsigned char*) strcpy((char*) buff+3, error::convert_to_sqlstate(sql_errno));
446
 
  pos+= strlen(error::convert_to_sqlstate(sql_errno));
 
437
  pos= (unsigned char*) strcpy((char*) buff+3, drizzle_errno_to_sqlstate(sql_errno));
 
438
  pos+= strlen(drizzle_errno_to_sqlstate(sql_errno));
447
439
 
448
440
  char *tmp= strncpy((char*)pos, err, DRIZZLE_ERRMSG_SIZE-1);
449
441
  tmp+= strlen((char*)pos);
478
470
*/
479
471
bool ClientMySQLProtocol::sendFields(List<Item> *list)
480
472
{
481
 
  List<Item>::iterator it(list->begin());
 
473
  List_iterator_fast<Item> it(*list);
482
474
  Item *item;
483
475
  unsigned char buff[80];
484
476
  String tmp((char*) buff,sizeof(buff),&my_charset_bin);
551
543
        pos[6]= 15;
552
544
        break;
553
545
 
554
 
      case DRIZZLE_TYPE_MICROTIME:
555
 
        pos[6]= 15;
556
 
        break;
557
 
 
558
546
      case DRIZZLE_TYPE_UUID:
559
547
        pos[6]= 15;
560
548
        break;
561
549
 
562
550
      case DRIZZLE_TYPE_BOOLEAN:
563
 
        pos[6]= DRIZZLE_COLUMN_TYPE_TINY;
 
551
        pos[6]= 15;
564
552
        break;
565
553
 
566
554
      case DRIZZLE_TYPE_DECIMAL:
611
599
{
612
600
  if (from->is_null())
613
601
    return store();
614
 
  if (from->type() == DRIZZLE_TYPE_BOOLEAN)
615
 
  {
616
 
    return store(from->val_int());
617
 
  }
618
 
 
619
602
  char buff[MAX_FIELD_WIDTH];
620
603
  String str(buff,sizeof(buff), &my_charset_bin);
621
604
 
992
975
 
993
976
static int init(drizzled::module::Context &context)
994
977
{  
 
978
  context.add(new MysqlProtocolStatus);
 
979
 
995
980
  /* Initialize random seeds for the MySQL algorithm with minimal changes. */
996
981
  time_t seed_time= time(NULL);
997
982
  random_seed1= seed_time % random_max;
1003
988
  context.add(mysql_password);
1004
989
 
1005
990
  listen_obj= new ListenMySQLProtocol("mysql_protocol", vm["bind-address"].as<std::string>(), true);
1006
 
  listen_obj->addCountersToTable();
1007
991
  context.add(listen_obj); 
1008
992
  context.registerVariable(new sys_var_constrained_value_readonly<in_port_t>("port", port));
1009
993
  context.registerVariable(new sys_var_constrained_value<uint32_t>("connect_timeout", connect_timeout));
1023
1007
{
1024
1008
  context("port",
1025
1009
          po::value<port_constraint>(&port)->default_value(3306),
1026
 
          _("Port number to use for connection or 0 for default to with MySQL "
 
1010
          N_("Port number to use for connection or 0 for default to with MySQL "
1027
1011
                              "protocol."));
1028
1012
  context("connect-timeout",
1029
1013
          po::value<timeout_constraint>(&connect_timeout)->default_value(10),
1030
 
          _("Connect Timeout."));
 
1014
          N_("Connect Timeout."));
1031
1015
  context("read-timeout",
1032
1016
          po::value<timeout_constraint>(&read_timeout)->default_value(30),
1033
 
          _("Read Timeout."));
 
1017
          N_("Read Timeout."));
1034
1018
  context("write-timeout",
1035
1019
          po::value<timeout_constraint>(&write_timeout)->default_value(60),
1036
 
          _("Write Timeout."));
 
1020
          N_("Write Timeout."));
1037
1021
  context("retry-count",
1038
1022
          po::value<retry_constraint>(&retry_count)->default_value(10),
1039
 
          _("Retry Count."));
 
1023
          N_("Retry Count."));
1040
1024
  context("buffer-length",
1041
1025
          po::value<buffer_constraint>(&buffer_length)->default_value(16384),
1042
 
          _("Buffer length."));
 
1026
          N_("Buffer length."));
1043
1027
  context("bind-address",
1044
1028
          po::value<string>()->default_value(""),
1045
 
          _("Address to bind to."));
 
1029
          N_("Address to bind to."));
1046
1030
  context("max-connections",
1047
1031
          po::value<uint32_t>(&ListenMySQLProtocol::mysql_counters->max_connections)->default_value(1000),
1048
 
          _("Maximum simultaneous connections."));
 
1032
          N_("Maximum simultaneous connections."));
1049
1033
  context("admin-ip-addresses",
1050
1034
          po::value<vector<string> >()->composing()->notifier(&ClientMySQLProtocol::mysql_compose_ip_addresses),
1051
 
          _("A restrictive IP address list for incoming admin connections."));
 
1035
          N_("A restrictive IP address list for incoming admin connections."));
 
1036
}
 
1037
 
 
1038
static int mysql_protocol_connection_count_func(drizzle_show_var *var, char *buff)
 
1039
{
 
1040
  var->type= SHOW_LONGLONG;
 
1041
  var->value= buff;
 
1042
  *((uint64_t *)buff)= ListenMySQLProtocol::mysql_counters->connectionCount;
 
1043
  return 0;
 
1044
}
 
1045
 
 
1046
static int mysql_protocol_connected_count_func(drizzle_show_var *var, char *buff)
 
1047
{
 
1048
  var->type= SHOW_LONGLONG;
 
1049
  var->value= buff;
 
1050
  *((uint64_t *)buff)= ListenMySQLProtocol::mysql_counters->connected;
 
1051
  return 0;
 
1052
}
 
1053
 
 
1054
static int mysql_protocol_failed_count_func(drizzle_show_var *var, char *buff)
 
1055
{
 
1056
  var->type= SHOW_LONGLONG;
 
1057
  var->value= buff;
 
1058
  *((uint64_t *)buff)= ListenMySQLProtocol::mysql_counters->failedConnections;
 
1059
  return 0;
 
1060
}
 
1061
 
 
1062
static st_show_var_func_container mysql_protocol_connection_count=
 
1063
  { &mysql_protocol_connection_count_func };
 
1064
 
 
1065
static st_show_var_func_container mysql_protocol_connected_count=
 
1066
  { &mysql_protocol_connected_count_func };
 
1067
 
 
1068
static st_show_var_func_container mysql_protocol_failed_count=
 
1069
  { &mysql_protocol_failed_count_func };
 
1070
 
 
1071
static drizzle_show_var mysql_protocol_status_variables[]= {
 
1072
  {"Connections",
 
1073
  (char*) &mysql_protocol_connection_count, SHOW_FUNC},
 
1074
  {"Connected",
 
1075
  (char*) &mysql_protocol_connected_count, SHOW_FUNC},
 
1076
  {"Failed_connections",
 
1077
  (char*) &mysql_protocol_failed_count, SHOW_FUNC},
 
1078
  {NULL, NULL, SHOW_LONGLONG}
 
1079
};
 
1080
 
 
1081
MysqlProtocolStatus::Generator::Generator(drizzled::Field **fields) :
 
1082
  plugin::TableFunction::Generator(fields)
 
1083
{
 
1084
  status_var_ptr= mysql_protocol_status_variables;
 
1085
}
 
1086
 
 
1087
bool MysqlProtocolStatus::Generator::populate()
 
1088
{
 
1089
  MY_ALIGNED_BYTE_ARRAY(buff_data, SHOW_VAR_FUNC_BUFF_SIZE, int64_t);
 
1090
  char * const buff= (char *) &buff_data;
 
1091
  drizzle_show_var tmp;
 
1092
 
 
1093
  if (status_var_ptr->name)
 
1094
  {
 
1095
    std::ostringstream oss;
 
1096
    string return_value;
 
1097
    const char *value;
 
1098
    int type;
 
1099
 
 
1100
    push(status_var_ptr->name);
 
1101
 
 
1102
    if (status_var_ptr->type == SHOW_FUNC)
 
1103
    {
 
1104
      ((drizzle_show_var_func)((st_show_var_func_container *)status_var_ptr->value)->func)(&tmp, buff);
 
1105
      value= buff;
 
1106
      type= tmp.type;
 
1107
    }
 
1108
    else
 
1109
    {
 
1110
      value= status_var_ptr->value;
 
1111
      type= status_var_ptr->type;
 
1112
    }
 
1113
 
 
1114
    switch(type)
 
1115
    {
 
1116
    case SHOW_LONGLONG:
 
1117
      oss << *(uint64_t*) value;
 
1118
      return_value= oss.str();
 
1119
      break;
 
1120
    default:
 
1121
      assert(0);
 
1122
    }
 
1123
    if (return_value.length())
 
1124
      push(return_value);
 
1125
    else
 
1126
      push(" ");
 
1127
 
 
1128
    status_var_ptr++;
 
1129
 
 
1130
    return true;
 
1131
  }
 
1132
  return false;
1052
1133
}
1053
1134
 
1054
1135
} /* namespace drizzle_plugin */
1062
1143
  "MySQL Protocol Module",
1063
1144
  PLUGIN_LICENSE_GPL,
1064
1145
  drizzle_plugin::init,             /* Plugin Init */
1065
 
  NULL, /* depends */
 
1146
  NULL, /* system variables */
1066
1147
  drizzle_plugin::init_options    /* config options */
1067
1148
}
1068
1149
DRIZZLE_DECLARE_PLUGIN_END;