/*
 * Copyright (c) 2016 MariaDB Corporation Ab
 *
 * Use of this software is governed by the Business Source License included
 * in the LICENSE.TXT file and at www.mariadb.com/bsl.
 *
 * Change Date: 2019-07-01
 *
 * On the date above, in accordance with the Business Source License, use
 * of this software will be governed by version 2 or later of the General
 * Public License.
 */

#include <shardrouter.h>

void
subsvc_set_state(SUBSERVICE* svc, subsvc_state_t state)
{
    if (state & SUBSVC_WAITING_RESULT)
    {
        /** Increase waiter count */
        atomic_add(&svc->n_res_waiting, 1);
    }

    svc->state |= state;
}

void
subsvc_clear_state(SUBSERVICE* svc, subsvc_state_t state)
{
    if (state & SUBSVC_WAITING_RESULT)
    {
        /** Decrease waiter count */
        atomic_add(&svc->n_res_waiting, -1);
    }

    svc->state &= ~state;
}

bool
get_shard_subsvc(SUBSERVICE** subsvc, ROUTER_CLIENT_SES* session, char* target)
{
    int i;

    if (subsvc == NULL || session == NULL || target == NULL)
    {
        return false;
    }

    for (i = 0; i < session->n_subservice; i++)
    {
        if (strcmp(session->subservice[i]->service->name, target) == 0)
        {

            if (SUBSVC_IS_OK(session->subservice[i]))
            {
                if (subsvc_is_valid(session->subservice[i]))
                {
                    *subsvc = session->subservice[i];
                    return true;
                }

                /**
                 * The service has failed
                 */

                subsvc_set_state(session->subservice[i], SUBSVC_FAILED);
            }
        }
    }

    return false;
}