sc2_connection.h

Go to the documentation of this file.

#pragma once

#include <string>
#include <mutex>
#include <condition_variable>
#include <deque>
#include <functional>
#include <atomic>

struct mg_connection;

namespace SC2APIProtocol {
    class Request;
    class Response;
}

namespace sc2 {


class Connection {
public:
    Connection();
    ~Connection();

    bool Connect(const std::string& address, int port, bool verbose = true);

    void Send(const SC2APIProtocol::Request* request);

    bool Receive(SC2APIProtocol::Response*& response, unsigned int timeout_ms);

    void PopResponse(SC2APIProtocol::Response*& response);

    void SetTimeoutCallback(std::function<void()> callback);

    void SetConnectionClosedCallback(std::function<void()> callback);

    bool HasConnection() const;

    void Disconnect();

    bool PollResponse();

    void PushResponse(SC2APIProtocol::Response*& response);


    std::function<void()> timeout_callback_;             
    std::function<void()> connection_closed_callback_;   

    mg_connection* connection_;                          

private:
    bool verbose_;                                   

    std::deque<SC2APIProtocol::Response*> queue_; 
    std::mutex mutex_;                               
    std::condition_variable condition_;              

    std::atomic_bool has_response_;                  
};

}