网络编程9.10数据库

数据库信息管理
 

#include <sqlite3.h>
#include <myhead.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__);
		return -1;
	}
	printf("添加工人成功\n");
	return 0;
}

int callback(void *arg,int n ,char **msgtext,char **msgtable)
{
	int i,j;
	for(i=0;i<n;i++)
	{
		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__);
		return -1;
	}
	return 0;
}
int show_all(void *arg,int n,char **msgtext,char **msgtable)
{
	int i,j;
	int *first_time = (int *)arg;
	if(*first_time==1)
	{
		for(i=0;i<n;i++)
		{
			printf("%s\t",msgtable[i]);
		}
		printf("\n");
		*first_time = 0;
	}
	for(i=0;i<n;i++)
	{
//		printf("%s\t",msgtext[i]?msgtext[i]:"NULL");
		printf("%s\t",msgtext[i]);
	}
	printf("\n");

	return 0;
}

int  do_showall(sqlite3 *ppdb)
{
	int	first_time = 1;
	char *errmsg = NULL;
	char sqlite[100] = "select * from Work";//输出所有工人信息

	if(sqlite3_exec(ppdb,sqlite,show_all,&first_time,&errmsg)!=0)
	{
		printf("SQLite error: %s\n", errmsg);  // 显示具体错误信息
		sqlite3_free(errmsg);  // 释放错误信息字符串

		return -1;
	}
	return 0;

}

int do_delete(sqlite3 *ppdb)
{
	int gh;
	char *errmsg = NULL;
	char sqlite[100];
	printf("请输入要删除的工人的工号:");
	scanf("%d",&gh);
	snprintf(sqlite,sizeof(sqlite),"delete from Work where gh=\"%d\";",gh);

	if(sqlite3_exec(ppdb,sqlite,NULL,NULL,&errmsg)!=0)
	{
		printf("SQLite error: %s\n", errmsg);  // 显示具体错误信息
		sqlite3_free(errmsg);  // 释放错误信息字符串

		return -1;
	}
 	return 0;
}

int do_update(sqlite3 *ppdb)
{
	int gh,choice;
	double salary;
	char name[20];
	char *errmsg = NULL;
	char sqlite[100];
	printf("请输入要修改信息的工号:");
	scanf("%d",&gh);
	getchar();

	printf("请输入选择修改的内容:\n");
	printf("1.修改姓名\n");
	printf("2.修改薪水\n");
	scanf("%d",&choice);
	
	switch(choice)
	{
	case 1:
		printf("请输入新姓名:");
		scanf("%s",name);

		snprintf(sqlite,sizeof(sqlite),"update Work set name = \"%s\" where gh=%d;",name,gh);

		break;
	case 2:
		printf("请输入新薪资:");
		scanf("%lf",&salary);

		snprintf(sqlite,sizeof(sqlite),"update Work set salary = \"%lf\" where gh=%d;",salary,gh);
		break;

	default:
		printf("无效的选择");
		return -1;
	}

		if(sqlite3_exec(ppdb,sqlite,NULL,NULL,&errmsg)!=0)
		{
			printf("SQLite error: %s\n", errmsg);  // 显示具体错误信息
			sqlite3_free(errmsg);  // 释放错误信息字符串

			return -1;
		}
	
}
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_update(ppdb);
		break;
	case 4:
		do_reseach(ppdb);//查找一个工人信息
		break;
	case 5:
		do_showall(ppdb);//输出所有工人信息
		break;
	case 0:
		exit(0);//退出
		break;
	default:
		printf("输入有误，请重新输入");
	}

}


int main(int argc,const char*argv[])
{
	sqlite3 *ppdb;
	//创建一个数据库
	int k=sqlite3_open("./hongming.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__);
		return -1;
	}
	while(1)
	{
		mnue(ppdb);
	}
	sqlite3_close(ppdb);

	return 0;
}