数据库工人增删改查系统

yangh1231
已于 2024-09-11 09:02:51 修改
阅读量140
点赞数 1
文章标签：数据库 jvm
于 2024-09-11 08:57:42 首次发布
本文链接：https://blog.csdn.net/yangh1231/article/details/142123899
版权
#include <myhead.h>
#include <sqlite3.h>

// 定义工人结构体，用于存储工人信息
typedef struct
{
    int gh;              // 工号
    char name[20];       // 姓名
    double salary;       // 薪水
} worker;

// 插入工人信息到数据库的函数
int do_insert(sqlite3 *ppDb)
{
    worker work;  // 创建工人信息结构体
    printf("请输入插入的工号：");
    scanf("%d", &work.gh);
    printf("请输入插入的姓名：");
    scanf("%s", work.name);
    printf("请输入插入的薪水：");
    scanf("%lf", &work.salary);

    char *errmsg = NULL;  // 用于存储 SQLite 错误信息
    char sqlite[100];     // 用于存储 SQL 语句
    // 构造插入 SQL 语句
    snprintf(sqlite, sizeof(sqlite), "insert into Work values(%d,\"%s\",%lf);", work.gh, work.name, work.salary);
    // 执行 SQL 语句，插入工人信息
    if (sqlite3_exec(ppDb, sqlite, NULL, NULL, &errmsg) != 0)
    {
        perror("sqlite3_exec");
        printf("错误码：%d 错误信息：%s, 出错行是：%d\n",
               sqlite3_errcode(ppDb), sqlite3_errmsg(ppDb), __LINE__);
        printf("错误信息：%s\n", errmsg);
        return -1;
    }
    printf("添加工人信息成功\n");
    return 0;
}

// 删除工人信息的函数
int do_delete(sqlite3 *ppDb)
{
    int dele_gh;  // 要删除的工人工号
    printf("请输入删除的工号：");
    scanf("%d", &dele_gh);

    char *errmsg = NULL;  // 用于存储 SQLite 错误信息
    char sqlite[100];     // 用于存储 SQL 语句
    // 构造删除 SQL 语句
    snprintf(sqlite, sizeof(sqlite), "delete from Work where gh == %d;", dele_gh);
    printf("%s", sqlite);
    // 执行 SQL 语句，删除工人信息
    if (sqlite3_exec(ppDb, sqlite, NULL, NULL, &errmsg) != 0)
    {
        perror("sqlite3_exec");
        printf("错误码：%d 错误信息：%s, 出错行是：%d\n",
               sqlite3_errcode(ppDb), sqlite3_errmsg(ppDb), __LINE__);
        printf("错误信息：%s\n", errmsg);
        return -1;
    }
    printf("删除工人信息成功\n");
    return 0;
}

// 修改工人工号的函数
int do_change(sqlite3 *ppDb)
{
    int change_gh;  // 修改前的工号
    int end_gh;     // 修改后的工号
    char *errmsg = NULL;  // 用于存储 SQLite 错误信息
    char sqlite[100];     // 用于存储 SQL 语句

    // 读取修改前后的工号
    printf("请输入修改前的工号：");
    scanf("%d", &change_gh);
    printf("请输入修改后的工号：");
    scanf("%d", &end_gh);
    getchar();  // 吸收多余的回车符

    // 查找要修改的工人信息
    snprintf(sqlite, sizeof(sqlite), "select * from Work where gh == %d;", change_gh);
    printf("%s\n", sqlite);
    if (sqlite3_exec(ppDb, sqlite, callback, NULL, &errmsg) != 0)
    {
        perror("sqlite3_exec");
        printf("错误码：%d 错误信息：%s, 出错行是：%d\n",
               sqlite3_errcode(ppDb), sqlite3_errmsg(ppDb), __LINE__);
        printf("错误信息：%s\n", errmsg);
        return -1;
    }

    // 更新工号
    snprintf(sqlite, sizeof(sqlite), "update Work set gh == %d where gh == %d;", end_gh, change_gh);
    printf("%s\n", sqlite);
    if (sqlite3_exec(ppDb, sqlite, NULL, NULL, &errmsg) != 0)
    {
        perror("sqlite3_exec");
        printf("错误码：%d 错误信息：%s, 出错行是：%d\n",
               sqlite3_errcode(ppDb), sqlite3_errmsg(ppDb), __LINE__);
        printf("错误信息：%s\n", errmsg);
        return -1;
    }
    printf("修改工人工号成功\n");

    // 查找修改后的工人信息
    snprintf(sqlite, sizeof(sqlite), "select * from Work where gh == %d;", end_gh);
    printf("%s\n", sqlite);
    if (sqlite3_exec(ppDb, sqlite, callback, NULL, &errmsg) != 0)
    {
        perror("sqlite3_exec");
        printf("错误码：%d 错误信息：%s, 出错行是：%d\n",
               sqlite3_errcode(ppDb), sqlite3_errmsg(ppDb), __LINE__);
        printf("错误信息：%s\n", errmsg);
        return -1;
    }
    return 0;
}

