思维导图
#include<header.h>
//添加员工信息
int do_add(sqlite3 *ppDb)
{
//准备SQL语句
int add_numb;
char add_name[20];
char add_sex[10];
double add_score;
printf("请输入要添加的学号>>>");
scanf("%d",&add_numb);
printf("请输入要添加的姓名>>>");
scanf("%s",add_name);
printf("请输入要添加的性别>>>");
scanf("%s",add_sex);
printf("请输入要添加的成绩>>>");
scanf("%lf",&add_score);
getchar();
char sql[128]="";
sprintf(sql,"insert into Worker (numb,name,sex,score) values (%d,\"%s\",\"%s\",%.2lf);",add_numb,add_name,add_sex,add_score);
printf("sql=%s\n",sql);
//执行SQL语句
char *errmsg=NULL;
if(sqlite3_exec(ppDb,sql,NULL,NULL,&errmsg)!=SQLITE_OK)
{
printf("sqlite3_exec error errmsg=%s\n",errmsg);
sqlite3_free(errmsg);//释放errmsg的空间
errmsg=NULL;
return -1;
}
puts("插入成功!!!");
return 0;
}
//删除员工信息
int do_delete(sqlite3 *ppDb)
{
//准备SQL语句
int del_numb;
printf("请输入要删除的信息>>>");
scanf("%d",&del_numb);//按照numb号删除
getchar();
char sql[128]="";
sprintf(sql,"delete from Worker where numb = %d;",del_numb);
printf("sql=%s\n",sql);
//执行SQL语句
char *errmsg=NULL;
if(sqlite3_exec(ppDb,sql,NULL,NULL,&errmsg)!=SQLITE_OK)
{
printf("sqlite3_exec error errmsg=%s\n",errmsg);
sqlite3_free(errmsg);//释放errmsg的空间
errmsg=NULL;
return -1;
}
puts("删除成功!!!");
return 0;
}
//修改员工信息
int do_update(sqlite3 *ppDb)
{
//准备SQL语句
int update_numb,add_numb;
printf("请输入原信息>>>");
scanf("%d",&add_numb);
getchar();
printf("请输入修改后的信息>>>");
scanf("%d",&update_numb);//按照numb号修改
getchar();
char sql[128]="";
sprintf(sql,"update Worker set numb=%d where numb = %d;",update_numb,add_numb);
printf("sql=%s\n",sql);
//执行SQL语句
char *errmsg=NULL;
if(sqlite3_exec(ppDb,sql,NULL,NULL,&errmsg)!=SQLITE_OK)
{
printf("sqlite3_exec error errmsg=%s\n",errmsg);
sqlite3_free(errmsg);//释放errmsg的空间
errmsg=NULL;
return -1;
}
puts("修改成功!!!");
return 0;
}
//定义查找员工信息的回调函数
int callback(void *arg,int cols,char **value_text,char **value_name)
{
if((*(int *)arg)==0)
{
//输出表头
for(int i=0;i<cols;i++)
{
printf("%s\t",value_name[i]);
}
puts("");
*((int *)arg)=1;//更改标识位
}
//输出当前记录信息
for(int i=0;i<cols;i++)
{
printf("%s\t",value_text[i]);
}
puts("");
return 0;
}
//查找员工信息
int do_search(sqlite3 *ppDb)
{
int flag=0;//标识位
//准备SQL语句
char sql[128]="select * from Worker";
char *errmsg=NULL;
//执行SQL语句
if(sqlite3_exec(ppDb,sql,callback,&flag,&errmsg)!=SQLITE_OK)
{
printf("sqlite3_exec error errmsg=%s\n",errmsg);
sqlite3_free(errmsg);//释放errmsg的空间
errmsg=NULL;
return -1;
}
puts("查找成功!!!");
return 0;
}
//定义查询结果函数
int do_search_table(sqlite3 *ppDb)
{
//准备SQL语句
char sql[128]="select * from Worker;";
char **res_ptr=NULL;//接收结果集的起始地址
int rows=0;//接收行数
int cols=0;//接收列数
char *errmsg=NULL;//接收错误信息
//调用函数执行SQL语句并获取结果集
if(sqlite3_get_table(ppDb,sql,&res_ptr,&rows,&cols,&errmsg)!=SQLITE_OK)
{
printf("sqlite3_get_table error errmsg=%s\n",errmsg);
sqlite3_free(errmsg);//释放errmsg的空间
errmsg=NULL;
return -1;
}
for(int i=0;i<rows+1;i++)//外行
{
for(int j=0;j<cols;j++)//内列
{
//输出当前位置的内容
printf("%s\t",*(res_ptr+(i*cols+j)));
}
puts("");
}
//释放结果集
sqlite3_free_table(res_ptr);
return 0;
}
int main(int argc, const char *argv[])
{
//1.打开或创建数据库
sqlite3 *ppDb=NULL;
if(sqlite3_open("./worker.db",&ppDb)!=SQLITE_OK)
{
printf("sqlite3_open error,errcode=%d,errmsg=%s\n",sqlite3_errcode(ppDb),sqlite3_errmsg(ppDb));
return -1;
}
puts("sqlite3_open success!!!");
//2.数据库的相关操作
//2.1创建数据表
//准备SQL语句
char sql[128]="create table if not exists Worker (numb int,name char,sex char,score double);";
char *errmsg=NULL;//记录返回的错误信息
//执行SQL语句
if(sqlite3_exec(ppDb,sql,NULL,NULL,&errmsg)!=SQLITE_OK)
{
printf("sqlite3_exec error errmsg=%s\n",errmsg);
sqlite3_free(errmsg);//释放errmsg的空间
errmsg=NULL;
return -1;
}
puts("create table success!!!");
//2.2数据库的增删改查操作
int menu=-1;//菜单选项
while(1)
{
system("clear");//将上面内容清屏
puts("\t\t=====公司员工管理系统=====");
puts("\t\t=====1.添加员工信息=====");
puts("\t\t=====2.删除员工信息=====");
puts("\t\t=====3.修改员工信息=====");
puts("\t\t=====4.查找员工信息=====");
puts("\t\t=====0.退出=====");
printf("输入待办业务");
scanf("%d",&menu);
getchar();
switch(menu)
{
case 1:
{
do_add(ppDb);//添加员工信息
}
break;
case 2:
{
do_delete(ppDb);//删除员工信息
}
break;
case 3:
{
do_update(ppDb);//修改员工信息
}
break;
case 4:
{
//do_search(ppDb);//查找员工信息
do_search_table(ppDb);
}
break;
case 0:goto END;
defalut:puts("输入错误,请重新输入!!!");
}
printf("按下任意键加回车清屏>>>");
while(getchar()!='\n');
}
END:
//3.关闭数据库
sqlite3_close(ppDb);
puts("sqlite3_close success!!!");
return 0;
}