mirror of https://github.com/sysown/proxysql
René Cannaò
6 years ago
committed by
GitHub
commit
a0bd85edf8
@ -0,0 +1,45 @@
|
||||
#ifndef __PROXYSQL_RESTAPI_H__
|
||||
#define __PROXYSQL_RESTAPI_H__
|
||||
|
||||
#include "proxy_defines.h"
|
||||
#include "proxysql.h"
|
||||
#include "cpp.h"
|
||||
#include <vector>
|
||||
|
||||
class SQLite3DB;
|
||||
|
||||
class Restapi_Row {
|
||||
public:
|
||||
unsigned int id;
|
||||
bool is_active;
|
||||
unsigned int interval_ms;
|
||||
std::string method;
|
||||
std::string uri;
|
||||
std::string script;
|
||||
std::string comment;
|
||||
unsigned int version;
|
||||
Restapi_Row(unsigned int _id, bool _is_active, unsigned int _in, const std::string& _method, const std::string& _uri, const std::string& _script, const std::string& _comment);
|
||||
};
|
||||
|
||||
class ProxySQL_Restapi {
|
||||
SQLite3DB* admindb;
|
||||
public:
|
||||
ProxySQL_Restapi(SQLite3DB* db);
|
||||
virtual ~ProxySQL_Restapi();
|
||||
|
||||
unsigned int last_version;
|
||||
unsigned int version;
|
||||
#ifdef PA_PTHREAD_MUTEX
|
||||
pthread_rwlock_t rwlock;
|
||||
#else
|
||||
rwlock_t rwlock;
|
||||
#endif
|
||||
std::vector<Restapi_Row> Restapi_Rows;
|
||||
void update_restapi_table(SQLite3_result *result);
|
||||
void load_restapi_to_runtime();
|
||||
void save_restapi_runtime_to_database(bool);
|
||||
void flush_restapi__from_memory_to_disk();
|
||||
void flush_restapi__from_disk_to_memory();
|
||||
};
|
||||
|
||||
#endif // #ifndef __PROXYSQL_RESTAPI_H__
|
||||
@ -0,0 +1,106 @@
|
||||
#include "proxysql_restapi.h"
|
||||
#include "proxysql.h"
|
||||
#include "proxysql_atomic.h"
|
||||
#include "cpp.h"
|
||||
|
||||
#include <sstream>
|
||||
|
||||
Restapi_Row::Restapi_Row(unsigned int _id, bool _is_active, unsigned int _in, const std::string& _method, const std::string& _uri, const std::string& _script, const std::string& _comment) :
|
||||
id(_id), is_active(_is_active), interval_ms(_in), method(_method), uri(_uri), script(_script), comment(_comment) {}
|
||||
|
||||
ProxySQL_Restapi::ProxySQL_Restapi(SQLite3DB* db) {
|
||||
assert(db);
|
||||
|
||||
admindb = db;
|
||||
#ifdef PA_PTHREAD_MUTEX
|
||||
pthread_rwlock_init(&rwlock,NULL);
|
||||
#else
|
||||
spinlock_rwlock_init(&rwlock);
|
||||
#endif
|
||||
version=0;
|
||||
}
|
||||
|
||||
|
||||
ProxySQL_Restapi::~ProxySQL_Restapi() {}
|
||||
|
||||
void ProxySQL_Restapi::update_restapi_table(SQLite3_result *resultset) {
|
||||
#ifdef PA_PTHREAD_MUTEX
|
||||
pthread_rwlock_wrlock(&rwlock);
|
||||
#else
|
||||
spin_wrlock(&rwlock);
|
||||
#endif
|
||||
// delete all current rows
|
||||
Restapi_Rows.clear();
|
||||
for (auto r : resultset->rows) {
|
||||
unsigned int id = strtoul(r->fields[0], NULL, 10);
|
||||
bool is_active=false;
|
||||
if (atoi(r->fields[1])) {
|
||||
is_active=true;
|
||||
}
|
||||
unsigned int interval_ms=strtoul(r->fields[2], NULL, 10);
|
||||
Restapi_Rows.push_back({id, is_active, interval_ms, r->fields[3], r->fields[4], r->fields[5], r->fields[6]});
|
||||
}
|
||||
|
||||
// increase version
|
||||
__sync_fetch_and_add(&version,1);
|
||||
// unlock
|
||||
#ifdef PA_PTHREAD_MUTEX
|
||||
pthread_rwlock_unlock(&rwlock);
|
||||
#else
|
||||
spin_wrunlock(&rwlock);
|
||||
#endif
|
||||
}
|
||||
|
||||
void ProxySQL_Restapi::flush_restapi__from_disk_to_memory() {
|
||||
admindb->wrlock();
|
||||
admindb->execute("DELETE FROM main.restapi_routes");
|
||||
admindb->execute("INSERT INTO main.restapi_routes SELECT * FROM disk.restapi_routes");
|
||||
admindb->wrunlock();
|
||||
}
|
||||
|
||||
void ProxySQL_Restapi::flush_restapi__from_memory_to_disk() {
|
||||
admindb->wrlock();
|
||||
admindb->execute("DELETE FROM disk.restapi_routes");
|
||||
admindb->execute("INSERT INTO disk.restapi_routes SELECT * FROM main.restapi_routes");
|
||||
admindb->wrunlock();
|
||||
}
|
||||
|
||||
void ProxySQL_Restapi::save_restapi_runtime_to_database(bool _runtime) {
|
||||
const char *query = _runtime ? "DELETE FROM main.runtime_restapi_routes" : "DELETE FROM main.restapi_routes";
|
||||
proxy_debug(PROXY_DEBUG_ADMIN, 4, "%s\n", query);
|
||||
admindb->execute(query);
|
||||
// read lock the scheduler
|
||||
#ifdef PA_PTHREAD_MUTEX
|
||||
pthread_rwlock_rdlock(&rwlock);
|
||||
#else
|
||||
spin_rdlock(&rwlock);
|
||||
#endif
|
||||
const char* table = _runtime ? " runtime_restapi_routes " : " restapi_routes ";
|
||||
for (auto r : Restapi_Rows) {
|
||||
std::stringstream ss;
|
||||
ss << "INSERT INTO " << table << " VALUES(" << r.id << "," << r.is_active << ","
|
||||
<< r.interval_ms << ",'" << r.method << "','" << r.uri << "','" << r.script << "','" << r.comment << "')";
|
||||
|
||||
admindb->execute(ss.str().c_str());
|
||||
}
|
||||
// unlock the scheduler
|
||||
#ifdef PA_PTHREAD_MUTEX
|
||||
pthread_rwlock_unlock(&rwlock);
|
||||
#else
|
||||
spin_rdunlock(&rwlock);
|
||||
#endif
|
||||
}
|
||||
|
||||
void ProxySQL_Restapi::load_restapi_to_runtime() {
|
||||
char *error=NULL;
|
||||
char *query=(char *)"SELECT * FROM restapi_routes";
|
||||
std::unique_ptr<SQLite3_result> resultset = std::unique_ptr<SQLite3_result>(admindb->execute_statement(query, &error));
|
||||
if (error) {
|
||||
proxy_error("Error on %s : %s\n", query, error);
|
||||
} else {
|
||||
update_restapi_table(resultset.get());
|
||||
save_restapi_runtime_to_database(true);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@ -0,0 +1,27 @@
|
||||
#!/usr/bin/env python
|
||||
#
|
||||
#./scripts/export_users.py '{"db":{"user":"root", "password":"a", "port":"3306", "host":"127.0.0.1"},"admin":{"user":"admin","password":"admin","port":"6032","host":"127.0.0.1"}}'
|
||||
#
|
||||
|
||||
import sys
|
||||
import subprocess
|
||||
import json
|
||||
from MySQLdb import _mysql
|
||||
|
||||
if len(sys.argv) > 1:
|
||||
params=json.loads(sys.argv[1])
|
||||
db_mysql=_mysql.connect(host=params['db']['host'],user=params['db']['user'],passwd=params['db']['password'],port=int(params['db']['port']))
|
||||
db_mysql.query('SELECT user, authentication_string from mysql.user')
|
||||
records=db_mysql.store_result().fetch_row(maxrows=0)
|
||||
db_proxy_admin=_mysql.connect(host=params['admin']['host'],user=params['admin']['user'],passwd=params['admin']['password'],port=int(params['admin']['port']))
|
||||
for row in records:
|
||||
db_proxy_admin.query('INSERT OR REPLACE INTO mysql_users (username, password) values ("'+str(row[0])+'","'+str(row[1])+'")')
|
||||
|
||||
result='{"num_records":"'+str(len(records))+'"}'
|
||||
try:
|
||||
subprocess.check_output(['mysql', '-u'+params['admin']['user'], '-p'+params['admin']['password'], '-h'+params['admin']['host'], '-P'+params['admin']['port'], '-e', 'load mysql users to runtime'],stderr= subprocess.STDOUT)
|
||||
except subprocess.CalledProcessError as e:
|
||||
result='"Error calling mysql: ' + e.output.replace("'", "") + '"'
|
||||
|
||||
print('{"params":'+str(sys.argv[1])+', "result":'+result+'}')
|
||||
|
||||
@ -0,0 +1,15 @@
|
||||
#!/usr/bin/env python
|
||||
|
||||
import sys
|
||||
import subprocess
|
||||
import json
|
||||
|
||||
if len(sys.argv) > 1:
|
||||
params=json.loads(sys.argv[1])
|
||||
out=''
|
||||
try:
|
||||
out=subprocess.check_output(['mysql', '-u'+params['user'], '-p'+params['password'], '-h'+params['host'], '-P'+params['port'], '-e', 'select * from stats.stats_memory_metrics'],stderr= subprocess.STDOUT)
|
||||
except subprocess.CalledProcessError as e:
|
||||
out="Error calling mysql: " + e.output.replace("'", "")
|
||||
print('{"params":'+sys.argv[1].encode('string-escape')+', "result":"'+out.encode('string-escape')+'"}')
|
||||
|
||||
Loading…
Reference in new issue