GongFuServer/script/merge/MergeServerLogic.lua

--合服逻辑

--[=[
    1.备份数据库, 方法见 MergeServerDefine
    2.先执行一次start.sh, 因为重启游戏服时会根据条件删除一些数据, 这样合服时要处理的数据就少了一些
    3.将/script 目录下 Main文件改名为其他名字, MainMerger改为Main，合服完毕后再改回来。
    4.检查MergeServerDefine中的MERGE_DB_TB数据库名是否正确, 如果正确则和正常启动游戏一样启动游戏服, 开始进行合服
    5.查看日志oss_merge, 看看是否有报错
    6.修改mysql区服列表的 port 和 dbName, 被合服的 port 和 dbname 要改为目标服一样
    7.修改linux上代表游戏服的bin*文件名, 防止 start.sh 启动时会把被合服也启动起来

    8.需要测试下邮件和cdk
]=]



local LuaMongo = _G.lua_mongo
local Config = require("Config")
local Log = require("common.Log")
local MergeServerDefine = require("merge.MergeServerDefine")
local MoZhuExcel = require("excel.mozhu")
local MailExcel = require("excel.mail")
local Util = require("common.Util")
local ItemDefine = require("bag.ItemDefine")

local primaryKeyTable = {}

--重复_id缓存表
local repeatIdTb = {}

--目标服和被合服所有玩家的名字缓存表
local allNameTb = {}

--同名数据
local repeatName2Data = {}

local num = 0

--次元魔珠数据缓存表
local mozhuDbCache = { roleList = {}, unionList = {}}

--创建一个mongo连接实例
local function clientMongoDb(addr)
    addr = addr or Config.DB_IP
    LuaMongo.client(addr)
end

--切换数据库
local function chooseMongoDb(dbName)
    if not dbName or dbName == "" then
        print(string.format("数据库名错误, dbName = %s\n", dbName))
        return
    end
	LuaMongo.auth(dbName, Config.DB_USER, Config.DB_PASS)
end

--生成新的 _id
local function generateNewId()
    local newId = LuaMongo.id()
    return newId
end

--清理部分缓存数据
local function cleanCache()
    primaryKeyTable = {}
    repeatIdTb = {}
    allNameTb = {}
    repeatName2Data = {}
    mozhuDbCache = { roleList = {}, unionList = {}}
end

--写日志
local function writeLog(tag, targetDb, sourceDb, sourceColl, info1, info2, docId, errInfo)
    local logInfo = string.format("tag: %s, targetDb: %s, sourceDb: %s, sourceColl: %s, info1: %s, info2: %s, docId: %s, errInfo: %s",
    tag, targetDb, sourceDb, sourceColl, info1, info2, docId, errInfo)

    Log.write(Log.LOGID_OSS_MERGE, logInfo)
end


-------------------------------------------------------------需要单独处理的部分数据---------------------------------------------------------------------


----------------恶魔之塔-------------------
local queryTowerByLv = { lv = nil }
local function getSourceServerTowerData(sourceDb, collectionName, level)
    local targetCollection = sourceDb .. collectionName
    queryTowerByLv.lv = level
    LuaMongo.find(targetCollection, queryTowerByLv, {})

    local data = {}
    if not LuaMongo.next(data) then
		return nil
	end
    return data
end


---------------------------------次元魔蛛/梼杌------------------------------------
local MozhuQueryFiles = { [MergeServerDefine.KEY_CIYUAN_MOZHU] = 1}
--读取

local function loadMoZhuDB(sourceDb, collectionName)
    local targetCollection = sourceDb .. collectionName
    LuaMongo.find(targetCollection, nil, MozhuQueryFiles)

    local data = {}
    if not LuaMongo.next(data) then
		return nil
	end

    local targetMoZhuData = data.ciyuanMoZhu
    for uuid, roleData in pairs(targetMoZhuData.role or {}) do
        mozhuDbCache.roleList[uuid] = roleData
    end

    for uuid, unionData in pairs(targetMoZhuData.union or {}) do
        mozhuDbCache.unionList[uuid] = unionData
        num = num + 1
    end
end

