使用C语言完成数据库的增删改
#include<myhead.h>
int do_add(sqlite3 *ppDb)
{
int numb;
char name[50];
int salary;
printf("请输入员工信息:工号、姓名、薪水\n");
scanf("%d %s %d",&numb,name,&salary);
char sql[128];
char *errmsg=NULL;
sprintf(sql,"INSERT INTO Worker values(%d,'%s',%d)",numb,name,salary);
if(sqlite3_exec(ppDb,sql,NULL,NULL,&errmsg)!=SQLITE_OK)
{
printf("exec error : %s\n", errmsg);
sqlite3_free(errmsg);
errmsg = NULL;
return -1;
}
printf("员工信息加入成功\n");
return 0;
}
int do_delete(sqlite3 *ppDb)
{
char buf[128];
printf("请输入要删除的员工工号:");
fgets(buf,sizeof(buf),stdin);
char sql[128]="DELETE form Worker where numb=";
strcat(sql,buf);
char *errmsg=NULL;
if(sqlite3_exec(ppDb,sql,NULL,NULL,&errmsg)!=SQLITE_OK)
{
printf("exec error : %s\n", errmsg);
sqlite3_free(errmsg);
errmsg = NULL;
return -1;
}
printf("删除成功\n");
return 0;
}
int do_update(sqlite3 *ppDb)
{
int salary,numb;
printf("请输入要更改薪资的员工的员工号:");
scanf("%d",&numb);
printf("请输入所要更改后的薪水:");
scanf("%d",&salary);
char sql[128];
sprintf(sql,"update Worker set salary=%d where numb=%d",salary,numb);
char *errmsg=NULL;
if(sqlite3_exec(ppDb,sql,NULL,NULL,&errmsg)!=SQLITE_OK)
{
printf("exec error : %s\n", errmsg);
sqlite3_free(errmsg);
errmsg = NULL;
return -1;
}
printf("更新成功\n");
return 0;
}
int main(int argc, const char *argv[])
{
//定义数据库句柄指针
sqlite3 * ppDb = NULL;
//打开数据库,如果数据库不存在,则创建数据库
//将数据库句柄由参数2返回
if(sqlite3_open("./my.db", &ppDb) != SQLITE_OK)
{
printf("sqlite3_open error,errcode=%d, errmsg = %s\n",\
sqlite3_errcode(ppDb), sqlite3_errmsg(ppDb));
return -1;
}
printf("sqlite3_open success\n");
//创建数据表
//1、准备sql语句
char sql[128] = "create table if not exists Worker(numb int, name char, salary double)";
char *errmsg = NULL; //存放执行sql语句口的错误信息
if(sqlite3_exec(ppDb, sql, NULL, NULL, &errmsg) != SQLITE_OK)
{
printf("exec error : %s\n", errmsg);
sqlite3_free(errmsg); //防止内存泄露
errmsg = NULL;
return -1;
}
printf("员工信息表创建成功\n");
int menu = -1; //菜单选项
for(;;)
{
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;
}