C/C++ Qt 資料庫與SqlTableModel元件應用

lyshark發表於2021-12-09

SqlTableModel 元件可以將資料庫中的特定欄位動態顯示在TableView表格元件中,通常設定QSqlTableModel類的變數作為資料模型後就可以顯示資料表內容,介面元件中則通過QDataWidgetMapper類例項設定為與某個資料庫欄位相關聯,則可以實現自動顯示欄位的內容,不僅是顯示,其還支援動態增刪改查等各種複雜操作,期間不需要使用任何SQL語句。

首先繪製好UI介面,本次案例介面稍顯複雜,左側是一個TableView元件,其他地方均為LineEdit元件與Button元件。

先來生成資料庫表記錄,此處我們只需要增加一個Student學生表,並插入兩條測試資料即可,執行以下程式碼完成資料建立。

#include <iostream>
#include <QSqlDatabase>
#include <QSqlError>
#include <QSqlQuery>
#include <QSqlRecord>
#include <QtSql>
#include <QDataWidgetMapper>

QSqlDatabase DB;                     // 資料庫連線
void MainWindow::InitSQL()
{
    QSqlDatabase db = QSqlDatabase::addDatabase("QSQLITE");
    db.setDatabaseName("./lyshark.db");
     if (!db.open())
     {
            return;
     }

    // 執行SQL建立表
    // https://Www.cnbloGs.com/lyshark
    db.exec("DROP TABLE Student");
    db.exec("CREATE TABLE Student ("
                    "id INTEGER PRIMARY KEY AUTOINCREMENT, "
                    "name VARCHAR(40) NOT NULL, "
                    "sex VARCHAR(40) NOT NULL, "
                    "age INTEGER NOT NULL,"
                    "mobile VARCHAR(40) NOT NULL,"
                    "city VARCHAR(40) NOT NULL)"
         );

    // 逐條插入資料
    db.exec("INSERT INTO Student(name,sex,age,mobile,city) ""VALUES ('lyshark.cnblogs.com','m','25','1234567890','beijing')");
    db.exec("INSERT INTO Student(name,sex,age,mobile,city) ""VALUES ('www.lyshark.com','x','22','4567890987','shanghai')");

    db.commit();
    db.close();
}

資料庫建立後表內記錄如下:

程式執行後我們將在MainWindow::MainWindow(QWidget *parent)建構函式內完成資料庫表記錄與TableView元件欄位的對應關係繫結,將資料庫繫結到QDataWidgetMapper物件上,繫結程式碼如下。

MainWindow::MainWindow(QWidget *parent) :QMainWindow(parent),ui(new Ui::MainWindow)
{
    ui->setupUi(this);

    // 開啟資料庫
    DB=QSqlDatabase::addDatabase("QSQLITE"); // 新增 SQL LITE資料庫驅動
    DB.setDatabaseName("./lyshark.db");     // 設定資料庫名稱
    if (!DB.open())
    {
        return;
    }

    // 開啟資料表
    tabModel=new QSqlTableModel(this,DB);                             // 資料表
    tabModel->setTable("Student");                                    // 設定資料表
    tabModel->setEditStrategy(QSqlTableModel::OnManualSubmit);        // 資料儲存方式,OnManualSubmit , OnRowChange
    tabModel->setSort(tabModel->fieldIndex("id"),Qt::AscendingOrder); // 排序
    if (!(tabModel->select()))                                        // 查詢資料
    {
       return;
    }

    // 設定欄位名稱
    tabModel->setHeaderData(tabModel->fieldIndex("id"),Qt::Horizontal,"Uid");
    tabModel->setHeaderData(tabModel->fieldIndex("name"),Qt::Horizontal,"Uname");
    tabModel->setHeaderData(tabModel->fieldIndex("sex"),Qt::Horizontal,"Usex");
    tabModel->setHeaderData(tabModel->fieldIndex("age"),Qt::Horizontal,"Uage");
    tabModel->setHeaderData(tabModel->fieldIndex("mobile"),Qt::Horizontal,"Umobile");
    tabModel->setHeaderData(tabModel->fieldIndex("city"),Qt::Horizontal,"Ucity");

    theSelection=new QItemSelectionModel(tabModel);                       // 關聯選擇模型
    ui->tableView->setModel(tabModel);                                    // 設定資料模型
    ui->tableView->setSelectionModel(theSelection);                       // 設定選擇模型
    ui->tableView->setSelectionBehavior(QAbstractItemView::SelectRows);   // 行選擇模式

    // 新增資料對映,將選中欄位對映到指定編輯框中
    // https://www.cnblogs.com/lysharK
    dataMapper= new QDataWidgetMapper();
    dataMapper->setModel(tabModel);
    dataMapper->setSubmitPolicy(QDataWidgetMapper::AutoSubmit);

    dataMapper->addMapping(ui->lineEdit_name,tabModel->fieldIndex("name"));          // 設定對映欄位
    dataMapper->addMapping(ui->lineEdit_mobile,tabModel->fieldIndex("mobile"));      // 第二個對映欄位
    dataMapper->toFirst();                                                           // 預設選中首條對映記錄

    // 繫結訊號,當滑鼠選擇時,在底部編輯框中輸出
    connect(theSelection,SIGNAL(currentRowChanged(QModelIndex,QModelIndex)),this,SLOT(on_currentRowChanged(QModelIndex,QModelIndex)));
    getFieldNames();
}

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

