/*******************************************************************************
 * Copyright 上海赜睿信息科技有限公司(Zilliz) - All Rights Reserved
 * Unauthorized copying of this file, via any medium is strictly prohibited.
 * Proprietary and confidential.
 ******************************************************************************/
#pragma once

#include <condition_variable>
#include <thread>
#include <mutex>
#include <queue>
#include <atomic>

namespace zilliz {
namespace vecwise {
namespace engine {

class Env {
public:
    Env();

    Env(const Env&) = delete;
    Env& operator=(const Env&) = delete;

    void schedule(void (*function_)(void* arg_), void* arg_);

    virtual void Stop();

    virtual ~Env();

    static Env* Default();

protected:
    void backgroud_thread_main();
    static void BackgroundThreadEntryPoint(Env* env) {
        env->backgroud_thread_main();
    }

    struct BGWork {
      explicit BGWork(void (*function_)(void*), void* arg_)
          : _function(function_), _arg(arg_)  {}

      void (* const _function)(void*);
      void* const _arg;
    };

    std::mutex _bg_work_mutex;
    std::condition_variable _bg_work_cv;
    std::queue<BGWork> _bg_work_queue;
    bool _bg_work_started;
    std::atomic<bool> _shutting_down;

}; // Env

} // namespace engine
} // namespace vecwise
} // namespace zilliz