IM项目：进阶版即时通讯项目---用户管理(纯业务)

本篇重要是对于该项目的用户管理模块做一个文档阐明，前面已经封装好了类的操作，因此本篇开始直接举行使用
本节仍旧不算很难，重要表现在各个接口的实现，和之前对于一些API的调用逻辑
服务重写

对于用户管理这个模块来说，重要涉及到的有下面的这几个服务

• 用户名注册
  
• 用户名登录
  
• 验证码获取
  
• 手机号注册
  
• 手机号登录
  
• 个人信息获取(单个)
  
• 个人信息获取(多个)
  
• 用户头像修改
  
• 用户昵称修改
  
• 用户签名修改
  
• 用户手机修改
  

因此，本质上必要重写的就是这11个接口

1. service UserService {
2.     rpc UserRegister(UserRegisterReq) returns (UserRegisterRsp);
3.     rpc UserLogin(UserLoginReq) returns (UserLoginRsp);
4.     rpc GetPhoneVerifyCode(PhoneVerifyCodeReq) returns (PhoneVerifyCodeRsp);
5.     rpc PhoneRegister(PhoneRegisterReq) returns (PhoneRegisterRsp);
6.     rpc PhoneLogin(PhoneLoginReq) returns (PhoneLoginRsp);
7.     rpc GetUserInfo(GetUserInfoReq) returns (GetUserInfoRsp);
8.     rpc GetMultiUserInfo(GetMultiUserInfoReq) returns (GetMultiUserInfoRsp);
9.     rpc SetUserAvatar(SetUserAvatarReq) returns (SetUserAvatarRsp);
10.     rpc SetUserNickname(SetUserNicknameReq) returns (SetUserNicknameRsp);
11.     rpc SetUserDescription(SetUserDescriptionReq) returns (SetUserDescriptionRsp);
12.     rpc SetUserPhoneNumber(SetUserPhoneNumberReq) returns (SetUserPhoneNumberRsp);
13. }

复制代码

用户注册

对于用户注册的处理逻辑来说，基本逻辑非常简朴，只必要从请求中把昵称和暗码取出来，然后判定是否合法、判定昵称是否存在，之后向数据库中新增数据，向elasticsearch中新增数据字段，响应给客户端即可

1. // 完成用户注册的接口
2. virtual void UserRegister(::google::protobuf::RpcController* controller,
3.     const ::im::UserRegisterReq* request,
4.     ::im::UserRegisterRsp* response,
5.     ::google::protobuf::Closure* done)
6. {
7.     LOG_DEBUG("收到用户注册请求！");
8.     brpc::ClosureGuard rpc_guard(done);
9.     // 定义一个错误处理函数，当出错的时候被调用
10.     auto err_response = [this, response](const std::string &rid,
11.         const std::string &errmsg) -> void {
12.         response->set_request_id(rid);
13.         response->set_success(false);
14.         response->set_errmsg(errmsg);
15.         return;
16.     };
18.     // 1. 从请求中取出昵称和密码
19.     std::string nickname = request->nickname();
20.     std::string password = request->password();
22.     // 2. 检查昵称是否合法（只能包含字母，数字，连字符-，下划线_，长度限制 3~15 之间）
23.     bool ret = nickname_check(nickname);
24.     if (ret == false)
25.     {
26.         LOG_ERROR("{} - 用户名长度不合法！", request->request_id());
27.         return err_response(request->request_id(), "用户名长度不合法！");
28.     }
30.     // 3. 检查密码是否合法（只能包含字母，数字，长度限制 6~15 之间）
31.     ret = password_check(password);
32.     if (ret == false)
33.     {
34.         LOG_ERROR("{} - 密码格式不合法！", request->request_id());
35.         return err_response(request->request_id(), "密码格式不合法！");
36.     }
38.     // 4. 根据昵称在数据库进行判断是否昵称已存在
39.     auto user = _mysql_user->select_by_nickname(nickname);
40.     if (user)
41.     {
42.         LOG_ERROR("{} - 用户名被占用- {}！", request->request_id(), nickname);
43.         return err_response(request->request_id(), "用户名被占用!");
44.     }
46.     // 5. 向数据库新增数据
47.     std::string uid = uuid();
48.     user = std::make_shared<User>(uid, nickname, password);
49.     ret = _mysql_user->insert(user);
50.     if (ret == false)
51.     {
52.         LOG_ERROR("{} - Mysql数据库新增数据失败！", request->request_id());
53.         return err_response(request->request_id(), "Mysql数据库新增数据失败!");
54.     }
56.     // 6. 向 ES 服务器中新增用户信息
57.     ret = _es_user->appendData(uid, "", nickname, "", "");
58.     if (ret == false) {
59.         LOG_ERROR("{} - ES搜索引擎新增数据失败！", request->request_id());
60.         return err_response(request->request_id(), "ES搜索引擎新增数据失败！");
61.     }
63.     // 7. 组织响应，进行成功与否的响应即可。
64.     response->set_request_id(request->request_id());
65.     response->set_success(true);
66. }

