QT 通用数据库数据导入导出方案

QT提供了通用的数据库接口可以很方便的处理各种数据库,在项目中经常会遇到数据库数据的导入导出的需求. 由于各个数据库的DDL差异比较大,无法做到导出表结构的通用方案,这个只能借助相应数据库特有的SQL语句或者工具来解决, 如postgresql的COPY命令和pg_dump工具等.
Navicat工具的导入导出功能做的非常不错,能够导出包含表结构和二进制数据的sql文件.
需解决问题:
1. 取出数据,组装insert语句
2. 对于二进制数据,无法做到像navicat那样可以直接执行sql语句执行. 只能转化为十六进制文本格式. 在导入时自动解析

3. 对于数据库建表语句, 目前无法提供通用的导出方式,只能借助工具

一. 数据导出

/**
 *@brief 导出数据库数据到文件中
 *@param path 文件路径
 */
void ExportDB(const QString &path)
{
QStringList vList;


// 导出数据库所有名称以auth开头的表
QStringList tables=gAuthDB.tables();
foreach(QString table,tables)
{
if(!table.startsWith("auth"))
continue;


QSqlQuery query(gAuthDB);
QString sql=QString("select * from %1").arg(table);
query.exec(sql);


QSqlRecord record=query.record();
while(query.next())
{
QString prefix=QString("insert into %1(").arg(table); // 记录属性字段名
QString suffix="values("; // 记录属性值


// 遍历属性字段
for(int i=0;i<record.count();i++)
{
QSqlField field=record.field(i);
QString fieldName=field.name();


switch(field.type())
{
case QVariant::String:
prefix+=fieldName;
suffix+=QString("'%1'").arg(query.value(i).toString());
break;
case QVariant::ByteArray:
{
prefix+=fieldName;
QByteArray data=query.value(i).toByteArray();
if(data.isNull())
{
suffix+="null";
}else
{
suffix+=QString("E'%1'").arg(data.toHex().data()); // blob数据按16进制格式导出
}
}
break;
default:
prefix+=fieldName;
suffix+=query.value(i).toString();
}


if(record.count()==1)
{
prefix+=")";
suffix+=")";
}else if(i!=record.count()-1)
{
prefix+=",";
suffix+=",";
}else if(i==record.count()-1)
{
prefix+=")";
suffix+=")";
}
}


// 组装sql语句 insert into auth_test values(0,'hello',E'003f')
QString iSql=QString("%1 %2;").arg(prefix).arg(suffix);
vList.append(iSql);
}
}


QFile file(path);
file.open(QIODevice::WriteOnly|QIODevice::Truncate);


// 将sql语句写入文件
QTextStream out(&file);
foreach(QString line,vList)
{
out<<line+"\n";
}
}

二. 数据导入

/**
 *@brief 读取sql文本内容,并写入至数据库
 *@param path sql文件路径
 */
void ImportDB(const QString &path)
{
QSqlQuery query(gAuthDB);
QFile file(path);
file.open(QFile::ReadOnly);


QTextStream in(&file);
while(!in.atEnd())
{
QString sql=in.readLine();


// 通过分析values(E'),判断是否有二进制数据,如没有直接运行sql语句,如有则需要将16进制文本转换为blob数据
QRegExp reg("E'([0-9a-f]{1,})'");
if(!sql.contains(reg))
{
query.exec(sql);
}else
{
int pos=0;
QStringList bList;


// 探索所有的blob字段
while((pos=reg.indexIn(sql,pos))!=-1)
{
bList.append(reg.cap(0));


QString blob=reg.cap(1);
pos+=reg.matchedLength();
}


// blob字段填充占位符
foreach(QString key,bList)
{
sql.replace(key,"?");
}


query.prepare(sql);


// 绑定占位符数据
for(int i=0;i<bList.size();i++)
{
// 去除E''
QString hexBlob=bList[i].mid(2,bList[i].size()-1);
// 还原16进制数据
QByteArray ba=QByteArray::fromHex(hexBlob.toLocal8Bit());


query.bindValue(i,ba);
}


query.exec();
}
}
}