工人系统的增删改查

#include<stdio.h>
#include<string.h>
#include<myhead.h>
#include<sqlite3.h>
typedef struct
{
	int gh;
	char name[20];
	double salarly;



}worker;
int do_insert(sqlite3 *ppdb)
{
	worker work;
	printf("输入插入的工号：");
	scanf("%d",&work.gh);
	printf("输入插入的名字：");
	scanf("%s",work.name);
	printf("输入插入的薪水：");
	scanf("%lf",&work.salarly);
	char * errmsg=NULL;
	char sqlite[100];
	snprintf(sqlite,sizeof(sqlite),"insert into Work values(%d,\"%s\",%lf);",work.gh,work.name,work.salarly);

	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");
}
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;
	}	
	printf("查找成功\n");
	return 0;





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

	int flag=1;
	if(sqlite3_exec(ppdb,sqlite,show_all,&flag,&errmsg)!=0)
	{
		perror("sqlite3_exec");
		printf("错误码：%d错误信息：%s,出错行：%d\n"
				,sqlite3_errcode(ppdb),sqlite3_errmsg(ppdb),__LINE__);

		return-1;
	}	



}
int do_dele(sqlite3*ppdb)
{

	char key[20];
	printf("请输入你要删除的工人姓名：");
	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__);

		return-1;
	}	


	return 0;



}
int do_change(sqlite3 *ppdb)
{
	
	worker work;
    char key[20];
    printf("请输入你要修改的工人姓名：");
    if (scanf("%s", key) != 1) {
        printf("输入错误\n");
        return -1;
    }
    getchar(); // 清除输入缓冲

    while(1)
    {
        int a;
        printf("请输入修改的项目：1:工号 2:姓名 3:薪水 4:退出 ");
        if (scanf("%d", &a) != 1) {
            printf("输入错误\n");
            getchar();
            continue;
        }
        getchar();

        char *errmsg = NULL;
        char sqlite[100];

        switch(a)
        {
            case 1:
                {
                    printf("输入新的工号:");
                    if (scanf("%d", &work.gh) != 1) {
                        printf("输入错误\n");
                        continue;
                    }
                    snprintf(sqlite, sizeof(sqlite), "UPDATE Work SET gh = %d WHERE name = \"%s\";",work.gh,key);
                    break;
                }
            case 2:
                {
                    printf("输入新的姓名:");
                    if (scanf("%19s", work.name) != 1) {
                        printf("输入错误\n");
                        continue;
                    }
                    snprintf(sqlite, sizeof(sqlite), "UPDATE Work SET name =\"%s\" WHERE name = \"%s\";",work.name,key);
                    break;
                }
            case 3:
                {
                    printf("输入新的薪水:");
                    if (scanf("%lf", &work.salarly) != 1) {
                        printf("输入错误\n");
                        continue;
                    }
                    snprintf(sqlite, sizeof(sqlite), "UPDATE Work SET salarly = %lf WHERE name = \"%s\";",work.salarly,key);
                    break;
                }
            case 4:
                return 0;
            default:
                printf("无效的选择\n");
                continue;
        }
	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;
	}	
		
	return 0;
	}
//sqlite3_free(errmsg);
}
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_dele(ppdb);
		break;
	case 3:
		do_change(ppdb);
		break;
	case 4:
		do_reseach(ppdb);
		break;
	case 0:
		exit(0);
	case 5:
		do_showall(ppdb);

		break;
	default:
		printf("输入有误\n");
	}


}
int main(int argc, const char *argv[])
{
	sqlite3 * ppdb;
	int k= sqlite3_open("./banzhang.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,salarly 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;
}