#include <myhead.h>
//添加学生信息函数
int do_add(sqlite3 *ppDb)
{
//准备sql语句
int add_numb = 0;
char add_name[128] = "";
double add_score = 0;
//提示并输入数据
printf("请输入学号:");
scanf("%d",&add_numb);
printf("请输入名字:");
scanf("%s",add_name);
printf("请输入分数:");
scanf("%lf",&add_score);
getchar();
char sql[128] = "";
sprintf(sql,"insert into Stu values(%d,\"%s\",%lf);",add_numb,add_name,add_score);
//定义接收错误信息的变量
char *errmsg = NULL;
//执行sql语句
if(sqlite3_exec(ppDb,sql,NULL,NULL,&errmsg) != SQLITE_OK)
{
printf("exec error:%s\n",errmsg);
sqlite3_free(errmsg);
return -1;
}
printf("插入成功\n");
return 0;
}
//定义回调函数查询处理后的结果集
int callback(void *arg,int cols,char **msgtext,char **msgheader)
{
//输出表头
if(*((int *)arg) == 0)
{
for(int i=0;i<cols;i++)
{
printf("%s\t",*(msgheader+i));
}
printf("\n");
(*((int *)arg))++;
}
//记录当前的输出内容
for(int i=0;i<cols;i++)
{
printf("%s\t",*(msgtext+i));
}
printf("\n");
return 0;
}
//查找函数
int do_search(sqlite3 *ppDb)
{
//准备sql语句
char sql[128] = "select * from Stu;";
//定义接收错误信息的变量
char *errmsg = NULL;
int flag = 0;//定义标识位
if(sqlite3_exec(ppDb,sql,callback,&flag,&errmsg) != SQLITE_OK)
{
printf("exec error:%s\n",errmsg);
sqlite3_free(errmsg);
return -1;
}
return 0;
}
//删除函数
int do_delete(sqlite3 *ppDb)
{
//准备sql语句
char sql[128] = "delete from Stu;";
//定义接收错误信息的变量
char *errmsg = NULL;
if(sqlite3_exec(ppDb,sql,callback,NULL,&errmsg) != SQLITE_OK)
{
printf("exec error:%s\n",errmsg);
sqlite3_free(errmsg);
return -1;
}
return 0;
}
//修改学生信息
int do_update(sqlite3 *ppDb)
{
int add_numb;
char add_name[100] = "";
printf("请输入要改动的名字:");
scanf("%s",add_name);
printf("请输入要改动对应的学号:");
scanf("%d",&add_numb);
getchar();
//准备sql语句
char sql[128] = "";
sprintf(sql,"update Stu set name=\"%s\" where numb=%d;",add_name,add_numb);
//定义接收错误信息的变量
char *errmsg = NULL;
if(sqlite3_exec(ppDb,sql,callback,NULL,&errmsg) != SQLITE_OK)
{
printf("exec error:%s\n",errmsg);
sqlite3_free(errmsg);
return -1;
}
return 0;
}
/**************主函数***************/
int main(int argc, const char *argv[])
{
//1.定义数据库句柄指针
sqlite3*ppDb = NULL;
//2.打开创建数据库
if(sqlite3_open("./my.db",&ppDb)!=SQLITE_OK)
{
fprintf(stderr, "sqlite3_open error:errmsg = %s, errcode = %d\n",\
sqlite3_errmsg(ppDb), sqlite3_errcode(ppDb));
return -1;
}
printf("sqlite3_open success\n");
//3.创建数据库表
//3.1准备sql语句
char sql[128]="create table if not exists Stu(numb int,name char,score double);";
char *errmsg = NULL;
//3.2执行sql语句
if(sqlite3_exec(ppDb,sql,NULL,NULL,&errmsg)!=SQLITE_OK)
{
printf("exec error:%s\n",errmsg);
return -1;
}
printf("create table Stu success\n");
//4.对数据表进行操作
int menu = -1;
while(1)
{
system("clear");
printf("\t\t============XXX 学生管理系统============\n");
printf("\t\t============1、添加学生信息=============\n");
printf("\t\t============2、删除学生信息=============\n");
printf("\t\t============3、修改学生信息=============\n");
printf("\t\t============4、查找学生信息=============\n");
printf("\t\t============0、退出系统=================\n");
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);
}
break;
case 0:goto END;
default:printf("您输入的功能有误请重新输入!!!\n");
}
printf("输入任意键,按下回车清屏\n");
while(getchar() != '\n');//循环读入字符,直到读到回车符结束循环
}
END:
//关闭服务器
sqlite3_close(ppDb);
return 0;
}