71 lines
2.1 KiB
C++
71 lines
2.1 KiB
C++
/*
|
|
* Copyright (c) 2018 MariaDB Corporation Ab
|
|
*
|
|
* Use of this software is governed by the Business Source License included
|
|
* in the LICENSE.TXT file and at www.mariadb.com/bsl11.
|
|
*
|
|
* Change Date: 2023-01-01
|
|
*
|
|
* On the date above, in accordance with the Business Source License, use
|
|
* of this software will be governed by version 2 or later of the General
|
|
* Public License.
|
|
*/
|
|
#pragma once
|
|
|
|
#include "hintrouterdefs.hh"
|
|
|
|
#include <deque>
|
|
#include <unordered_map>
|
|
#include <vector>
|
|
#include <string>
|
|
|
|
#include <maxscale/router.hh>
|
|
#include "dcb.hh"
|
|
|
|
using std::string;
|
|
|
|
class HintRouter;
|
|
|
|
class HintRouterSession : public maxscale::RouterSession
|
|
{
|
|
public:
|
|
typedef std::unordered_map<string, Dcb> BackendMap; // All backends, indexed by name
|
|
typedef std::vector<Dcb> BackendArray;
|
|
typedef std::vector<SERVER_REF*> RefArray;
|
|
typedef BackendMap::value_type MapElement;
|
|
typedef BackendArray::size_type size_type;
|
|
|
|
HintRouterSession(MXS_SESSION* pSession,
|
|
HintRouter* pRouter,
|
|
const BackendMap& backends
|
|
);
|
|
|
|
~HintRouterSession();
|
|
|
|
void close();
|
|
|
|
int32_t routeQuery(GWBUF* pPacket);
|
|
|
|
void clientReply(GWBUF* pPacket, DCB* pBackend);
|
|
|
|
void handleError(GWBUF* pMessage,
|
|
DCB* pProblem,
|
|
mxs_error_action_t action,
|
|
bool* pSuccess);
|
|
|
|
private:
|
|
HintRouterSession(const HintRouterSession&); // denied
|
|
HintRouterSession& operator=(const HintRouterSession&); // denied
|
|
private:
|
|
bool route_by_hint(GWBUF* pPacket, HINT* current_hint, bool ignore_errors);
|
|
bool route_to_slave(GWBUF* pPacket, bool print_errors);
|
|
void update_connections();
|
|
|
|
HintRouter* m_router;
|
|
BackendMap m_backends; // all connections
|
|
Dcb m_master; // connection to master
|
|
BackendArray m_slaves; // connections to slaves
|
|
size_type m_n_routed_to_slave; // packets routed to a single slave, used for rr
|
|
size_type m_surplus_replies; // how many replies should be ignored
|
|
};
|