网络编程 day7 作业

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/socket.h>
#include <sys/ipc.h>
#include <sys/sem.h>
#include <sys/msg.h>
#include <sys/shm.h>
#include <sys/un.h>
#include <sys/select.h>
#include <arpa/inet.h>
#include <semaphore.h>
#include <arpa/inet.h>
#include <fcntl.h>
#include <pthread.h>
#include <semaphore.h>
#include <wait.h>
#include <signal.h>
#include <poll.h>
#include <errno.h>
#include <sqlite3.h>

//封装添加学生信息函数
int do_add(sqlite3 *ppDb)
{
    //准备sql语句
    int add_numb = 0;
    char add_name[20] = "";
    double add_score = 0;

    //提示并输入数据
    printf("请输入学号：");
    scanf("%d", &add_numb);
    printf("请输入姓名：");
    scanf("%s", add_name);
    printf("请输入成绩：");
    scanf("%lf", &add_score);
    getchar();

    char sq1[128] = "";
    sprintf(sq1,"insert into Stu values(%d, \"%s\", %lf);", add_numb,add_name,add_score);
    
    //定义接收错误信息的变量
    char *errmsg = NULL;
    
    //执行sq1语句
    if(sqlite3_exec(ppDb, sq1, 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)
{
    //准备sq2语句
	char sq2[128] = "select * from Stu;";
    //定义错误信息容器
    char *errmsg = NULL;

    int flag = 0;        //定义标识位

    //执行sq2语句
    if(sqlite3_exec(ppDb, sq2, 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 sq3[128] = "";
    sprintf(sq3,"delete from Stu where numb=%d\;", add_numb);
    //定义错误信息容器
    char *errmsg = NULL;

    int flag = 0;        //定义标识位

    //执行sql语句
    if(sqlite3_exec(ppDb, sq3, callback, &flag, &errmsg) != SQLITE_OK)
    {
        printf("exec error:%s\n", errmsg);

        sqlite3_free(errmsg);       //释放错误信息的空间
        return -1;
    }


    return 0;
}

//定义修改函数
int do_update(sqlite3 *ppDb)
{
    //准备sq4语句
    char sq4[128] = "";
    sprintf(sq4,"update Stu name=%s\ where num=%d\;",add_name,add_numb);
    //定义错误信息容器
    char *errmsg = NULL;

    int flag = 0;        //定义标识位

    //执行sql语句
    if(sqlite3_exec(ppDb, sq4, callback, &flag, &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 *sql =  "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);
        sqlite3_free(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;
}