#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);

    QStringList labels;
    labels<<"股票名字"<<"市值"<<"收盘价"<<"涨跌幅"<<"排名"<<"交易时间"<<"排名变化";
    m_model->setHorizontalHeaderLabels(labels);
}

AnalysisByRedisCache::~AnalysisByRedisCache()
{
    delete ui;
}

void AnalysisByRedisCache::on_pushButton_loadDataSQLRedis_clicked()
{
    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);
    qDebug() << "Cached query executed in" << timer.elapsed() << "ms";
    m_result = result;
}

// 滑动窗口计算排名变化
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;
    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)));
                rank_val = amount_rank.toInt();
                QStandardItem * amount_rank_item = new QStandardItem(QString::number(rank_rate));
                if(rank_rate > 0.9){
                    amount_rank_item->setData(QColor("red"),Qt::BackgroundColorRole);
                }
                items.append(amount_rank_item);

            }
            if(rank_val == 0 && amount_rank.toInt() != 0){
                rank_val = amount_rank.toInt();
            }

            m_model->appendRow(items);
        }

    }
    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);

    int batch_num = ui->comboBox_batchNum->currentText().toInt(); // 批量分析个股的数量
    // 拿出个股交易额排名的前N名，进行缓存分析
    emit getStockNamesSignal(batch_num);

}