C++ Mysql对接实现登录注册

C++ Mysql对接、简单实现登录注册

明镜止水
环境：Ubuntu20.04、C++、Mysql8.0

---

---

表结构：

由于打算集成到自己开发Cat服务器程序中去，为提高性能，后续打算引入Mysql连接池的概念，故使用类来封装。

Cat_Mysql类的设计

//Mysql相关类放这

#ifndef _CAT_MYSQL__H_
#define _CAT_MYSQL__H_

#include <iostream>
#include <mysql.h>
#include <iomanip>
#include <string>

class Cat_Mysql
{
private:
    const char*    m_host; // 主机地址
    const char*    m_username; // 用户名
    const char*    m_password; // 密码
    const char*    m_database; // 连接的数据库名称database
    unsigned int   m_port; // Mysql端口号，默认3306
    const char*    m_unix_socket; // unix连接标识，一般默认为NULL
    unsigned long  m_clientflag; // 客户端连接标志，一般默认为0

private:
    MYSQL m_Mysql; // Mysql连接句柄
    MYSQL_RES* m_SelectResult = NULL; // 用于存储Mysql查询结果集合

public:
//构造函数
    Cat_Mysql();
//析构函数
    ~Cat_Mysql();

public:
//初始化数据库
    bool my_Init();
//释放数据库资源
    void my_FreeConnect();
//sql语句执行
    bool my_RunSql(const char* sql);
//sql语句查询
    bool my_SelectData(const char* sql);
//打印结果
    void my_PrintData();

//具体功能函数放这
public:
//注册验证
    bool my_Register(std::string& un, std::string& pw);
//登录验证
    bool my_Login(std::string& un, std::string& pw);
};

#endif

Cat_Mysql类成员函数的实现

• 这里暂时先在构造函数中直接写数据库信息，后面迁移到服务器程序的时候，可以把数据库相关信息写到配置文件中去，再调个文件读写对象的成员函数来读取就行；
• 为了防止Mysql连接资源忘记释放，干脆把释放函数写到析构函数中去，对象销毁时，会自动释放；对于每一步操作可能会失败的情况，尽量打印出来，方便排错，后续再把日志系统整合进去；
• 几个功能函数暂时留着，等后面开发需求都清楚后，一些可能重复用的功能可以单独写进去，比如sql查询、sql执行等，尽量简洁会好点。

//Mysql相关功能实现函数放这

#include <iostream>
#include <mysql.h>
#include <vector>
#include <string>

#include "cat_mysql.h"

using namespace std;

//构造函数
Cat_Mysql::Cat_Mysql()
{
    m_host = "localhost";
	m_username = "root";
	m_password = "123456";
	m_database = "qiye";
	m_port = 3306;
	m_unix_socket = NULL;
	m_clientflag = 0;

	my_Init();
}

//析构函数
Cat_Mysql::~Cat_Mysql()
{
	//释放Mysql资源
	my_FreeConnect();
}

//初始化数据库
bool Cat_Mysql::my_Init()
{
    // Mysql初始化函数，返回Mysql句柄
	if (!mysql_init(&m_Mysql))
	{
		printf("Mysql初始化失败!\n");
		return false;
	}

	// 连接Mysql数据库，检测是否连接成功，成功返回连接句柄,失败返回 NULL
	if (!mysql_real_connect(&m_Mysql, m_host, m_username, m_password,
		m_database, m_port, m_unix_socket, m_clientflag)) 
    {
		printf("数据库连接失败!\n");
		return false;
	}

	// 连接成功
	return true;
}