// 回调函数，用于输出查询结果
int callback(void *arg, int n, char **msgtext, char **msgtable)
{
    int i;
    // 输出列名
    for (i = 0; i < n; i++)
    {
        printf("%s\t", msgtable[i]);
    }
    printf("\n");

    // 输出查询到的行
    for (i = 0; i < n; i++)
    {
        printf("%s\t", msgtext[i]);
    }
    printf("\n");
    printf("查找成功\n");
    return 0;
}

// 查找工人信息的函数
int do_reseach(sqlite3 *ppDb)
{
    char key[20];  // 存储要查找的工人姓名
    printf("请输入你要查找的工人姓名：");
    scanf("%s", key);

    char *errmsg = NULL;  // 用于存储 SQLite 错误信息
    char sqlite[100];     // 用于存储 SQL 语句
    // 构造查询 SQL 语句
    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__);
        printf("错误信息：%s\n", errmsg);
        return -1;
    }
    return 0;
}

// 回调函数，输出所有工人信息
int show_all(void *arg, int n, char **msgtext, char **msgtable)
{
    int i;

    // 打印列名
    for (i = 0; i < n; i++) {
        printf("%s\t", msgtable[i]);
    }
    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;  // 用于存储 SQLite 错误信息
    char sqlite[100] = "select * from Work";  // 查询所有工人信息的 SQL 语句
    if (sqlite3_exec(ppDb, sqlite, show_all, NULL, &errmsg) != 0)
    {
        perror("sqlite3_exec");
        printf("错误码：%d 错误信息：%s, 出错行是：%d\n",
               sqlite3_errcode(ppDb), sqlite3_errmsg(ppDb), __LINE__);
        printf("错误信息：%s\n", errmsg);
        return -1;
    }
    return 0;
}

// 显示菜单
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_delete(ppDb);  // 删除工人信息
            break;
        case 3:
            do_change(ppDb);  // 修改工人信息
            break;
        case 4:
            do_reseach(ppDb);  // 查找工人信息
            break;
        case 5:
            do_showall(ppDb);  // 输出所有工人信息
            break;
        case 0:
            exit(0);  // 退出
        default:
            printf("输入有误，请重新输入\n");
    }
}

// 主函数
int main(int argc, const char *argv[])
{
    // 1、创建或打开数据库文件
    sqlite3 *ppDb;
    int k = sqlite3_open("./work.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");

    // 2、创建表格
    char *errmsg = NULL;  // 用于存储 SQLite 错误信息
    char sqlite[100] = "create table if not exists Work(gh int, name char, salary double);";  // 创建表格的 SQL 语句
    if (sqlite3_exec(ppDb, sqlite, NULL, NULL, &errmsg) != 0)
    {
        perror("sqlite3_exec");
        printf("错误码：%d 错误信息：%s, 出错行是：%d\n",
               sqlite3_errcode(ppDb), sqlite3_errmsg(ppDb), __LINE__);
        printf("错误信息：%s\n", errmsg);
        return -1;
    }

    // 3、显示菜单并执行相应操作
    while (1)
    {
        mnue(ppDb);  // 调用菜单函数
    }

    // 4、关闭数据库文件
    sqlite3_close(ppDb);
    return 0;
}