1.22 第九章 数据库day2

思维导图

 

#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;
}