玩家信息

dev/data/player.lua

@@ -0,0 +1,48 @@
+--[[
+Descripttion:玩家信息
+version:
+Author: Neo,Huang
+Date: 2021-09-15 19:47:54
+LastEditors: Neo,Huang
+LastEditTime: 2021-09-15 19:47:55
+--]]
+local code = require("code")
+local timeUtil = require("utils.timeUtil")
+local moduleData = require("data.module")
+
+local MODULE_NAME = "player"
+
+local root = {}
+
+-- 更新验证码
+function root:update_verify_code(uid)
+    if is_empty(uid) or is_robot(uid) then
+        return
+    end
+    local r = {}
+    for i = 1, 6 do
+        table.insert(r, string.char(math.random(48, 57)))
+    end
+    local vcode = table.concat(r)
+    local vcodeInfo = {vcode = vcode, timeout = timeUtil.now(uid) + 60}
+    moduleData:hset(uid, MODULE_NAME, "vcodeInfo", vcodeInfo)
+    return vcode
+end
+
+-- 获取验证码
+function root:get_verify_code(uid)
+    if is_empty(uid) or is_robot(uid) then
+        return
+    end
+    local vcodeInfo = moduleData:hget_json(uid, MODULE_NAME, "vcodeInfo")
+    if is_empty(vcodeInfo) or is_empty(vcodeInfo.vcode) then
+        return code.VERIFY_CODE.VERIFY_CODE
+    end
+    local currTime = timeUtil.now(uid)
+    if currTime > (vcodeInfo.timeout or 0) then
+        return code.VERIFY_CODE.TIME_OUT
+    end
+    return code.OK, vcodeInfo.vcode
+end
+
+return root

dev/data/user.lua

@@ -29,6 +29,9 @@ function root:user_init_register_info(uid, info)
     moduleData:hset(uid, MODULE_NAME, "deviceId", info.deviceId or "")
     moduleData:hset(uid, MODULE_NAME, "channel", info.channel)
 
+    moduleData:hset(uid, MODULE_NAME, "phone", info.phone)
+    moduleData:hset(uid, MODULE_NAME, "shareCode", info.shareCode)
+
     moduleData:hset(uid, MODULE_NAME, "nickname", info.nickname or string.format("玩家%06d", uid))
     -- 非测试环境随机密码
     local password = info.password
@@ -69,4 +72,21 @@ function root:get_status(uid)
     return moduleData:hget(uid, MODULE_NAME, "status")
 end
 
+-- 获取当前邀请码
+function root:get_share_code(uid)
+    if is_empty(uid) then
+        return
+    end
+    return moduleData:hget(uid, MODULE_NAME, "shareCode")
+end
+
+-- 绑定邀请码
+function root:band_share_code(uid, sharecode)
+    if is_empty(uid) then
+        return false
+    end
+    moduleData:hset(uid, MODULE_NAME, "bandShareCode", sharecode)
+    return true
+end
+
 return root

dev/modules/sharecode.lua

@@ -0,0 +1,16 @@
+-- 要求码
+local root = class("sharecode", require("base.baseModule"))
+
+function root:ctor(sharecode)
+    root.super.ctor(self, sharecode, "sharecode", "sharecode", true, false)
+    self.sharecode = sharecode
+end
+
+function root:mysql_get_init_columns()
+    return {
+        sharecode = "varchar(20) NOT NULL COMMENT '邀请码'",
+        uid = "bigint(20) unsigned NOT NULL COMMENT '用户ID'"
+    }
+end
+
+return root

dev/modules/user.lua

@@ -1,5 +1,8 @@
 local code = require("code")
 local util_player = require("utils.util_player")
+local util_global = require("utils.util_global")
+
+local playerData = require("data.player")
 
 local root = class("user", require("base.baseModule"))
 
@@ -30,7 +33,11 @@ function root:mysql_get_init_columns()
         registerTime = "int(11) DEFAULT 0 COMMENT '注册时间'",
         registerVersion = "varchar(45) DEFAULT NULL COMMENT '注册版本'",
         ip = "varchar(45) DEFAULT NULL COMMENT 'IP'",
