实现一个简单的 mysql 工具

2023-10-20 18:21:54 浏览数 (1)

无论在 windows 下还是 linux 下,我们每次去连接 mysql 的时候都会运行一个叫做 mysql 的命令,本文就模仿制作一个类似的程序,实现可以在里面执行 DML 和 DQL 语句。具体代码的实现请参考程序。


代码实现

代码语言:javascript复制
#include <stdio.h>
#include <mysql.h>
#include <string.h>
#include <ctype.h>
int main(int argc, char* argv[])
{
unsigned int err;
MYSQL* mysql;
// 初始化 mysql 句柄
mysql = mysql_init(NULL);
if (mysql == NULL)
{
err = mysql_errno(mysql);
printf(“mysql_init error, code = %dn”, err);
return err;
}
// 连接服务器
mysql = mysql_real_connect(mysql, “127.0.0.1”, “root”, “******“, “mydb61”, 0, NULL, 0);
if (NULL == mysql)
{
err = mysql_errno(mysql);
printf(“mysql_real_connect error, code = %dn”, err);
return err;
}
char sql[1024];
while (1)
{
printf(“mysql> “);
fgets(sql, sizeof(sql), stdin);
sql[strlen(sql) - 1] = ‘’;
// 判断是否是输入了退出指令
if (strcmp(sql, “EXIT”) == 0  strcmp(sql, “QUIT”) == 0
strcmp(sql, “exit”) == 0  strcmp(sql, “quit”) == 0)
{
break;
}
// 判断是否是执行了 select 或 show 或 desc 开头的命令
if (strncmp(sql, “SELECT”, 6) == 0  strncmp(sql, “select”, 6) == 0
strncmp(sql, “SHOW”, 4) == 0  strncmp(sql, “show”, 4) == 0
strncmp(sql, “DESC”, 4) == 0  strncmp(sql, “desc”, 4) == 0)
{
// 执行 sql 语句,如果要查询包含二进制数据的,要用 mysql_real_query
err = mysql_query(mysql, sql);
if (err != 0)
{
err = mysql_errno(mysql);
printf(“mysql_query error, code = %dn”, err);
return err;
}
// 获取查询出来的数据有多少个列
unsigned int num, i = 0;
num = mysql_field_count(mysql);
// 获取结果集
MYSQL_RES* result;
result = mysql_store_result(mysql);
if (NULL == result)
{
err = mysql_errno(mysql);
printf(“mysql_query error, code = %dn”, err);
return err;
}
/*
与 mysql_field_count() 功能一致,但 mysql_field_count() 是返回上一次执行语句(query)结果列的数目
而 mysql_num_fields() 是从结果集中获取返回列数
*/
// 获取表头内容
MYSQL_FIELD *fields;
fields = mysql_fetch_fields(result);
for(i = 0; i < num; i  )
{
   printf(“%8st”, fields[i].name);
}
putchar(10);
// 读取结果集
MYSQL_ROW row;
while ((row = mysql_fetch_row(result)))
{
for (i = 0; i < num; i  )
{
printf(“%8st”, row[i]);
}
putchar(10);
}
// 释放结果集
mysql_free_result(result);
}
else
{
// 执行 DQL 语句
err = mysql_query(mysql, sql);
if (err != 0)
{
err = mysql_errno(mysql);
printf(“mysql_query error, code = %dn”, err);
return err;
}
my_ulonglong row = mysql_affected_rows(mysql);
printf(“Query OK, %d row affected”, row);
}
putchar(10);
}
// 关闭 mysql 句柄
mysql_close(mysql);
return 0;
}

编译程序

编译指令:gcc mysql_tool.c -o mysql_tool -I /usr/include/mysql -L /usr/lib64/mysql -l mysqlclient -l stdc -l pthread -l dl -l rt

执行效果

我们分别执行了两个 select 语句,可以正确的得到查询结果,并且执行了一个 update 语句,可以返回有多少行受影响了。这就是我们模仿的一个 mysql 客户端工具。

0 人点赞