1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
/* - mode: c; c-basic-offset: 2; indent-tabs-mode: nil; -*-
* vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
*
* Copyright (C) 2011 David Shrewsbury
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
*/
#ifndef PLUGIN_SLAVE_QUEUE_CONSUMER_H
#define PLUGIN_SLAVE_QUEUE_CONSUMER_H
#include <plugin/slave/queue_thread.h>
#include <plugin/slave/sql_executor.h>
#include <drizzled/session.h>
namespace drizzled
{
namespace message
{
class Transaction;
}
}
namespace slave
{
class QueueConsumer : public QueueThread, public SQLExecutor
{
public:
QueueConsumer() :
QueueThread(),
SQLExecutor("slave", "replication"),
_check_interval(5)
{ }
bool init();
bool process();
void shutdown();
void setSleepInterval(uint32_t seconds)
{
_check_interval= seconds;
}
uint32_t getSleepInterval()
{
return _check_interval;
}
/**
* Update applier status in state table.
*
* @param err_msg Error message string
* @param status false = STOPPED, true = RUNNING
*/
void setApplierState(const std::string &err_msg, bool status);
private:
typedef std::vector<uint64_t> TrxIdList;
/** Number of seconds to sleep between checking queue for messages */
uint32_t _check_interval;
bool getListOfCompletedTransactions(TrxIdList &list);
bool getMessage(drizzled::message::Transaction &transaction,
std::string &commit_id,
uint64_t trx_id,
uint32_t segment_id);
/**
* Convert the given Transaction message into equivalent SQL.
*
* @param[in] transaction Transaction protobuf message to convert.
* @param[in,out] aggregate_sql Buffer for total SQL for this transaction.
* @param[in,out] segmented_sql Buffer for carried over segmented statements.
*
* @retval true Success
* @retval false Failure
*/
bool convertToSQL(const drizzled::message::Transaction &transaction,
std::vector<std::string> &aggregate_sql,
std::vector<std::string> &segmented_sql);
/**
* Execute a batch of SQL statements.
*
* @param sql Batch of SQL statements to execute.
* @param commit_id Commit ID value to store in state table.
*
* @retval true Success
* @retval false Failure
*/
bool executeSQLWithCommitId(std::vector<std::string> &sql,
const std::string &commit_id);
/**
* Remove messages for a given transaction from the queue.
*
* @param trx_id Transaction ID for the messages to remove.
*
* @retval true Success
* @retval false Failure
*/
bool deleteFromQueue(uint64_t trx_id);
/**
* Determine if a Statement message is an end message.
*
* @retval true Is an end Statement message
* @retval false Is NOT an end Statement message
*/
bool isEndStatement(const drizzled::message::Statement &statement);
};
} /* namespace slave */
#endif /* PLUGIN_SLAVE_QUEUE_CONSUMER_H */
|