繫結成功後執行程式即可看到如下效果,資料庫中的記錄被對映到了元件內.

當使用者點選TableView元件內的某一行記錄時,則觸發MainWindow::on_currentRowChanged函式。

  • 執行獲取name/mobile欄位,並放入對映資料集中的 lineEdit編輯框中
void MainWindow::on_currentRowChanged(const QModelIndex &current, const QModelIndex &previous)
{
    Q_UNUSED(previous);

    dataMapper->setCurrentIndex(current.row());      // 更細資料對映的行號
    int curRecNo=current.row();                      // 獲取行號
    QSqlRecord  curRec=tabModel->record(curRecNo);   // 獲取當前記錄

    QString uname = curRec.value("name").toString();     // 取出資料
    QString mobile = curRec.value("mobile").toString();

    ui->lineEdit_name->setText(uname);                   // 設定到編輯框
    ui->lineEdit_mobile->setText(mobile);
}

執行效果如下:

增加插入與刪除記錄實現方法都是呼叫TabModel提供的預設函式,通過獲取當前選中行號,並對該行號執行增刪改查方法即可。

// 新增一條記錄
// https://www.cnblogS.com/lyshark
void MainWindow::on_pushButton_add_clicked()
{
    tabModel->insertRow(tabModel->rowCount(),QModelIndex());                 // 在末尾新增一個記錄
    QModelIndex curIndex=tabModel->index(tabModel->rowCount()-1,1);          // 建立最後一行的ModelIndex
    theSelection->clearSelection();                                          // 清空選擇項
    theSelection->setCurrentIndex(curIndex,QItemSelectionModel::Select);     // 設定剛插入的行為當前選擇行

    int currow=curIndex.row();                                              // 獲得當前行
    tabModel->setData(tabModel->index(currow,0),1000+tabModel->rowCount()); // 自動生成編號
    tabModel->setData(tabModel->index(currow,2),"M");                       // 預設為男
    tabModel->setData(tabModel->index(currow,3),"0");                       // 預設年齡0
}

// 插入一條記錄
void MainWindow::on_pushButton_insert_clicked()
{
    QModelIndex curIndex=ui->tableView->currentIndex();
    int currow=curIndex.row();                                              // 獲得當前行

    tabModel->insertRow(curIndex.row(),QModelIndex());
    tabModel->setData(tabModel->index(currow,0),1000+tabModel->rowCount()); // 自動生成編號

    theSelection->clearSelection();                                         // 清除已有選擇
    theSelection->setCurrentIndex(curIndex,QItemSelectionModel::Select);
}

// 刪除一條記錄
void MainWindow::on_pushButton_delete_clicked()
{
    QModelIndex curIndex=theSelection->currentIndex();  // 獲取當前選擇單元格的模型索引
    tabModel->removeRow(curIndex.row());                // 刪除最後一行
}

