← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to plugin/mysql_protocol/mysql_protocol.h

  • Committer: Mark Atwood
  • Date: 2011-08-11 03:05:03 UTC
  • mfrom: (2385.1.12 refactor4)
  • Revision ID: me@mark.atwood.name-20110811030503-rp9xjihc5x3y0x4q
merge lp:~olafvdspek/drizzle/refactor4

Show diffs side-by-side

added added

removed removed

Lines of Context:
plugin/mysql_protocol/mysql_protocol.h
17
17
 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
18
18
 */
19
19
 
20
 
#ifndef PLUGIN_MYSQL_PROTOCOL_MYSQL_PROTOCOL_H
21
 
#define PLUGIN_MYSQL_PROTOCOL_MYSQL_PROTOCOL_H
 
20
#pragma once
22
21
 
23
22
#include <drizzled/plugin/listen_tcp.h>
24
23
#include <drizzled/plugin/client.h>
25
24
#include <drizzled/atomics.h>
26
 
#include "drizzled/plugin/table_function.h"
 
25
#include <drizzled/plugin/table_function.h>
27
26
 
28
27
#include "net_serv.h"
29
28
 
30
 
namespace drizzle_plugin
31
 
{
32
 
void compose_ip_addresses(std::vector<std::string> options);
 
29
namespace drizzle_plugin {
33
30
 
34
31
class ProtocolCounters
35
32
{
36
 
  public:
37
 
    ProtocolCounters():
 
33
public:
 
34
  ProtocolCounters() :
38
35
      max_connections(1000)
39
 
    { }
40
 
    drizzled::atomic<uint64_t> connectionCount;
41
 
    drizzled::atomic<uint64_t> adminConnectionCount;
42
 
    drizzled::atomic<uint64_t> failedConnections;
43
 
    drizzled::atomic<uint64_t> connected;
44
 
    drizzled::atomic<uint64_t> adminConnected;
45
 
    uint32_t max_connections;
 
36
  { }
 
37
 
 
38
  drizzled::atomic<uint64_t> connectionCount;
 
39
  drizzled::atomic<uint64_t> failedConnections;
 
40
  drizzled::atomic<uint64_t> connected;
 
41
  uint32_t max_connections;
46
42
};
47
43
 
48
44
typedef drizzled::constrained_check<uint32_t, 300, 1> timeout_constraint;
51
47
 
52
48
class ListenMySQLProtocol: public drizzled::plugin::ListenTcp
53
49
{
54
 
protected:
55
 
  const std::string _hostname;
56
 
  bool _using_mysql41_protocol;
57
 
 
58
50
public:
59
 
  ListenMySQLProtocol(std::string name,
60
 
                      const std::string &hostname,
61
 
                      bool using_mysql41_protocol):
 
51
  ListenMySQLProtocol(std::string name, const std::string &hostname) :
62
52
   drizzled::plugin::ListenTcp(name),
63
 
   _hostname(hostname),
64
 
   _using_mysql41_protocol(using_mysql41_protocol)
 
53
   _hostname(hostname)
65
54
  { }
66
 
  virtual ~ListenMySQLProtocol();
67
 
  virtual const std::string getHost(void) const;
68
 
  virtual in_port_t getPort(void) const;
 
55
  virtual const std::string getHost() const;
 
56
  virtual in_port_t getPort() const;
69
57
  virtual drizzled::plugin::Client *getClient(int fd);
70
 
  static ProtocolCounters *mysql_counters;
71
 
  virtual ProtocolCounters *getCounters(void) const { return mysql_counters; }
72
 
  void addCountersToTable(void);
 
58
  static ProtocolCounters mysql_counters;
 
59
  virtual ProtocolCounters& getCounters() const { return mysql_counters; }
 
60
  void addCountersToTable();
 
61
protected:
 
62
  const std::string _hostname;
73
63
};
74
64
 
75
65
class ClientMySQLProtocol: public drizzled::plugin::Client
78
68
  NET net;
79
69
  drizzled::String packet;
80
70
  uint32_t client_capabilities;
81
 
  bool is_admin_connection;
82
 
  bool _using_mysql41_protocol;
 
71
  bool _is_interactive;
83
72
 
84
 
  bool checkConnection(void);
85
 
  bool netStoreData(const unsigned char *from, size_t length);
 
73
  bool checkConnection();
 
74
  void netStoreData(const void*, size_t);
86
75
  void writeEOFPacket(uint32_t server_status, uint32_t total_warn_count);
87
76
  unsigned char *storeLength(unsigned char *packet, uint64_t length);
88
77
  void makeScramble(char *scramble);
89
78
 
90
79
public:
91
 
  ClientMySQLProtocol(int fd, bool _using_mysql41_protocol, ProtocolCounters *set_counters);
 
80
  ClientMySQLProtocol(int fd, ProtocolCounters&);
92
81
  virtual ~ClientMySQLProtocol();
93
82
 
94
 
  ProtocolCounters *counters;
95
 
 
96
 
  virtual int getFileDescriptor(void);
 
83
  bool isInteractive() const
 
84
  {
 
85
    return _is_interactive;
 
86
  }
 
87
 
 
88
  ProtocolCounters& counters;
 
89
 
 
90
  virtual int getFileDescriptor();
97
91
  virtual bool isConnected();
98
 
  virtual bool isReading(void);
99
 
  virtual bool isWriting(void);
100
 
  virtual bool flush(void);
101
 
  virtual void close(void);
102
 
 
103
 
  virtual bool authenticate(void);
104
 
  virtual bool readCommand(char **packet, uint32_t *packet_length);
105
 
 
106
 
  virtual void sendOK(void);
107
 
  virtual void sendEOF(void);
 
92
  virtual bool flush();
 
93
  virtual void close();
 
94
 
 
95
  virtual bool authenticate();
 
96
  virtual bool readCommand(char **packet, uint32_t& packet_length);
 
97
 
 
98
  virtual void sendOK();
 
99
  virtual void sendEOF();
108
100
  virtual void sendError(const drizzled::error_t sql_errno, const char *err);
109
101
 
110
 
  virtual bool sendFields(drizzled::List<drizzled::Item> *list);
 
102
  virtual void sendFields(drizzled::List<drizzled::Item>&);
111
103
 
112
104
  using Client::store;
113
 
  virtual bool store(drizzled::Field *from);
114
 
  virtual bool store(void);
115
 
  virtual bool store(int32_t from);
116
 
  virtual bool store(uint32_t from);
117
 
  virtual bool store(int64_t from);
118
 
  virtual bool store(uint64_t from);
119
 
  virtual bool store(double from, uint32_t decimals, drizzled::String *buffer);
120
 
  virtual bool store(const char *from, size_t length);
 
105
  virtual void store(drizzled::Field*);
 
106
  virtual void store();
 
107
  virtual void store(int32_t from);
 
108
  virtual void store(uint32_t from);
 
109
  virtual void store(int64_t from);
 
110
  virtual void store(uint64_t from);
 
111
  virtual void store(double from, uint32_t decimals, drizzled::String *buffer);
 
112
  virtual void store(const char*, size_t);
121
113
 
122
 
  virtual bool haveError(void);
123
 
  virtual bool haveMoreData(void);
124
 
  virtual bool wasAborted(void);
125
 
  virtual bool isAdminAllowed(void);
126
 
  static std::vector<std::string> mysql_admin_ip_addresses;
127
 
  static void mysql_compose_ip_addresses(std::vector<std::string> options);
 
114
  virtual bool haveError();
 
115
  virtual bool wasAborted();
128
116
};
129
117
 
130
118
} /* namespace drizzle_plugin */
131
119
 
132
 
#endif /* PLUGIN_MYSQL_PROTOCOL_MYSQL_PROTOCOL_H */