--随机词条
local function randomCiTiao(data)
    local weekTime = Util.getWeekStartTime(os.time())
    -- if data.citiaoTime and data.citiaoTime == weekTime then
    --     return
    -- end

    data.citiao = {}
    local citiao_rare_max

    if not citiao_rare_max then
        -- 防止策划删减 词条
        citiao_rare_max = 0
        for k, v in pairs(MoZhuExcel.citiao) do
            if v then
                citiao_rare_max = citiao_rare_max + v.quanzhong
            end
        end
    end

    local getRare = 0
    for i=1, 1 do
        local random = math.random(1, citiao_rare_max - getRare)
        for k, v in pairs(MoZhuExcel.citiao) do
            if not data.citiao[k] and random <= v.quanzhong then
                data.citiao[k] = i
                getRare = getRare + v.quanzhong
                break
            elseif not data.citiao[k] then
                -- 防止循环最后一个是 已经随机过的  导致 随机不出来
                random = random - v.quanzhong
            end
        end
    end
    data.citiaoTime = weekTime
end

--初始化魔珠db
local function initMoZhuDB(data)
    data.time = os.time()
    data.unionRank = {}
    data.roleRank = {}
    data.role = {}
    data.union= {}

    --词条
    randomCiTiao(data)
end

--插入
local function insertTargetMoZhuDB(targetDb)
    local collectionName = MergeServerDefine.COLLECTIONS[3]
    local targetCollection = targetDb .. collectionName

    LuaMongo.find(targetCollection, nil, MozhuQueryFiles)
    local data = {}
    if not LuaMongo.next(data) then
        return
	end
    

    if not data.ciyuanMoZhu then
        data.ciyuanMoZhu = {}
        initMoZhuDB(data.ciyuanMoZhu)
    end

    local targetMoZhuData = data.ciyuanMoZhu
    local roleRankLen = #targetMoZhuData.roleRank
    local unionRankLen = #targetMoZhuData.unionRank

    --玩家
    for uuid, roleData in pairs(mozhuDbCache.roleList) do
        targetMoZhuData.role[uuid] = roleData

        roleRankLen = roleRankLen + 1
        targetMoZhuData.roleRank[roleRankLen] = uuid
    end

    --公会
    for uuid, unionData in pairs(mozhuDbCache.unionList) do
        targetMoZhuData.union[uuid] = unionData

        unionRankLen = unionRankLen + 1
        targetMoZhuData.unionRank[unionRankLen] = uuid
    end

    --排序
    table.sort(targetMoZhuData.roleRank, function(a, b)
        local roleA = targetMoZhuData.role[a]
        local roleB = targetMoZhuData.role[b]
        return roleA.hurt > roleB.hurt
    end)

    table.sort(targetMoZhuData.unionRank, function(a, b)
        local unionA = targetMoZhuData.union[a]
        local unionB = targetMoZhuData.union[b]
        if unionA.hurt > unionB.hurt then
            return true
        elseif unionA.hurt == unionB.hurt then
            return unionA.time < unionB.time
        end
        return false
    end)

    --更新
	LuaMongo.update(targetCollection, {}, {
        ["$set"] = {
            [MergeServerDefine.KEY_CIYUAN_MOZHU] = targetMoZhuData
        },
    })

    mozhuDbCache = nil
end
-----------------------------------------------------------------------------



---------------------------------------玩家同名------------------------------
local NameQueryFiles = { ["name"] = 1 }

--加载目标库所有玩家的名字
local function loadTargetCharName(targetDb)
    local collectionName = MergeServerDefine.COLLECTIONS[1]
    local targetCollection = targetDb .. collectionName

    LuaMongo.find(targetCollection, nil, NameQueryFiles)

    while true do
        local data = {}
        if not LuaMongo.next(data) then
            break
        end
        allNameTb[data.name] = '1'
    end
end

--得到一个邮件数据
local mailConfig = MailExcel.mail[MergeServerDefine.MAIL_ID]
local function frontMail(receiverUuid)
    local mail = {}
	mail.type = 1
	mail.receiverUuid = receiverUuid
	mail.title = mailConfig.title
	mail.content =  mailConfig.content
	mail.read = nil
	mail.senderName = mailConfig.title
    mail.head = 0
    mail.fbTime = nil
    mail.fbContent = nil
    mail.time = os.time()
    mail.get = nil
    mail.flag = 1
    mail.items = { { ItemDefine.ITEM_ZUANSHI_ID, MergeServerDefine.MAIL_NUM } }
    return mail
