数据库实现工人数据增删改查

#include <myhead.h>
#include <sqlite3.h>
typedef struct
{
    int gh;
    char name[20];
    double salary;
}Worker;

int do_insert(sqlite3 *ppDb)
{
    Worker work;
    printf("请输入插入的工号：");
    scanf("%d",&work.gh);
    printf("请输入插入的姓名：");
    scanf("%s",work.name);
    printf("请输入插入的薪水：");
    scanf("%lf",&work.salary);

    char *errmsg = NULL;//存储错误信息
    char sqlite[100]; //插入一个工人信息
    snprintf(sqlite,sizeof(sqlite),"insert into Work values(%d,\"%s\",%lf);",work.gh,work.name,work.salary);
    if(sqlite3_exec(ppDb,sqlite,NULL,NULL,&errmsg)!=0)//实现插入一个工人信息
    {
        perror("sqlite3_exec");
        printf("错误码：%d错误信息：%s,出错行是：%d\n",
                sqlite3_errcode(ppDb),sqlite3_errmsg(ppDb),__LINE__);
        printf("错误信息：%s\n",errmsg);
        sqlite3_free(errmsg);//释放内存
        errmsg = NULL;
        return -1;
    }
    printf("添加工人信息成功\n");
    return 0;
}
int callback(void *arg,int n,char **msgtext ,char **msgtable)//回调 输出查找到的工人信息
{
    int i,j;
    for(i = 0;i<n;i++)//n是列
    {
        printf("%s\t",*(msgtable+i));
    }
    printf("\n");
    for(j = 0;j<n;j++)
    {
        printf("%s\t",*(msgtext+j));
    }
    printf("\n");
    printf("查找成功\n");
    return 0;
}
int do_reseach(sqlite3 *ppDb)
{
    char key[20];
    printf("请输入你要查找的工人姓名：");
    scanf("%s",key);
    char *errmsg = NULL;//存储错误信息
    char sqlite[100]; //查找一个工人信息
    snprintf(sqlite,sizeof(sqlite),"select * from Work where name==\"%s\";",key);

    if(sqlite3_exec(ppDb,sqlite,callback,NULL,&errmsg)!=0)//查找一个工人信息
    {
        perror("sqlite3_exec");
        printf("错误码：%d错误信息：%s,出错行是：%d\n",
                sqlite3_errcode(ppDb),sqlite3_errmsg(ppDb),__LINE__);
        printf("错误信息：%s\n",errmsg);
        sqlite3_free(errmsg);//释放内存
        errmsg = NULL;
        return -1;
    }
    return 0;    
}
int show_all(void *arg,int n,char **msgtext,char **msgtable)//回调函数显示所有工人信息
{
    int i,j;
    if(*(int *)arg==0)
    {
        for(i = 0;i<n;i++)
        {
            printf("%s\t",*(msgtable+i));
        }
        *((int *)arg)=1;
    }
    printf("\n");

    for(i = 0;i<n;i++)
    {
        printf("%s\t",*(msgtext+i));
    }
    printf("\n");
    return 0;
}
int do_showall(sqlite3 *ppDb)
{
    char *errmsg = NULL;//存储错误信息
    char sqlite[100] = "select * from Work"; //输出所有工人信息
    int flag=0;
    if(sqlite3_exec(ppDb,sqlite,show_all,&flag,&errmsg)!=0)//输出所有工人信息
    {
        perror("sqlite3_exec");
        printf("错误码：%d错误信息：%s,出错行是：%d\n",
                sqlite3_errcode(ppDb),sqlite3_errmsg(ppDb),__LINE__);
        printf("错误信息：%s\n",errmsg);
        sqlite3_free(errmsg);//释放内存
        errmsg = NULL;
        return -1;
    }    
    return 0;
}
int do_delet(sqlite3 *ppDb)
{
    int gh;
    char *errmsg;
    char sqlite[100] = "delete  from Work";
    printf("请输入删除的工号：");
    scanf("%d",&gh);
    snprintf(sqlite,sizeof(sqlite),"delete  from Work where gh==%d;",gh);
    if(sqlite3_exec(ppDb,sqlite,NULL,NULL,&errmsg)!=0)
    {
        perror("sqlite3_exec");
        printf("错误码：%d错误信息：%s,出错行是：%d\n",
                sqlite3_errcode(ppDb),sqlite3_errmsg(ppDb),__LINE__);
        printf("错误信息：%s\n",errmsg);
        sqlite3_free(errmsg);//释放内存
        errmsg = NULL;
        return -1;
    }
    printf("删除工人信息成功\n");
    return 0;
}
int do_change(sqlite3 *ppDb)

