mirror of
https://gitee.com/fasiondog/hikyuu.git
synced 2024-12-02 20:08:26 +08:00
422 lines
12 KiB
C++
422 lines
12 KiB
C++
/*
|
||
* StockManager.cpp
|
||
*
|
||
* Created on: 2011-11-9
|
||
* Author: fasiondog
|
||
*/
|
||
|
||
#ifndef NOMINMAX
|
||
#define NOMINMAX
|
||
#endif
|
||
|
||
#include "GlobalInitializer.h"
|
||
#include <chrono>
|
||
#include <fmt/format.h>
|
||
#include <boost/lexical_cast.hpp>
|
||
#include <boost/algorithm/string.hpp>
|
||
|
||
#include "utilities/IniParser.h"
|
||
#include "utilities/util.h"
|
||
#include "StockManager.h"
|
||
#include "GlobalTaskGroup.h"
|
||
#include "data_driver/kdata/cvs/KDataTempCsvDriver.h"
|
||
#include "data_driver/base_info/sqlite/SQLiteBaseInfoDriver.h"
|
||
#include "data_driver/base_info/mysql/MySQLBaseInfoDriver.h"
|
||
#include "data_driver/block_info/qianlong/QLBlockInfoDriver.h"
|
||
#include "data_driver/kdata/hdf5/H5KDataDriver.h"
|
||
#include "data_driver/kdata/tdx/TdxKDataDriver.h"
|
||
#include "data_driver/kdata/mysql/MySQLKDataDriver.h"
|
||
|
||
namespace hku {
|
||
|
||
StockManager* StockManager::m_sm = nullptr;
|
||
|
||
void StockManager::quit() {
|
||
releaseThreadPool();
|
||
if (m_sm) {
|
||
delete m_sm;
|
||
m_sm = nullptr;
|
||
}
|
||
}
|
||
|
||
StockManager::StockManager() {}
|
||
StockManager::~StockManager() {
|
||
auto tg = getGlobalTaskGroup();
|
||
if (tg && !tg->done()) {
|
||
releaseThreadPool();
|
||
}
|
||
fmt::print("Quit Hikyuu system!\n\n");
|
||
}
|
||
|
||
StockManager& StockManager::instance() {
|
||
if (!m_sm) {
|
||
m_sm = new StockManager();
|
||
}
|
||
return (*m_sm);
|
||
}
|
||
|
||
Parameter default_preload_param() {
|
||
Parameter param;
|
||
param.set<bool>("day", true);
|
||
param.set<bool>("week", false);
|
||
param.set<bool>("month", false);
|
||
param.set<bool>("quarter", false);
|
||
param.set<bool>("halfyear", false);
|
||
param.set<bool>("year", false);
|
||
param.set<bool>("min", false);
|
||
param.set<bool>("min5", false);
|
||
param.set<bool>("min15", false);
|
||
param.set<bool>("min30", false);
|
||
param.set<bool>("min60", false);
|
||
return param;
|
||
}
|
||
|
||
Parameter default_other_param() {
|
||
Parameter param;
|
||
param.set<string>("tmpdir", ".");
|
||
param.set<string>("logger", "");
|
||
return param;
|
||
}
|
||
|
||
void StockManager::init(const Parameter& baseInfoParam, const Parameter& blockParam,
|
||
const Parameter& kdataParam, const Parameter& preloadParam,
|
||
const Parameter& hikyuuParam) {
|
||
m_baseInfoDriverParam = baseInfoParam;
|
||
m_blockDriverParam = blockParam;
|
||
m_kdataDriverParam = kdataParam;
|
||
m_preloadParam = preloadParam;
|
||
m_hikyuuParam = hikyuuParam;
|
||
|
||
// 创建内部线程池
|
||
// 不能同过 GlobalInitializer 初始化全局线程池
|
||
// 原因是 std::thread 无法在 dllmain 中创建使用，会造成死锁
|
||
initThreadPool();
|
||
|
||
// 获取临时路径信息
|
||
try {
|
||
m_tmpdir = hikyuuParam.get<string>("tmpdir");
|
||
} catch (...) {
|
||
m_tmpdir = "";
|
||
}
|
||
|
||
try {
|
||
m_datadir = hikyuuParam.get<string>("datadir");
|
||
} catch (...) {
|
||
m_datadir = "";
|
||
}
|
||
|
||
m_stockDict.clear();
|
||
m_marketInfoDict.clear();
|
||
m_stockTypeInfo.clear();
|
||
|
||
string funcname(" [StockManager::init]");
|
||
|
||
//加载证券基本信息
|
||
BaseInfoDriverPtr base_info = DataDriverFactory::getBaseInfoDriver(baseInfoParam);
|
||
HKU_CHECK(base_info, "Failed get base info driver!");
|
||
base_info->loadBaseInfo();
|
||
|
||
//获取板块驱动
|
||
m_blockDriver = DataDriverFactory::getBlockDriver(blockParam);
|
||
|
||
//获取K线数据驱动并预加载指定的数据
|
||
HKU_INFO("Loading KData...");
|
||
std::chrono::system_clock::time_point start_time = std::chrono::system_clock::now();
|
||
|
||
KDataDriverPtr kdata_driver = DataDriverFactory::getKDataDriver(m_kdataDriverParam);
|
||
|
||
setKDataDriver(kdata_driver);
|
||
|
||
// add special Market, for temp csv file
|
||
m_marketInfoDict["TMP"] =
|
||
MarketInfo("TMP", "Temp Csv file", "temp load from csv file", "000001", Null<Datetime>());
|
||
|
||
std::chrono::duration<double> sec = std::chrono::system_clock::now() - start_time;
|
||
HKU_INFO("{:<.2f}s Loaded Data.", sec.count());
|
||
}
|
||
|
||
void StockManager::setKDataDriver(const KDataDriverPtr& driver) {
|
||
if (!driver) {
|
||
HKU_ERROR("kdata driver is null!");
|
||
return;
|
||
}
|
||
|
||
if (m_kdataDriverParam == driver->getParameter()) {
|
||
} else {
|
||
m_kdataDriverParam = driver->getParameter();
|
||
}
|
||
|
||
bool preload_day = false;
|
||
try {
|
||
preload_day = m_preloadParam.get<bool>("day");
|
||
if (preload_day)
|
||
HKU_INFO("Preloading all day kdata to buffer!");
|
||
} catch (...) {
|
||
preload_day = false;
|
||
}
|
||
|
||
bool preload_week = false;
|
||
try {
|
||
preload_week = m_preloadParam.get<bool>("week");
|
||
if (preload_week)
|
||
HKU_INFO("Preloading all week kdata to buffer!");
|
||
} catch (...) {
|
||
preload_week = false;
|
||
}
|
||
|
||
bool preload_month = false;
|
||
try {
|
||
preload_month = m_preloadParam.get<bool>("month");
|
||
if (preload_week)
|
||
HKU_INFO("Preloading all month kdata to buffer!");
|
||
} catch (...) {
|
||
preload_month = false;
|
||
}
|
||
|
||
bool preload_quarter = false;
|
||
try {
|
||
preload_quarter = m_preloadParam.get<bool>("quarter");
|
||
if (preload_quarter)
|
||
HKU_INFO("Preloading all quarter kdata to buffer!");
|
||
} catch (...) {
|
||
preload_quarter = false;
|
||
}
|
||
|
||
bool preload_halfyear = false;
|
||
try {
|
||
preload_halfyear = m_preloadParam.get<bool>("halfyear");
|
||
if (preload_halfyear)
|
||
HKU_INFO("Preloading all halfyear kdata to buffer!");
|
||
} catch (...) {
|
||
preload_halfyear = false;
|
||
}
|
||
|
||
bool preload_year = false;
|
||
try {
|
||
preload_year = m_preloadParam.get<bool>("year");
|
||
if (preload_year)
|
||
HKU_INFO("Preloading all year kdata to buffer!");
|
||
} catch (...) {
|
||
preload_year = false;
|
||
}
|
||
|
||
bool preload_min = false;
|
||
try {
|
||
preload_min = m_preloadParam.get<bool>("min");
|
||
if (preload_min)
|
||
HKU_INFO("Preloading all 1 min kdata to buffer!");
|
||
} catch (...) {
|
||
preload_min = false;
|
||
}
|
||
|
||
bool preload_min5 = false;
|
||
try {
|
||
preload_min5 = m_preloadParam.get<bool>("min5");
|
||
if (preload_min5)
|
||
HKU_INFO("Preloading all 5 min kdata to buffer!");
|
||
} catch (...) {
|
||
preload_min5 = false;
|
||
}
|
||
|
||
bool preload_min15 = false;
|
||
try {
|
||
preload_min15 = m_preloadParam.get<bool>("min15");
|
||
if (preload_min15)
|
||
HKU_INFO("Preloading all 15 min kdata to buffer!");
|
||
} catch (...) {
|
||
preload_min15 = false;
|
||
}
|
||
|
||
bool preload_min30 = false;
|
||
try {
|
||
preload_min30 = m_preloadParam.get<bool>("min30");
|
||
if (preload_min30)
|
||
HKU_INFO("Preloading all 30 min kdata to buffer!");
|
||
} catch (...) {
|
||
preload_min30 = false;
|
||
}
|
||
|
||
bool preload_min60 = false;
|
||
try {
|
||
preload_min60 = m_preloadParam.get<bool>("min60");
|
||
if (preload_min60)
|
||
HKU_INFO("Preloading all 60 min kdata to buffer!");
|
||
} catch (...) {
|
||
preload_min60 = false;
|
||
}
|
||
|
||
for (auto iter = m_stockDict.begin(); iter != m_stockDict.end(); ++iter) {
|
||
if (iter->second.market() == "TMP")
|
||
continue;
|
||
|
||
iter->second.setKDataDriver(driver);
|
||
|
||
if (preload_day)
|
||
iter->second.loadKDataToBuffer(KQuery::DAY);
|
||
|
||
if (preload_week)
|
||
iter->second.loadKDataToBuffer(KQuery::WEEK);
|
||
|
||
if (preload_month)
|
||
iter->second.loadKDataToBuffer(KQuery::MONTH);
|
||
|
||
if (preload_quarter)
|
||
iter->second.loadKDataToBuffer(KQuery::QUARTER);
|
||
|
||
if (preload_halfyear)
|
||
iter->second.loadKDataToBuffer(KQuery::HALFYEAR);
|
||
|
||
if (preload_year)
|
||
iter->second.loadKDataToBuffer(KQuery::YEAR);
|
||
|
||
if (preload_min)
|
||
iter->second.loadKDataToBuffer(KQuery::MIN);
|
||
|
||
if (preload_min5)
|
||
iter->second.loadKDataToBuffer(KQuery::MIN5);
|
||
|
||
if (preload_min15)
|
||
iter->second.loadKDataToBuffer(KQuery::MIN15);
|
||
|
||
if (preload_min30)
|
||
iter->second.loadKDataToBuffer(KQuery::MIN30);
|
||
|
||
if (preload_min60)
|
||
iter->second.loadKDataToBuffer(KQuery::MIN60);
|
||
}
|
||
}
|
||
|
||
string StockManager::tmpdir() const {
|
||
return m_tmpdir;
|
||
}
|
||
|
||
string StockManager::datadir() const {
|
||
return m_datadir;
|
||
}
|
||
|
||
Stock StockManager::getStock(const string& querystr) const {
|
||
Stock result;
|
||
string query_str = querystr;
|
||
to_upper(query_str);
|
||
auto iter = m_stockDict.find(query_str);
|
||
if (iter != m_stockDict.end()) {
|
||
return iter->second;
|
||
}
|
||
return result;
|
||
}
|
||
|
||
MarketInfo StockManager::getMarketInfo(const string& market) const {
|
||
string market_tmp = market;
|
||
to_upper(market_tmp);
|
||
auto iter = m_marketInfoDict.find(market_tmp);
|
||
if (iter != m_marketInfoDict.end()) {
|
||
return iter->second;
|
||
}
|
||
return Null<MarketInfo>();
|
||
}
|
||
|
||
StockTypeInfo StockManager::getStockTypeInfo(uint32 type) const {
|
||
auto iter = m_stockTypeInfo.find(type);
|
||
if (iter != m_stockTypeInfo.end()) {
|
||
return iter->second;
|
||
}
|
||
return Null<StockTypeInfo>();
|
||
}
|
||
|
||
MarketList StockManager::getAllMarket() const {
|
||
MarketList result;
|
||
auto iter = m_marketInfoDict.begin();
|
||
for (; iter != m_marketInfoDict.end(); ++iter) {
|
||
result.push_back(iter->first);
|
||
}
|
||
return result;
|
||
}
|
||
|
||
Block StockManager::getBlock(const string& category, const string& name) {
|
||
return m_blockDriver ? m_blockDriver->getBlock(category, name) : Block();
|
||
}
|
||
|
||
BlockList StockManager::getBlockList(const string& category) {
|
||
return m_blockDriver ? m_blockDriver->getBlockList(category) : BlockList();
|
||
}
|
||
|
||
BlockList StockManager::getBlockList() {
|
||
return m_blockDriver ? m_blockDriver->getBlockList() : BlockList();
|
||
}
|
||
|
||
DatetimeList StockManager::getTradingCalendar(const KQuery& query, const string& market) {
|
||
Stock stock = getStock("SH000001");
|
||
size_t start_ix = 0, end_ix = 0;
|
||
DatetimeList result;
|
||
if (stock.getIndexRange(query, start_ix, end_ix)) {
|
||
result = stock.getDatetimeList(start_ix, end_ix, query.kType());
|
||
}
|
||
return result;
|
||
}
|
||
|
||
Stock StockManager::addTempCsvStock(const string& code, const string& day_filename,
|
||
const string& min_filename, price_t tick, price_t tickValue,
|
||
int precision, size_t minTradeNumber, size_t maxTradeNumber) {
|
||
string new_code(code);
|
||
to_upper(new_code);
|
||
Stock result("TMP", new_code, day_filename, STOCKTYPE_TMP, true, Datetime(199901010000),
|
||
Null<Datetime>(), tick, tickValue, precision, minTradeNumber, maxTradeNumber);
|
||
|
||
KDataTempCsvDriver* p = new KDataTempCsvDriver(day_filename, min_filename);
|
||
result.setKDataDriver(KDataDriverPtr(p));
|
||
result.loadKDataToBuffer(KQuery::DAY);
|
||
result.loadKDataToBuffer(KQuery::MIN);
|
||
|
||
if (!loadStock(result)) {
|
||
//加入失败，返回Null<Stock>
|
||
return Null<Stock>();
|
||
}
|
||
|
||
return result;
|
||
}
|
||
|
||
void StockManager::removeTempCsvStock(const string& code) {
|
||
string query_str = "TMP" + code;
|
||
to_upper(query_str);
|
||
auto iter = m_stockDict.find(query_str);
|
||
if (iter != m_stockDict.end()) {
|
||
m_stockDict.erase(iter);
|
||
}
|
||
}
|
||
|
||
bool StockManager::loadStock(const Stock& stock) {
|
||
string market_code(stock.market_code());
|
||
to_upper(market_code);
|
||
if (m_stockDict.find(market_code) != m_stockDict.end()) {
|
||
HKU_ERROR("The stock had exist! {}", market_code);
|
||
return false;
|
||
}
|
||
|
||
m_stockDict[market_code] = stock;
|
||
return true;
|
||
}
|
||
|
||
bool StockManager::loadMarketInfo(const MarketInfo& marketInfo) {
|
||
string market = marketInfo.market();
|
||
to_upper(market);
|
||
if (m_marketInfoDict.find(market) != m_marketInfoDict.end()) {
|
||
HKU_ERROR("The marketInfo had exist! {}", market);
|
||
return false;
|
||
}
|
||
|
||
m_marketInfoDict[market] = marketInfo;
|
||
return true;
|
||
}
|
||
|
||
bool StockManager::loadStockTypeInfo(const StockTypeInfo& stkTypeInfo) {
|
||
if (m_stockTypeInfo.find(stkTypeInfo.type()) != m_stockTypeInfo.end()) {
|
||
HKU_ERROR("The stockTypeInfo had exist! {}", stkTypeInfo.type());
|
||
return false;
|
||
}
|
||
|
||
m_stockTypeInfo[stkTypeInfo.type()] = stkTypeInfo;
|
||
return true;
|
||
}
|
||
|
||
} // namespace hku
|