eth-rpc-proxy/nginx-proxy/lua/cache/cache.lua

local json = require("cjson")
local cache_rules = require("cache.cache_rules")

local _M = {}

local cache_dicts = {
    permanent = ngx.shared.rpc_cache,
    short = ngx.shared.rpc_cache_short,
    minimal = ngx.shared.rpc_cache_minimal
}

-- Helper function to decode JSON once and cache result
local function get_decoded_body(body_data)
    if not body_data then return nil end
    
    -- If already decoded (table), return as is
    if type(body_data) == "table" then
        return body_data
    end
    
    local ok, body_json = pcall(json.decode, body_data)
    if not ok or not body_json or not body_json.method then
        return nil
    end
    
    return body_json
end

-- Unified cache function that handles all cache operations
function _M.check_cache(chain, network, body_data)
    local decoded_body = get_decoded_body(body_data)
    if not decoded_body then
        return {
            cache_type = nil,
            cache_key = nil,
            ttl = nil,
            cached_response = nil
        }
    end
    
    local cache_info = cache_rules.get_cache_info(chain, network, decoded_body)
    if not cache_info or cache_info.cache_type == "none" or cache_info.ttl == 0 then
        return {
            cache_type = nil,
            cache_key = nil,
            ttl = nil,
            cached_response = nil
        }
    end
    
    local cache_type = cache_info.cache_type
    local ttl = cache_info.ttl
    
    local cache_key = chain .. ":" .. network .. ":" .. ngx.md5(body_data)
    local shared_dict = cache_dicts[cache_type]
    
    local cached_response = shared_dict:get(cache_key)
    local stats_dict = ngx.shared.cache_stats
    
    if cached_response then
        ngx.log(ngx.INFO, "Cache hit (", cache_type, ") for key: ", cache_key)
        -- Increment cache hit counter
        stats_dict:incr("cache_hits_" .. cache_type, 1, 0)
        stats_dict:incr("cache_hits_total", 1, 0)
    else
        ngx.log(ngx.INFO, "Cache miss (", cache_type, ") for key: ", cache_key)
        -- Increment cache miss counter
        stats_dict:incr("cache_misses_" .. cache_type, 1, 0)
        stats_dict:incr("cache_misses_total", 1, 0)
    end
    
    -- Increment total requests counter
    stats_dict:incr("total_requests_" .. cache_type, 1, 0)
    stats_dict:incr("total_requests_all", 1, 0)
    
    return {
        cache_type = cache_type,
        cache_key = cache_key,
        ttl = ttl,
        cached_response = cached_response
    }
end

-- Save function that uses cache_info
function _M.save_to_cache(cache_info, response_body)
    if not cache_info.cache_type or not cache_info.cache_key or not cache_info.ttl then
        return false
    end
    
    local shared_dict = cache_dicts[cache_info.cache_type]
    if not shared_dict then
        ngx.log(ngx.ERR, "Invalid cache_type: ", cache_info.cache_type)
        return false
    end
    
    local success, err = shared_dict:set(cache_info.cache_key, response_body, cache_info.ttl)
    if success then
        ngx.log(ngx.INFO, "Cached response (", cache_info.cache_type, ") for key: ", cache_info.cache_key, " with TTL: ", cache_info.ttl, " seconds")
    else
        ngx.log(ngx.ERR, "Failed to cache response (", cache_info.cache_type, "): ", err)
    end
    return success
end

return _M