//提交注册信息
bool Cat_Mysql::my_Register(std::string& un, std::string& pw)
{
	string sql = "SELECT * FROM user";
	//查询，判断是否已有该用户名
    if (mysql_query(&m_Mysql, sql.c_str()))
	{
		printf("查询失败!\n");
		return false;
	}
	m_SelectResult = mysql_store_result(&m_Mysql);
	//获取行，row[1]表示第二个字段username，拿它判断就行
	MYSQL_ROW row;
	while (row = mysql_fetch_row(m_SelectResult))
	{
		//要依据表的结构字段来使用，不然查询很麻烦，简单写的话，账号和密码是同一行就是正常登录
		if (row[1] == un) {
			cout << "该用户被注册!" << endl;
			return false;
		}
	}

	//正式注册
	sql = "INSERT INTO user(username, password) VALUES(\""+ un + "\", \"" + pw + "\")";
	cout << "执行sql语句：" << sql << endl;
    // mysql_query数据库操作函数，传入SQL语句执行。成功返回0
	if (mysql_query(&m_Mysql, sql.c_str()))
	{
		printf("sql语句执行失败!\n");
		return false;
	}

	printf("sql语句执行成功!\n");
	return true;
}

//提交登录信息
bool Cat_Mysql::my_Login(std::string& un, std::string& pw)
{
	string sql = "SELECT * FROM user";
	cout << "执行sql语句：" << sql << endl;

	//查询
    if (mysql_query(&m_Mysql, sql.c_str()))
	{
		printf("查询失败!\n");
		return false;
	}
 
	//获取结果集
	m_SelectResult = mysql_store_result(&m_Mysql);
	if (!m_SelectResult) {
		printf("没有该用户\n");
		return false;
	}

	//获取行
	MYSQL_ROW row;
	while (row = mysql_fetch_row(m_SelectResult))
	{
		//要依据表的结构字段来使用，不然查询很麻烦，简单写的话，账号和密码是同一行就是正常登录
		cout << "row[1]: " << row[1] << " " << "row[2]: " << row[2] <<endl;
		if (row[1] == un && row[2] == pw) return true;
	}

	cout << "账号或是密码错误!" << endl;
	return false;
}

//释放
void Cat_Mysql::my_FreeConnect() 
{
	cout << "====================================" << endl;
	printf("开始回收Mysql连接!\n");
	// 释放存储的查询结果集合
	mysql_free_result(m_SelectResult);
	// 释放Mysql连接资源
	mysql_close(&m_Mysql);
    return ;
}

//==================================================================
//以下后面根据具体功能再来实现
//执行sql语句
bool Cat_Mysql::my_RunSql(const char* sql){}
//查询sql数据
bool Cat_Mysql::my_SelectData(const char* sql){}
//打印信息
void Cat_Mysql::my_PrintData(){}

主函数main

主函数先这样写，后续打算用QT写界面，就不用这么麻烦了，先写两个函数去调Cat_Mysql的成员函数

//程序入口

#include <iostream>
#include <fstream>
#include <string>
#include <time.h>

#include "cat_mysql.h"

using namespace std;

Cat_Mysql my;

int main()
{
    cout << "====================================" << endl;
    //islogin=true代表登录，flag=0代表退出
    int flag = 1;
    bool islogin = false;
    while (flag)
    {
        int index;
        cout << "1. 注册\n2. 登录\n其余为退出\n请输入要执行的操作：";
        cin >> index;
        switch (index)
        {
        case 1:
            cat_Register();
            break;
        
        case 2:
            if (cat_Login())
            {
                islogin = true;
                flag = 0;
            }
            break;

        default:
            flag = 0;
            break;
        } 
    }

    if (islogin)
        cout << "已登陆!!!" << endl;
    else
        cout << "啥也没发生!" << endl;
    return 0;
}

//注册
bool cat_Register()
{
    string username, password;
    cout << "注册时，请不要有空格和特殊符号！" << endl;

    cout << "select a username: " ; cin >> username;
    cout << "select a password: " ; cin >> password;

    if (my.my_Register(username, password))
    {
        printf("注册成功！\n");
    }
    else
    {
        printf("注册失败！\n");
        return false;
    }

    return true;
}

//登录
bool cat_Login()
{
    string username, password, un, pw;
    //正常流程的登录
    cout << "登陆时，请不要有空格和特殊符号！" << endl;
    cout << "Enter username: "; cin >> username;
    cout << "Enter password: "; cin >> password;

    if(my.my_Login(username, password))
    {
        printf("登录成功！\n");
        return true;
    }
    
    return false;
}

编译运行和效果

g++ app/cat_mysql.cpp app/main.cpp -I/usr/include/mysql -lmysqlclient -I include -o f

./f

登录

注册