AAMS/gettxt.cpp at master · zhangxy567/AAMS · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
#include "basis.h"
#include "func.h"

int gettxt()
{
    MYSQL *conn;
    MYSQL_RES *res;
    MYSQL_ROW row;
    unsigned int num_fields;
    unsigned int i;

    conn = mysql_init(NULL);
    // 连接数据库
    if (mysql_real_connect(conn, "localhost", "root", "root", "mysql", 3306, NULL, 0) == NULL)
    {
        fprintf(stderr, "%s\n", mysql_error(conn));
        mysql_close(conn);
        return 1;
    }


    // 执行 SQL 查询
    if (mysql_query(conn, "SELECT user FROM mysql.user;"))
    {
        fprintf(stderr, "%s\n", mysql_error(conn));
        mysql_close(conn);
        return 1;
    }

    // 获取结果集
    res = mysql_store_result(conn);
    num_fields = mysql_num_fields(res);


    FILE *file = fopen("user.txt", "w"); //this place need be same at main_func.cpp at 216 line
    if (file == NULL)
    {
        fprintf(stderr, "Could not open file for writing\n");
        mysql_free_result(res);
        mysql_close(conn);
        exit(1);
    }


    // 遍历每一行
    while ((row = mysql_fetch_row(res)) != NULL)
    {

        for(i = 0; i < num_fields; i++)
        {
            string str = row[i];
            string prefix2 = "2";
            string prefix3 = "1";
            fprintf(file, "%s ", row[i] ? row[i] : "NULL");
//1 to root
//2 to stu
//3 to teacher
            if(     strcmp(row[i],"root") == 0 )
            {
                fprintf(file, "1");
            }
            else if(  startsWith(str , prefix2) )
            {
                fprintf(file, "2");
            }
            else if(  startsWith(str , prefix3) )
            {
                fprintf(file, "3");

            }
            else
            {
                fprintf(file, "4");

            }


            fprintf(file, "\n");
        }

    }


    // 清理资源
    mysql_free_result(res);
    mysql_close(conn);
    fclose(file);
    return 0;
}