网络编程作业

完成数据库的增删改

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <sqlite3.h>

int  do_insert(sqlite3 *db)
{
	int id;
	int score;
	char name[10] = "";
	char sql[128] = "";
	char * errmsg = NULL;
	printf("请输入学生的ID：");
	scanf("%d",&id);
	printf("请输入学生的名字：");
	scanf("%s",name);
	printf("请输入学生的成绩：");
	scanf("%d",&score);
	sprintf(sql,"insert into stu values (%d, '%s', %d)",id,name,score);
	if(sqlite3_exec(db, sql, NULL, NULL, &errmsg) != SQLITE_OK)
	{
		printf("line:%d sqlite3_exec:%s\n", __LINE__, errmsg);
		return -1;
	}
	printf("insert table stu success\n");

}

int do_delete(sqlite3 *db)
{
	int id;
	char sql[128] = "";
	char * errmsg = NULL;
	printf("请输入要删除学生的ID：");
	scanf("%d",&id);
	sprintf(sql,"delete from stu where id=%d",id);
	if(sqlite3_exec(db, sql, NULL, NULL, &errmsg) != SQLITE_OK)
	{
		printf("line:%d sqlite3_exec:%s\n", __LINE__, errmsg);
		return -1;
	}
	printf("delete table stu success\n");

}

int do_update(sqlite3 *db)
{
	int id;
	int score;
	int c;
	char name[10] = "";
	char sql[128] = "";
	char * errmsg = NULL;
	printf("请输入要修改的学生的ID：");
	scanf("%d",&id);
	printf("--------------1.姓名--------------\n");
	printf("--------------2.成绩--------------\n");
	printf("请输入>>> ");
	scanf("%d",&c);
	switch(c)
	{
	case 1:
		{
			printf("请输入该学生的姓名：");
			scanf("%s",name);
			sprintf(sql,"update stu set name=%s where id=%d",name,id);
			if(sqlite3_exec(db, sql, NULL, NULL, &errmsg) != SQLITE_OK)
			{
				printf("line:%d sqlite3_exec:%s\n", __LINE__, errmsg);
				return -1;
			}
			printf("update table stu success\n");
			break;
		}
	case 2:
		{
			printf("请输入该学生的成绩：");
			scanf("%d",&score);
			sprintf(sql,"update stu set score=%d where id=%d",score,id);
			if(sqlite3_exec(db, sql, NULL, NULL, &errmsg) != SQLITE_OK)
			{
				printf("line:%d sqlite3_exec:%s\n", __LINE__, errmsg);
				return -1;
			}
			printf("update table stu success\n");
			break;
		}
	default:
		{
			printf("输入错误，请重新输入\n");
		}
	}

}
int main(int argc, const char *argv[])
{
	sqlite3* db = NULL;
	if(sqlite3_open("./my.db", &db) != SQLITE_OK)
	{
		printf("line:%d  sqlite3_open:[%d] %s\n",__LINE__, sqlite3_errcode(db), sqlite3_errmsg(db));
		return -1;
	}
	printf("sqlite3_open success\n");

	char sql[128] = "create table if not exists stu (id int primary key, name char, score double)";
	char * errmsg = NULL;
	if(sqlite3_exec(db, sql, NULL, NULL, &errmsg) != SQLITE_OK)
	{
		printf("line:%d sqlite3_exec:%s\n", __LINE__, errmsg);
		return -1;
	}
	printf("create table stu success\n");

	int c = 0;
	while(1)
	{
		printf("-----------------------------\n");
		printf("------------1.插入-----------\n");
		printf("------------2.删除-----------\n");
		printf("------------3.修改-----------\n");
		printf("------------4.查询-----------\n");
		printf("------------5.退出-----------\n");
		printf("-----------------------------\n");
		printf("请输入>>> ");
		scanf("%d",&c);
		while(getchar()!=10);
		
		switch(c)
		{
		case 1:
			do_insert(db);
			break;
		case 2:
			do_delete(db);
			break;
		case 3:
			do_update(db);
			break;
		case 4:
			//do_select();
			break;
		case 5:
			goto END;
		default:
			printf("输入错误，请重新输入\n");
		}
	}

END:

	if(sqlite3_close(db) != SQLITE_OK)
	{

		printf("line:%d  sqlite3_close:[%d] %s\n",__LINE__, sqlite3_errcode(db), sqlite3_errmsg(db));
		return -1;
	}
	printf("sqlite3_close success\n");

	db = NULL;

	return 0;
}