forked from mkoppanen/php-tokyo_tyrant
-
Notifications
You must be signed in to change notification settings - Fork 1
/
failover.c
144 lines (117 loc) · 3.99 KB
/
failover.c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
/*
+----------------------------------------------------------------------+
| PHP Version 5 / Tokyo tyrant |
+----------------------------------------------------------------------+
| Copyright (c) 2009 Mikko Koppanen |
+----------------------------------------------------------------------+
| This source file is subject to version 3.0 of the PHP license, |
| that is bundled with this package in the file LICENSE, and is |
| available through the world-wide-web at the following url: |
| http://www.php.net/license/3_0.txt. |
| If you did not receive a copy of the PHP license and are unable to |
| obtain it through the world-wide-web, please send a note to |
| [email protected] so we can mail you a copy immediately. |
+----------------------------------------------------------------------+
| Authors: Mikko Koppanen <[email protected]> |
+----------------------------------------------------------------------+
*/
#include "php_tokyo_tyrant_session.h"
#include "ext/standard/php_rand.h"
long php_tt_server_fail(int op, char *host, int port TSRMLS_DC)
{
char *key = NULL;
zend_bool result;
int key_len;
zval *fail_count, **ptrptr;
if (!TOKYO_G(failures)) {
TOKYO_G(failures) = malloc(sizeof(HashTable));
if (!TOKYO_G(failures)) {
return 0;
}
zend_hash_init(TOKYO_G(failures), 5, NULL, ZVAL_INTERNAL_PTR_DTOR, 1);
}
key = php_tt_hash_key(host, port, 0.0, &key_len TSRMLS_CC);
if (zend_hash_find(TOKYO_G(failures), key, key_len + 1, (void **)&ptrptr) == SUCCESS) {
fail_count = *ptrptr;
if (op == PHP_TT_GET) {
efree(key);
return Z_LVAL_P(fail_count);
} else if (op == PHP_TT_INCR) {
Z_LVAL_P(fail_count)++;
} else {
Z_LVAL_P(fail_count)--;
}
} else {
if (op == PHP_TT_GET) {
efree(key);
return 0;
}
fail_count = malloc(sizeof(zval));
INIT_PZVAL(fail_count);
ZVAL_LONG(fail_count, (op == PHP_TT_INCR) ? 1 : 0);
}
result = (zend_hash_update(TOKYO_G(failures), key, key_len + 1, (void *)&fail_count, sizeof(zval *), NULL) == SUCCESS);
efree(key);
return Z_LVAL_P(fail_count);
}
void php_tt_server_fail_incr(char *host, int port TSRMLS_DC)
{
php_tt_server_fail(PHP_TT_INCR, host, port TSRMLS_CC);
}
void php_tt_server_fail_decr(char *host, int port TSRMLS_DC)
{
php_tt_server_fail(PHP_TT_DECR, host, port TSRMLS_CC);
}
zend_bool php_tt_server_ok(char *host, int port TSRMLS_DC)
{
long fail_count = php_tt_server_fail(PHP_TT_GET, host, port TSRMLS_CC);
/* Server is always ok if failover is disabled */
if (!TOKYO_G(allow_failover)) {
return 1;
}
/* If this matches, execute a health check on servers and restore them to pool if they are fine */
if ((php_rand(TSRMLS_C) % TOKYO_G(health_check_divisor)) == (php_rand(TSRMLS_C) % TOKYO_G(health_check_divisor))) {
php_tt_health_check(TSRMLS_C);
}
if (fail_count < TOKYO_G(fail_threshold)) {
return 1;
} else {
return 0;
}
}
static int _php_tt_check_servers(zval **ppzval, int num_args, va_list args, zend_hash_key *hash_key)
{
char *key, host[4096];
double timeout;
int port;
if (hash_key->nKeyLength == 0 || hash_key->nKeyLength >= 4096) {
return ZEND_HASH_APPLY_REMOVE;
}
key = estrdup(hash_key->arKey);
memset(host, '\0', 4096);
if (sscanf(key, "%s %d %lf", host, &port, &timeout) == 3) {
int value;
uint_fast64_t index;
TCRDB *rdb = tcrdbnew();
value = tcrdbopen(rdb, host, port);
if (!value) {
efree(key);
tcrdbdel(rdb);
return ZEND_HASH_APPLY_KEEP;
}
index = tcrdbtblgenuid(rdb);
tcrdbdel(rdb);
if (index == -1) {
efree(key);
return ZEND_HASH_APPLY_KEEP;
}
}
efree(key);
return ZEND_HASH_APPLY_REMOVE;
}
void php_tt_health_check(TSRMLS_D)
{
if (TOKYO_G(failures) && zend_hash_num_elements(TOKYO_G(failures)) > 0) {
zend_hash_apply_with_arguments(TOKYO_G(failures), (apply_func_args_t)_php_tt_check_servers, 0);
}
}