varnish-cache/vmod/vmod_directors_shard_dir.c
0
/*-
1
 * Copyright 2009-2016 UPLEX - Nils Goroll Systemoptimierung
2
 * All rights reserved.
3
 *
4
 * Authors: Nils Goroll <nils.goroll@uplex.de>
5
 *          Geoffrey Simmons <geoff.simmons@uplex.de>
6
 *          Julian Wiesener <jw@uplex.de>
7
 *
8
 * SPDX-License-Identifier: BSD-2-Clause
9
 *
10
 * Redistribution and use in source and binary forms, with or without
11
 * modification, are permitted provided that the following conditions
12
 * are met:
13
 * 1. Redistributions of source code must retain the above copyright
14
 *    notice, this list of conditions and the following disclaimer.
15
 * 2. Redistributions in binary form must reproduce the above copyright
16
 *    notice, this list of conditions and the following disclaimer in the
17
 *    documentation and/or other materials provided with the distribution.
18
 *
19
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
20
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
21
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22
 * ARE DISCLAIMED.  IN NO EVENT SHALL AUTHOR OR CONTRIBUTORS BE LIABLE
23
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
24
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
25
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
26
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29
 * SUCH DAMAGE.
30
 */
31
32
/*lint -e801 */
33
34
#include "config.h"
35
36
#include <stdlib.h>
37
#include <stdio.h>
38
#include <time.h>
39
#include <limits.h>
40
41
#include "cache/cache.h"
42
43
#include "vbm.h"
44
#include "vrnd.h"
45
46
#include "vcc_directors_if.h"
47
#include "vmod_directors_shard_dir.h"
48
49
struct shard_be_info {
50
        unsigned        hostid;
51
        unsigned        healthy;
52
        double          changed;        // when
53
};
54
55
/*
56
 * circle walk state for shard_next
57
 *
58
 * pick* cut off the search after having seen all possible backends
59
 */
60
struct shard_state {
61
        const struct vrt_ctx    *ctx;
62
        struct sharddir *shardd;
63
        uint32_t                idx;
64
65
        struct vbitmap          *picklist;
66
        unsigned                pickcount;
67
68
        struct shard_be_info    previous;
69
        struct shard_be_info    last;
70
};
71
72
void
73 7
sharddir_debug(struct sharddir *shardd, const uint32_t flags)
74
{
75 7
        CHECK_OBJ_NOTNULL(shardd, SHARDDIR_MAGIC);
76 7
        shardd->debug_flags = flags;
77 7
}
78
79
void
80 21
sharddir_log(struct vsl_log *vsl, enum VSL_tag_e tag,  const char *fmt, ...)
81
{
82
        va_list ap;
83
84 21
        va_start(ap, fmt);
85 21
        if (vsl != NULL)
86 10
                VSLbv(vsl, tag, fmt, ap);
87
        else
88 11
                VSLv(tag, NO_VXID, fmt, ap);
89 21
        va_end(ap);
90 21
}
91
92
static int
93 104
shard_lookup(const struct sharddir *shardd, const uint32_t key)
94
{
95 104
        CHECK_OBJ_NOTNULL(shardd, SHARDDIR_MAGIC);
96
97 104
        const uint32_t n = shardd->n_points;
98 104
        uint32_t i, idx = UINT32_MAX, high = n, low = 0;
99
100 104
        assert (n < idx);
101
102 104
        do {
103 635
            i = (high + low) / 2 ;
104 635
            if (shardd->hashcircle[i].point == key)
105 1
                idx = i;
106 634
            else if (i == n - 1)
107 18
                idx = n - 1;
108 616
            else if (shardd->hashcircle[i].point < key &&
109 326
                     shardd->hashcircle[i+1].point >= key)
110 68
                idx = i + 1;
111 548
            else if (shardd->hashcircle[i].point > key)
112 580
                if (i == 0)
113 17
                    idx = 0;
114
                else
115 273
                    high = i;
116
            else
117 258
                low = i;
118 635
        } while (idx == UINT32_MAX);
119
120 104
        return (idx);
121
}
122
123
static int
124 208
shard_next(struct shard_state *state, VCL_INT skip, VCL_BOOL healthy)
125
{
126 208
        int c, chosen = -1;
127
        VCL_BACKEND be;
128
        vtim_real changed;
129
        struct shard_be_info *sbe;
130
131 208
        AN(state);
132 208
        CHECK_OBJ_NOTNULL(state->shardd, SHARDDIR_MAGIC);
133
134 208
        if (state->pickcount >= state->shardd->n_backend)
135 3
                return (-1);
136
137 328
        while (state->pickcount < state->shardd->n_backend && skip >= 0) {
138
139 327
                c = state->shardd->hashcircle[state->idx].host;
140
141 327
                if (!vbit_test(state->picklist, c)) {
142
143 220
                        vbit_set(state->picklist, c);
144 220
                        state->pickcount++;
145
146 220
                        sbe = NULL;
147 220
                        be = state->shardd->backend[c].backend;
148 220
                        AN(be);
149 220
                        if (VRT_Healthy(state->ctx, be, &changed)) {
150 214
                                if (skip-- == 0) {
151 204
                                        chosen = c;
152 204
                                        sbe = &state->last;
153 204
                                } else {
154 10
                                        sbe = &state->previous;
155
                                }
156
157 220
                        } else if (!healthy && skip-- == 0) {
158 0
                                chosen = c;
159 0
                                sbe = &state->last;
160 0
                        }
161 220
                        if (sbe == &state->last &&
162 204
                            state->last.hostid != UINT_MAX)
163 100
                                memcpy(&state->previous, &state->last,
164
                                    sizeof(state->previous));
165
166 220
                        if (sbe) {
167 214
                                sbe->hostid = c;
168 214
                                sbe->healthy = 1;
169 214
                                sbe->changed = changed;
170 214
                        }
171 220
                        if (chosen != -1)
172 204
                                break;
173 16
                }
174
175 123
                if (++(state->idx) == state->shardd->n_points)
176 19
                        state->idx = 0;
177
        }
178 205
        return (chosen);
179 208
}
180
181
void
182 25
sharddir_new(struct sharddir **sharddp, const char *vcl_name,
183
    const struct vmod_directors_shard_param *param)