{
    char name[20];
    char fond_name[20];
    printf("请输修改员工的名字：");
    scanf("%s",fond_name);
    printf("修改后的姓名：");
    scanf("%s",name);
    char *errmsg=NULL;
    char sqlite[100];
    snprintf(sqlite,sizeof(sqlite),"update Work set name=\"%s\"where name=\"%s\";",name,fond_name);
    if(sqlite3_exec(ppDb,sqlite,NULL,NULL,&errmsg)!=0)
    {
        perror("sqlite3_exec");
        printf("错误码：%d错误信息：%s,出错行是：%d\n",
                sqlite3_errcode(ppDb),sqlite3_errmsg(ppDb),__LINE__);
        printf("错误信息：%s\n",errmsg);
        sqlite3_free(errmsg);//释放内存
        errmsg = NULL;
        return -1;
    }
    printf("修改成功\n");
    return 0;
}

int do_allshow(sqlite3 *ppDb)
{
    char **errmsg=NULL;
    char sqlite[100] = "select * from Work"; 
    char **pazRuselt;//指向table的指针
    int rnpow;//行
    int rncolumn;//列
    if(sqlite3_get_table(ppDb,sqlite,&pazRuselt,&rnpow,&rncolumn,errmsg)!=SQLITE_OK)
    {
        perror("sqlite3_exec");
        printf("错误码：%d错误信息：%s,出错行是：%d\n",
                sqlite3_errcode(ppDb),sqlite3_errmsg(ppDb),__LINE__);
        printf("错误信息：%s\n",*errmsg);
        sqlite3_free(errmsg);//释放内存
        errmsg = NULL;    
        return -1;
    }
    int i,j;
    for(i = 0;i<=rnpow;i++)
    {
        for(j = 0;j<rncolumn;j++)
        {
            printf("%s\t",*(pazRuselt+i*rncolumn+j));
        }
        printf("\n");
    }
    sqlite3_free_table(pazRuselt);
    return 0;
    
}

void mnue(sqlite3 *ppDb)
{
    int ch;
    printf("\t\t\t***********工人阶级为主导的社会主义国家*****************\n");
    printf("\t\t\t1、插入一个工人信息\n");
    printf("\t\t\t2、删除一个工人信息\n");
    printf("\t\t\t3、修改一个工人信息\n");
    printf("\t\t\t4、查找一个工人信息\n");
    printf("\t\t\t5、输出所有工人信息\n");
    printf("\t\t\t0.退出\n");
    printf("请输入你的选择：");
    scanf("%d",&ch);
    getchar();//吸收回车
    switch(ch)
    {
        case 1:
        do_insert(ppDb);//插入一个工人信息
        break;
        case 2:
        do_delet(ppDb);
        break;
        case 3:do_change(ppDb);
        
        break;
        case 4:
        do_reseach(ppDb);//查找一个工人信息
        break;
        case 0:
        exit(0);//退出
        break;
        case 5:
        //do_showall(ppDb);//输出所有工人信息
        do_allshow(ppDb);//显示所有信息

        break;
        default:
        printf("输入有误，请重新输入\n");
    }
}
int main(int argc, const char *argv[])
{
    //1、创建或者打开数据库文件
    sqlite3 *ppDb;
    int k = sqlite3_open("./banzhang.db",&ppDb);//打开或者创建一个数据库
    if(k!=SQLITE_OK)
    {
        perror("sqlite3_open");
        printf("错误码：%d错误信息：%s,出错行是：%d\n",
                sqlite3_errcode(ppDb),sqlite3_errmsg(ppDb),__LINE__);
        return -1;
    }
    printf("数据库打开成功\n");
    //2、创建表格
    char *errmsg = NULL;//存储错误信息
    char sqlite[100] = "create table if not exists Work(gh int,name char,salary double);"; //创建一个表格
    if(sqlite3_exec(ppDb,sqlite,NULL,NULL,&errmsg)!=0)
    {
        perror("sqlite3_exec");
        printf("错误码：%d错误信息：%s,出错行是：%d\n",
                sqlite3_errcode(ppDb),sqlite3_errmsg(ppDb),__LINE__);
        printf("错误信息：%s\n",errmsg);
        return -1;
    }
    while(1)
    {
        mnue(ppDb);
    }



    sqlite3_close(ppDb);
    return 0;
}