box-server/dev/base/baseModule.lua

--[[
Descripttion:基础模块类
version:
Author: Neo,Huang
Date: 2022-07-04 19:07:54
LastEditors: Neo,Huang
LastEditTime: 2022-07-04 19:13:19
--]]
local lib_game_redis = require("lib_game_redis")
local lib_game_mysql = require("lib_game_mysql")
local redisUtil = require("utils.redisUtil")
local sqlUtil = require("utils.sqlUtil")

local root = class("base")

function root:ctor(uid, mdlName, keyName, isSaveDB, isPersonal)
    self.uid = uid
    self.mdlName = mdlName
    self.keyName = keyName
    self.isSaveDB = isSaveDB
    self.isPersonal = isPersonal or true
end

-- 模块sql属性
function root:mysql_get_init_columns()
    return {}
end

-- 初始化模块表
function root:mysql_get_table_info()
    return {
        mdlName = self.mdlName,
        keyName = self.keyName,
        isSaveDB = self.isSaveDB,
        columnNameOptions = self:mysql_get_init_columns()
    }
end

function root:getKeyValue()
    return self[self.keyName]
end
-- 获取模块key
function root:get_module_redis_key()
    local moduleKey = string.format("mdl:%s:%s", self.mdlName, tostring(self:getKeyValue()))
    return moduleKey
end
-- 模块数据库表名
function root:get_table_name()
    return string.format("mdl_%s", tostring(self.mdlName))
end

-- 获取玩家模块数据
function root:get_data_from_db()
    local moduleKey = self:get_module_redis_key()
    local res = redisUtil.hgetall(moduleKey)
    if not is_empty(res) then
        return res
    end

    if self.isSaveDB then
        local sql = sqlUtil.select_table_by_key(self:get_table_name(), self.keyName, self:getKeyValue())
        res = lib_game_mysql:query(sql)
        log.info("get_data_from_db sql[%s] res[%s]", tostring(sql), tostring(res))
        if res and res[1] then
            -- add to redis
            redisUtil.hmset(moduleKey, res[1])
            -- get again
            res = redisUtil.hgetall(moduleKey)
            if not is_empty(res) then
                return res
            end
        end
        return res
    end
end

-- 字段类型 - json
function root:is_json(column)
    return string.match(string.upper(column), "JSON")
end
-- 字段类型 - int
function root:is_number(column)
    return string.match(string.upper(column), "INT")
end

-- 字段值
function root:get_column_db_value(column)
    if is_empty(column) then
        return
    end
    local key = self:get_module_redis_key()
    local val = lib_game_redis:hget(key, column)
    log.info("get_column_db_value key[%s] column[%s] val[%s]", tostring(key), tostring(column), tostring(val))
    if self:is_json(column) then
        val = string.format("'%s'", cjson_encode(val))
    elseif self:is_number(column) then
        val = tonumber(val or 0)
    else
        val = string.format("'%s'", val)
    end
    return val
end

-- 备份模块数据
function root:backup_to_db()
    local options = self:mysql_get_init_columns()
    if is_empty(options) then
        return false
    end
    local colums = {}
    local values = {}
    for k, v in pairs(options) do
        table.insert(colums, k)
        table.include(values, self:get_column_db_value(k))
    end

    local tbName = self:get_table_name()
    local sql =
        string.format("replace into %s (%s) value (%s);", tbName, table.concat(colums, ","), table.concat(values, ","))
    local ret = lib_game_mysql:query(sql)
    if not is_empty(ret) then
        log.info("backup_to_db sql[%s] ret[%s]", tostring(sql), tostring(ret))
    end
    return ret
end
-- 删除模块热数据
function root:del_hot_data()
    local key = self:get_module_redis_key()
    local ret = lib_game_redis:del(key)
    log.info("del_hot_data key[%s] ret[%s]", tostring(key), tostring(ret))
end

----------------------------------------
-- 实时数据
----------------------------------------
local moduleData = require("data.module")
-- 获取数据类型
function root:get_sub_key_type(subKey)
    if subKey == nil then
        return
    end

    local mpOptions = self:mysql_get_init_columns()
    for k, v in pairs(mpOptions) do
        if tostring(k) == tostring(subKey) then
            if self:is_number(v) then
                return "int"
            end
            if self:is_json(v) then
                return "json"
            end
            return "string"
        end
    end
end
-- 获取数据
function root:redis_get_key_info(subKey, uid)
    if subKey == nil then
        return
    end
    uid = uid or self.uid
    local info = moduleData:hget(uid, self.mdlName, subKey)
    local tyKey = self:get_sub_key_type(subKey)
    if tyKey then
        if tyKey == "json" then
            if info then
                info = cjson_decode(info)
            end
            -- 主要针对C++中的NULL
            if is_empty(info) then
                info = {}
            end
        end
        if tyKey == "int" then
            info = tonumber(info or 0)
        end
    end
    return info
end
-- 设置数据
function root:redis_update_key_info(subKey, info, uid)
    uid = uid or self.uid
    return moduleData:hset(uid, self.mdlName, subKey, info)
end

-- +1
function root:redis_hincrby(subKey, uid)
    uid = uid or self.uid
    return moduleData:hincrby(uid, self.mdlName, subKey)
end

return root