Add configuration capabilities for persistent connections.
This commit is contained in:
@ -619,6 +619,7 @@ hashtable_memory_fns(monitorhash,strdup,NULL,free,NULL);
|
||||
}
|
||||
else if (!strcmp(type, "server"))
|
||||
{
|
||||
SERVER *server = obj->element;
|
||||
char *address;
|
||||
char *port;
|
||||
char *protocol;
|
||||
@ -631,6 +632,8 @@ hashtable_memory_fns(monitorhash,strdup,NULL,free,NULL);
|
||||
monuser = config_get_value(obj->parameters,
|
||||
"monitoruser");
|
||||
monpw = config_get_value(obj->parameters, "monitorpw");
|
||||
server->persistpoolmax = config_get_value(obj->parameters, "persistpoolmax");
|
||||
server->persistmaxtime = config_get_value(obj->parameters, "persistmaxtime");
|
||||
|
||||
if (address && port && protocol)
|
||||
{
|
||||
@ -677,6 +680,10 @@ hashtable_memory_fns(monitorhash,strdup,NULL,free,NULL);
|
||||
"monitorpw")
|
||||
&& strcmp(params->name,
|
||||
"type")
|
||||
&& strcmp(params->name,
|
||||
"persistpoolmax")
|
||||
&& strcmp(params->name,
|
||||
"persistmaxtime")
|
||||
)
|
||||
{
|
||||
serverAddParameter(obj->element,
|
||||
|
@ -56,6 +56,7 @@
|
||||
#include <stdarg.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
#include <time.h>
|
||||
#include <dcb.h>
|
||||
#include <spinlock.h>
|
||||
#include <server.h>
|
||||
@ -1297,7 +1298,9 @@ dcb_close(DCB *dcb)
|
||||
{
|
||||
char *user;
|
||||
user = session_getUser(dcb->session);
|
||||
if (user && dcb->server)
|
||||
if (user && dcb->server
|
||||
&& dcb->server->persistpoolmax
|
||||
&& dcb_persistent_clean_count(dcb) < dcb->server->persistpoolmax)
|
||||
{
|
||||
dcb->user = strdup(user);
|
||||
spinlock_acquire(&dcb->server->persistlock);
|
||||
@ -2311,6 +2314,46 @@ dcb_null_auth(DCB *dcb, SERVER *server, SESSION *session, GWBUF *buf)
|
||||
return 0;
|
||||
}
|
||||
|
||||
/**
|
||||
* Check persistent pool for expiry or excess size and count
|
||||
*
|
||||
* @param dcb The DCB being closed.
|
||||
*/
|
||||
int
|
||||
dcb_persistent_clean_count(DCB *dcb)
|
||||
{
|
||||
int count = 0;
|
||||
if (dcb)
|
||||
{
|
||||
SERVER *server = dcb->server;
|
||||
DCB *previousdcb = NULL;
|
||||
DCB *persistentdcb = server->persistent;
|
||||
|
||||
while (persistentdcb) {
|
||||
if (count >= server->persistpoolmax || (persistentdcb->last_read + server->persistmaxtime) < time(NULL))
|
||||
{
|
||||
if (previousdcb) {
|
||||
previousdcb->nextpersistent = persistentdcb->nextpersistent;
|
||||
}
|
||||
else
|
||||
{
|
||||
dcb->nextpersistent = persistentdcb->nextpersistent;
|
||||
}
|
||||
/** Call possible callback for this DCB in case of close */
|
||||
dcb_call_callback(persistentdcb, DCB_REASON_CLOSE);
|
||||
dcb_add_to_zombieslist(persistentdcb);
|
||||
atomic_add(&server->stats.n_persistent, -1);
|
||||
}
|
||||
else
|
||||
{
|
||||
count++;
|
||||
}
|
||||
persistentdcb = persistentdcb->nextpersistent;
|
||||
}
|
||||
}
|
||||
return count;
|
||||
}
|
||||
|
||||
/**
|
||||
* Return DCB counts optionally filtered by usage
|
||||
*
|
||||
|
@ -140,39 +140,35 @@ SERVER *ptr;
|
||||
DCB *
|
||||
server_get_persistent(SERVER *server, char *user, const char *protocol)
|
||||
{
|
||||
DCB *dcb, *previous;
|
||||
int rc;
|
||||
DCB *dcb, *previous = NULL;
|
||||
|
||||
spinlock_acquire(&server->persistlock);
|
||||
dcb = server->persistent;
|
||||
previous = NULL;
|
||||
while (dcb) {
|
||||
/* Test for expired, free and remove from list if it is */
|
||||
if (dcb->user && dcb->protoname && 0 == strcmp(dcb->user, user) && 0 == strcmp(dcb->protoname, protocol))
|
||||
{
|
||||
if (NULL == previous)
|
||||
{
|
||||
server->persistent = dcb->nextpersistent;
|
||||
}
|
||||
else
|
||||
{
|
||||
previous->nextpersistent = dcb->nextpersistent;
|
||||
}
|
||||
free(dcb->user);
|
||||
dcb->user = NULL;
|
||||
spinlock_release(&server->persistlock);
|
||||
atomic_add(&server->stats.n_persistent, -1);
|
||||
atomic_add(&server->stats.n_current, 1);
|
||||
return dcb;
|
||||
}
|
||||
previous = dcb;
|
||||
dcb = dcb->nextpersistent;
|
||||
}
|
||||
if (NULL != server->persistent)
|
||||
if (server->persistent && dcb_persistent_clean_count(server->persistent) && server->persistent)
|
||||
{
|
||||
/* Change user, remove DCB from list, release spinlock, return dcb */
|
||||
spinlock_acquire(&server->persistlock);
|
||||
dcb = server->persistent;
|
||||
while (dcb) {
|
||||
if (dcb->user && dcb->protoname && 0 == strcmp(dcb->user, user) && 0 == strcmp(dcb->protoname, protocol))
|
||||
{
|
||||
if (NULL == previous)
|
||||
{
|
||||
server->persistent = dcb->nextpersistent;
|
||||
}
|
||||
else
|
||||
{
|
||||
previous->nextpersistent = dcb->nextpersistent;
|
||||
}
|
||||
free(dcb->user);
|
||||
dcb->user = NULL;
|
||||
spinlock_release(&server->persistlock);
|
||||
atomic_add(&server->stats.n_persistent, -1);
|
||||
atomic_add(&server->stats.n_current, 1);
|
||||
return dcb;
|
||||
}
|
||||
previous = dcb;
|
||||
dcb = dcb->nextpersistent;
|
||||
}
|
||||
spinlock_release(&server->persistlock);
|
||||
}
|
||||
spinlock_release(&server->persistlock);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
@ -716,7 +712,7 @@ SERVER_PARAM *param;
|
||||
}
|
||||
|
||||
/**
|
||||
* Retreive a parameter value from a server
|
||||
* Retrieve a parameter value from a server
|
||||
*
|
||||
* @param server The server we are looking for a parameter of
|
||||
* @param name The name of the parameter we require
|
||||
|
Reference in New Issue
Block a user