// 儲存修改資料
void MainWindow::on_pushButton_save_clicked()
{
    bool res=tabModel->submitAll();
    if (!res)
    {
        std::cout << "save as ok" << std::endl;
    }
}

// 恢復原始狀態
void MainWindow::on_pushButton_reset_clicked()
{
    tabModel->revertAll();
}

增刪改查實現如下:

針對與排序與過濾的實現方式如下,同樣是呼叫了標準函式。

// 以Combox中的欄位對目標 升序排列
void MainWindow::on_pushButton_ascending_clicked()
{
    tabModel->setSort(ui->comboBox->currentIndex(),Qt::AscendingOrder);
    tabModel->select();
}

// 以Combox中的欄位對目標 降序排列
// https://www.Cnblogs.com/LyShark
void MainWindow::on_pushButton_descending_clicked()
{
    tabModel->setSort(ui->comboBox->currentIndex(),Qt::DescendingOrder);
    tabModel->select();
}

// 過濾出所有男記錄
void MainWindow::on_pushButton_filter_man_clicked()
{
    tabModel->setFilter(" sex = 'M' ");
}
// 恢復預設過濾器
void MainWindow::on_pushButton_default_clicked()
{
    tabModel->setFilter("");
}

過濾效果如下所示:

批量修改某個欄位,其實現原理是首先通過i<tabModel->rowCount()獲取記錄總行數,然後通過aRec.setValue設定指定欄位數值,並最終tabModel->submitAll()提交到表格中。

void MainWindow::on_pushButton_clicked()
{
    if (tabModel->rowCount()==0)
        return;

    for (int i=0;i<tabModel->rowCount();i++)
    {
        QSqlRecord aRec=tabModel->record(i);            // 獲取當前記錄
        aRec.setValue("age",ui->lineEdit->text());      // 設定資料
        tabModel->setRecord(i,aRec);
    }
    tabModel->submitAll();                              // 提交修改
}

迴圈修改實現效果如下:

上方程式碼中,如果需要修改或增加特定行或記錄我們只需要點選相應的按鈕,並在選中行直接編輯即可實現向資料庫中插入資料,而有時我們不希望通過在原表上操作,而是通過新建窗體並在窗體中完成增刪改,此時就需要使用Dialog窗體並配合原生SQL語句來實現對記錄的操作了。

以增加為例,主窗體中直接彈出增加選項卡,並填寫相關引數,直接提交即可。

// https://www.cnblogs.com/LyShark
void MainWindow::on_pushButton_insert_clicked()
{
    QSqlQuery query;
    query.exec("select * from Student where id =-1");    // 查詢欄位資訊,是否存在
    QSqlRecord curRec=query.record();                    // 獲取當前記錄,實際為空記錄
    curRec.setValue("id",qryModel->rowCount()+1001);

    WindowPtr->setWindowFlags(flags | Qt::MSWindowsFixedSizeDialogHint); // 設定對話方塊固定大小
    WindowPtr->setInsertRecord(curRec);                                  // 插入記錄
    int ret=WindowPtr->exec();                                           // 以模態方式顯示對話方塊
    if (ret==QDialog::Accepted)                                          // OK鍵被按下
    {
        query.prepare("INSERT INTO Student(id,name,sex,age,mobile,city)"
                      " VALUES(:Id, :Name, :Sex, :Age, :Mobile, :City)");

        query.bindValue(":Id",recData.value("id"));
        query.bindValue(":Name",recData.value("name"));
        query.bindValue(":Sex",recData.value("sex"));
        query.bindValue(":Age",recData.value("age"));
        query.bindValue(":Mobile",recData.value("mobile"));
        query.bindValue(":City",recData.value("city"));

        if (query.exec())
        {
            QString sqlStr=qryModel->query().executedQuery(); // 執行過的SELECT語句
            qryModel->setQuery(sqlStr);                       // 重新查詢資料
        }
    }
    delete WindowPtr;
}

Dialog增加效果如下:

相關文章