#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);
return -1;
}
printf("添加工人信息成功\n");
return 0;
}
int do_delete(sqlite3 *ppDb)
{
int dele_gh;
worker work;
printf("请输入删除的工号:");
scanf("%d",&dele_gh);
char *errmsg = NULL;//存储错误信息
char sqlite[100]; //删除一个工人信息
snprintf(sqlite,sizeof(sqlite),"delete from Work where gh == %d;",dele_gh);
printf("%s",sqlite);
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;
}
printf("删除工人信息成功\n");
return 0;
}
int do_change(sqlite3 *ppDb)
{
int change_gh;
int end_gh;
char *errmsg = NULL;//存储错误信息
char sqlite[100]; //修改一个工人信息
worker work;
//***************************************************
printf("请输入修改前的工号:");
scanf("%d",&change_gh);
printf("请输入修改后的工号:");
scanf("%d",&end_gh);
getchar();
//****************************************************
snprintf(sqlite,sizeof(sqlite),"select * from Work where gh==%d;",change_gh);
printf("%s\n",sqlite);
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);
return -1;
}
snprintf(sqlite,sizeof(sqlite),"update Work set gh==%d where gh == %d;",end_gh,change_gh);
printf("%s\n",sqlite);
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;
}
printf("修改工人工号成功\n");
snprintf(sqlite,sizeof(sqlite),"select * from Work where gh==%d;",end_gh);
printf("%s\n",sqlite);
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);
return -1;
}
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);
return -1;
}
return 0;
}
int show_all(void *arg, int n, char **msgtext, char **msgtable) // 回调函数显示所有工人信息
{
int i;
// 打印列名
for (i = 0; i < n; i++) {
printf("%s\t", msgtable[i]);
}
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"; //输出所有工人信息
if(sqlite3_exec(ppDb,sqlite,show_all,NULL,&errmsg)!=0)//输出所有工人信息
{
perror("sqlite3_exec");
printf("错误码:%d错误信息:%s,出错行是:%d\n",
sqlite3_errcode(ppDb),sqlite3_errmsg(ppDb),__LINE__);
printf("错误信息:%s\n",errmsg);
return -1;
}
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_delete(ppDb);
break;
case 3:
do_change(ppDb);
break;
case 4:
do_reseach(ppDb);//查找一个工人信息
break;
case 0:
exit(0);//退出
break;
case 5:
do_showall(ppDb);//输出所有工人信息
break;
default:
printf("输入有误,请重新输入\n");
}
}
int main(int argc, const char *argv[])
{
//1、创建或者打开数据库文件
sqlite3 *ppDb;
int k = sqlite3_open("./work.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;
}