复制代码

用户登录

用户登录的处理逻辑也比较简朴，在收到请求之后，提取出对应的字段，然后到数据库中来获取看暗码是否一致，之后到Redis中判定用户是否已经登录，假如没有登录，那么就参加信息即可

1. // 用户登录
2. virtual void UserLogin(::google::protobuf::RpcController* controller,
3.     const ::im::UserLoginReq* request,
4.     ::im::UserLoginRsp* response,
5.     ::google::protobuf::Closure* done)
6. {
7.     LOG_DEBUG("收到用户登录请求！");
8.     brpc::ClosureGuard rpc_guard(done);
9.     auto err_response = [this, response](const std::string &rid,
10.         const std::string &errmsg) -> void {
11.         response->set_request_id(rid);
12.         response->set_success(false);
13.         response->set_errmsg(errmsg);
14.         return;
15.     };
17.     // 1. 从请求中取出昵称和密码
18.     std::string nickname = request->nickname();
19.     std::string password = request->password();
21.     // 2. 通过昵称获取用户信息，进行密码是否一致的判断
22.     auto user = _mysql_user->select_by_nickname(nickname);
23.     if (!user || password != user->password())
24.     {
25.         LOG_ERROR("{} - 用户名或密码错误 - {}-{}！", request->request_id(), nickname, password);
26.         return err_response(request->request_id(), "用户名或密码错误!");
27.     }
29.     // 3. 根据 redis 中的登录标记信息是否存在判断用户是否已经登录。
30.     bool ret = _redis_status->exists(user->user_id());
31.     if (ret == true)
32.     {
33.         LOG_ERROR("{} - 用户已在其他地方登录 - {}！", request->request_id(), nickname);
34.         return err_response(request->request_id(), "用户已在其他地方登录!");
35.     }
37.     // 4. 构造会话 ID，生成会话键值对，向 redis 中添加会话信息以及登录标记信息
38.     std::string ssid = uuid();
39.     _redis_session->append(ssid, user->user_id());
41.     // 5. 组织响应，返回生成的会话 ID
42.     response->set_request_id(request->request_id());
43.     response->set_login_session_id(ssid);
44.     response->set_success(true);
45. }

复制代码

验证码获取

验证码服务获取就必要用到前面的关于验证码的SDK了，对于验证码的SDK服务来说，可以使用提前封装好的的内容，传入一个手机号，一个验证码就可以了，这时候就表现出了提前封装的好处

