/*-

 * Copyright 2009-2016 UPLEX - Nils Goroll Systemoptimierung

 * All rights reserved.

 *

 * Authors: Julian Wiesener <jw@uplex.de>

 *          Nils Goroll <slink@uplex.de>

 *

 * SPDX-License-Identifier: BSD-2-Clause

 *

 * Redistribution and use in source and binary forms, with or without

 * modification, are permitted provided that the following conditions

 * are met:

 * 1. Redistributions of source code must retain the above copyright

 *    notice, this list of conditions and the following disclaimer.

 * 2. Redistributions in binary form must reproduce the above copyright

 *    notice, this list of conditions and the following disclaimer in the

 *    documentation and/or other materials provided with the distribution.

 *

 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND

 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE

 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE

 * ARE DISCLAIMED.  IN NO EVENT SHALL AUTHOR OR CONTRIBUTORS BE LIABLE

 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL

 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS

 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)

 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT

 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY

 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF

 * SUCH DAMAGE.

 */

struct vbitmap;

struct shard_circlepoint {

        uint32_t                point;

        unsigned int            host;

};

struct shard_backend {

        VCL_BACKEND             backend;

        union {

                const char      *ident;

                void            *freeptr;

        };

        VCL_DURATION            rampup;

        uint32_t                replicas;

};

struct vmod_directors_shard_param;

#define SHDBG_LOOKUP     1

#define SHDBG_CIRCLE    (1<<1)

#define SHDBG_RAMPWARM  (1<<2)

struct sharddir {

        unsigned                                magic;

#define SHARDDIR_MAGIC                          0xdbb7d59f

        uint32_t                                debug_flags;

        pthread_rwlock_t                        mtx;

        unsigned                                n_backend;

        unsigned                                l_backend;

        struct shard_backend                    *backend;

        const char                              *name;

        struct shard_circlepoint                *hashcircle;

        const struct vmod_directors_shard_param *param;

        VCL_DURATION                            rampup_duration;

        VCL_REAL                                warmup;

        uint32_t                                n_points;

};

/* VRT_priv_task() id offsets */

enum shard_task_off_e {

        task_off_param = 0,

        task_off_cfg = 1

};

static inline VCL_BACKEND

{

}

#define SHDBG(flag, shardd, ...)                                        \

        do {                                                            \

                if ((shardd)->debug_flags & (flag))                     \

                        VSL(SLT_Debug, NO_VXID, "vmod_directors: shard: " \

                            __VA_ARGS__);                               \

        } while (0)

#define shard_log(vsl, tag, name, fmt, ...)                             \

        sharddir_log(vsl, tag, "vmod_directors: shard %s: " fmt,        \

            name, __VA_ARGS__)

#define shard_fail(ctx, name, fmt, ...)                         \

        VRT_fail(ctx, "vmod_directors: shard %s: " fmt, name, __VA_ARGS__)

#define shard_err(vsl, name, fmt, ...)                          \

        shard_log(vsl, SLT_Error, name, fmt, __VA_ARGS__)

#define shard_err0(vsl, name, msg)                      \

        shard_log(vsl, SLT_Error, name, "%s", msg)

#define shard_notice(vsl, name, fmt, ...)               \

        shard_log(vsl, SLT_Notice, name, fmt, __VA_ARGS__)

void sharddir_debug(struct sharddir *shardd, const uint32_t flags);

void sharddir_log(struct vsl_log *, enum VSL_tag_e tag,  const char *fmt, ...);

void sharddir_new(struct sharddir **sharddp, const char *vcl_name,

    const struct vmod_directors_shard_param *param);

void sharddir_set_param(struct sharddir *shardd,

    const struct vmod_directors_shard_param *param);

void sharddir_release(struct sharddir *shardd);

void sharddir_delete(struct sharddir **sharddp);

void sharddir_rdlock(struct sharddir *shardd);

void sharddir_wrlock(struct sharddir *shardd);

void sharddir_unlock(struct sharddir *shardd);

VCL_BOOL sharddir_any_healthy(VRT_CTX, struct sharddir *, VCL_TIME *);

VCL_BACKEND sharddir_pick_be(VRT_CTX, struct sharddir *, uint32_t, VCL_INT,

   VCL_REAL, VCL_BOOL, VCL_ENUM healthy);

/* in shard_cfg.c */

void shardcfg_backend_clear(struct sharddir *shardd);

void shardcfg_delete(const struct sharddir *shardd);

VCL_DURATION shardcfg_get_rampup(const struct sharddir *shardd, unsigned host);