网络编程 9.10

完成代码剩余功能

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;

    char sqlite[256];

    snprintf(sqlite, sizeof(sqlite), "INSERT INTO Work (gh, name, salary) VALUES (%d, '%s', %f);", work.gh, work.name, work.salary);

   

    if (sqlite3_exec(ppDb, sqlite, NULL, NULL, &errmsg) != SQLITE_OK) {

        printf("错误码：%d 错误信息：%s\n", sqlite3_errcode(ppDb), sqlite3_errmsg(ppDb));

        return -1;

    }

    printf("添加工人信息成功\n");

    return 0;

}

int callback(void *arg, int n, char **msgtable, char **msgtext) {

    for (int i = 0; i < n; i++) {

        printf("%s\t", msgtable[i]);

    }

    printf("\n");

    return 0;

}

int do_research(sqlite3 *ppDb) {

    char key[20];

    printf("请输入你要查找的工人姓名: ");

    scanf("%s", key);

    char *errmsg = NULL;

    char sqlite[256];

    snprintf(sqlite, sizeof(sqlite), "SELECT * FROM Work WHERE name = '%s';", key);

    if (sqlite3_exec(ppDb, sqlite, callback, NULL, &errmsg) != SQLITE_OK) {

        printf("错误码：%d 错误信息：%s\n", sqlite3_errcode(ppDb), sqlite3_errmsg(ppDb));

        return -1;

    }

    return 0;

}

int do_delete(sqlite3 *ppDb) {

    int gh;

    printf("请输入要删除的工号: ");

    scanf("%d", &gh);

    char *errmsg = NULL;

    char sqlite[256];

    snprintf(sqlite, sizeof(sqlite), "DELETE FROM Work WHERE gh = %d;", gh);

    if (sqlite3_exec(ppDb, sqlite, NULL, NULL, &errmsg) != SQLITE_OK) {

        printf("错误码：%d 错误信息：%s\n", sqlite3_errcode(ppDb), sqlite3_errmsg(ppDb));

        return -1;

    }

    printf("删除工人信息成功\n");

    return 0;

}

int do_update(sqlite3 *ppDb) {

    int gh;

    Worker work;

    printf("请输入要修改的工号: ");

    scanf("%d", &gh);

    printf("请输入新的姓名: ");

    scanf("%s", work.name);

    printf("请输入新的薪水: ");

    scanf("%lf", &work.salary);

    char *errmsg = NULL;

    char sqlite[256];

    snprintf(sqlite, sizeof(sqlite), "UPDATE Work SET name = '%s', salary = %f WHERE gh = %d;", work.name, work.salary, gh);

    if (sqlite3_exec(ppDb, sqlite, NULL, NULL, &errmsg) != SQLITE_OK) {

        printf("错误码：%d 错误信息：%s\n", sqlite3_errcode(ppDb), sqlite3_errmsg(ppDb));

        return -1;

    }

    printf("修改工人信息成功\n");

    return 0;

}

int do_showall(sqlite3 *ppDb) {

    char *errmsg = NULL;

    char sqlite[100] = "SELECT * FROM Work;";

    if (sqlite3_exec(ppDb, sqlite, callback, NULL, &errmsg) != SQLITE_OK) {

        printf("错误码：%d 错误信息：%s\n", sqlite3_errcode(ppDb), sqlite3_errmsg(ppDb));

        return -1;

    }

    return 0;

}

void menu(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);

    switch (ch) {

        case 1:

            do_insert(ppDb);

            break;

        case 2:

            do_delete(ppDb);

            break;

        case 3:

            do_update(ppDb);

            break;

        case 4:

            do_research(ppDb);

            break;

        case 5:

            do_showall(ppDb);

            break;

        case 0:

            exit(0);

        default:

            printf("输入有误，请重新输入\n");

    }

}

int main(int argc, const char *argv[]) {

    sqlite3 *ppDb;

    int k = sqlite3_open("./banzhang.db", &ppDb);

    if (k != SQLITE_OK) {

        printf("错误码：%d 错误信息：%s\n", sqlite3_errcode(ppDb), sqlite3_errmsg(ppDb));

        return -1;

    }

    printf("数据库打开成功\n");

    char *errmsg = NULL;

    char sqlite[100] = "CREATE TABLE IF NOT EXISTS Work (gh INT, name CHAR, salary DOUBLE);";

    if (sqlite3_exec(ppDb, sqlite, NULL, NULL, &errmsg) != SQLITE_OK) {

        printf("错误码：%d 错误信息：%s\n", sqlite3_errcode(ppDb), sqlite3_errmsg(ppDb));

        return -1;

    }

    while (1) {

        menu(ppDb);

    }

    sqlite3_close(ppDb);

    return 0;

}