#include <myhead.h>
#include <sqlite3.h>
typedef struct
{
int id;
char name[20];
int age;
int money;
}woker;
int callbake(void *arg,int n,char **a,char **b)//回调 输出查找到的工人信息
{
for(int i =0;i<n;i++)
{
printf("%s\t",*(b+i));
}
putchar(10);
for(int i=0;i<n;i++)
{
printf("%s\t",*(a+i));
}
putchar(10);
printf("查找成功\n");
return 0;
}
int do_insert(sqlite3 *ppDb)
{
woker work;
printf("请输入工号\n");
scanf("%d",&work.id);
printf("请输入工姓名\n");
scanf("%s",work.name);
printf("请输入年龄\n");
scanf("%d",&work.age);
printf("请输入薪资\n");
scanf("%d",&work.money);
char *errmsg = NULL;//存储错误信息
char sqlite[100];//插入一个工人信息
snprintf(sqlite,sizeof(sqlite),"insert into Work values(%d ,\"%s\", %d, %d);",work.id,work.name,work.age,work.money);
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");
}
int do_reserch(sqlite3 *ppDb)
{
printf("请输入要查的员工姓名\n");
char key[100];
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,callbake,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");
}
int do_delete(sqlite3 *ppDb)
{
printf("请输入要删除的员工信息\n");
char key[20];
scanf("%s",key);
char *errmsg = NULL;
char sqlite[100];
snprintf(sqlite,sizeof(sqlite),"delete from Work where name == \"%s\";",key);
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");
}
int do_updata(sqlite3 *ppDb)
{
printf("请输入你要修改的员工姓名\n");
char key[20];
scanf("%s",key);
printf("请输入你想修改员工的什么信息\n");
printf("\t1.工号\n");
printf("\t2.姓名\n");
printf("\t3.年龄\n");
printf("\t4.薪资\n");
printf("\t5.全部修改\n");
int n;
scanf("%d",&n);
getchar();
char *errmsg = NULL;
char sqlite[100];
switch(n)
{
case 1:
printf("新工号\n");
int a;
scanf("%d",&a);
getchar();
snprintf(sqlite,sizeof(sqlite),"update Work set id =%d where name == \"%s\";",a,key);
break;
case 2:
printf("新姓名\n");
char arr[20];
scanf("%s",arr);
getchar();
snprintf(sqlite,sizeof(sqlite),"update Work set name =\"%s\" where name == \"%s\";",arr,key);
break;
case 3:
printf("新年龄\n");
int b;
scanf("%d",&b);
getchar();
snprintf(sqlite,sizeof(sqlite),"update Work set age = %d where name == \"%s\";",b,key);
break;
case 4:
printf("新薪资\n");
int c;
scanf("%d",&c);
getchar();
snprintf(sqlite,sizeof(sqlite),"update Work set money = %d where name == \"%s\";",c,key);
break;
case 5:
break;
}
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");
}
void mnue(sqlite3 *ppDb)
{
printf("--------狗熊岭员工信息表--------\n");
printf("\t1,插入一个成员的信息\n");
printf("\t2,删除一个成员的信息\n");
printf("\t3,修改一个成员的信息\n");
printf("\t4,查找一个成员的信息\n");
printf("\t0.退出\n");
printf("请输入你的选择\n");
int n;
scanf("%d",&n);
getchar();
switch(n)
{
case 1:
do_insert(ppDb);
break;
case 2:
do_delete(ppDb);
break;
case 3:
do_updata(ppDb);
break;
case 4:
do_reserch(ppDb);
break;
case 5:
//do_showall(ppDb);
break;
case 0:
exit(0);
break;
default:
printf("输入错误\n");
}
}
int main(int argc, const char *argv[])
{
sqlite3 *ppDb;//1.创建或打开一个数据库文件
int k = sqlite3_open("./Workdata.dp",&ppDb);
if(k != SQLITE_OK)
{
perror("sqlit3_open");
printf("错误码:%d错误信息：%s, 出错行是%d\n",
sqlite3_errcode(ppDb),sqlite3_errmsg(ppDb),LINE);
return -1;
}
printf("数据库打开成功\n");
char *errmsg = NULL;
char sqlite[100] = "create table if not exists Work(id int,name char,age int,money int);";//创建一个表格
//printf("");
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;
}