#ifndef FRIENDMODEL_H_
#define FRIENDMODEL_H_
#include<vector>
#include"user.hpp"
using namespace std;
//好友信息方法
class FriendModel{
public:
void insert(int userid,int friendid);
vector<User> query(int userid);
private:
};
#endif
friendmodel.cpp
#include"friendmodel.hpp"
#include"db.hpp"
void FriendModel::insert(int userid,int friendid){
char sql[128] = {0};
sprintf(sql,"insert into Friend(userid,friendid) values(%d,%d)",userid,friendid);
MySQL mysql;
if(mysql.connect()){
mysql.update(sql);
}
}
vector<User> FriendModel::query(int userid){
// 1.组装sql语句
char sql[256] = {0};
sprintf(sql, "select a.id,a.name,a.state from User a inner join Friend b on b.friendid = a.id where b.userid = %d", userid);
vector<User> vec;
MySQL mysql;
if (mysql.connect())
{
MYSQL_RES *res = mysql.query(sql);
if (res != nullptr)
{
MYSQL_ROW row;
while((row = mysql_fetch_row(res)) != nullptr){
User user;
user.setID(atoi(row[0]));
user.setname(row[1]);
user.setstate(row[2]);
vec.push_back(user);
}
mysql_free_result(res);
}
}
return vec;
}
chatservice.hpp 中添加部分
#include"friendmodel.hpp"
//添加好友
void addFriend(const TcpConnectionPtr &conn,json &js,Timestamp time);
FriendModel _friendmodel;
chatservice.cpp 部分
绑定自己绑
void ChatService::login(const TcpConnectionPtr &conn,json &js,Timestamp time){
int id = js["id"].get<int>();
string pwd = js["password"];
User user = _usermodel.query(id);
if (user.getID() == id && user.getpassword() == pwd)
{
if (user.getstate() == "online")
{
// 该用户已经登录，不允许重复登录
json response;
response["msgid"] = LOGIN_MSG_ACK;
response["errno"] = 2;
response["errmsg"] = "this account is using, input another!";
conn->send(response.dump());
}
else
{
//添加作用域，限制锁的粒度
{
lock_guard<mutex> lock(_connMutex);
//记录用户连接
_userConnMap.insert({id,conn});
}
// 登录成功，更新用户状态信息 state offline=>online
user.setstate("online");
_usermodel.updateState(user);
json response;
response["msgid"] = LOGIN_MSG_ACK;
response["errno"] = 0;
response["id"] = user.getID();
response["name"] = user.getname();
//查询用户是否有离线消息
vector<string> vecofflinemsg = _offlineMsgmodel.query(id);
if(!vecofflinemsg.empty()){
response["offlinemsg"] = vecofflinemsg;
//清空离线消息
_offlineMsgmodel.remove(id);
}
vector<User> uservec = _friendmodel.query(id);
if(!uservec.empty()){
vector<string> vecfriend;
for(User &user:uservec){
json js;
js["id"] = user.getID();
js["name"] = user.getname();
js["state"] = user.getstate();
最后
本人也收藏了一份Java面试核心知识点来应付面试，借着这次机会可以送给我的读者朋友们:
目录：
Java面试核心知识点
一共有30个专题，足够读者朋友们应付面试啦，也节省朋友们去到处搜刮资料自己整理的时间！
Java面试核心知识点
