代码功能:
从sql.txt中读出SQL语句,并一一执行,若有一条执行失败,则所有执行全部不生效。回滚到执行前的状态。
#include <stdio.h>
#include <sqlite3.h>
#define SQL_FILE "./sql.txt"
int commitSQL(sqlite3 *db)
{
int ret=0;
FILE *fp;
char buf[100]={0};
char *errmsg = 0;
char **azResult;
int i, nRow=0, nColumn=0;
fp = fopen(SQL_FILE, "r");
if(NULL == fp)
return -1;
ret = sqlite3_exec(db,"BEGIN EXCLUSIVE", NULL,NULL, &errmsg);
if(ret != SQLITE_OK)
{
printf("ret = %d, BEGIN EXCLUSIVE: %s", ret, errmsg);
}
while( NULL != fgets(buf,100, fp))
{
printf("%s\n", buf);
if(0 == strncmp(buf,"select", 6))
{
ret += sqlite3_get_table(db, buf, &azResult, &nRow, &nColumn,&errmsg);
if(ret != SQLITE_OK)
{
printf("ret = %d, search table fail: %s\n", ret, errmsg);
}
printf("row:%d column=%d\n",nRow, nColumn);
for(i=nColumn;i<(nRow+1)*nColumn;i++)
{
printf("azResult[%d] = %s\n", i, azResult[i]);
}
sqlite3_free_table(azResult);
}
else
ret += sqlite3_exec(db, buf, 0, 0, &errmsg);
if(SQLITE_OK != ret)
{
printf("ret = %d, sql: %s\n", ret, errmsg);
break;
}
}
if(ret == SQLITE_OK)
{
ret=sqlite3_exec(db,"COMMIT", NULL,NULL, &errmsg);
if(ret != SQLITE_OK)
{
printf("ret = %d, COMMIT: %s\n", ret, errmsg);
}
}
else
{
ret=sqlite3_exec(db,"ROLLBACK", NULL,NULL, &errmsg);
if(ret != SQLITE_OK)
{
printf("ret = %d, ROLLBACK: %s\n", ret, errmsg);
}
sqlite3_free(errmsg);
fclose(fp);
return -2;
}
sqlite3_free(errmsg);
fclose(fp);
return ret;
}
int main()
{
sqlite3 *db;
char *dbPath="test.db";
int ret = 0;
ret = sqlite3_open(dbPath, &db);
if(ret == SQLITE_OK)
{
printf("open succeed\n");
}
else
{
printf("open fail\n");
}
ret = commitSQL(db);
printf("ret = %d\n", ret);
sqlite3_close(db);
}