184
{
185
        struct sharddir *shardd;
186
187 25
        AN(vcl_name);
188 25
        AN(sharddp);
189 25
        AZ(*sharddp);
190 25
        ALLOC_OBJ(shardd, SHARDDIR_MAGIC);
191 25
        AN(shardd);
192 25
        *sharddp = shardd;
193 25
        shardd->name = vcl_name;
194 25
        shardd->param = param;
195 25
        PTOK(pthread_rwlock_init(&shardd->mtx, NULL));
196 25
}
197
198
void
199 3
sharddir_set_param(struct sharddir *shardd,
200
    const struct vmod_directors_shard_param *param)
201
{
202 3
        CHECK_OBJ_NOTNULL(shardd, SHARDDIR_MAGIC);
203 3
        shardd->param = param;
204 3
}
205
206
void
207 5
sharddir_release(struct sharddir *shardd)
208
{
209 5
        CHECK_OBJ_NOTNULL(shardd, SHARDDIR_MAGIC);
210 5
        shardcfg_backend_clear(shardd);
211 5
}
212
213
void
214 5
sharddir_delete(struct sharddir **sharddp)
215
{
216
        struct sharddir *shardd;
217
218 5
        TAKE_OBJ_NOTNULL(shardd, sharddp, SHARDDIR_MAGIC);
219 5
        shardcfg_delete(shardd);
220 5
        PTOK(pthread_rwlock_destroy(&shardd->mtx));
221 5
        FREE_OBJ(shardd);
222 5
}
223
224
void
225 150
sharddir_rdlock(struct sharddir *shardd)
226
{
227 150
        CHECK_OBJ_NOTNULL(shardd, SHARDDIR_MAGIC);
228 150
        PTOK(pthread_rwlock_rdlock(&shardd->mtx));
229 150
}
230
231
void
232 44
sharddir_wrlock(struct sharddir *shardd)
233
{
234 44
        CHECK_OBJ_NOTNULL(shardd, SHARDDIR_MAGIC);
235 44
        PTOK(pthread_rwlock_wrlock(&shardd->mtx));
236 44
}
237
238
void
239 194
sharddir_unlock(struct sharddir *shardd)
240
{
241 194
        CHECK_OBJ_NOTNULL(shardd, SHARDDIR_MAGIC);
242 194
        PTOK(pthread_rwlock_unlock(&shardd->mtx));
243 194
}
244
245
static inline void
246 104
validate_alt(VRT_CTX, const struct sharddir *shardd, VCL_INT *alt)
247
{
248 104
        const VCL_INT alt_max = shardd->n_backend - 1;
249
250 104
        if (*alt < 0) {
251 0
                shard_err(ctx->vsl, shardd->name,
252
                    "invalid negative parameter alt=%ld, set to 0", *alt);
253 0
                *alt = 0;
254 104
        } else if (*alt > alt_max) {
255 3
                shard_err(ctx->vsl, shardd->name,
256
                    "parameter alt=%ld limited to %ld", *alt, alt_max);
257 3
                *alt = alt_max;
258 3
        }
259 104
}
260
261
static inline void
262 104
init_state(struct shard_state *state,
263
    VRT_CTX, struct sharddir *shardd, struct vbitmap *picklist)