1. // 验证码获取服务
2. virtual void GetPhoneVerifyCode(::google::protobuf::RpcController* controller,
3.     const ::im::PhoneVerifyCodeReq* request,
4.     ::im::PhoneVerifyCodeRsp* response,
5.     ::google::protobuf::Closure* done)
6. {
7.     LOG_DEBUG("收到短信验证码获取请求！");
8.     brpc::ClosureGuard rpc_guard(done);
9.     auto err_response = [this, response](const std::string &rid,
10.         const std::string &errmsg) -> void {
11.         response->set_request_id(rid);
12.         response->set_success(false);
13.         response->set_errmsg(errmsg);
14.         return;
15.     };
17.     // 1. 从请求中取出手机号码
18.     std::string phone = request->phone_number();
20.     // 2. 验证手机号码格式是否正确（必须以 1 开始，第二位 3~9 之间，后边 9 个数字字符）
21.     bool ret = phone_check(phone);
22.     if (ret == false)
23.     {
24.         LOG_ERROR("{} - 手机号码格式错误 - {}！", request->request_id(), phone);
25.         return err_response(request->request_id(), "手机号码格式错误!");
26.     }
28.     // 3. 生成 4 位随机验证码
29.     std::string code_id = uuid();
30.     std::string code = vcode();
32.     // 4. 基于短信平台 SDK 发送验证码
33.     ret = _dms_client->send(phone, code);
34.     if (ret == false)
35.     {
36.         LOG_ERROR("{} - 短信验证码发送失败 - {}！", request->request_id(), phone);
37.         return err_response(request->request_id(), "短信验证码发送失败!");
38.     }
40.     // 5. 构造验证码 ID，添加到 redis 验证码映射键值索引中
41.     _redis_codes->append(code_id, code);
43.     // 6. 组织响应，返回生成的验证码 ID
44.     response->set_request_id(request->request_id());
45.     response->set_success(true);
46.     response->set_verify_code_id(code_id);
47.     LOG_DEBUG("获取短信验证码处理完成！");
48. }

复制代码

手机号注册

基本流程是，从请求中提取出手机号和验证码，判定手机号和验证码合法，然后在数据库中查询手机号是否已经被注册过，然后在MySQL和elasticsearch中新增用户信息即可，最后响应

1. // 手机号注册
2. virtual void PhoneRegister(::google::protobuf::RpcController* controller,
3.     const ::im::PhoneRegisterReq* request,
4.     ::im::PhoneRegisterRsp* response,
5.     ::google::protobuf::Closure* done)
6. {
7.     LOG_DEBUG("收到手机号注册请求！");
8.     brpc::ClosureGuard rpc_guard(done);
9.     auto err_response = [this, response](const std::string &rid,
10.         const std::string &errmsg) -> void {
11.         response->set_request_id(rid);
12.         response->set_success(false);
13.         response->set_errmsg(errmsg);
14.         return;
15.     };
17.     // 1. 从请求中取出手机号码和验证码,验证码ID
18.     std::string phone = request->phone_number();
19.     std::string code_id = request->verify_code_id();
20.     std::string code = request->verify_code();
22.     // 2. 检查注册手机号码是否合法
23.     bool ret = phone_check(phone);
24.     if (ret == false)
25.     {
26.         LOG_ERROR("{} - 手机号码格式错误 - {}！", request->request_id(), phone);
27.         return err_response(request->request_id(), "手机号码格式错误!");
28.     }
30.     // 3. 从 redis 数据库中进行验证码 ID-验证码一致性匹配
31.     auto vcode = _redis_codes->code(code_id);
32.     if (vcode != code)
33.     {
34.         LOG_ERROR("{} - 验证码错误 - {}-{}！", request->request_id(), code_id, code);
35.         return err_response(request->request_id(), "验证码错误!");
36.     }
38.     // 4. 通过数据库查询判断手机号是否已经注册过
39.     auto user = _mysql_user->select_by_phone(phone);
40.     if (user)
41.     {
42.         LOG_ERROR("{} - 该手机号已注册过用户 - {}！", request->request_id(), phone);
43.         return err_response(request->request_id(), "该手机号已注册过用户!");
44.     }
46.     // 5. 向数据库新增用户信息
47.     std::string uid = uuid();
48.     user = std::make_shared<User>(uid, phone);
49.     ret = _mysql_user->insert(user);
50.     if (ret == false)
51.     {
52.         LOG_ERROR("{} - 向数据库添加用户信息失败 - {}！", request->request_id(), phone);
53.         return err_response(request->request_id(), "向数据库添加用户信息失败!");
54.     }
56.     // 6. 向 ES 服务器中新增用户信息
57.     ret = _es_user->appendData(uid, phone, uid, "", "");
58.     if (ret == false)
59.     {
60.         LOG_ERROR("{} - ES搜索引擎新增数据失败！", request->request_id());
61.         return err_response(request->request_id(), "ES搜索引擎新增数据失败！");
62.     }
64.     // 7. 组织响应，进行成功与否的响应即可。
65.     response->set_request_id(request->request_id());
66.     response->set_success(true);
67. }

