milvus/cpp/src/db/Env.cpp

#include <assert.h>
#include "Env.h"

namespace zilliz {
namespace vecwise {
namespace engine {

Env::Env() : _bg_work_started(false) {
}

void Env::schedule(void (*function_)(void* arg_), void* arg_) {
    std::unique_lock<std::mutex> lock(_bg_work_mutex);

    if (!_bg_work_started) {
        _bg_work_started = true;
        std::thread bg_thread(Env::BackgroundThreadEntryPoint, this);
        bg_thread.detach();
    }

    if (_bg_work_queue.empty()) {
        _bg_work_cv.notify_one();
    }

    _bg_work_queue.emplace(function_, arg_);
}

void Env::backgroud_thread_main() {
    while (true) {
        std::unique_lock<std::mutex> lock(_bg_work_mutex);
        while (_bg_work_queue.empty()) {
            _bg_work_cv.wait(lock);
        }

        assert(!_bg_work_queue.empty());
        auto bg_function = _bg_work_queue.front()._function;
        void* bg_arg = _bg_work_queue.front()._arg;
        _bg_work_queue.pop();

        lock.unlock();
        bg_function(bg_arg);
    }
}

Env::~Env() {}

Env* Env::Default() {
    static Env env;
    return &env;
}

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