264
{
265 104
        AN(picklist);
266
267 104
        state->ctx = ctx;
268 104
        state->shardd = shardd;
269 104
        state->idx = UINT32_MAX;
270 104
        state->picklist = picklist;
271
272
        /* healthy and changed only defined for valid hostids */
273 104
        state->previous.hostid = UINT_MAX;
274 104
        state->last.hostid = UINT_MAX;
275 104
}
276
277
/* basically same as vdir_any_healthy
278
 * - XXX we should embed a vdir
279
 * - XXX should we return the health state of the actual backend
280
 *   for healthy=IGNORE ?
281
 */
282
VCL_BOOL
283 16
sharddir_any_healthy(VRT_CTX, struct sharddir *shardd, VCL_TIME *changed)
284
{
285 16
        unsigned i, retval = 0;
286
        VCL_BACKEND be;
287
        vtim_real c;
288
289 16
        CHECK_OBJ_NOTNULL(shardd, SHARDDIR_MAGIC);
290 16
        sharddir_rdlock(shardd);
291 16
        if (changed != NULL)
292 7
                *changed = 0;
293 16
        for (i = 0; i < shardd->n_backend; i++) {
294 16
                be = shardd->backend[i].backend;
295 16
                CHECK_OBJ_NOTNULL(be, DIRECTOR_MAGIC);
296 16
                retval = VRT_Healthy(ctx, be, &c);
297 16
                if (changed != NULL && c > *changed)
298 7
                        *changed = c;
299 16
                if (retval)
300 16
                        break;
301 0
        }
302 16
        sharddir_unlock(shardd);
303 16
        return (retval);
304
}
305
306
/*
307
 * core function for the director backend/resolve method
308
 */
309
310
static VCL_BACKEND
311 104
sharddir_pick_be_locked(VRT_CTX, const struct sharddir *shardd, uint32_t key,
312
    VCL_INT alt, VCL_REAL warmup, VCL_BOOL rampup, VCL_ENUM healthy,
313
    struct shard_state *state)
