test/hikyuu2
1
0
Fork 0
mirror of https://gitee.com/fasiondog/hikyuu.git synced 2024-12-02 11:58:21 +08:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

mysql 支持分时(continnue)

Browse Source
This commit is contained in:
fasiondog 2023-12-03 02:09:51 +08:00
parent 82c944b86e
commit 580e9108c3
4 changed files with 66 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

62
hikyuu_cpp/hikyuu/data_driver/kdata/mysql/MySQLKDataDriver.cpp
View File

@ -126,7 +126,7 @@ size_t MySQLKDataDriver::getCount(const string& market, const string& code, KQue
size_t result = 0;
try {
result = m_connect->queryInt(
result = m_connect->queryNumber(
fmt::format("select count(1) from {}", _getTableName(market, code, kType)), 0);
} catch (...) {
// 表可能不存在, 不打印异常信息
@ -164,4 +164,64 @@ bool MySQLKDataDriver::getIndexRangeByDate(const string& market, const string& c
return true;
}
TimeLineList MySQLKDataDriver::getTimeLineList(const string& market, const string& code,
const KQuery& query) {
TimeLineList result;
HKU_IF_RETURN(query.queryType() >= KQuery::INVALID, result);
string table = fmt::format("`{}_time`.`{}`", market, code);
to_lower(table);
HKU_INFO("getTimeLineList {}", table);
SQLStatementPtr st;
try {
if (query.queryType() == KQuery::DATE) {
st = m_connect->getStatement(fmt::format(
"select `date`, `price`, `vol` from {} where date >= {} and date < {} order by date",
table, query.startDatetime().number(), query.endDatetime().number()));
} else {
size_t total = m_connect->queryNumber(fmt::format("select count(1) from {}", table), 0);
size_t start = m_connect->queryNumber(
fmt::format("select count(1) from {} where date<{}", table, query.start()), 0);
size_t end = m_connect->queryNumber(
fmt::format("select count(1) from {} where date<{}", table, query.end()), 0);
st = m_connect->getStatement(
fmt::format("select `date`, `price`, `vol` from {} order by date limit {}, {}", table,
start, end - start));
}
} catch (const std::exception& e) {
HKU_ERROR("Failed get {} timeline! {}", table, e.what());
return result;
} catch (...) {
HKU_ERROR("Failed get {} timeline!", table);
return result;
}
try {
m_connect->transaction();
st->exec();
while (st->moveNext()) {
uint64_t date = 0;
double price = 0.0, vol = 0.0;
try {
st->getColumn(0, date, price, vol);
result.emplace_back(Datetime(date), price, vol);
} catch (const std::exception& e) {
HKU_ERROR("Failed get {} timeline! {}", table, e.what());
} catch (...) {
HKU_ERROR("Failed get {} timeline!", table, code);
}
}
m_connect->commit();
} catch (const std::exception& e) {
m_connect->rollback();
} catch (...) {
m_connect->rollback();
}
return result;
}
} /* namespace hku */

3
hikyuu_cpp/hikyuu/data_driver/kdata/mysql/MySQLKDataDriver.h
View File

@ -48,6 +48,9 @@ public:
virtual KRecordList getKRecordList(const string& market, const string& code,
const KQuery& query) override;
virtual TimeLineList getTimeLineList(const string& market, const string& code,
const KQuery& query) override;
private:
string _getTableName(const string& market, const string& code, KQuery::KType ktype);
KRecordList _getKRecordList(const string& market, const string& code, KQuery::KType kType,

1
hikyuu_pywrap/_TimeLineRecord.cpp
View File

@ -22,6 +22,7 @@ void export_TimeLineReord() {
class_<TimeLineRecord>("TimeLineRecord", "分时线记录，属性可读写", init<>())
.def(init<const Datetime&, price_t, price_t>())
.def(self_ns::str(self))
.def(self_ns::repr(self))
.def_readwrite("date", &TimeLineRecord::datetime, "日期时间")
.def_readwrite("price", &TimeLineRecord::price, "价格")
.def_readwrite("vol", &TimeLineRecord::vol, "成交量")

1
hikyuu_pywrap/_TransRecord.cpp
View File

@ -22,6 +22,7 @@ void export_TransRecord() {
class_<TransRecord>("TransRecord", init<>())
.def(init<const Datetime&, price_t, price_t, TransRecord::DIRECT>())
.def(self_ns::str(self))
.def(self_ns::repr(self))
.def_readwrite("date", &TransRecord::datetime, "时间")
.def_readwrite("price", &TransRecord::price, "价格")
.def_readwrite("vol", &TransRecord::vol, "成交量")