/*
 * Copyright (C) 2008-2011 TrinityCore <http://www.trinitycore.org/>
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License as published by the
 * Free Software Foundation; either version 2 of the License, or (at your
 * option) any later version.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
 * more details.
 *
 * You should have received a copy of the GNU General Public License along
 * with this program. If not, see <http://www.gnu.org/licenses/>.
 */

#ifndef _CALLBACK_H
#define _CALLBACK_H

#include <ace/Future.h>
#include <ace/Future_Set.h>
#include "QueryResult.h"

typedef ACE_Future<QueryResult> QueryResultFuture;
typedef ACE_Future<PreparedQueryResult> PreparedQueryResultFuture;

/*! A simple template using ACE_Future to manage callbacks from the thread and object that
    issued the request. <ParamType> is variable type of parameter that is used as parameter
    for the callback function.
*/
template <typename Result, typename ParamType>
class QueryCallback
{
    public:
        QueryCallback() {}

        void SetFutureResult(ACE_Future<Result> value)
        {
            result = value;
        }

        ACE_Future<Result> GetFutureResult()
        {
            return result;
        }

        int IsReady()
        {
            return result.ready();
        }

        void GetResult(Result& res)
        {
            result.get(res);
        }

        void FreeResult()
        {
            result.cancel();
        }

        void SetParam(ParamType value)
        {
            param = value;
        }

        ParamType GetParam()
        {
            return param;
        }

    private:
        ACE_Future<Result> result;
        ParamType param;
};

template <typename Result, typename ParamType1, typename ParamType2>
class QueryCallback_2
{
    public:
        QueryCallback_2() {}

        void SetFutureResult(ACE_Future<Result> value)
        {
            result = value;
        }

        ACE_Future<Result> GetFutureResult()
        {
            return result;
        }

        int IsReady()
        {
            return result.ready();
        }

        void GetResult(Result& res)
        {
            result.get(res);
        }

        void FreeResult()
        {
            result.cancel();
        }

        void SetFirstParam(ParamType1 value)
        {
            param_1 = value;
        }

        void SetSecondParam(ParamType2 value)
        {
            param_2 = value;
        }

        ParamType1 GetFirstParam()
        {
            return param_1;
        }

        ParamType2 GetSecondParam()
        {
            return param_2;
        }

    private:
        ACE_Future<Result> result;
        ParamType1 param_1;
        ParamType2 param_2;
};

#endif