复制代码

手机号登录

基本逻辑是，提取信息，检查手机号在数据库中是否存在，然后从Redis中查看验证码，然后查看是否已经登录，然后分配一个会话ID，响应

1. // 手机号登录
2. virtual void PhoneLogin(::google::protobuf::RpcController* controller,
3.     const ::im::PhoneLoginReq* request,
4.     ::im::PhoneLoginRsp* response,
5.     ::google::protobuf::Closure* done)
6. {
7.     LOG_DEBUG("收到手机号登录请求！");
8.     brpc::ClosureGuard rpc_guard(done);
9.     auto err_response = [this, response](const std::string &rid,
10.         const std::string &errmsg) -> void {
11.         response->set_request_id(rid);
12.         response->set_success(false);
13.         response->set_errmsg(errmsg);
14.         return;
15.     };
17.     // 1. 从请求中取出手机号码和验证码 ID，以及验证码。
18.     std::string phone = request->phone_number();
19.     std::string code_id = request->verify_code_id();
20.     std::string code = request->verify_code();
22.     // 2. 检查注册手机号码是否合法
23.     bool ret = phone_check(phone);
24.     if (ret == false)
25.     {
26.         LOG_ERROR("{} - 手机号码格式错误 - {}！", request->request_id(), phone);
27.         return err_response(request->request_id(), "手机号码格式错误!");
28.     }
30.     // 3. 根据手机号从数据数据进行用户信息查询，判断用用户是否存在
31.     auto user = _mysql_user->select_by_phone(phone);
32.     if (!user)
33.     {
34.         LOG_ERROR("{} - 该手机号未注册用户 - {}！", request->request_id(), phone);
35.         return err_response(request->request_id(), "该手机号未注册用户!");
36.     }
38.     // 4. 从 redis 数据库中进行验证码 ID-验证码一致性匹配
39.     auto vcode = _redis_codes->code(code_id);
40.     if (vcode != code)
41.     {
42.         LOG_ERROR("{} - 验证码错误 - {}-{}！", request->request_id(), code_id, code);
43.         return err_response(request->request_id(), "验证码错误!");
44.     }
45.     _redis_codes->remove(code_id);
47.     // 5. 根据 redis 中的登录标记信息是否存在判断用户是否已经登录。
48.     ret = _redis_status->exists(user->user_id());
49.     if (ret == true)
50.     {
51.         LOG_ERROR("{} - 用户已在其他地方登录 - {}！", request->request_id(), phone);
52.         return err_response(request->request_id(), "用户已在其他地方登录!");
53.     }
55.     // 6. 构造会话 ID，生成会话键值对，向 redis 中添加会话信息以及登录标记信息
56.     std::string ssid = uuid();
57.     _redis_session->append(ssid, user->user_id());
59.     // 7. 组织响应，返回生成的会话 ID
60.     response->set_request_id(request->request_id());
61.     response->set_login_session_id(ssid);
62.     response->set_success(true);
63. }

复制代码

获取用户信息

这个调用逻辑相比较前面而言复杂一点，涉及到了从文件模块举行下载文件的过程，基本的流程是，根据要查询的这个ID信息，然后去到MySQL数据库中匹配查找，将查找到的结果填充到信息字段当中，之后再根据对应的ID信息去到文件存储模块中下载对应的头像信息的文件，最后团体举行返回

