Files
apache-http-server/modules/cache/mod_socache_redis.c
Jim Jagielski 696e56b117 Merge r1768070, r1768120, r1768225, r1769712, r1769737, r1774610, r1828624, r1828626, r1855701 from trunk:
And the socache provider interface for Redis


And provide some nice usage stats. Could be prettier, for sure.

Needs LOGNO #s

reorganize mod_socache_redis so it compiles w/o apr-redis

It looks like this was the original intent. The provider
is not registered if apr-redis is missing.

It was previously failing to compile due to the dirconf being
hidden behind the #ifdef guard but not the routine module
code that used it.




Fix some HTML tags
</br> --> <br />

Couls also be <br> or <br/> but some (very) old clients don't like it.
httpd is not really consistent on it. (personnlaly <br> would be just fine for me but <br /> is the most commonly used form)

related to http://svn.apache.org/viewvc?view=revision&revision=1774602


Use apr_pcalloc in the 'create' function. All the fields are initialized in the 'init' function, but it is It is cleaner and more future proof.

Small style correction (remove extra spaces)

Initial doc for the sosache_redis module.

Corrections and improvements welcome.

Update compatibility notes for mod_socache_redis
Submitted by: jim, covener, jailletc36, jim, jailletc36, jailletc36, jailletc36
Reviewed by: jim, minfrin, covener


git-svn-id: https://svn.apache.org/repos/asf/httpd/httpd/branches/2.4.x@1855817 13f79535-47bb-0310-9956-ffa450edef68
2019-03-19 12:23:15 +00:00

487 lines
17 KiB
C

/* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "httpd.h"
#include "http_config.h"
#include "http_protocol.h"
#include "apr.h"
#include "apu_version.h"
#include "ap_socache.h"
#include "ap_mpm.h"
#include "http_log.h"
#include "apr_strings.h"
#include "mod_status.h"
typedef struct {
apr_uint32_t ttl;
apr_uint32_t rwto;
} socache_rd_svr_cfg;
/* apr_redis support requires >= 1.6 */
#if APU_MAJOR_VERSION > 1 || \
(APU_MAJOR_VERSION == 1 && APU_MINOR_VERSION > 5)
#define HAVE_APU_REDIS 1
#endif
/* The underlying apr_redis system is thread safe.. */
#define RD_KEY_LEN 254
#ifndef RD_DEFAULT_SERVER_PORT
#define RD_DEFAULT_SERVER_PORT 6379
#endif
#ifndef RD_DEFAULT_SERVER_MIN
#define RD_DEFAULT_SERVER_MIN 0
#endif
#ifndef RD_DEFAULT_SERVER_SMAX
#define RD_DEFAULT_SERVER_SMAX 1
#endif
#ifndef RD_DEFAULT_SERVER_TTL
#define RD_DEFAULT_SERVER_TTL apr_time_from_sec(15)
#endif
#ifndef RD_DEFAULT_SERVER_RWTO
#define RD_DEFAULT_SERVER_RWTO apr_time_from_sec(5)
#endif
module AP_MODULE_DECLARE_DATA socache_redis_module;
#ifdef HAVE_APU_REDIS
#include "apr_redis.h"
struct ap_socache_instance_t {
const char *servers;
apr_redis_t *rc;
const char *tag;
apr_size_t taglen; /* strlen(tag) + 1 */
};
static const char *socache_rd_create(ap_socache_instance_t **context,
const char *arg,
apr_pool_t *tmp, apr_pool_t *p)
{
ap_socache_instance_t *ctx;
*context = ctx = apr_pcalloc(p, sizeof *ctx);
if (!arg || !*arg) {
return "List of server names required to create redis socache.";
}
ctx->servers = apr_pstrdup(p, arg);
return NULL;
}
static apr_status_t socache_rd_init(ap_socache_instance_t *ctx,
const char *namespace,
const struct ap_socache_hints *hints,
server_rec *s, apr_pool_t *p)
{
apr_status_t rv;
int thread_limit = 0;
apr_uint16_t nservers = 0;
char *cache_config;
char *split;
char *tok;
socache_rd_svr_cfg *sconf = ap_get_module_config(s->module_config,
&socache_redis_module);
ap_mpm_query(AP_MPMQ_HARD_LIMIT_THREADS, &thread_limit);
/* Find all the servers in the first run to get a total count */
cache_config = apr_pstrdup(p, ctx->servers);
split = apr_strtok(cache_config, ",", &tok);
while (split) {
nservers++;
split = apr_strtok(NULL,",", &tok);
}
rv = apr_redis_create(p, nservers, 0, &ctx->rc);
if (rv != APR_SUCCESS) {
ap_log_error(APLOG_MARK, APLOG_CRIT, rv, s, APLOGNO(03473)
"Failed to create Redis Object of '%d' size.",
nservers);
return rv;
}
/* Now add each server to the redis */
cache_config = apr_pstrdup(p, ctx->servers);
split = apr_strtok(cache_config, ",", &tok);
while (split) {
apr_redis_server_t *st;
char *host_str;
char *scope_id;
apr_port_t port;
rv = apr_parse_addr_port(&host_str, &scope_id, &port, split, p);
if (rv != APR_SUCCESS) {
ap_log_error(APLOG_MARK, APLOG_CRIT, rv, s, APLOGNO(03474)
"Failed to Parse redis Server: '%s'", split);
return rv;
}
if (host_str == NULL) {
ap_log_error(APLOG_MARK, APLOG_CRIT, rv, s, APLOGNO(03475)
"Failed to Parse Server, "
"no hostname specified: '%s'", split);
return APR_EINVAL;
}
if (port == 0) {
port = RD_DEFAULT_SERVER_PORT;
}
rv = apr_redis_server_create(p,
host_str, port,
RD_DEFAULT_SERVER_MIN,
RD_DEFAULT_SERVER_SMAX,
thread_limit,
sconf->ttl,
sconf->rwto,
&st);
if (rv != APR_SUCCESS) {
ap_log_error(APLOG_MARK, APLOG_CRIT, rv, s, APLOGNO(03476)
"Failed to Create redis Server: %s:%d",
host_str, port);
return rv;
}
rv = apr_redis_add_server(ctx->rc, st);
if (rv != APR_SUCCESS) {
ap_log_error(APLOG_MARK, APLOG_CRIT, rv, s, APLOGNO(03477)
"Failed to Add redis Server: %s:%d",
host_str, port);
return rv;
}
split = apr_strtok(NULL,",", &tok);
}
ctx->tag = apr_pstrcat(p, namespace, ":", NULL);
ctx->taglen = strlen(ctx->tag) + 1;
/* socache API constraint: */
AP_DEBUG_ASSERT(ctx->taglen <= 16);
return APR_SUCCESS;
}
static void socache_rd_destroy(ap_socache_instance_t *context, server_rec *s)
{
/* noop. */
}
/* Converts (binary) id into a key prefixed by the predetermined
* namespace tag; writes output to key buffer. Returns non-zero if
* the id won't fit in the key buffer. */
static int socache_rd_id2key(ap_socache_instance_t *ctx,
const unsigned char *id, unsigned int idlen,
char *key, apr_size_t keylen)
{
char *cp;
if (idlen * 2 + ctx->taglen >= keylen)
return 1;
cp = apr_cpystrn(key, ctx->tag, ctx->taglen);
ap_bin2hex(id, idlen, cp);
return 0;
}
static apr_status_t socache_rd_store(ap_socache_instance_t *ctx, server_rec *s,
const unsigned char *id, unsigned int idlen,
apr_time_t expiry,
unsigned char *ucaData, unsigned int nData,
apr_pool_t *p)
{
char buf[RD_KEY_LEN];
apr_status_t rv;
apr_uint32_t timeout;
if (socache_rd_id2key(ctx, id, idlen, buf, sizeof(buf))) {
return APR_EINVAL;
}
timeout = apr_time_sec(expiry - apr_time_now());
if (timeout <= 0) {
return APR_EINVAL;
}
rv = apr_redis_setex(ctx->rc, buf, (char*)ucaData, nData, timeout, 0);
if (rv != APR_SUCCESS) {
ap_log_error(APLOG_MARK, APLOG_CRIT, rv, s, APLOGNO(03478)
"scache_rd: error setting key '%s' "
"with %d bytes of data", buf, nData);
return rv;
}
return APR_SUCCESS;
}
static apr_status_t socache_rd_retrieve(ap_socache_instance_t *ctx, server_rec *s,
const unsigned char *id, unsigned int idlen,
unsigned char *dest, unsigned int *destlen,
apr_pool_t *p)
{
apr_size_t data_len;
char buf[RD_KEY_LEN], *data;
apr_status_t rv;
if (socache_rd_id2key(ctx, id, idlen, buf, sizeof buf)) {
return APR_EINVAL;
}
/* ### this could do with a subpool, but _getp looks like it will
* eat memory like it's going out of fashion anyway. */
rv = apr_redis_getp(ctx->rc, p, buf, &data, &data_len, NULL);
if (rv) {
if (rv != APR_NOTFOUND) {
ap_log_error(APLOG_MARK, APLOG_ERR, rv, s, APLOGNO(03479)
"scache_rd: 'retrieve' FAIL");
}
return rv;
}
else if (data_len > *destlen) {
ap_log_error(APLOG_MARK, APLOG_ERR, rv, s, APLOGNO(03480)
"scache_rd: 'retrieve' OVERFLOW");
return APR_ENOMEM;
}
memcpy(dest, data, data_len);
*destlen = data_len;
return APR_SUCCESS;
}
static apr_status_t socache_rd_remove(ap_socache_instance_t *ctx, server_rec *s,
const unsigned char *id,
unsigned int idlen, apr_pool_t *p)
{
char buf[RD_KEY_LEN];
apr_status_t rv;
if (socache_rd_id2key(ctx, id, idlen, buf, sizeof buf)) {
return APR_EINVAL;
}
rv = apr_redis_delete(ctx->rc, buf, 0);
if (rv != APR_SUCCESS) {
ap_log_error(APLOG_MARK, APLOG_DEBUG, rv, s, APLOGNO(03481)
"scache_rd: error deleting key '%s' ",
buf);
}
return rv;
}
static void socache_rd_status(ap_socache_instance_t *ctx, request_rec *r, int flags)
{
apr_redis_t *rc = ctx->rc;
int i;
for (i = 0; i < rc->ntotal; i++) {
apr_redis_server_t *rs;
apr_redis_stats_t *stats;
char *role;
apr_status_t rv;
char *br = (!(flags & AP_STATUS_SHORT) ? "<br />" : "");
rs = rc->live_servers[i];
ap_rprintf(r, "Redis server: %s:%d [%s]%s\n", rs->host, (int)rs->port,
(rs->status == APR_RC_SERVER_LIVE) ? "Up" : "Down",
br);
rv = apr_redis_stats(rs, r->pool, &stats);
if (rv != APR_SUCCESS)
continue;
if (!(flags & AP_STATUS_SHORT)) {
ap_rprintf(r, "<b>General::</b> Version: <i>%u.%u.%u</i> [%u bits], PID: <i>%u</i>, Uptime: <i>%u hrs</i> <br />\n",
stats->major, stats->minor, stats->patch, stats->arch_bits,
stats->process_id, stats->uptime_in_seconds/3600);
ap_rprintf(r, "<b>Clients::</b> Connected: <i>%d</i>, Blocked: <i>%d</i> <br />\n",
stats->connected_clients, stats->blocked_clients);
ap_rprintf(r, "<b>Memory::</b> Total: <i>%" APR_UINT64_T_FMT "</i>, Max: <i>%" APR_UINT64_T_FMT "</i>, Used: <i>%" APR_UINT64_T_FMT "</i> <br />\n",
stats->total_system_memory, stats->maxmemory, stats->used_memory);
ap_rprintf(r, "<b>CPU::</b> System: <i>%u</i>, User: <i>%u</i><br />\n",
stats->used_cpu_sys, stats->used_cpu_user );
ap_rprintf(r, "<b>Connections::</b> Recd: <i>%" APR_UINT64_T_FMT "</i>, Processed: <i>%" APR_UINT64_T_FMT "</i>, Rejected: <i>%" APR_UINT64_T_FMT "</i> <br />\n",
stats->total_connections_received, stats->total_commands_processed,
stats->rejected_connections);
ap_rprintf(r, "<b>Cache::</b> Hits: <i>%" APR_UINT64_T_FMT "</i>, Misses: <i>%" APR_UINT64_T_FMT "</i> <br />\n",
stats->keyspace_hits, stats->keyspace_misses);
ap_rprintf(r, "<b>Net::</b> Input bytes: <i>%" APR_UINT64_T_FMT "</i>, Output bytes: <i>%" APR_UINT64_T_FMT "</i> <br />\n",
stats->total_net_input_bytes, stats->total_net_output_bytes);
if (stats->role == APR_RS_SERVER_MASTER)
role = "master";
else if (stats->role == APR_RS_SERVER_SLAVE)
role = "slave";
else
role = "unknown";
ap_rprintf(r, "<b>Misc::</b> Role: <i>%s</i>, Connected Slaves: <i>%u</i>, Is Cluster?: <i>%s</i> \n",
role, stats->connected_clients,
(stats->cluster_enabled ? "yes" : "no"));
ap_rputs("<hr><br />\n", r);
}
else {
ap_rprintf(r, "Version: %u.%u.%u [%u bits], PID: %u, Uptime: %u hrs %s\n",
stats->major, stats->minor, stats->patch, stats->arch_bits,
stats->process_id, stats->uptime_in_seconds/3600, br);
ap_rprintf(r, "Clients:: Connected: %d, Blocked: %d %s\n",
stats->connected_clients, stats->blocked_clients, br);
ap_rprintf(r, "Memory:: Total: %" APR_UINT64_T_FMT ", Max: %" APR_UINT64_T_FMT ", Used: %" APR_UINT64_T_FMT " %s\n",
stats->total_system_memory, stats->maxmemory, stats->used_memory,
br);
ap_rprintf(r, "CPU:: System: %u, User: %u %s\n",
stats->used_cpu_sys, stats->used_cpu_user , br);
ap_rprintf(r, "Connections:: Recd: %" APR_UINT64_T_FMT ", Processed: %" APR_UINT64_T_FMT ", Rejected: %" APR_UINT64_T_FMT " %s\n",
stats->total_connections_received, stats->total_commands_processed,
stats->rejected_connections, br);
ap_rprintf(r, "Cache:: Hits: %" APR_UINT64_T_FMT ", Misses: %" APR_UINT64_T_FMT " %s\n",
stats->keyspace_hits, stats->keyspace_misses, br);
ap_rprintf(r, "Net:: Input bytes: %" APR_UINT64_T_FMT ", Output bytes: %" APR_UINT64_T_FMT " %s\n",
stats->total_net_input_bytes, stats->total_net_output_bytes, br);
if (stats->role == APR_RS_SERVER_MASTER)
role = "master";
else if (stats->role == APR_RS_SERVER_SLAVE)
role = "slave";
else
role = "unknown";
ap_rprintf(r, "Misc:: Role: %s, Connected Slaves: %u, Is Cluster?: %s %s\n",
role, stats->connected_clients,
(stats->cluster_enabled ? "yes" : "no"), br);
}
}
}
static apr_status_t socache_rd_iterate(ap_socache_instance_t *instance,
server_rec *s, void *userctx,
ap_socache_iterator_t *iterator,
apr_pool_t *pool)
{
return APR_ENOTIMPL;
}
static const ap_socache_provider_t socache_mc = {
"redis",
0,
socache_rd_create,
socache_rd_init,
socache_rd_destroy,
socache_rd_store,
socache_rd_retrieve,
socache_rd_remove,
socache_rd_status,
socache_rd_iterate,
};
#endif /* HAVE_APU_REDIS */
static void* create_server_config(apr_pool_t* p, server_rec* s)
{
socache_rd_svr_cfg *sconf = apr_palloc(p, sizeof(socache_rd_svr_cfg));
sconf->ttl = RD_DEFAULT_SERVER_TTL;
sconf->rwto = RD_DEFAULT_SERVER_RWTO;
return sconf;
}
static const char *socache_rd_set_ttl(cmd_parms *cmd, void *dummy,
const char *arg)
{
apr_interval_time_t ttl;
socache_rd_svr_cfg *sconf = ap_get_module_config(cmd->server->module_config,
&socache_redis_module);
if (ap_timeout_parameter_parse(arg, &ttl, "s") != APR_SUCCESS) {
return apr_pstrcat(cmd->pool, cmd->cmd->name,
" has wrong format", NULL);
}
if ((ttl < apr_time_from_sec(0)) || (ttl > apr_time_from_sec(3600))) {
return apr_pstrcat(cmd->pool, cmd->cmd->name,
" can only be 0 or up to one hour.", NULL);
}
/* apr_redis_server_create needs a ttl in usec. */
sconf->ttl = ttl;
return NULL;
}
static const char *socache_rd_set_rwto(cmd_parms *cmd, void *dummy,
const char *arg)
{
apr_interval_time_t rwto;
socache_rd_svr_cfg *sconf = ap_get_module_config(cmd->server->module_config,
&socache_redis_module);
if (ap_timeout_parameter_parse(arg, &rwto, "s") != APR_SUCCESS) {
return apr_pstrcat(cmd->pool, cmd->cmd->name,
" has wrong format", NULL);
}
if ((rwto < apr_time_from_sec(0)) || (rwto > apr_time_from_sec(3600))) {
return apr_pstrcat(cmd->pool, cmd->cmd->name,
" can only be 0 or up to one hour.", NULL);
}
/* apr_redis_server_create needs a ttl in usec. */
sconf->rwto = rwto;
return NULL;
}
static void register_hooks(apr_pool_t *p)
{
#ifdef HAVE_APU_REDIS
ap_register_provider(p, AP_SOCACHE_PROVIDER_GROUP, "redis",
AP_SOCACHE_PROVIDER_VERSION,
&socache_mc);
#endif
}
static const command_rec socache_redis_cmds[] =
{
AP_INIT_TAKE1("RedisConnPoolTTL", socache_rd_set_ttl, NULL, RSRC_CONF,
"TTL used for the connection pool with the Redis server(s)"),
AP_INIT_TAKE1("RedisTimeout", socache_rd_set_rwto, NULL, RSRC_CONF,
"R/W timeout used for the connection with the Redis server(s)"),
{NULL}
};
AP_DECLARE_MODULE(socache_redis) = {
STANDARD20_MODULE_STUFF,
NULL, /* create per-dir config structures */
NULL, /* merge per-dir config structures */
create_server_config, /* create per-server config structures */
NULL, /* merge per-server config structures */
socache_redis_cmds, /* table of config file commands */
register_hooks /* register hooks */
};