314
{
315
        VCL_BACKEND be;
316
        VCL_DURATION chosen_r, alt_r;
317
318 104
        CHECK_OBJ_NOTNULL(shardd, SHARDDIR_MAGIC);
319 104
        CHECK_OBJ_NOTNULL(ctx, VRT_CTX_MAGIC);
320 104
        AN(ctx->vsl);
321 104
        assert(shardd->n_backend > 0);
322
323 104
        assert(shardd->hashcircle);
324
325 104
        validate_alt(ctx, shardd, &alt);
326
327 104
        state->idx = shard_lookup(shardd, key);
328 104
        assert(state->idx < UINT32_MAX);
329
330 104
        SHDBG(SHDBG_LOOKUP, shardd, "lookup key %x idx %u host %u",
331
            key, state->idx, shardd->hashcircle[state->idx].host);
332
333 104
        if (alt > 0) {
334 81
                if (shard_next(state, alt - 1,
335 54
                    healthy == VENUM(ALL) ? 1 : 0) == -1) {
336 0
                        if (state->previous.hostid != UINT_MAX) {
337 0
                                be = sharddir_backend(shardd,
338 0
                                    state->previous.hostid);
339 0
                                AN(be);
340 0
                                return (be);
341
                        }
342 0
                        return (NULL);
343
                }
344 27
        }
345
346 104
        if (shard_next(state, 0, healthy == VENUM(IGNORE) ? 0 : 1) == -1) {
347 0
                if (state->previous.hostid != UINT_MAX) {
348 0
                        be = sharddir_backend(shardd, state->previous.hostid);
349 0
                        AN(be);
350 0
                        return (be);
351
                }
352 0
                return (NULL);
353
        }
354
355 104
        be = sharddir_backend(shardd, state->last.hostid);
356 104
        AN(be);
357
358 104
        if (warmup == -1)
359 103
                warmup = shardd->warmup;
360
361
        /* short path for cases we dont want ramup/warmup or can't */
362 104
        if (alt > 0 || healthy == VENUM(IGNORE) || (!rampup && warmup == 0) ||
363 77
            shard_next(state, 0, 1) == -1)
364 31
                return (be);
365
366 73
        assert(alt == 0);
367 73
        assert(state->previous.hostid != UINT_MAX);
368 73
        assert(state->last.hostid != UINT_MAX);
369 73
        assert(state->previous.hostid != state->last.hostid);
370 73
        assert(be == sharddir_backend(shardd, state->previous.hostid));
371
372 73
        chosen_r = shardcfg_get_rampup(shardd, state->previous.hostid);
373 73
        alt_r = shardcfg_get_rampup(shardd, state->last.hostid);
374
375 73
        SHDBG(SHDBG_RAMPWARM, shardd, "chosen host %u rampup %f changed %f",
376
            state->previous.hostid, chosen_r,
377
            ctx->now - state->previous.changed);
378 73
        SHDBG(SHDBG_RAMPWARM, shardd, "alt host %u rampup %f changed %f",
379
            state->last.hostid, alt_r,
380
            ctx->now - state->last.changed);
381
382 73
        if (ctx->now - state->previous.changed < chosen_r) {
383
                /*
384
                 * chosen host is in rampup
385
                 * - no change if alternative host is also in rampup or the dice
386
                 *   has rolled in favour of the chosen host
387
                 */
388 3
                if (!rampup ||
389 2
                    ctx->now - state->last.changed < alt_r ||
390 2
                    VRND_RandomTestableDouble() * chosen_r <
391 1
                    (ctx->now - state->previous.changed))
392 1
                        return (be);
393 1
        } else {
394
                /* chosen host not in rampup - warmup ? */
395 71
                if (warmup == 0 || VRND_RandomTestableDouble() > warmup)
396 71
                        return (be);
397
        }
398
399 1
        be = sharddir_backend(shardd, state->last.hostid);
400 1
        return (be);
401 104
}
402
403
VCL_BACKEND
404 104
sharddir_pick_be(VRT_CTX, struct sharddir *shardd, uint32_t key, VCL_INT alt,
405
    VCL_REAL warmup, VCL_BOOL rampup, VCL_ENUM healthy)
406
{
407
        VCL_BACKEND be;
408
        struct shard_state state[1];
409
        unsigned picklist_sz;
410
411 104
        CHECK_OBJ_NOTNULL(ctx, VRT_CTX_MAGIC);
412 104
        CHECK_OBJ_NOTNULL(shardd, SHARDDIR_MAGIC);
413
414 104
        sharddir_rdlock(shardd);
415
416 104
        if (shardd->n_backend == 0) {
417 0
                shard_err0(ctx->vsl, shardd->name, "no backends");
418 0
                sharddir_unlock(shardd);
419 0
                return (NULL);
420
        }
421
422 104
        picklist_sz = VBITMAP_SZ(shardd->n_backend);
423 104
        char picklist_spc[picklist_sz];
424
425 104
        memset(state, 0, sizeof(state));
426 104
        init_state(state, ctx, shardd, vbit_init(picklist_spc, picklist_sz));
427
428 208
        be = sharddir_pick_be_locked(ctx, shardd, key, alt, warmup, rampup,
429 104
            healthy, state);
430 104
        sharddir_unlock(shardd);
431
432 104
        vbit_destroy(state->picklist);
433 104
        return (be);
434 104
}