#include "FaceDataImgDao.h"

FaceDataImgDao::FaceDataImgDao(QObject *parent) : BaseDao(parent)
{

}

QVector<QVariantMap> FaceDataImgDao::findAllRecord()
{
    // 新建查询
    QSqlQuery query(ConnectionManager::getInstance()->getConnection());

    // 查询语句
    QString sql = "SELECT * FROM FACE_DATA_IMAGE";

    // 执行查询
    query.exec(sql);

    // 返回结果
    QVector<QVariantMap> result;

    // 遍历查询结果
    while (query.next()) {
        QVariantMap item;

        item.insert("id", query.value("id").toString());
        item.insert("person_id", query.value("person_id").toString());
        item.insert("face_image", query.value("face_image").toString());
        result.append(item);
    }

//    LOG(DEBUG) << QString("查询FACE_DATA_IMAGE表的所有记录[%1]").arg(result.size()).toStdString();

    return result;
}

QVariantMap FaceDataImgDao::findRecordById(QString id)
{
    // 新建查询
    QSqlQuery query(ConnectionManager::getInstance()->getConnection());

    // 查询语句
    QString sql = QString("SELECT * FROM FACE_DATA_IMAGE WHERE ID = '%1'").arg(id);

    // 执行查询
    query.exec(sql);

    // 返回结果
    QVariantMap result;

    // 获取结果
    if (query.next())
    {
        result.insert("id", query.value("id").toString());
        result.insert("person_id", query.value("person_id").toString());
        result.insert("face_image", query.value("face_image").toString());
    }

//    LOG(DEBUG) << QString("根据id查询FACE_DATA_IMAGE表的记录[%1][%2]").arg(result.size()).arg(sql).toStdString();
    return result;
}


QVariantMap FaceDataImgDao::findRecordByPersonId(QString personId)
{
    // 新建查询
    QSqlQuery query(ConnectionManager::getInstance()->getConnection());

    // 查询语句
    QString sql = QString("SELECT * FROM FACE_DATA_IMAGE WHERE PERSON_ID = '%1'").arg(personId);

    // 执行查询
    query.exec(sql);

    // 返回结果
    QVariantMap result;

    if (query.next())
    {
        result.insert("id", query.value("id").toString());
        result.insert("person_id", query.value("person_id").toString());
        result.insert("face_image", query.value("face_image").toString());
    }

//    LOG(TRACE) << QString("根据id查询FACE_DATA_IMAGE表的记录[personId=%1][%2]").arg(personId).arg(sql).toStdString();
    return result;
}


QVector<QVariantMap> FaceDataImgDao::findRecordsByProperty(QString properName, QVariant properValue)
{
    // 新建查询
    QSqlQuery query(ConnectionManager::getInstance()->getConnection());

    // 查询语句
    QString sql = "SELECT * FROM FACE_DATA_IMAGE WHERE %1 = '%2'";
    sql = sql.arg(properName).arg(properValue.toString());

    // 执行查询
    query.exec(sql);

    // 获取结果集的大小
    int count = 0;

    // 返回结果
    QVector<QVariantMap> result;

    // 遍历查询结果
    while (query.next()) {
        QVariantMap item;

        count++;

        item.insert("id", query.value("id").toString());
        item.insert("person_id", query.value("person_id").toString());
        item.insert("face_image", query.value("face_image").toString());
        result.append(item);
    }

//    LOG(DEBUG) << QString("根据属性值查询FACE_DATA_IMAGE表的记录[%1][%2]").arg(count).arg(sql).toStdString();
    return result;
}

QString FaceDataImgDao::save(QVariantMap object)
{
    // 新建查询
    QSqlQuery query(ConnectionManager::getInstance()->getConnection());

    qulonglong id = ConnectionManager::getInstance()->generateId();

    // INSERT语句
    QString sql = QString("INSERT INTO FACE_DATA_IMAGE "
                          "(ID, PERSON_ID, FACE_IMAGE) "
                          "VALUES ('%1', '%2', :face)")
            .arg(id).arg(object.value("person_id").toString());

    query.prepare(sql);
    query.bindValue(":face", object.value("face_image"));

    // 开启事务
    ConnectionManager::getInstance()->getConnection().transaction();

    // 执行插入
    bool success = query.exec();

    // 结束事务
    ConnectionManager::getInstance()->getConnection().commit();

//    LOG(DEBUG) << QString("保存人脸图片[%1][%2]").arg(success).arg(sql).toStdString();

    // 返回结果
    if (success == true)
    {
        return QString("%1").arg(id);
    }
    else
    {
        return "-1";
    }
}

bool FaceDataImgDao::edit(QVariantMap newObject, QString id)
{
    // 新建查询
    QSqlQuery query(ConnectionManager::getInstance()->getConnection());

    // UPDATE语句
    if (!newObject.contains("face_image")){
        return false;
    }
    QString sql = QString("UPDATE FACE_DATA_IMAGE SET FACE_IMAGE = :faceImage WHERE ID = '%1'").arg(id);

    query.prepare(sql);
    query.bindValue(":faceImage", newObject.value("face_image"));

    // 开启事务
    ConnectionManager::getInstance()->getConnection().transaction();

    // 执行更新
    bool success = query.exec();

//    LOG(DEBUG) << QString("编辑人脸图片[%1][%2]").arg(success).arg(sql).toStdString();

    // 结束事务
    ConnectionManager::getInstance()->getConnection().commit();

    // 返回结果
    return success;
}


bool FaceDataImgDao::dele(QString id)
{
    // 新建查询
    QSqlQuery query(ConnectionManager::getInstance()->getConnection());

    // DELETE
    QString sql = QString("DELETE FROM FACE_DATA_IMAGE WHERE ID = '%1'").arg(id);

    // 开启事务
    ConnectionManager::getInstance()->getConnection().transaction();

    // 执行更新
    bool success = query.exec(sql);

//    LOG(DEBUG) << QString("删除人脸图片[%1][%2]").arg(success).arg(sql).toStdString();

    // 结束事务
    ConnectionManager::getInstance()->getConnection().commit();

    // 返回结果
    return success;
}