end


--给因同名而修改名字的玩家发送补偿邮件
local function sendMail(targetDb)
    local coll = MergeServerDefine.COLLECTIONS[2]
    local targetCollection = targetDb .. coll

    for oldName, v in pairs(repeatName2Data) do
        local mail = frontMail(v.uuid)
        local _, err = pcall(LuaMongo.insert, targetCollection, mail)

        writeLog(MergeServerDefine.CHANGENAME_TAG, targetDb, v.sourceDb, coll, oldName, v.newName, v.uuid, err or "")
    end
end


--------------------------------更新合服时间------------------------------------
local function updateMergeTime(targetDb)
    local targetCollection = targetDb .. MergeServerDefine.COLLECTIONS[3]
    local now = os.time()
    LuaMongo.update(targetCollection, {}, {
        ["$set"] = {
            [MergeServerDefine.KEY_MERGE_TIME] = now
        },
    })
end

----------------------------------------------------------------------------------------------------------------------------------





-----------------------------------------------------char集合中_id的重复检测和处理-------------------------------------------------

--用于检测不同数据库的char集合中的_id是否有重复, 一般来说通过mongo生成的_id基本不会重复(暂时不用)
local function primaryKeyRepeatCheck()
    print("================================检查char集合_id的重复性开始=======================================\n")

    local collectionName = MergeServerDefine.COLLECTIONS[1]
    for k, dbName in ipairs(MergeServerDefine.MERGE_DB_TB) do
        local targetCollection = dbName .. collectionName
        LuaMongo.find(targetCollection, nil, nil, {_id = 1})
        while true do
            local data = {}
            if not LuaMongo.next(data) then
                break
            end
            if k ~= 1 then
                if primaryKeyTable[data._id] then
                    repeatIdTb[data._id] = generateNewId()

                    Log.write(Log.LOGID_OSS_MERGE, string.format("result: %s, targetColl: %s, repeatId: %s, newId: %s",
                    "Repeat", targetCollection, data._id, repeatIdTb[data._id]))
                else
                    primaryKeyTable[data._id] = '1'
                end
            else
                primaryKeyTable[data._id] = '1'
            end
        end
    end
    print("================================检查char集合_id的重复性结束=======================================\n")
end

--好友集合
local function handleFriend(data)
    if repeatIdTb[data.uuid1] then
        data.uuid1 = repeatIdTb[data.uuid1]
    end

    if repeatIdTb[data.uuid2] then
        data.uuid2 = repeatIdTb[data.uuid2]
    end
end

--邮件集合
local function handleMail(data)
    if repeatIdTb[data.receiverUuid] then
        data.receiverUuid = repeatIdTb[data.receiverUuid]
    end
end

--公会集合
local function handleUnion(data)
    local tb = {}
    for id, v in pairs(data.member) do
        if repeatIdTb[id] then
            tb[id] = v
        end
    end

    for id, v in pairs(tb) do
        data.member[id] = nil
        data.member[repeatIdTb[id]] = v
    end


    tb = {}
    for id, v in pairs(data.apply) do
        if repeatIdTb[id] then
            tb[id] = v
        end
    end

    for id, v in pairs(tb) do
        data.apply[id] = nil
        data.apply[repeatIdTb[id]] = v
    end

    if data.presidentUuid and repeatIdTb[data.presidentUuid] then
        data.presidentUuid = repeatIdTb[data.presidentUuid]
    end
end

--星空争霸集合
local function handleStar(data)
    if repeatIdTb[data.uuid] then
        data.uuid = repeatIdTb[data.uuid]
    end
end

--单人竞技场集合
local function handleJJC(data)
    if not data.monsterOutID and repeatIdTb[data._id] then
        data._id = repeatIdTb[data._id]
    end
end

--战斗视频集合
local function handleVideo(data)
    local uuid = data.combatInfo.attacker.uuid
    if repeatIdTb[uuid] then
        data.combatInfo.attacker.uuid = repeatIdTb[uuid]
    end