1. // 获取用户信息
2. virtual void GetUserInfo(::google::protobuf::RpcController* controller,
3.     const ::im::GetUserInfoReq* request,
4.     ::im::GetUserInfoRsp* response,
5.     ::google::protobuf::Closure* done)
6. {
7.     LOG_DEBUG("收到获取单个用户信息请求！");
8.     brpc::ClosureGuard rpc_guard(done);
9.     auto err_response = [this, response](const std::string &rid,
10.         const std::string &errmsg) -> void {
11.         response->set_request_id(rid);
12.         response->set_success(false);
13.         response->set_errmsg(errmsg);
14.         return;
15.     };
17.     // 1. 从请求中取出用户 ID
18.     std::string uid = request->user_id();
20.     // 2. 通过用户 ID，从数据库中查询用户信息
21.     auto user = _mysql_user->select_by_id(uid);
22.     if (!user)
23.     {
24.         LOG_ERROR("{} - 未找到用户信息 - {}！", request->request_id(), uid);
25.         return err_response(request->request_id(), "未找到用户信息!");
26.     }
28.     // 3. 根据用户信息中的头像 ID，从文件服务器获取头像文件数据，组织完整用户信息
29.     UserInfo *user_info = response->mutable_user_info();
30.     user_info->set_user_id(user->user_id());
31.     user_info->set_nickname(user->nickname());
32.     user_info->set_description(user->description());
33.     user_info->set_phone(user->phone());
34.    
35.     // 如果用户有头像信息
36.     if (!user->avatar_id().empty())
37.     {
38.         // 从信道管理对象中，获取到连接了文件管理子服务的channel
39.         auto channel = _mm_channels->choose(_file_service_name);
40.         if (!channel)
41.         {
42.             LOG_ERROR("{} - 未找到文件管理子服务节点 - {} - {}！",
43.                 request->request_id(), _file_service_name, uid);
44.             return err_response(request->request_id(), "未找到文件管理子服务节点!");
45.         }
46.         // 进行文件子服务的rpc请求，进行头像文件下载
47.         im::FileService_Stub stub(channel.get());
48.         im::GetSingleFileReq req;
49.         im::GetSingleFileRsp rsp;
50.         req.set_request_id(request->request_id());
51.         req.set_file_id(user->avatar_id());
52.         brpc::Controller cntl;
53.         stub.GetSingleFile(&cntl, &req, &rsp, nullptr);
54.         if (cntl.Failed() == true || rsp.success() == false)
55.         {
56.             LOG_ERROR("{} - 文件子服务调用失败：{}！", request->request_id(), cntl.ErrorText());
57.             return err_response(request->request_id(), "文件子服务调用失败!");
58.         }
59.         user_info->set_avatar(rsp.file_data().file_content());
60.     }
61.     // 4. 组织响应，返回用户信息
62.     response->set_request_id(request->request_id());
63.     response->set_success(true);
64. }

复制代码

获取多个用户的信息

这个其实和前面的情况非常类似，只不外区别是必要使用一个map来举行管理映射，这点在proto文件中就可以提前举行定义

1. message GetMultiUserInfoRsp {
2.     string request_id = 1;
3.     bool success = 2;
4.     string errmsg = 3;
5.     map<string, UserInfo> users_info = 4;
6. }

复制代码

