微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

c语言从文件中读入格式化数据并存入sqlite3

这周做数据库作业,其中一个小部分是从文件中读入格式化数据并存入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[n]的形式,实参就是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,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 insertData(sqlite3 *,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);
    sqlite3_exec(db,Student,Apply,0);

    //插入数据
    int college[3] = { STR,STR,NUM };
    int apply[4] = {NUM,STR};
    int student[4] = {NUM,NUM,NUM};

    if(!insertData(db,"dbcollege.txt",200,"College",3,college))
        printf("插入College表失败!\n");
    if(!insertData(db,"dbapply.txt","Apply",4,apply))
        printf("插入Apply表失败!\n");
    if(!insertData(db,"dbstudent.txt","Student",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 *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,&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,int *format){
    FILE *file;
    if (!(file = fopen(fileName,"r"))){
        printf("打开文件%s失败!\n",fileName);
        return 0;
    }
    char *sql = 0;
    sqlite3_exec(db,"begin;",0);
    while (myGetsql(file,tableName,&sql,length,format)){
        if (sqlITE_OK != sqlite3_exec(db,sql,0))
            printf("%s 语句执行失败!\n",sql);
    }
    fclose(file);
    sqlite3_exec(db,"commit;",0);
    return 1;
}

原文地址:https://www.jb51.cc/sqlite/199567.html

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。

相关推荐