c语言从文件中读入格式化数据并存入sqlite3
2015-06-12 22:59
926 查看
这周做数据库作业,其中一个小部分是从文件中读入格式化数据并存入sqlite3中。一开始只是为了完成实验内容而写了一个专门的函数来完成。 但写完之后发现有时候我们确实是需要从文件中读入格式化数据的。所以想写一个通用的函数,通过给定的参数来从文件中读入数据。 先是各个函数的解读,然后是所有的代码。
//************************************ // Method: insertData 从文件中读取记录并插入数据库 // FullName: insertData // Access: public // Returns: int // Qualifier: // Parameter: sqlite3 * db sqlite3实例指针 // Parameter: char * fileName 要读取内容的文件名 // Parameter: int maxSize 每行记录的最大长度 // Parameter: char * tableName 要插入数据的表名 // Parameter: int length 字段数 // Parameter: int * format 每个字段的格式,数字:1,字符(串):0 //************************************ int insertData(sqlite3 *db, char *fileName, int maxSize, char *tableName, int length, int *format);
这个函数就是封装的读取插入数据的最终形态。参数简要描述如注释里面说的那样。前4个参数应该都好理解,可能需要解释的就是最后两个参数了。 int length 指的是你要插入数据的表(tableName)里一共有多少字段。 int *format 这其实是一个整型数组。你在定义的时候可以使用format 的形式,实参就是format。这个数组代表的是数据库表中每一个字段是数字还是字符串。了解数据库的同学应该都知道,简单的DBMS的sql语句在插入数值时其实只有三种:数字,字符串,NULL。而数字和NULL的插入方式是一样的。 如下是插入数字的sql语句:
insert into tableName (number)values(12);
这句话想tableName中插入了一组记录,插入的类型是数字。 如下是插入字符串的sql语句:
insert into talbeName (string) values ('hello world');
应该看出区别的了,两者的唯一区别就是字符串多了一对单(双)引号。因为在c语言中,字符串只能用双引号("")来括起来,所以我们在c语言里的sql语句只能使用单引号('')了。 那么上面的format就好理解了。按照顺序,如果第i个字段(从0计数)为字符串形式,则format[i] = 0,如果第i个字段为数字形势,则format[i] = 1; 当然,这个函数时假设插入的数据中是按顺序包含每一个字段。 如果需要只插入某些,可以自行修改。
//************************************ // Method: myGetSQL 通过文件内容获取的sql语句 // FullName: myGetSQL // Access: public // Returns: int // Qualifier: insertData() // Parameter: FILE * file 导入数据的文件名 // Parameter: int maxSize 文件里每行记录的最大长度 // Parameter: char * tableName 需要插入数据库的表名 // Parameter: char ** SQL sql执行语句 // Parameter: int fields 字段数 // Parameter: int * format 每个字段的格式,数字:1,字符(串):0 //************************************ int myGetSQL(FILE *file, int maxSize, char *tableName, char **SQL, int fields, int *format) ;
这个函数的作用就是从文件中获取格式化的内容并返回sql语句。很容易理解。 这个函数的唯一调用者就是insertData()函数了。
//************************************ // Method: join 将字符串b接在字符串a后面 // FullName: join // Access: public // Returns: void // Qualifier: // Parameter: char * * a 原始字符串 // Parameter: char * * b 需要添加的字符串 //************************************ void join(char **a, char **b);
这个函数的作用就是把一个字符串添加到另一个字符串的后面,相当于c语言中的字符串的"+"操作符了。 接下来就是全部的代码了。
#include "stdio.h" #include "io.h" #include "stdlib.h" #include "string.h" #include "sqlite3.h" #include "time.h" #define STR 0 #define NUM 1 void join(char **, char**); int myGetSQL(FILE *, int , char *, char **, int, int *); int insertData(sqlite3 *, char *, int, char *, int, int *); int main(int argc,char *argv[]) { clock_t t1 = clock(); sqlite3 *db; int rc; char *ErrMsg = 0; const char *dbName = "DBforCourse.db3"; if(!access(dbName,0)){ if(0!=remove("DBforCourse.db3")){ printf("删除文件失败!\n"); } } rc = sqlite3_open("DBforCourse.db3", &db); if (rc){ fprintf(stderr, "Can't open database: %s\n", sqlite3_errmsg(db)); exit(0); } //创建数据库表 char *College = "create table College(cName VARCHAR(20),state VARCHAR(20),enrollment int)"; char *Student = "create table Student(sID int,sName VARCHAR(20),GPA real,sizeHS int)"; char *Apply = "create table Apply(sID int,cName VARCHAR(20),major VARCHAR(20),decision VARCHAR(10))"; sqlite3_exec(db,College,0,0,0); sqlite3_exec(db,Student,0,0,0); sqlite3_exec(db,Apply,0,0,0); //插入数据 int college[3] = { STR, STR, NUM }; int apply[4] = {NUM,STR,STR,STR}; int student[4] = {NUM,STR,NUM,NUM}; if(!insertData(db,"dbcollege.txt",200,"College",3,college)) printf("插入College表失败!\n"); if(!insertData(db,"dbapply.txt",200,"Apply",4,apply)) printf("插入Apply表失败!\n"); if(!insertData(db,"dbstudent.txt",200,"Student",4,student)) printf("插入Student表失败!\n"); sqlite3_close(db); clock_t t2 = clock(); printf("耗时%ld毫秒\n",t2-t1); } //************************************ // Method: myGetSQL 通过文件内容获取的sql语句 // FullName: myGetSQL // Access: public // Returns: int // Qualifier: // Parameter: FILE * file 导入数据的文件名 // Parameter: int maxSize 文件里每行记录的最大长度 // Parameter: char * tableName 需要插入数据库的表名 // Parameter: char ** SQL sql执行语句 // Parameter: int fields 字段数 // Parameter: int * format 每个字段的格式,数字:1,字符(串):0 //************************************ int myGetSQL(FILE *file, int maxSize, char *tableName, char **SQL, int fields, int *format) { int i, s = 0, k = 0; char *query[fields]; char input[maxSize]; char *enter; char *a0 = "insert into "; char *a1 = " values("; char *a2 = ","; char *a3 = "'"; char *a4 = "',"; char *a5 = ")"; *SQL = a0; join(SQL, &tableName); join(SQL, &a1); if (fgets(input, maxSize, file)){ i = 0; //删除末尾的\n符 if (enter = strchr(input, '\n')) *enter = '\0'; query[i] = strtok(input, ","); while (query[++i] = strtok(0, ",")){} for (s = 0; s < fields - 1; s++){ if (*(format + s)){//数字 join(SQL, &query[s]); join(SQL, &a2); } else{//字符串 join(SQL, &a3); join(SQL, &query[s]); join(SQL, &a4); } } if (*(format + s)){//数字 join(SQL, &query[s]); } else{//字符串 join(SQL, &a3); join(SQL, &query[s]); join(SQL, &a3); } join(SQL, &a5); k = 1; } else k = 0; return k; } //************************************ // Method: join 将字符串b接在字符串a后面 // FullName: join // Access: public // Returns: void // Qualifier: // Parameter: char * * a 原始字符串 // Parameter: char * * b 需要添加的字符串 //************************************ void join(char **a, char **b) { size_t size = strlen(*a) + strlen(*b) + 1; char *result = (char *)malloc(size); strcpy(result, *a); strcat(result, *b); free(*a); *a = result; } //************************************ // Method: insertData 从文件中读取记录并插入数据库 // FullName: insertData // Access: public // Returns: int // Qualifier: // Parameter: sqlite3 * db sqlite3实例指针 // Parameter: char * fileName 要读取内容的文件名 // Parameter: int maxSize 每行记录的最大长度 // Parameter: char * tableName 要插入数据的表名 // Parameter: int length 字段数 // Parameter: int * format 每个字段的格式,数字:1,字符(串):0 //************************************ int insertData(sqlite3 *db, char *fileName, int maxSize, char *tableName, int length, int *format){ FILE *file; if (!(file = fopen(fileName, "r"))){ printf("打开文件%s失败!\n", fileName); return 0; } char *SQL = 0; sqlite3_exec(db, "begin;", 0, 0, 0); while (myGetSQL(file, maxSize, tableName, &SQL, length, format)){ if (SQLITE_OK != sqlite3_exec(db, SQL, 0, 0, 0)) printf("%s 语句执行失败!\n", SQL); } fclose(file); sqlite3_exec(db, "commit;", 0, 0, 0); return 1; }
相关文章推荐
- C++常成员函数 - const 关键字
- C、C++一次将整个文件读入内存
- c++ 用类模版实现链表(c++语言程序设计第四版示例代码)
- dilworth解决的题目poj3636,poj1631
- 嵌入式C常见笔试题:字符串
- 再读effective c++
- C语言中动态的申请二(三)及多维数组的使用方法及释放方法
- C语言字符指针和字符数组的区别
- VC++动态链接库编程
- C++ 纯虚方法
- 转 值得推荐的C/C++框架和库
- 单链表的逆置-C++实现
- chapter12test4
- C++ Primer学习2:细节:引用
- leetcode系列题目——Set Matrix Zeroes
- C++二叉树翻转
- c++中的4种类型转换
- zerglurker的C语言教程008——循环语句详解
- C++中的explicit
- 【学习笔记】【C语言】循环结构-do while