account_manager.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <string>
#include <unordered_map>
#include <optional>
#include <mutex>
#include <vector>
#include "app/model/account.hpp"
 
namespace fix40 {
 
// 前向声明
class IStore;
 
class AccountManager {
public:
    // -------------------------------------------------------------------------
    // 构造函数
    // -------------------------------------------------------------------------
 
    AccountManager();
 
    explicit AccountManager(IStore* store);
 
    ~AccountManager() = default;
 
    // 禁用拷贝
    AccountManager(const AccountManager&) = delete;
    AccountManager& operator=(const AccountManager&) = delete;
 
    // -------------------------------------------------------------------------
    // 账户管理
    // -------------------------------------------------------------------------
 
    Account createAccount(const std::string& accountId, double initialBalance);
 
    std::optional<Account> getAccount(const std::string& accountId) const;
 
    bool hasAccount(const std::string& accountId) const;
 
    std::vector<std::string> getAllAccountIds() const;
 
    size_t size() const;
 
    // -------------------------------------------------------------------------
    // 保证金操作
    // -------------------------------------------------------------------------
 
    bool freezeMargin(const std::string& accountId, double amount);
 
    bool unfreezeMargin(const std::string& accountId, double amount);
 
    bool confirmMargin(const std::string& accountId, double frozenAmount, double usedAmount);
 
    bool releaseMargin(const std::string& accountId, double amount);
 
    // -------------------------------------------------------------------------
    // 盈亏操作
    // -------------------------------------------------------------------------
 
    bool updatePositionProfit(const std::string& accountId, double profit);
 
    bool addCloseProfit(const std::string& accountId, double profit);
 
    // -------------------------------------------------------------------------
    // 清理方法
    // -------------------------------------------------------------------------
 
    void clear();
 
private:
    void persistAccount(const Account& account);
 
    std::unordered_map<std::string, Account> accounts_;
 
    IStore* store_;
 
    mutable std::mutex mutex_;
};
 
} // namespace fix40