← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to client/drizzledump.h

  • Committer: Andrew Hutchings
  • Date: 2010-09-22 19:58:16 UTC
  • mto: (1792.1.1 build)
  • mto: This revision was merged to the branch mainline in revision 1793.
  • Revision ID: andrew@linuxjedi.co.uk-20100922195816-uc3aud8va2sxponn
Put drizzle and mysql processes in seperate classes/files

Show diffs side-by-side

added added

removed removed

Lines of Context:
client/drizzledump.h
1
 
/* -*- mode: c++; c-basic-offset: 2; indent-tabs-mode: nil; -*-
2
 
 *  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
3
 
 *
4
 
 *  Copyright (C) 2010 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_DRIZZLEDUMP_H
21
 
#define CLIENT_DRIZZLEDUMP_H
22
 
 
23
 
class DrizzleDumpDatabase;
24
 
 
25
 
class DrizzleDumpIndex
26
 
{
27
 
  public:
28
 
    std::string indexName;
29
 
 
30
 
    DrizzleDumpIndex(std::string &index) :
31
 
      indexName(index)
32
 
    { }
33
 
 
34
 
    ~DrizzleDumpIndex()
35
 
    {
36
 
      columns.clear();
37
 
    }
38
 
 
39
 
 
40
 
    bool isPrimary;
41
 
    bool isUnique;
42
 
    bool isHash;
43
 
 
44
 
    std::vector<std::string> columns;
45
 
    friend std::ostream& operator <<(std::ostream &os, const DrizzleDumpIndex &obj);
46
 
};
47
 
 
48
 
class DrizzleDumpField
49
 
{
50
 
  public:
51
 
    std::stringstream errmsg;
52
 
 
53
 
    DrizzleDumpField(std::string &field) :
54
 
      fieldName(field)
55
 
    { }
56
 
 
57
 
    friend std::ostream& operator <<(std::ostream &os, const DrizzleDumpField &obj);
58
 
    std::string fieldName;
59
 
 
60
 
    std::string type;
61
 
    uint32_t length;
62
 
    bool isNull;
63
 
    bool isUnsigned;
64
 
    bool isAutoIncrement;
65
 
    bool defaultIsNull;
66
 
    bool convertDateTime;
67
 
    std::string defaultValue;
68
 
    std::string collation;
69
 
 
70
 
    /* For enum type */
71
 
    std::string enumValues;
72
 
 
73
 
    /* For decimal/double */
74
 
    uint32_t decimalPrecision;
75
 
    uint32_t decimalScale;
76
 
 
77
 
    void dateTimeConvert(const char* oldDefault);
78
 
    void setCollate(const char* newCollate);
79
 
    void setType(const char* raw_type, const char* collation);
80
 
};
81
 
 
82
 
class DrizzleDumpTable
83
 
{
84
 
  public:
85
 
    std::stringstream errmsg;
86
 
 
87
 
    DrizzleDumpTable(std::string &table) :
88
 
      tableName(table)
89
 
    { }
90
 
 
91
 
    ~DrizzleDumpTable()
92
 
    {
93
 
      fields.clear();
94
 
      indexes.clear();
95
 
    }
96
 
 
97
 
    bool populateFields(drizzle_con_st &connection);
98
 
    bool populateIndexes(drizzle_con_st &connection);
99
 
    std::vector<DrizzleDumpField*> fields;
100
 
    std::vector<DrizzleDumpIndex*> indexes;
101
 
 
102
 
    friend std::ostream& operator <<(std::ostream &os, const DrizzleDumpTable &obj);
103
 
    std::string tableName;
104
 
    std::string engineName;
105
 
    std::string collate;
106
 
 
107
 
    void setCollate(const char* newCollate);
108
 
    void setEngine(const char* newEngine);
109
 
 
110
 
    // Currently MySQL only, hard to do in Drizzle
111
 
    uint64_t autoIncrement;
112
 
    DrizzleDumpDatabase* database;
113
 
};
114
 
 
115
 
class DrizzleDumpDatabase
116
 
{
117
 
  public:
118
 
    std::stringstream errmsg;
119
 
 
120
 
    DrizzleDumpDatabase(const std::string &database) :
121
 
      databaseName(database)
122
 
    { }
123
 
 
124
 
    ~DrizzleDumpDatabase()
125
 
    {
126
 
      tables.clear();
127
 
    }
128
 
 
129
 
    friend std::ostream& operator <<(std::ostream &os, const DrizzleDumpDatabase &obj);
130
 
 
131
 
    bool populateTables(drizzle_con_st &connection);
132
 
    std::vector<DrizzleDumpTable*> tables;
133
 
 
134
 
    void setCollate(const char* newCollate);
135
 
    const std::string databaseName;
136
 
    std::string collate;
137
 
};
138
 
 
139
 
class DrizzleDumpData
140
 
{
141
 
  DrizzleDumpTable *table;
142
 
  drizzle_con_st *connection;
143
 
  std::stringstream errmsg;
144
 
  drizzle_result_st *result;
145
 
 
146
 
  public:
147
 
    DrizzleDumpData(drizzle_con_st &conn, DrizzleDumpTable *dataTable);
148
 
    ~DrizzleDumpData();
149
 
    friend std::ostream& operator <<(std::ostream &os, const DrizzleDumpData &obj);
150
 
 
151
 
    /* For 0000-00-00 -> NULL conversion */
152
 
    std::string convertDate(const char* oldDate) const;
153
 
    /* For xx:xx:xx -> INT conversion */
154
 
    long convertTime(const char* oldTime) const;
155
 
    std::string convertHex(const char* from, size_t from_size) const;
156
 
    std::string escape(const char* from, size_t from_size) const;
157
 
};
158
 
 
159
 
#endif /* CLIENT_DRIZZLEDUMP_H */