MySQL代理结果字段操作

2014-1-30 10:18:45
收藏：0
阅读：108
评论：1

我有一个MYSQL服务器和MYSQL-PROXY，我正在尝试操纵作为对SELECT查询响应发送的结果。我已经在lua中编写了这段代码：

function string.starts(String,Start)
   return string.sub(String,1,string.len(Start))==Start
end

function read_query_result(inj)
        local fn = 1
        local fields = inj.resultset.fields
        while fields[fn] do
            fn = fn + 1
        end
        fn = fn - 1
        print("FIELD NUMBER: " .. fn)
        for row in inj.resultset.rows do
            print ("--------------")
            for i = 1, fn do
                if (string.starts(fields[i].name,"TEST")) then
                    row[i]="TESTED"
                end
                print ("DATA: " ..  fields[i].name .. " -> " .. row[i])
            end
        end
        return proxy.PROXY_SEND_RESULT
end

我可以正确读取字段名和值。我可以检测到我想要修改结果的条件，但我无法获取发送到客户端的数据。

我看到两个问题：

  • 我正在设置局部行变量的值，但我没有找到设置实际结果集（inj.Resultset.row[i]或类似）的方法。
  • return proxy.PROXY_SEND_RESULT有问题，因为我发现当我注释掉那句话时，我能够看到结果，如果我取消注释它，我会得到一个错误。

我没有找到参考示例代码。

点赞
用户2935222
用户2935222
  • 需要将数据插入表中
  • PROXY_SEND_RESULT 要求设置 proxy.response.type。

以下是正确的模块：

function read_query_result(inj)
        local fn = 1
        local fields = inj.resultset.fields
        proxy.response.resultset = {fields = {}, rows = {}}
        while fields[fn] do
            table.insert(proxy.response.resultset.fields, {type = proxy.MYSQL_TYPE_STRING, name = fields[fn].name})
            fn = fn + 1
        end
        fn = fn - 1
        for row in inj.resultset.rows do
            for i = 1, fn do
                if (string.starts(fields[i].name,"TEST")) then
                    row[i]="TESTED"
                end
            end
            table.insert(proxy.response.resultset.rows, row )
        end
        proxy.response.type = proxy.MYSQLD_PACKET_OK
        return proxy.PROXY_SEND_RESULT
end
2014-01-30 16:10:07

作者:

用户2935222

用户2935222
推荐文章