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

}