PowerModeler/source/dbStructureModel.cpp

#include "dbStructureModel.h"
#include "mainwindow.h"
#include "sqlQueryExecutor.h"
#include "logger.h"
// #include <QSqlDatabase>
// #include <QSqlQuery>

DBStructureModel::DBStructureModel(QObject* parent)
    : QAbstractItemModel(parent)
    , m_rootNode(new DBStructureNode(RootNode, "Root"))
    , m_pMainWindow(nullptr)
{

}

DBStructureModel::~DBStructureModel()
{
    delete m_rootNode;
}

void DBStructureModel::setMainWindow(MainWindow* window)
{
    m_pMainWindow = window;
}

DBStructureNode* DBStructureModel::getNode(const QModelIndex& index) const
{
    if(index.isValid())
        return static_cast<DBStructureNode*>(index.internalPointer());
    else
        return m_rootNode;
}

DBStructureNode* DBStructureModel::getConnectionNode(const QString& name) const
{
    for(int i = 0; i < m_rootNode->childCount(); ++i)
    {
        DBStructureNode* node = m_rootNode->child(i);
        if(node->type() == ConnectionNode && node->name() == name)
            return node;
    }
    return nullptr;
}

DBStructureNode* DBStructureModel::getModelNode(DBStructureNode* connNode, int modelID) const
{
    if(!connNode)
        return nullptr;

    for(int i = 0; i < connNode->childCount(); ++i)
    {
        DBStructureNode* node = connNode->child(i);
        if(node->type() == TableNode && node->data(Qt::UserRole + NodeDataRole::ID).toInt() == modelID)
            return node;
    }
    return nullptr;
}

QModelIndex DBStructureModel::index(int row, int column, const QModelIndex& parent) const
{
    if(!hasIndex(row, column, parent))
        return QModelIndex();

    DBStructureNode* parentNode = getNode(parent);
    DBStructureNode* childNode = parentNode->child(row);
    if(childNode)
        return createIndex(row, column, childNode);
    else
        return QModelIndex();
}

QModelIndex DBStructureModel::parent(const QModelIndex& index) const
{
    if(!index.isValid())
        return QModelIndex();

    DBStructureNode* childNode = static_cast<DBStructureNode*>(index.internalPointer());
    DBStructureNode* parentNode = childNode->parentNode();
    if(parentNode == m_rootNode)
        return QModelIndex();

    return createIndex(parentNode->row(), 0 ,parentNode);
}

int DBStructureModel::rowCount(const QModelIndex& parent) const
{
    DBStructureNode* parentNode = getNode(parent);
    return parentNode->childCount();
}

int DBStructureModel::columnCount(const QModelIndex& parent) const
{
    return DBStructureNode::ColumnCount;
}

QVariant DBStructureModel::data(const QModelIndex& index, int role) const
{
    if(!index.isValid())
        return QVariant();

    DBStructureNode* node = static_cast<DBStructureNode*>(index.internalPointer());

    switch (role)
    {
    case Qt::DisplayRole:
        return node->columnData(DBStructureNode::ColumnName);
    case Qt::DecorationRole:
        return node->icon();
    default:
        return node->data(role);
    }
}

QVariant DBStructureModel::headerData(int section, Qt::Orientation orientation, int role) const
{
    if(orientation == Qt::Horizontal && role == Qt::DisplayRole)
    {
        switch(section)
        {
        case DBStructureNode::ColumnName:
            return "Name";
        default:
            return QVariant();
        }
    }

    return QVariant();
}

void DBStructureModel::addConnection(const QString& name, const QString& dbType)
{
    beginInsertRows(QModelIndex(), m_rootNode->childCount(), m_rootNode->childCount()); //链接节点是根节点的(m_rootNode)的子节点，根节点的索引是QModelIndex()

    DBStructureNode* connectionNode = new DBStructureNode(ConnectionNode, name, m_rootNode);
    m_rootNode->appendChild(connectionNode);

    endInsertRows(); //该语句之后会触发rowsInserted(const QModelIndex &parent, int first, int last)信号，通知视图刷新对应行
}

void DBStructureModel::removeConnection(const QString& name)
{
    DBStructureNode* connNode = getConnectionNode(name);
    if(!connNode)
        return;

    beginRemoveRows(QModelIndex(), connNode->row(), connNode->row());
    m_rootNode->removeChild(connNode);
    endRemoveRows();
}

QModelIndex DBStructureModel::getConnNodeIndex(const QString& name)
{
    DBStructureNode* connNode = getConnectionNode(name);
    if(!connNode)
        return QModelIndex();
    else
        return index(connNode->row(), 0, QModelIndex());
}

