diff options
Diffstat (limited to 'src/common')
| -rw-r--r-- | src/common/Threading/Callback.h | 209 | ||||
| -rw-r--r-- | src/common/Utilities/StartProcess.cpp | 4 | ||||
| -rw-r--r-- | src/common/Utilities/Timer.h | 4 | 
3 files changed, 5 insertions, 212 deletions
| diff --git a/src/common/Threading/Callback.h b/src/common/Threading/Callback.h deleted file mode 100644 index f7eab57ddda..00000000000 --- a/src/common/Threading/Callback.h +++ /dev/null @@ -1,209 +0,0 @@ -/* - * Copyright (C) 2008-2016 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 <future> -#include "QueryResult.h" - -typedef std::future<QueryResult> QueryResultFuture; -typedef std::promise<QueryResult> QueryResultPromise; -typedef std::future<PreparedQueryResult> PreparedQueryResultFuture; -typedef std::promise<PreparedQueryResult> PreparedQueryResultPromise; - -#define CALLBACK_STAGE_INVALID uint8(-1) - -template <typename Result, typename ParamType, bool chain = false> -class QueryCallback -{ -    public: -        QueryCallback() : _param(), _stage(chain ? 0 : CALLBACK_STAGE_INVALID)  { } - -        //! The parameter of this function should be a resultset returned from either .AsyncQuery or .AsyncPQuery -        void SetFutureResult(std::future<Result> value) -        { -            _result = std::move(value); -        } - -        std::future<Result>& GetFutureResult() -        { -            return _result; -        } - -        int IsReady() -        { -            return _result.valid() && _result.wait_for(std::chrono::seconds(0)) == std::future_status::ready; -        } - -        void GetResult(Result& res) -        { -            res = _result.get(); -        } - -        void FreeResult() -        { -            // Nothing to do here, the constructor of std::future will take care of the cleanup -        } - -        void SetParam(ParamType value) -        { -            _param = value; -        } - -        ParamType GetParam() -        { -            return _param; -        } - -        //! Resets the stage of the callback chain -        void ResetStage() -        { -            if (!chain) -                return; - -            _stage = 0; -        } - -        //! Advances the callback chain to the next stage, so upper level code can act on its results accordingly -        void NextStage() -        { -            if (!chain) -                return; - -            ++_stage; -        } - -        //! Returns the callback stage (or CALLBACK_STAGE_INVALID if invalid) -        uint8 GetStage() -        { -            return _stage; -        } - -        //! Resets all underlying variables (param, result and stage) -        void Reset() -        { -            SetParam(ParamType()); -            FreeResult(); -            ResetStage(); -        } - -    private: -        std::future<Result> _result; -        ParamType _param; -        uint8 _stage; - -        QueryCallback(QueryCallback const& right) = delete; -        QueryCallback& operator=(QueryCallback const& right) = delete; -}; - -template <typename Result, typename ParamType1, typename ParamType2, bool chain = false> -class QueryCallback_2 -{ -    public: -        QueryCallback_2() : _stage(chain ? 0 : CALLBACK_STAGE_INVALID) { } - -        //! The parameter of this function should be a resultset returned from either .AsyncQuery or .AsyncPQuery -        void SetFutureResult(std::future<Result> value) -        { -            _result = std::move(value); -        } - -        std::future<Result>& GetFutureResult() -        { -            return _result; -        } - -        int IsReady() -        { -            return _result.valid() && _result.wait_for(std::chrono::seconds(0)) == std::future_status::ready; -        } - -        void GetResult(Result& res) -        { -            res = _result.get(); -        } - -        void FreeResult() -        { -            // Nothing to do here, the constructor of std::future will take care of the cleanup -        } - -        void SetFirstParam(ParamType1 value) -        { -            _param_1 = value; -        } - -        void SetSecondParam(ParamType2 value) -        { -            _param_2 = value; -        } - -        ParamType1 GetFirstParam() -        { -            return _param_1; -        } - -        ParamType2 GetSecondParam() -        { -            return _param_2; -        } - -        //! Resets the stage of the callback chain -        void ResetStage() -        { -            if (!chain) -                return; - -            _stage = 0; -        } - -        //! Advances the callback chain to the next stage, so upper level code can act on its results accordingly -        void NextStage() -        { -            if (!chain) -                return; - -            ++_stage; -        } - -        //! Returns the callback stage (or CALLBACK_STAGE_INVALID if invalid) -        uint8 GetStage() -        { -            return _stage; -        } - -        //! Resets all underlying variables (param, result and stage) -        void Reset() -        { -            SetFirstParam(NULL); -            SetSecondParam(NULL); -            FreeResult(); -            ResetStage(); -        } - -    private: -        std::future<Result> _result; -        ParamType1 _param_1; -        ParamType2 _param_2; -        uint8 _stage; - -        QueryCallback_2(QueryCallback_2 const& right) = delete; -        QueryCallback_2& operator=(QueryCallback_2 const& right) = delete; -}; - -#endif diff --git a/src/common/Utilities/StartProcess.cpp b/src/common/Utilities/StartProcess.cpp index 29feeaa2714..f35c6de3b5c 100644 --- a/src/common/Utilities/StartProcess.cpp +++ b/src/common/Utilities/StartProcess.cpp @@ -92,6 +92,7 @@ static int CreateChildProcess(T waiter, std::string const& executable,              // With binding stdin              return execute(run_exe(boost::filesystem::absolute(executable)),                  set_args(args), +                inherit_env(),                  bind_stdin(*inputSource),                  bind_stdout(file_descriptor_sink(outPipe.sink, close_handle)),                  bind_stderr(file_descriptor_sink(errPipe.sink, close_handle))); @@ -101,6 +102,7 @@ static int CreateChildProcess(T waiter, std::string const& executable,              // Without binding stdin              return execute(run_exe(boost::filesystem::absolute(executable)),                  set_args(args), +                inherit_env(),                  bind_stdout(file_descriptor_sink(outPipe.sink, close_handle)),                  bind_stderr(file_descriptor_sink(errPipe.sink, close_handle)));          } @@ -237,7 +239,7 @@ public:      }  }; -TC_COMMON_API std::shared_ptr<AsyncProcessResult> +std::shared_ptr<AsyncProcessResult>      StartAsyncProcess(std::string executable, std::vector<std::string> args,                        std::string logger, std::string input_file, bool secure)  { diff --git a/src/common/Utilities/Timer.h b/src/common/Utilities/Timer.h index cdce08caaf0..f66bb90c98e 100644 --- a/src/common/Utilities/Timer.h +++ b/src/common/Utilities/Timer.h @@ -25,9 +25,9 @@ inline uint32 getMSTime()  {      using namespace std::chrono; -    static const system_clock::time_point ApplicationStartTime = system_clock::now(); +    static const steady_clock::time_point ApplicationStartTime = steady_clock::now(); -    return uint32(duration_cast<milliseconds>(system_clock::now() - ApplicationStartTime).count()); +    return uint32(duration_cast<milliseconds>(steady_clock::now() - ApplicationStartTime).count());  }  inline uint32 getMSTimeDiff(uint32 oldMSTime, uint32 newMSTime) | 
