制作工人薪资管理系统

#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),errmsg,__LINE__);
		return -1;
	}
	printf("添加信息成功\n");
	return 0;
}
int callback(void *arg,int n,char **msgtext,char **msgtable)//回调输出查找的工人信息
{
	for(int i=0;i<n;i++)
	{
		printf("%s\t",*(msgtable+i));
	}
	putchar(10);
	for(int 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),errmsg,__LINE__);
		return -1;
	}
}
int show_all(void *arg,int n,char **msgtext,char **msgtable)//回调函数显示所有工人信息
{
	for(int i=0;i<n;i++)
	{
		printf("%s\t",*(msgtable+i));
	}
	putchar(10);
	//for(int i=0;i<n;i++)
	//{
		for(int j=0;j<n;j++)
		{
			printf("%s\t",*(msgtext+j));
		}
		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),errmsg,__LINE__);
		return -1;
	}	
}
int do_dele(sqlite3 *ppdb)
{
	char a[20];
	printf("请输入要删除的工人：");
	scanf("%s",a);
	char *errmsg=NULL;
	char sqlite[100];
	snprintf(sqlite,sizeof(sqlite),"delete from Work where name==\"%s\";",a);//删除一个工人信息
	if(sqlite3_exec(ppdb,sqlite,NULL,NULL,&errmsg)!=0)
	{
		perror("sqlite_exec");
		printf("错误码：%d,错误行：%s,错误信息：%d\n",sqlite3_errcode(ppdb),errmsg,__LINE__);
		return -1;
	}
}
int do_change(sqlite3 *ppdb)
{
	char a[20];
	int b;
	printf("请输入要修改的工人：");
	scanf("%s",a);
	printf("请输入修改后的工资：");
	scanf("%d",&b);
	char *errmsg;
	char sqlite[100];
	snprintf(sqlite,sizeof(sqlite),"update Work set selary=%d where name=\"%s\";",b,a);//修改工人信息
	if(sqlite3_exec(ppdb,sqlite,NULL,NULL,&errmsg)!=0)
	{
		perror("sqlite_exec");
		printf("错误码：%d,错误行：%s,错误信息：%d\n",sqlite3_errcode(ppdb),errmsg,__LINE__);
		return -1;
	}
}
void mnue(sqlite3 *ppdb)
{
	int a;
	printf("\t\t\t*********1、插入一个信息\n");
	printf("\t\t\t*********2、删除一个信息\n");
	printf("\t\t\t*********3、修改一个信息\n");
	printf("\t\t\t*********4、查找一个信息\n");
	printf("\t\t\t*********5、输出所有信息\n");
	printf("\t\t\t*********0、退出\n");
	printf("请输入你的选择：");
	scanf("%d",&a);
	getchar();//吸收回车
	switch(a)
	{
		case 1:
			do_insert(ppdb);//插入工人信息
		break;
		case 2:
			do_dele(ppdb);//删除工人信息
			do_showall(ppdb);
		break;
		case 3:
			do_change(ppdb);//修改工人信息
			do_showall(ppdb);
		break;
		case 4:
			do_reseach(ppdb);//查找一个工人信息
		break;
		case 5:
			do_showall(ppdb);//输出所有工人信息
		break;
		case 0:
		exit(0);//退出
		default:
		printf("输入有误，请重新输入\n");
	}
}
int main(int argc, const char *argv[])
{
	//创建或打开数据库文件
	sqlite3 *ppdb;
	int k=sqlite3_open("./lili.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");
	//创建一个表格
	char *errmsg=NULL;//存储错误信息
	char sqlite[100]="create table if not exists Work(gh int,name char,selary double);";
	if(sqlite3_exec(ppdb,sqlite,NULL,NULL,&errmsg)!=0)
	{
		perror("sqlite3_exec");
		printf("错误码：%d,错误信息：%s,错误行：%d\n",sqlite3_errcode(ppdb),errmsg,__LINE__);
		return -1;
	}
	while(1)
	{
		mnue(ppdb);//数据库指针作为参数
	}
	sqlite3_close(ppdb);//关闭数据库
	return 0;
}