1. // 获取多个用户的信息
2. virtual void GetMultiUserInfo(::google::protobuf::RpcController* controller,
3.     const ::im::GetMultiUserInfoReq* request,
4.     ::im::GetMultiUserInfoRsp* response,
5.     ::google::protobuf::Closure* done)
6. {
7.     LOG_DEBUG("收到批量用户信息获取请求！");
8.     brpc::ClosureGuard rpc_guard(done);
9.     // 1. 定义错误回调
10.     auto err_response = [this, response](const std::string &rid,
11.         const std::string &errmsg) -> void {
12.         response->set_request_id(rid);
13.         response->set_success(false);
14.         response->set_errmsg(errmsg);
15.         return;
16.     };
18.     // 2. 从请求中取出用户ID --- 列表
19.     std::vector<std::string> uid_lists;
20.     for (int i = 0; i < request->users_id_size(); i++)
21.     {
22.         uid_lists.push_back(request->users_id(i));
23.     }
25.     // 3. 从数据库进行批量用户信息查询
26.     auto users = _mysql_user->select_multi_users(uid_lists);
27.     if (users.size() != request->users_id_size())
28.     {
29.         LOG_ERROR("{} - 从数据库查找的用户信息数量不一致 {}-{}！",
30.             request->request_id(), request->users_id_size(), users.size());
31.         return err_response(request->request_id(), "从数据库查找的用户信息数量不一致!");
32.     }
34.     // 4. 批量从文件管理子服务进行文件下载
35.     auto channel = _mm_channels->choose(_file_service_name);
36.     if (!channel)
37.     {
38.         LOG_ERROR("{} - 未找到文件管理子服务节点 - {}！", request->request_id(), _file_service_name);
39.         return err_response(request->request_id(), "未找到文件管理子服务节点!");
40.     }
41.     im::FileService_Stub stub(channel.get());
42.     im::GetMultiFileReq req;
43.     im::GetMultiFileRsp rsp;
44.     req.set_request_id(request->request_id());
45.     for (auto &user : users)
46.     {
47.         if (user.avatar_id().empty())
48.             continue;
49.         req.add_file_id_list(user.avatar_id());
50.     }
51.     brpc::Controller cntl;
52.     stub.GetMultiFile(&cntl, &req, &rsp, nullptr);
53.     if (cntl.Failed() == true || rsp.success() == false)
54.     {
55.         LOG_ERROR("{} - 文件子服务调用失败：{} - {}！", request->request_id(),
56.             _file_service_name, cntl.ErrorText());
57.         return err_response(request->request_id(), "文件子服务调用失败!");
58.     }
60.     // 5. 组织响应（）
61.     for (auto &user : users)
62.     {
63.         auto user_map = response->mutable_users_info();//本次请求要响应的用户信息map
64.         auto file_map = rsp.mutable_file_data(); //这是批量文件请求响应中的map
65.         UserInfo user_info;
66.         user_info.set_user_id(user.user_id());
67.         user_info.set_nickname(user.nickname());
68.         user_info.set_description(user.description());
69.         user_info.set_phone(user.phone());
70.         user_info.set_avatar((*file_map)[user.avatar_id()].file_content());
71.         (*user_map)[user_info.user_id()] = user_info;
72.     }
74.     response->set_request_id(request->request_id());
75.     response->set_success(true);
76. }

复制代码

设置用户头像

这个调用逻辑也比较简朴，和用户注册比较像
添加人头像-提取信息-判定-添加数据库-添加elasticsearch-响应

1. // 设置用户头像
2. virtual void SetUserAvatar(::google::protobuf::RpcController* controller,
3.     const ::im::SetUserAvatarReq* request,
4.     ::im::SetUserAvatarRsp* response,
5.     ::google::protobuf::Closure* done)
6. {
7.     LOG_DEBUG("收到用户头像设置请求！");
8.     brpc::ClosureGuard rpc_guard(done);
9.     auto err_response = [this, response](const std::string &rid,
10.         const std::string &errmsg) -> void {
11.         response->set_request_id(rid);
12.         response->set_success(false);
13.         response->set_errmsg(errmsg);
14.         return;
15.     };
17.     // 1. 从请求中取出用户 ID 与头像数据
18.     std::string uid = request->user_id();
20.     // 2. 从数据库通过用户 ID 进行用户信息查询，判断用户是否存在
21.     auto user = _mysql_user->select_by_id(uid);
22.     if (!user)
23.     {
24.         LOG_ERROR("{} - 未找到用户信息 - {}！", request->request_id(), uid);
25.         return err_response(request->request_id(), "未找到用户信息!");
26.     }
28.     // 3. 上传头像文件到文件子服务，
29.     auto channel = _mm_channels->choose(_file_service_name);
30.     if (!channel)
31.     {
32.         LOG_ERROR("{} - 未找到文件管理子服务节点 - {}！", request->request_id(), _file_service_name);
33.         return err_response(request->request_id(), "未找到文件管理子服务节点!");
34.     }
35.     im::FileService_Stub stub(channel.get());
36.     im::PutSingleFileReq req;
37.     im::PutSingleFileRsp rsp;
38.     req.set_request_id(request->request_id());
39.     req.mutable_file_data()->set_file_name("");
40.     req.mutable_file_data()->set_file_size(request->avatar().size());
41.     req.mutable_file_data()->set_file_content(request->avatar());
42.     brpc::Controller cntl;
43.     stub.PutSingleFile(&cntl, &req, &rsp, nullptr);
44.     if (cntl.Failed() == true || rsp.success() == false)
45.     {
46.         LOG_ERROR("{} - 文件子服务调用失败：{}！", request->request_id(), cntl.ErrorText());
47.         return err_response(request->request_id(), "文件子服务调用失败!");
48.     }
49.     std::string avatar_id = rsp.file_info().file_id();
51.     // 4. 将返回的头像文件 ID 更新到数据库中
52.     user->avatar_id(avatar_id);
53.     bool ret = _mysql_user->update(user);
54.     if (ret == false)
55.     {
56.         LOG_ERROR("{} - 更新数据库用户头像ID失败 ：{}！", request->request_id(), avatar_id);
57.         return err_response(request->request_id(), "更新数据库用户头像ID失败!");
58.     }
60.     // 5. 更新 ES 服务器中用户信息
61.     ret = _es_user->appendData(user->user_id(), user->phone(),
62.         user->nickname(), user->description(), user->avatar_id());
63.     if (ret == false)
64.     {
65.         LOG_ERROR("{} - 更新搜索引擎用户头像ID失败 ：{}！", request->request_id(), avatar_id);
66.         return err_response(request->request_id(), "更新搜索引擎用户头像ID失败!");
67.     }
69.     // 6. 组织响应，返回更新成功与否
70.     response->set_request_id(request->request_id());
71.     response->set_success(true);
72. }