-        status = "int(11) DEFAULT 0 COMMENT '账号状态 0:正常 1:封号 2:注销'"
+        status = "int(11) DEFAULT 0 COMMENT '账号状态 0:正常 1:封号 2:注销'",
+        phone = "varchar(45) DEFAULT NULL COMMENT '绑定手机号'",
+        bandShareCode = "varchar(45) DEFAULT NULL COMMENT '绑定推广码'",
+        shareCode = "varchar(45) DEFAULT NULL COMMENT '我的推广码'",
+        steamLink = "varchar(45) DEFAULT NULL COMMENT 'steam交易链接'"
     }
 end
 
@@ -40,7 +47,65 @@ end
 -- 获取自己的信息
 function root:itf_get_info(role, msg)
     local info = util_player:get_player_info(self.uid)
-    return code.OK, {baseInfo = info}
+    return code.OK, {playerInfo = info}
+end
+
+-- 获取验证码
+function root:itf_get_verify_code(role, msg)
+    local vcode = playerData:update_verify_code(self.uid)
+    local ret = {}
+    if IS_TEST then
+        ret.vcode = vcode
+    else
+        -- 发短信
+    end
+    return code.OK, ret
+end
+
+-- 更新主播邀请码
+function root:itf_update_band_share_code(role, msg)
+    local shareCode = msg.shareCode
+    if is_empty(shareCode) then
+        return code.PARAMTER_ERROR
+    end
+    if not util_global:is_sharecode_active(shareCode) then
+        -- 邀请码不存在
+        return code.SHARE_CODE.NOT_FOUND
+    end
+    self:redis_update_key_info("bandShareCode", shareCode)
+    return code.OK
+end
+
+-- 更新自己邀请码
+function root:itf_update_share_code(role, msg)
+    local shareCode = msg.shareCode
+    if is_empty(shareCode) then
+        shareCode = util_global:gen_share_code()
+    end
+    if not util_global:is_sharecode_active(shareCode) then
+        -- 邀请码已存在
+        return code.SHARE_CODE.EXIST
+    end
+    self:redis_update_key_info("shareCode", shareCode)
+    return code.OK
+end
+
+-- 更新steam交易链接
+function root:itf_update_steam_link(role, msg)
+    local link = msg.link
+    local vcode = msg.vcode
+    if is_empty(link) or is_empty(vcode) then
+        return code.PARAMTER_ERROR
+    end
+    local errcode, _vcode = playerData:get_verify_code(self.uid)
+    if code.is_not_ok(errcode) then
+        return errcode
+    end
+    if vcode ~= _vcode then
+        return code.VERIFY_CODE.NOT_MATCH
+    end
+    self:redis_update_key_info("steamLink", link)
+    return code.OK
 end
 
 return root

dev/utils/util_global.lua

@@ -1,5 +1,5 @@
 --[[
-Descripttion:玩家事件
+Descripttion:全局
 version:
 Author: Neo,Huang
 Date: 2022-07-05 19:59:16
@@ -7,6 +7,9 @@ LastEditors: Neo,Huang
 LastEditTime: 2022-07-05 20:08:31
 --]]
 local lib_game_redis = require("lib_game_redis")
+local lib_game_mysql = require("lib_game_mysql")
+
+local userData = require("data.user")
 
 local MODULE_NAME = "global"
 
@@ -30,4 +33,56 @@ function root:gen_mail_id()
     return id
 end
 
+-- 创建邀请码
+function root:gen_share_code()
+    local function _gen_code()
+        local r = {}
+        for i = 1, 5 do
+            table.insert(r, string.char(math.random(65, 90)))
+        end
+        return table.concat(r)
+    end
+    local sharecode = _gen_code()
+    while self:is_sharecode_active(sharecode) do
+        sharecode = _gen_code()
+    end
+    -- 绑定
+    lib_game_redis:sadd("sharecode", sharecode)
+
+    return sharecode
+end
+
+-- 邀请码是否使用
+function root:is_sharecode_active(sharecode)
+    if is_empty(sharecode) then
+        return false
+    end
+    local ret = lib_game_redis:sismerber("sharecode", sharecode)
+    return ret == 1
+end
+-- 删除邀请码
+function root:del_sharecode(sharecode)
+    local sql = string.format("delete from `mdl_sharecode` where `sharecode` = '%s'", tostring(sharecode))
+    lib_game_mysql:query(sql)
+    lib_game_redis:srem("sharecode", sharecode)
+end
+-- 更新邀请码
+function root:update_sharecode(uid, sharecode)
+    if is_empty(uid) or is_empty(sharecode) then
+        return false
+    end
+    -- 删除旧邀请码
+    local shareCode = userData:get_share_code(uid)
+    if not is_empty(shareCode) then
+        self:del_sharecode(shareCode)
+    end
+    -- 更新
+    lib_game_redis:sadd("sharecode", sharecode)
+
+    local values = string.format("sharecode='%s', uid=%s", tostring(sharecode), tostring(uid))
+    local sql = string.format("insert into mdl_sharecode set %s;", tostring(values))
+    lib_game_mysql:query(sql)
+    return true
+end
+
 return root

