~drizzle-trunk/drizzle/development

2187.3.2 by Andrew Hutchings
Separate the server detection functions into a .h file
1
/* -*- mode: c++; c-basic-offset: 2; indent-tabs-mode: nil; -*-
2
 *  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
3
 *
4
 *  Copyright (C) 2011 Andrew Hutchings
5
 *
6
 *  This program is free software; you can redistribute it and/or modify
7
 *  it under the terms of the GNU General Public License as published by
8
 *  the Free Software Foundation; version 2 of the License.
9
 *
10
 *  This program is distributed in the hope that it will be useful,
11
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
12
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13
 *  GNU General Public License for more details.
14
 *
15
 *  You should have received a copy of the GNU General Public License
16
 *  along with this program; if not, write to the Free Software
17
 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
18
 */
19
20
#ifndef CLIENT_SERVER_DETECT_H
21
#define CLIENT_SERVER_DETECT_H
22
23
#include <boost/regex.hpp>
24
25
class ServerDetect
26
{
27
  public:
28
    enum server_type {
29
      SERVER_MYSQL_FOUND,
30
      SERVER_DRIZZLE_FOUND,
31
      SERVER_UNKNOWN_FOUND
32
    };
33
34
    server_type getServerType() { return type; }
35
    std::string& getServerVersion() { return version; }
36
37
    ServerDetect(drizzle_con_st *connection) :
38
      type(SERVER_UNKNOWN_FOUND),
39
      version("")
40
    {
41
      boost::match_flag_type flags = boost::match_default;
42
43
      boost::regex mysql_regex("(5\\.[0-9]+\\.[0-9]+)");
44
      boost::regex drizzle_regex("(20[0-9]{2}\\.(0[1-9]|1[012])\\.[0-9]+)");
45
46
      version= drizzle_con_server_version(connection);
47
48
      if (regex_search(version, mysql_regex, flags))
49
        type= SERVER_MYSQL_FOUND;
50
      else if (regex_search(version, drizzle_regex, flags))
51
        type= SERVER_DRIZZLE_FOUND;
52
      else
53
        type= SERVER_UNKNOWN_FOUND;
54
    }
55
56
  private:
57
    server_type type;
58
    std::string version;
59
};
60
61
#endif /* CLIENT_SERVER_DETECT_H */