复制代码

添加用户昵称

这个调用逻辑也比较简朴，和用户注册比较像
添加用户昵称-提取信息-判定-添加数据库-添加elasticsearch-响应

1. // 设置用户昵称
2. virtual void SetUserNickname(::google::protobuf::RpcController* controller,
3.     const ::im::SetUserNicknameReq* request,
4.     ::im::SetUserNicknameRsp* response,
5.     ::google::protobuf::Closure* done)
6. {
7.     LOG_DEBUG("收到用户昵称设置请求！");
8.     brpc::ClosureGuard rpc_guard(done);
9.     auto err_response = [this, response](const std::string &rid,
10.         const std::string &errmsg) -> void {
11.         response->set_request_id(rid);
12.         response->set_success(false);
13.         response->set_errmsg(errmsg);
14.         return;
15.     };
17.     // 1. 从请求中取出用户 ID 与新的昵称
18.     std::string uid = request->user_id();
19.     std::string new_nickname = request->nickname();
21.     // 2. 判断昵称格式是否正确
22.     bool ret = nickname_check(new_nickname);
23.     if (ret == false)
24.     {
25.         LOG_ERROR("{} - 用户名长度不合法！", request->request_id());
26.         return err_response(request->request_id(), "用户名长度不合法！");
27.     }
29.     // 3. 从数据库通过用户 ID 进行用户信息查询，判断用户是否存在
30.     auto user = _mysql_user->select_by_id(uid);
31.     if (!user)
32.     {
33.         LOG_ERROR("{} - 未找到用户信息 - {}！", request->request_id(), uid);
34.         return err_response(request->request_id(), "未找到用户信息!");
35.     }
37.     // 4. 将新的昵称更新到数据库中
38.     user->nickname(new_nickname);
39.     ret = _mysql_user->update(user);
40.     if (ret == false)
41.     {
42.         LOG_ERROR("{} - 更新数据库用户昵称失败 ：{}！", request->request_id(), new_nickname);
43.         return err_response(request->request_id(), "更新数据库用户昵称失败!");
44.     }
46.     // 5. 更新 ES 服务器中用户信息
47.     ret = _es_user->appendData(user->user_id(), user->phone(),
48.         user->nickname(), user->description(), user->avatar_id());
49.     if (ret == false)
50.     {
51.         LOG_ERROR("{} - 更新搜索引擎用户昵称失败 ：{}！", request->request_id(), new_nickname);
52.         return err_response(request->request_id(), "更新搜索引擎用户昵称失败!");
53.     }
55.     // 6. 组织响应，返回更新成功与否
56.     response->set_request_id(request->request_id());
57.     response->set_success(true);
58. }

复制代码

设置用户签名

这个和前面基本一样，这里就不写了
设置手机号

这个和前面基本一样，这里就不写了
服务搭建

这里和前面的头脑一样，都是用一个类来控礼服务的搭建，考虑到篇幅原因，这里就不写代码了，大概就是下面两个类

1. class UserServer;
2. class UserServerBuilder;

复制代码

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。

继续阅读请点击广告