#include "analysisbyrediscache.h"
|
#include "ui_analysisbyrediscache.h"
|
#include <QDateTime>
|
#include <QDebug>
|
#include <QElapsedTimer>
|
|
AnalysisByRedisCache::AnalysisByRedisCache(QWidget *parent) :
|
QMainWindow(parent),
|
ui(new Ui::AnalysisByRedisCache)
|
{
|
ui->setupUi(this);
|
|
ui->dateEdit->setDate(QDate::currentDate().addMonths(-1));
|
|
if(m_dbCache.initMySQL("127.0.0.1",3306,"root","root","stock_plan")){
|
qDebug()<<"mysql conn ok";
|
}else{
|
qDebug()<<"mysql conn fail";
|
}
|
|
if(m_dbCache.initRedis("127.0.0.1",6379)){
|
qDebug()<<"redis conn ok";
|
}else{
|
qDebug()<<"redis conn fail";
|
}
|
|
m_model = new QStandardItemModel(this);
|
ui->tableView->setModel(m_model);
|
|
m_modelResult = new QStandardItemModel(this);
|
ui->tableView_2->setModel(m_modelResult);
|
|
QStringList labels;
|
labels<<"股票名字"<<"市值"<<"收盘价"<<"涨跌幅"<<"排名"<<"交易时间"<<"排名变化"<<"底部反转时间"<<"一阳穿四线时间";
|
m_model->setHorizontalHeaderLabels(labels);
|
m_modelResult->setHorizontalHeaderLabels(labels);
|
}
|
|
AnalysisByRedisCache::~AnalysisByRedisCache()
|
{
|
delete ui;
|
}
|
|
void AnalysisByRedisCache::on_pushButton_loadDataSQLRedis_clicked()
|
{
|
ui->label_load->setText("缓存加载中...");
|
QString begin_date = ui->dateEdit->text();
|
QString limit_str = ui->comboBox->currentText();
|
QString sql = QString("select * from stock_day_info where time_trade > '%1' limit %2;").arg(begin_date).arg(limit_str);
|
qDebug()<<"sql:"<<sql;
|
QElapsedTimer timer;
|
timer.start();
|
|
auto result = m_dbCache.query(sql);
|
|
qDebug() << "Query executed in" << timer.elapsed() << "ms";
|
qDebug() << "Retrieved" << result.size() << "rows";
|
|
// 第二次查询应该从缓存获取
|
timer.restart();
|
result = m_dbCache.query(sql);
|
qint64 resTime = timer.elapsed();
|
qDebug() << "Cached query executed in" << resTime << "ms";
|
m_result = result;
|
ui->label_load->setText(QString("缓存加载%2条,耗时%1ms").arg(QString::number(resTime)).arg(QString::number(result.size())));
|
|
}
|
|
// 滑动窗口计算排名变化
|
void AnalysisByRedisCache::calcWindowRank(int start)
|
{
|
|
// 滑动窗口排名，6个要2+的前100
|
QVector<int> ranks;
|
for(int i=start;i<m_model->rowCount();++i){
|
ranks.append(m_model->item(i,4)->text().toInt());
|
if(i >= 6+start){
|
// 先统计排名计数
|
int cntBig=0, cntSmall=0;
|
for(int j=0;j<ranks.size();++j){
|
if(ranks.at(j) > 100){
|
cntBig++;
|
}else{
|
cntSmall++;
|
}
|
}
|
// 统计对比
|
if(cntSmall == 2 && ranks.back() <= 100){
|
// 将当前6个元素定为符合的窗口，背景色设置为紫色
|
for(int k=0;k<6;++k){
|
m_model->item(i-k,4)->setData(QColor("red"),Qt::BackgroundColorRole);
|
}
|
}
|
|
// 清除窗口第一个元素
|
ranks.pop_front();
|
}
|
|
}
|
|
}
|
void AnalysisByRedisCache::searchAndAnalysisData(QString name)
|
{
|
QElapsedTimer timer;
|
timer.start();
|
qDebug()<<"m_result size:"<<m_result.size()<<name;
|
|
int rank_val = 0;
|
int cnt = 0;
|
int bigFlag = false;
|
int bigRankUp = false; // 突然从很低的排名上升到大排名
|
bool big_3000 = false;
|
bool big_2000 = false;
|
for(auto it:m_result){
|
if(it["name"].toString() == name){
|
// qDebug()<<it["name"].toString()<<it["close"].toString()<<it["amount_rank"].toString()<<it["time_trade"].toString();
|
cnt++;
|
QList<QStandardItem*> items;
|
items.append(new QStandardItem(it["name"].toString()));
|
double market_capital = it["market_capital"].toDouble() / 100000000;
|
items.append(new QStandardItem(QString::number(market_capital)));
|
items.append(new QStandardItem(it["close"].toString()));
|
|
QStandardItem *percentItem = new QStandardItem(it["percent"].toString());
|
if(percentItem->text().toDouble() > 0){
|
percentItem->setData(QColor("red"),Qt::DecorationRole); // 添加装饰颜色为红色
|
percentItem->setData(QColor("red"),Qt::TextColorRole); // 添加文本颜色为红色
|
items.at(0)->setData(QColor("red"),Qt::TextColorRole); // 设置股票名字为红色
|
|
}else if(percentItem->text().toDouble() < 0){
|
percentItem->setData(QColor("green"),Qt::DecorationRole); // 添加装饰颜色为红色
|
percentItem->setData(QColor("green"),Qt::TextColorRole); // 设置背景颜色为绿色
|
items.at(0)->setData(QColor("green"),Qt::TextColorRole); // 设置股票名字为绿色
|
}
|
items.append(percentItem);
|
//items.append(new QStandardItem(it["percent"].toString()));
|
items.append(new QStandardItem(it["amount_rank"].toString()));
|
items.append(new QStandardItem(it["time_trade"].toString()));
|
QString amount_rank = it["amount_rank"].toString();
|
if(rank_val != 0){
|
double rank_rate = 1 - (amount_rank.toDouble()/rank_val);
|
//items.append(new QStandardItem(QString::number(rank_rate)));
|
|
QStandardItem * amount_rank_item = new QStandardItem(QString::number(rank_rate));
|
if(rank_rate > 0.9){
|
amount_rank_item->setData(QColor("red"),Qt::BackgroundColorRole);
|
bigFlag = true;
|
}
|
if(rank_rate > 0.7 && rank_val >= 1000 && amount_rank.toInt() < 1000){
|
bigRankUp = true;
|
bigFlag = true;
|
if(rank_rate > 0.9){
|
// 紫色
|
amount_rank_item->setData(QColor(170, 85, 255),Qt::BackgroundColorRole);
|
}else{
|
// 粉色
|
amount_rank_item->setData(QColor("pink"),Qt::BackgroundColorRole);
|
}
|
|
|
|
if(rank_val >= 3000){
|
amount_rank_item->setData(QColor(85,170,255),Qt::BackgroundColorRole);
|
big_3000 = true;
|
}
|
else if(rank_val >= 2000){
|
amount_rank_item->setData(QColor(85,255,255),Qt::BackgroundColorRole);
|
big_2000 = true;
|
}
|
|
|
|
|
}
|
items.append(amount_rank_item);
|
rank_val = amount_rank.toInt(); // 到后面再赋值
|
|
}
|
if(rank_val == 0 && amount_rank.toInt() != 0){
|
rank_val = amount_rank.toInt();
|
}
|
|
m_model->appendRow(items);
|
if(bigFlag){
|
QList<QStandardItem*> items_2;
|
for(int i=0;i<items.size();++i){
|
items_2.append(new QStandardItem(items.at(i)->text()));
|
if(i==6 && items.at(i)->text().toDouble() > 0.95){
|
items_2.back()->setData(QColor("red"),Qt::BackgroundColorRole);
|
}else if(i==6 && bigRankUp){
|
if(items.at(i)->text().toDouble() > 0.9){
|
items_2.back()->setData(QColor(170, 85, 255),Qt::BackgroundColorRole);
|
}else{
|
items_2.back()->setData(QColor("pink"),Qt::BackgroundColorRole);
|
}
|
|
if(big_3000){
|
items_2.back()->setData(QColor(85,170,255),Qt::BackgroundColorRole);
|
}
|
if(big_2000){
|
items_2.back()->setData(QColor(85,255,255),Qt::BackgroundColorRole);
|
}
|
|
}
|
}
|
m_modelResult->appendRow(items_2);
|
bigFlag = false;
|
bigRankUp = false;
|
big_3000 = false;
|
big_2000 = false;
|
}
|
}
|
|
}
|
qDebug()<<"cnt="<<cnt;
|
qDebug() << "Query executed in" << timer.elapsed() << "ms";
|
qDebug()<<"滑动窗口排名-cnt:"<<m_model->rowCount()<<cnt<<"start:"<<m_model->rowCount() - cnt + 1;
|
calcWindowRank(m_model->rowCount() - cnt + 1);
|
}
|
void AnalysisByRedisCache::on_pushButton_analysisByRedis_clicked()
|
{
|
// 单个分析
|
QString name = ui->lineEdit_name->text();
|
searchAndAnalysisData(name);
|
|
}
|
|
void AnalysisByRedisCache::on_pushButton_batchAnalysis_clicked()
|
{
|
|
// 批量分析
|
// 重设模型的行数为0，清空一下数据
|
m_model->setRowCount(0);
|
m_modelResult->setRowCount(0);
|
|
int batch_num = ui->comboBox_batchNum->currentText().toInt(); // 批量分析个股的数量
|
// 拿出个股交易额排名的前N名，进行缓存分析
|
emit getStockNamesSignal(batch_num);
|
|
}
|
