MXS-2027: Store LOAD DATA state inside MXS_SESSION

By storing the data gathere by readwritesplit inside the session, the
protocol will be aware of the state of the LOAD DATA LOCAL INFILE
execution. This prevents misinterpretation of the data which previously
led to closed connections, effectively rendering LOAD DATA LOCAL INFILE
unusable.

This change is a temporary solution to a problem that needs to be solved
at the protocol level. The changes required to implement this are too big
to add into a bug fix release.
This commit is contained in:
Markus Mäkelä 2018-08-28 14:19:06 +03:00
parent e38e08089a
commit 5f4aa46552
No known key found for this signature in database
GPG Key ID: 72D48FCE664F7B19
5 changed files with 18 additions and 2 deletions

View File

@ -181,6 +181,7 @@ typedef struct session
bool qualifies_for_pooling; /**< Whether this session qualifies for the connection pool */
SessionStmtQueue* last_statements; /*< The N last statements by the client */
session_close_t close_reason; /**< Reason why the session was closed */
bool load_active; /**< Data streaming state (for LOAD DATA LOCAL INFILE) */
skygw_chk_t ses_chk_tail;
} MXS_SESSION;
@ -557,4 +558,14 @@ session_dump_statements_t session_get_dump_statements();
*/
const char* session_get_close_reason(const MXS_SESSION* session);
static inline void session_set_load_active(MXS_SESSION* session, bool value)
{
session->load_active = value;
}
static inline bool session_is_load_active(const MXS_SESSION* session)
{
return session->load_active;
}
MXS_END_DECLS

View File

@ -130,6 +130,7 @@ static MXS_SESSION* session_alloc_body(SERVICE* service, DCB* client_dcb,
session->stmt.target = NULL;
session->qualifies_for_pooling = false;
session->close_reason = SESSION_CLOSE_NONE;
session->load_active = false;
MXS_CONFIG *config = config_get_global_options();
// If MaxScale is running in Oracle mode, then autocommit needs to

View File

@ -1622,7 +1622,7 @@ static int route_by_statement(MXS_SESSION* session, uint64_t capabilities, GWBUF
/**
* Update the currently command being executed.
*/
if (!proto->changing_user)
if (!proto->changing_user && !session_is_load_active(session))
{
update_current_command(session->client_dcb, packetbuf);
}
@ -1631,7 +1631,8 @@ static int route_by_statement(MXS_SESSION* session, uint64_t capabilities, GWBUF
{
ss_dassert(GWBUF_IS_CONTIGUOUS(packetbuf));
if (rcap_type_required(capabilities, RCAP_TYPE_TRANSACTION_TRACKING))
if (rcap_type_required(capabilities, RCAP_TYPE_TRANSACTION_TRACKING) &&
!session_is_load_active(session))
{
if (session_trx_is_ending(session))
{

View File

@ -1204,6 +1204,7 @@ static void clientReply(MXS_ROUTER *instance,
{
// Server responded with an error to the LOAD DATA LOCAL INFILE
rses->load_data_state = LOAD_DATA_INACTIVE;
session_set_load_active(backend_dcb->session, true);
}
if (backend->get_reply_state() == REPLY_STATE_DONE)

View File

@ -1220,6 +1220,7 @@ bool handle_got_target(RWSplit *inst, RWSplitSession *rses,
/** The first packet contains the actual query and the server
* will respond to it */
rses->load_data_state = LOAD_DATA_ACTIVE;
session_set_load_active(rses->client_dcb->session, true);
}
else if (rses->load_data_state == LOAD_DATA_END)
{
@ -1227,6 +1228,7 @@ bool handle_got_target(RWSplit *inst, RWSplitSession *rses,
* to which the server responds with an OK or an ERR packet */
ss_dassert(gwbuf_length(querybuf) == 4);
rses->load_data_state = LOAD_DATA_INACTIVE;
session_set_load_active(rses->client_dcb->session, false);
}
}