client_state.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <string>
#include <vector>
#include <mutex>
#include <atomic>
#include <functional>
#include <unordered_map>
#include <filesystem>
#include <chrono>
 
namespace fix40::client {
 
enum class ConnectionState {
    DISCONNECTED,
    CONNECTING,
    CONNECTED,
    LOGGING_IN,
    LOGGED_IN,
    ERROR
};
 
struct AccountInfo {
    double balance = 0.0;         
    double available = 0.0;       
    double frozenMargin = 0.0;    
    double usedMargin = 0.0;      
    double positionProfit = 0.0;  
    double closeProfit = 0.0;     
    double dynamicEquity = 0.0;   
    double riskRatio = 0.0;       
};
 
struct PositionInfo {
    std::string instrumentId;
    int64_t longPosition = 0;
    double longAvgPrice = 0.0;
    int64_t shortPosition = 0;
    double shortAvgPrice = 0.0;
    double profit = 0.0;
    bool quantitiesValid = false; 
};
 
enum class OrderState {
    PENDING_NEW,
    NEW,
    PARTIALLY_FILLED,
    FILLED,
    CANCELED,
    REJECTED
};
 
struct OrderInfo {
    std::string clOrdID;
    std::string orderId;
    std::string symbol;
    std::string side;           
    double price = 0.0;
    int64_t orderQty = 0;
    int64_t filledQty = 0;
    double avgPx = 0.0;
    OrderState state = OrderState::PENDING_NEW;
    std::string text;           
    std::string updateTime;
};
 
class ClientState {
public:
    using StateChangeCallback = std::function<void()>;
 
    ClientState() = default;
 
    // =========================================================================
    // 连接状态
    // =========================================================================
    
    void setConnectionState(ConnectionState state);
    ConnectionState getConnectionState() const;
    std::string getConnectionStateString() const;
    
    void setUserId(const std::string& userId);
    std::string getUserId() const;
 
    // =========================================================================
    // 账户信息
    // =========================================================================
    
    void updateAccount(const AccountInfo& info);
    AccountInfo getAccount() const;
 
    // =========================================================================
    // 持仓信息
    // =========================================================================
    
    void updatePosition(const PositionInfo& pos);
    void setPositions(const std::vector<PositionInfo>& positions);
    std::vector<PositionInfo> getPositions() const;
    void clearPositions();
 
    // =========================================================================
    // 订单信息
    // =========================================================================
    
    void addOrder(const OrderInfo& order);
    void updateOrder(const std::string& clOrdID, const OrderInfo& order);
    std::vector<OrderInfo> getOrders() const;
    std::vector<OrderInfo> getActiveOrders() const;
    void clearOrders();
 
    void setOrders(const std::vector<OrderInfo>& orders);
    
    void saveOrders(const std::string& filepath = "");
    
    void loadOrders(const std::string& filepath = "");
 
    // =========================================================================
    // 合约搜索结果
    // =========================================================================
    
    void setSearchResults(const std::vector<std::string>& results);
    std::vector<std::string> getSearchResults() const;
 
    // =========================================================================
    // 状态变更通知
    // =========================================================================
    
    void setOnStateChange(StateChangeCallback callback);
    void notifyStateChange();
 
    // =========================================================================
    // 消息/错误
    // =========================================================================
    
    void setLastError(const std::string& error);
    std::string getLastError() const;
    
    void addMessage(const std::string& msg);
    std::vector<std::string> getMessages() const;
 
private:
    mutable std::mutex mutex_;
    
    // 连接状态
    std::atomic<ConnectionState> connectionState_{ConnectionState::DISCONNECTED};
    std::string userId_;
    
    // 账户信息
    AccountInfo account_;
    
    // 持仓信息
    std::vector<PositionInfo> positions_;
    
    // 订单信息
    std::unordered_map<std::string, OrderInfo> orders_;
    // 订单插入顺序（用于稳定展示顺序/保存顺序）
    std::vector<std::string> orderSequence_;
    
    // 合约搜索结果
    std::vector<std::string> searchResults_;
    
    // 消息
    std::vector<std::string> messages_;
    std::string lastError_;
    
    // 状态变更回调
    StateChangeCallback onStateChange_;
    
    // 节流：上次通知时间
    mutable std::chrono::steady_clock::time_point lastNotifyTime_;
};
 
} // namespace fix40::client