This script shows how to measure client traffic by user.

MySQL Proxy is a simple program that sits between your client and MySQL server that can monitor, analyze or transform their communication. See http://forge.mysql.com/wiki/MySQL_Proxy for more information.

---
-- user_bandwidth.lua
--  
-- measures bandwidth by user
--
proxy.global.bandwidth = proxy.global.bandwidth or {}
 
local session_user 
 
function read_auth( auth )
    session_user = auth.username
    proxy.global.bandwidth[session_user] = 
        proxy.global.bandwidth[session_user] or 0
end
 
function read_query (packet )
    -- just to show how we can block a user query
    -- when the quota has been exceeded
    if proxy.global.bandwidth[session_user] > 10000 
       and session_user ~= 'root'
    then
        return error_result('you have exceeded your query quota')  
    end
    proxy.global.bandwidth[session_user ] = 
        proxy.global.bandwidth[session_user] + packet:len()
    proxy.queries:append(1, packet )
    return proxy.PROXY_SEND_QUERY
end
 
function read_query_result(inj)
    local fields = inj.resultset.fields
    local rows = inj.resultset.rows
    if fields  then
        for i = 1, #fields do
            proxy.global.bandwidth[session_user] = 
                proxy.global.bandwidth[session_user] + 
                    (fields[i] and fields[i].name:len() or 0)
        end
        if rows then 
            for row in rows do
                for i = 1, #fields do
                    proxy.global.bandwidth[session_user] = 
                        proxy.global.bandwidth[session_user] + 
                            (row[i] and row[i]:len() or 0)
                end
            end
        end
    end
    print (session_user .. ' -> ' .. 
        proxy.global.bandwidth[session_user])
end
 
function error_result (msg)
    proxy.response = {
        type        = proxy.MYSQLD_PACKET_ERR,
        errmsg      = msg,
        errcode     = 7777,
        sqlstate    = 'X7777',
    }
    return proxy.PROXY_SEND_RESULT
end
 

Current Tags

You must be logged in to tag this tool