daydaytalk-fwutils/target/fwutils/token.lua

local config = require("fwutils.config")
local codec = require("fastweb.codec")
local utils = require("utils")
local cjson = require("cjson")
local redis_pool = require("redis.pool")
local M = {}
local function make_key(id,token)
    if type(id) == "number" then
        id = string.format("%d",id)
    end
    return config.server_id.."_token_"..id.."_"..token
end
M.__parse_token = function(token)
    if token == nil or token == "" then
        return false,"token is required"
    end
    local token_data = codec.aes_de(config.token.key,utils.hex_to_bytes(token),config.token.algorithm,config.token.mode)
    if token_data == nil or token_data == "" then
        return false,"token is invalid(a)"
    end
    local token_data_json = cjson.decode(utils.hex_to_bytes(token_data))
    if token_data_json == nil or token_data_json.id == nil or token_data_json.id <= 0 then
        return false,"token is invalid(no id)"
    end
    return true,token_data_json 
end
M.get = function(token)
    -- 解析TOKEN
    local result,token_data_json = M.__parse_token(token)
    if not result then
        return false,token_data_json
    end
     

    -- 获取TOKEN数据
    local conn = redis_pool.new(_G[config.db.redis_pool_name]):get()

    local data = conn:get(make_key(token_data_json["id"],token))
    if data ~= nil and data ~= "" then
        return true,cjson.decode(data)
    end
    return false,"token is invalid"
end
M.set = function(token,data)
    -- 解析TOKEN
    local result,token_data_json = M.__parse_token(token)
    if not result then
        return false,token_data_json
    end
    if data == nil then
        return false,"data is required"
    end
    local conn = redis_pool.new(_G[config.db.redis_pool_name]):get()
    conn:setex(make_key(token_data_json["id"],token),config.token.expire,cjson.encode(data))
    return true
end
M.del_by_id = function(id)
    if id == nil or id <= 0 then
        return false,"id is required"
    end
    local conn = redis_pool.new(_G[config.db.redis_pool_name]):get()
    local keys = conn:keys(config.other.SERVER_ID.."_token_"..string.format("%d",id).."_*")
    for _,key in ipairs(keys) do
        conn:del(key)
    end
    return true
end
M.del_by_token = function(token)
    -- 解析TOKEN
    local result,token_data_json = M.__parse_token(token)
    if not result then
        return false,token_data_json
    end
    local conn = redis_pool.new(_G[config.db.redis_pool_name]):get()
    conn:del(make_key(token_data_json["id"],token))
    return true
end
M.create = function(id,data)
    if id == nil or id <= 0 then
        return false,"id is required"
    end
    local key_data = {
        id = id,
        create_time = fw.now_msec()
    }
    local token = codec.aes_en(config.token.key,cjson.encode(key_data),config.token.algorithm,config.token.mode)
    if M.set(token,data) == false then
        return false,"create token failed"
    end 
    return true,token
end
return M