void DBStructureModel::addDataModel(const QString& connection, Model& model)
{
    QModelIndex connIndex = getConnNodeIndex(connection);
    if(!connIndex.isValid())
    {
        LOG_ERROR("DBStructureModel", QString::fromWCharArray(L"获取ModelIndex失败，节点名称:%1").arg(connection));
        return;
    }

    DBStructureNode* connNode = getConnectionNode(connection);
    if(!connNode)
    {
        LOG_ERROR("DBStructureModel", QString::fromWCharArray(L"获取Node对象失败，节点名称:%1").arg(connection));
        return;
    }

    beginInsertRows(connIndex, connNode->childCount(), connNode->childCount());

    DBStructureNode* modelNode = new DBStructureNode(TableNode, model.name, connNode);
    modelNode->setData(Qt::UserRole + NodeDataRole::ID, model.id);
    for(int groupID : model.groups)
    {
        //QString groupName = SqlQueryExecutor::instance().getAttributeGroupName(connection, groupID);
        AttributeGroup group = SqlQueryExecutor::instance().getAttributeGroupData(connection, groupID);
        if(group.name.isEmpty())
            continue;
        DBStructureNode* groupNode = new DBStructureNode(GroupNode, group.name, modelNode);
        groupNode->setData(Qt::UserRole + NodeDataRole::ID, groupID);
        groupNode->setData(Qt::UserRole + NodeDataRole::Type, group.isPublic);
        modelNode->appendChild(groupNode);
    }
    connNode->appendChild(modelNode);

    endInsertRows();
}

void DBStructureModel::removeDataModel(DBStructureNode* modelNode)
{
    //可以不对node在进行一系列的检查，因为该函数调取之前已有相关逻辑
    int modelID = modelNode->data(Qt::UserRole + NodeDataRole::ID).toInt();
    //先从数据库中删除
    DBStructureNode* connNode = modelNode->parentNode();
    if(!(connNode && connNode->type() == ConnectionNode))
    {
        if(m_pMainWindow)
        {
            QString error = QString::fromWCharArray(L"删除失败，未找到所在链接节点信息");
            m_pMainWindow->showMessageDialog(type_information, QString::fromWCharArray(L"失败"), error);
        }
        return;
    }
    bool result = SqlQueryExecutor::instance().removeModel(connNode->name(), modelID);
    if(!result)
    {
        if(m_pMainWindow)
        {
            QString error = QString::fromWCharArray(L"删除失败，详情可见日志文件");
            m_pMainWindow->showMessageDialog(type_information, QString::fromWCharArray(L"失败"), error);
        }
        return;
    }
    else //从列表中删除
    {
        beginRemoveRows(index(connNode->row(), 0, QModelIndex()), modelNode->row(), modelNode->row());
        connNode->removeChild(modelNode);
        endRemoveRows();
    }
}

void DBStructureModel::addDataGroup(const QString& connection, int modelID, QVector<int> groups)
{
    DBStructureNode* connNode = getConnectionNode(connection);
    if(!connNode)
    {
        LOG_ERROR("DBStructureModel", QString::fromWCharArray(L"添加数组时获取ConnectionNode失败，节点名称:%1").arg(connection));
        return;
    }

    DBStructureNode* modelNode = getModelNode(connNode, modelID);
    if(!connNode)
    {
        LOG_ERROR("DBStructureModel", QString::fromWCharArray(L"添加数组时获取ModelNode失败，节点ID:%1").arg(modelID));
        return;
    }

    QModelIndex connIndex = index(connNode->row(), 0, QModelIndex());
    QModelIndex modelIndex = index(modelNode->row(), 0, connIndex);
    beginInsertRows(modelIndex, modelNode->childCount(), modelNode->childCount() + groups.count() - 1);

    for(int groupID : groups)
    {
        //QString groupName = SqlQueryExecutor::instance().getAttributeGroupName(connection, groupID);
        AttributeGroup group = SqlQueryExecutor::instance().getAttributeGroupData(connection, groupID);
        if(group.name.isEmpty())
            continue;
        DBStructureNode* groupNode = new DBStructureNode(GroupNode, group.name, modelNode);
        groupNode->setData(Qt::UserRole + NodeDataRole::ID, groupID);
        groupNode->setData(Qt::UserRole + NodeDataRole::Type, group.isPublic);
        modelNode->appendChild(groupNode);
    }

    endInsertRows();
}