dev/utils/util_player.lua

@@ -24,4 +24,23 @@ function root:get_player_info(uid)
     return info
 end
 
+-- 玩家基础信息
+function root:get_base_info(uid)
+    local info = {
+        uid = uid,
+        nickname = moduleData:hget(uid, "user", "nickname"),
+        icon = moduleData:hget(uid, "user", "headUrl"),
+        level = moduleData:hget_int(uid, "player", "level"),
+        vipLevel = moduleData:hget_int(uid, "player", "vipLevel"),
+        registerVersion = moduleData:hget(uid, "user", "registerVersion"),
+        activeDays = moduleData:hget(uid, "player", "activeDays"),
+        phone = moduleData:hget(uid, "user", "phone"),
+        bandShareCode = moduleData:hget(uid, "user", "bandShareCode"),
+        shareCode = moduleData:hget(uid, "user", "shareCode"),
+        steamLink = moduleData:hget(uid, "user", "steamLink"),
+        isRealname = false
+    }
+    return info
+end
+
 return root

nodes/game/interface/user.lua

@@ -9,6 +9,10 @@ LastEditTime: 2022-07-05 20:29:29
 local code = require "code"
 local timeUtil = require("utils.timeUtil")
 local util_user = require("utils.util_user")
+local util_global = require("utils.util_global")
+
+local playerData = require("data.player")
+local userData = require("data.user")
 
 local root = {}
 
@@ -26,4 +30,24 @@ function root:after_user_self_info(role, msg)
     util_user:user_proto_notify(role.uid, "on_user_system_info", msg)
 end
 
+-- 获取验证码
+function root:user_get_verify_code(role, msg)
+    return role.user:itf_get_verify_code(role, msg)
+end
+
+-- 更新主播邀请码
+function root:user_update_band_share_code(role, msg)
+    return role.user:itf_update_band_share_code(role, msg)
+end
+
+-- 更新自己邀请码
+function root:user_update_share_code(role, msg)
+    return role.user:itf_update_share_code(role, msg)
+end
+
+-- 更新steam交易链接
+function root:user_update_steam_link(role, msg)
+    return role.user:itf_update_steam_link(role, msg)
+end
+
 return root

nodes/login/controllers/usr.lua

@@ -1,12 +1,13 @@
-local code = require "code"
-local tokenUtil = require "utils.tokenUtil"
+local code = require("code")
+local tokenUtil = require("utils.tokenUtil")
 local serverLogUtil = require("utils.serverLogUtil")
 local util_user = require("utils.util_user")
-local accountModule = require("modules.account")
 local util_global = require("utils.util_global")
-
 local lib_game_mysql = require("lib_game_mysql")
 local lib_game_redis = require("lib_game_redis")
+
+local accountModule = require("modules.account")
+
 local moduleData = require("data.module")
 local userData = require("data.user")
 
@@ -159,8 +160,18 @@ function root.usr_register_by_phone(msg)
     err = moduleData:hset(account, "account", "uid", uid)
     err = accountObj:backup_to_db()
 
+    -- 生产推广码
+    msg.shareCode = util_global:gen_share_code()
+    local values = string.format("sharecode='%s', uid=%s", tostring(msg.shareCode), tostring(uid))
+    local sql = string.format("insert into mdl_sharecode set %s;", tostring(values))
+    lib_game_mysql:query(sql)
+
     -- 正常注册
     userData:user_init_register_info(uid, msg)
+    -- 绑定主播邀请码
+    if not is_empty(msg.sharecode) and util_global:is_sharecode_active(msg.sharecode) then
+        userData:band_share_code(uid, msg.sharecode)
+    end
     -- 注册埋点
     serverLogUtil.logRegister(uid, msg.channel, msg.version, msg.sysVer or "", msg.uuid, msg.udid, msg.ip)