end

----------------------------------------------------合并数据----------------------------------------------------------

local writeQueue = {}
local function flushWriteQueue()
    for _, task in ipairs(writeQueue) do
        local ok, err = pcall(LuaMongo.insert, task.collection, task.data)
        if not ok then
            Log.write(Log.LOGID_OSS_MERGE, task.collection, err)
        end
    end
    writeQueue = {}
end

-- 每积累100条数据批量写入
local function bufferedInsert(collection, data)
    table.insert(writeQueue, {collection=collection, data=data})
    if #writeQueue >= 100 then
        flushWriteQueue()
    end
end


local function processCollection(sourceDb, targetDb, coll)
    local sourceCollection = sourceDb .. coll
    local targetCollection = targetDb .. coll

    local errNum = 0
    local allNum = 0
    LuaMongo.find(sourceCollection, nil, {})
    while true do
        local data = {}
        if not LuaMongo.next(data) then
            break
        end


        --直接插入目标库
        local ok, err = pcall(LuaMongo.insert, targetCollection, data)
        if not ok then
            errNum = errNum + 1
            writeLog(MergeServerDefine.ERR_TAG, targetDb, sourceDb, coll, 1, 1, data._id, err)
        end

        allNum = allNum + 1
    end

    writeLog(MergeServerDefine.SUCC_TAG, targetDb, sourceDb, coll, allNum, errNum, "", "")
end

--合并char集合
local function processCharColl(sourceDb, targetDb, coll)
    local sourceCollection = sourceDb .. coll
    local targetCollection = targetDb .. coll

    local errNum = 0
    local allNum = 0
    LuaMongo.find(sourceCollection, nil, {})
    while true do
        local data = {}
        if not LuaMongo.next(data) then
            break
        end

        --同名检测
        if allNameTb[data.name] then
            num = num + 1
            local newName = MergeServerDefine.NEW_NAME .. num

            --统计数据
            repeatName2Data[data.name] = {
                uuid = data._id,
                newName = newName,
                sourceDb = sourceDb,
            }

            --改名
            data.name = newName
        end
        allNameTb[data.name] = '1'

        --插入目标库
        local ok, err = pcall(LuaMongo.insert, targetCollection, data)
        if not ok then
            errNum = errNum + 1
            writeLog(MergeServerDefine.ERR_TAG, targetDb, sourceDb, coll, 1, 1, data._id, err)
        end

        allNum = allNum + 1
    end

    writeLog(MergeServerDefine.SUCC_TAG, targetDb, sourceDb, coll, allNum, errNum, "", "")
end


local function startMergeServer(dbList)
    assert(dbList and next(dbList), "====数据错误====")

    cleanCache()

    --目标集合数据库名
    local targetDb = dbList[1]

    loadTargetCharName(targetDb)

    for i = 2, #dbList do
        local sourceDb = dbList[i]
        LuaMongo.auth(sourceDb, Config.DB_USER, Config.DB_PASS)

        for _, coll in ipairs(MergeServerDefine.COLLECTIONS) do
            if not MergeServerDefine.NO_INSERT_COLLECTIONS[coll] then

                print(string.format("========================开始处理集合: %s, %s\n", sourceDb, coll))
                if coll == MergeServerDefine.COLLECTIONS[1] then
                    processCharColl(sourceDb, targetDb, coll)
                elseif coll == MergeServerDefine.COLLECTIONS[3] then
                    loadMoZhuDB(sourceDb, coll)
                else
                    processCollection(sourceDb, targetDb, coll)
                end
            end
        end
    end

    --处理那些需要单独处理的数据
    insertTargetMoZhuDB(targetDb)

    sendMail(targetDb)

    updateMergeTime(targetDb)
end


--合服开始函数
function Start()
    --创建mongo连接实例
    clientMongoDb()

    --检查char集合_id的重复性
    --primaryKeyRepeatCheck()


    print("========================合服开始=====================")


    for i, dbList in ipairs(MergeServerDefine.MERGE_DB_TB) do
        startMergeServer(dbList)
    end


    print("========================合服结束=====================")
end