void DBStructureModel::removeDataGroup(DBStructureNode* groupNode)
{
    int groupID = groupNode->data(Qt::UserRole + NodeDataRole::ID).toInt();
    //先从数据库中删除
    DBStructureNode* modelNode = groupNode->parentNode();
    if(!(modelNode && modelNode->type() == TableNode))
    {
        if(m_pMainWindow)
        {
            QString error = QString::fromWCharArray(L"删除失败，未找到所在链接节点信息");
            m_pMainWindow->showMessageDialog(type_information, QString::fromWCharArray(L"失败"), error);
        }
        return;
    }
    int modelID = modelNode->data(Qt::UserRole + NodeDataRole::ID).toInt();
    DBStructureNode* connNode = modelNode->parentNode();
    if(!(connNode && connNode->type() == ConnectionNode))
    {
        if(m_pMainWindow)
        {
            QString error = QString::fromWCharArray(L"删除失败，未找到所在链接节点信息");
            m_pMainWindow->showMessageDialog(type_information, QString::fromWCharArray(L"失败"), error);
        }
        return;
    }
    bool result = SqlQueryExecutor::instance().removeAttributeGroup(connNode->name(), modelID, groupID);
    if(!result)
    {
        if(m_pMainWindow)
        {
            QString error = QString::fromWCharArray(L"删除失败，详情可见日志文件");
            m_pMainWindow->showMessageDialog(type_information, QString::fromWCharArray(L"失败"), error);
        }
        return;
    }
    else //从列表中删除
    {
        QModelIndex connIndex = index(connNode->row(), 0, QModelIndex());
        beginRemoveRows(index(modelNode->row(), 0, connIndex), groupNode->row(), groupNode->row());
        modelNode->removeChild(groupNode);
        endRemoveRows();
    }
}

void DBStructureModel::refreshStructure_Connection(const QString& connection)
{
    DBStructureNode* connNode = getConnectionNode(connection);
    if(!connNode)
        return;

    //先删除当前链接节点下的节点
    if(connNode->childCount() > 0)
    {
        beginRemoveRows(index(connNode->row(), 0, QModelIndex()), 0, connNode->childCount() - 1);
        connNode->removeAllChildren();
        endRemoveRows(); //该语句之后会触发rowsRemoved(const QModelIndex &parent, int first, int last)信号，通知视图刷新对应行
    }

    QSqlDatabase db = QSqlDatabase::database(connection);
    if(!db.isOpen())
    {
        if(connNode->status() == Connect)
        {
            connNode->setStatus(Disconnect);
            QModelIndex topLeft = index(connNode->row(), 0, QModelIndex());
            QModelIndex bottomRight = index(connNode->row(), 0, QModelIndex());
            emit dataChanged(topLeft, bottomRight, {Qt::DecorationRole});
        }
        return;
    }

    //重新加载数据节点
    /*QString strSql = "SELECT * FROM basic.attribute_group ORDER BY id ASC";
    QSqlQuery query(db);
    if(!query.exec(strSql))
        return;

    QVector<DBStructureNode*> tables;
    while(query.next())
    {
        QString tableName = query.value(2).toString();
        tables << new DBStructureNode(TableNode, tableName, connNode);
    }*/

    const QVector<Model> models = SqlQueryExecutor::instance().getModels(connection);
    beginInsertRows(index(connNode->row(), 0, QModelIndex()), 0, models.count() - 1);

    for(const Model& model : models)
    {
        DBStructureNode* modelNode = new DBStructureNode(TableNode, model.name, connNode);
        modelNode->setData(Qt::UserRole + NodeDataRole::ID, model.id);
        for(int groupID : model.groups)
        {
            //QString groupName = SqlQueryExecutor::instance().getAttributeGroupName(connection, groupID);
            AttributeGroup group = SqlQueryExecutor::instance().getAttributeGroupData(connection, groupID);
            if(group.name.isEmpty())
                continue;
            DBStructureNode* groupNode = new DBStructureNode(GroupNode, group.name, modelNode);
            groupNode->setData(Qt::UserRole + NodeDataRole::ID, groupID);
            groupNode->setData(Qt::UserRole + NodeDataRole::Type, group.isPublic);
            modelNode->appendChild(groupNode);
        }
        connNode->appendChild(modelNode);
    }

    endInsertRows();

    if(connNode->status() == Disconnect)
    {
        connNode->setStatus(Connect);
        QModelIndex topLeft = index(connNode->row(), 0, QModelIndex());
        QModelIndex bottomRight = index(connNode->row(), 0, QModelIndex());
        emit dataChanged(topLeft, bottomRight, {Qt::DecorationRole});
    }

}