run_scope.h

Go to the documentation of this file.

// SPDX-License-Identifier: Apache-2.0
#pragma once
 
#include <entropic/entropic_export.h>
 
#include <atomic>
#include <cstddef>
#include <string>
 
namespace entropic {
 
ENTROPIC_EXPORT const std::atomic<bool>* current_run_cancel();
 
ENTROPIC_EXPORT bool current_run_cancelled();
 
ENTROPIC_EXPORT bool cancel_current_run();
 
class ENTROPIC_EXPORT RunCancelScope {
public:
    explicit RunCancelScope(std::atomic<bool>* token);
 
    ~RunCancelScope();
 
    RunCancelScope(const RunCancelScope&) = delete;
    RunCancelScope& operator=(const RunCancelScope&) = delete;
 
private:
    std::atomic<bool>* previous_;
};
 
ENTROPIC_EXPORT std::string current_run_session();
 
class ENTROPIC_EXPORT RunSessionScope {
public:
    explicit RunSessionScope(std::string key);
 
    ~RunSessionScope();
 
    RunSessionScope(const RunSessionScope&) = delete;
    RunSessionScope& operator=(const RunSessionScope&) = delete;
 
private:
    std::string previous_;  
};
 
using RunStreamFn = void (*)(const char* chunk, size_t len, void* user_data);
 
ENTROPIC_EXPORT bool emit_to_current_run_stream(const char* chunk, size_t len);
 
class ENTROPIC_EXPORT RunStreamScope {
public:
    RunStreamScope(RunStreamFn fn, void* user_data);
 
    ~RunStreamScope();
 
    RunStreamScope(const RunStreamScope&) = delete;
    RunStreamScope& operator=(const RunStreamScope&) = delete;
 
private:
    RunStreamFn previous_fn_;  
    void* previous_ud_;        
};
 
}  // namespace entropic