网站首页lua/js

lua函数库

发布时间:2017-08-13 07:33:49编辑:slayer.hover阅读(4807)

           习惯了PHP开发的同学,在刚使用lua来coding的时候一定非常不爽,各种常用的函数库都没了。 

    有问题就解决,没有问题的,就先创造问题再解决,我们自己来写lua的仿PHP函数库。

    下面列出一些常用的函数代码及测试用例:

    https://github.com/slayerhover/ucenterbylua/blob/master/application/library/function.lua


    ----function.lua代码如下:
    
    local cjson    = require "cjson"
    local http    = require "resty.http"
    local _M = { _VERSION="1.0.0"};
    
    ----判断table是否为空
    function _M.empty(t)
        if t == nil or _G.next(t) == nil then
            return true
        else
            return false
        end
    end
    
    ----格式化输出,调试代码使用
    function _M.dump(t)    
            local function parse_array(tab)
                local str = ''
                for _, v in pairs(tab) do
                    str    =    str .. '\t\t' .. _ .. ' => ' .. v .. '\n'
                end            
                return str
            end
            
            local str = type(t);        
            if str=='table' then        
                str = str .. '(' .. #t .. ')' .. '\n{\n' 
                for k,v in pairs(t) do
                    if type(v)=="table" then
                        str = str .. '\t' .. k .. ' = > {\n' .. parse_array(v) .. '\t}' .. '\n'
                    else
                        str = str .. '\t' .. k .. ' => ' .. (v) ..  '\n'
                    end
                end
            else
                str = str .. '\n{\n' .. tostring(t) .. '\n'
            end        
            str = str .. '}'
            
            ngx.say('\n' .. str .. '\n')
    end    
    
    ----curl获取网址内容
    function _M.curl(url, params, method)
        local httpc    =    http.new()
        httpc:set_timeout(2000) --2秒超时
        local resp, err = httpc:request_uri(url, {  
            method    = method or 'GET',
            body     = params,
            ssl_verify = false, --兼容https
            headers = {  
                ["User-Agent"]  = "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/40.0.2214.111 Safari/537.36",
                ["Content-Type"]= "application/x-www-form-urlencoded"
                
            }  
        })
        if not resp then          
            return false
        end  
              
        httpc:close()
        return resp.body
    end
    
    ----获取table大小 
    function _M.sizeof(t)
        local count = 0
        for _, v in pairs(t) do
            count = count + 1
        end            
        return count;
    end
    function _M.count(t)
        local count = 0
        for _, v in pairs(t) do
            count = count + 1
        end            
        return count;
    end
    
    ----json_encode,返回字符串
    function _M.json_encode(t)
        cjson.encode_empty_table_as_object(true)
        local str = cjson.encode(t)
        return str
    end
    
    ----json_decode,返回table
    function _M.json_decode(str)
        local json = false
        pcall(function(str) json = cjson.decode(str) end, str)
        return json
    end
    
    ----分隔字符串,返回table
    function _M.explode(split, str)
        local str_split_tab = {}
        while true do
            local idx = string.find(str,split,1,true);
            if nil~=idx then
                local insert_str = '';
                if 1==idx then
                    insert_str = string.sub(str, 1,idx + #split - 1);
                else
                    insert_str = string.sub(str, 1,idx - 1);
                end
                if (insert_str ~= split) and (nil ~= insert_str or '' ~= insert_str) then
                    table.insert(str_split_tab,insert_str);
                end
                str = string.sub(str,idx + #split,-1);
            else
                if nil ~= str or '' ~= str then
                    table.insert(str_split_tab,str);
                end
                break;
            end
        end
        return str_split_tab;
    end
    
    ----合并table,返回字符串
    function _M.implode(split, t)
        local tab = {}
        for _, v in pairs(t) do
            table.insert(tab, v)
        end            
        return table.concat(tab, split);
    end
    
    ----字符串查找替换
    function _M.str_replace(str, find, replace)
        local res,res_count = string.gsub(str,find,replace);
        return res,res_count
    end
    
    ----字符串查找位置
    function _M.strpos(str, find)
        local res,res_end = string.find(str, find)
        if ''==res or nil==res then
            return false;
        end    
        return res;
    end
    
    ----字符串查找位置,不区分大小写
    function _M.stripos(str, find)
        local str = string.lower(str);
        local find= string.lower(find);
        
        local res,res_end = string.find(str, find)
        if ''==res or nil==res then
            return false;
        end    
        return res;
    end
    
    ----正则匹配一次
    function _M.preg_match(regex, str)
        local res,err = ngx.re.match(str, regex, "io")
        if res then
            return res;
        else
            ngx.log(ngx.ERR, "error: ", err)
            return false;
        end
    end
    
    ----正则匹配所有
    function _M.preg_match_all(regex, str)
        local it,err = ngx.re.gmatch(str, regex, "io")
        if not it then
            ngx.log(ngx.ERR, "error: ", err)
            return false;
        end
        local res = {};    
        while true do
            local m, err = it()
            if err then
                ngx.log(ngx.ERR, "error: ", err)
                return false;
            end 
            if not m then
                break
            end
            table.insert(res, m);
        end            
        return res;
    end
    
    ----正则替换
    function _M.preg_replace(regex, replacement, str, option)
        local newstr, n, err = ngx.re.gsub(str, regex, replacement, option)
        if newstr then
            return newstr;        
            -- newstr == "[hello,h], [world,w]"
            -- n == 2
        else
            ngx.log(ngx.ERR, "error: ", err)
            return false;
        end
    end
    
    ----清除空白字符
    function _M.trim(str)
        str = _M:ltrim( str );
        str = _M:rtrim( str );
        return str;
    end
    
    ----清除左侧空白字符
    function _M.ltrim( str )
        if ''==str or nil==str then
            return str;
        end
        local len = string.len( str );
        
        local substart = 1;
        local  lenadd = 1;
        while ( string.find ( str," ",lenadd) == lenadd ) do
            substart = substart + 1;
            lenadd= lenadd + 1;
        end
        
        str=string.sub ( str ,substart ,len);
        
        local substart1 = 1;
        local lenadd1 = 1;
        len=string.len(str);
        while(string.find(str,"%s",lenadd1 )==lenadd1) do
                    substart1 = substart1 +1
                    lenadd1 = lenadd1 +1
        end
        
        str=string.sub(str,substart1,len);
        
        return str;
    end
    
    ----清除右侧空白字符
    function _M.rtrim( str )
        if ''==str or nil==str then
            return str;
        end
        local len = string.len(str);
        
        local substart = len;
        local  lenadd = len;
        while ( string.find ( str,"%s",lenadd) == lenadd ) do
            substart = substart - 1;
            lenadd= lenadd - 1;
        end
        str =string.sub(str , 1,substart );
        len=string.len(str);
        local substart1 = len;
        local lenadd1 = len;
        while(string.find(str," ",lenadd1 )==lenadd1) do
                    substart1 = substart1 -1
                    lenadd1 = lenadd1 -1
        end
        
        str=string.sub(str,1,substart);
        return str;
    end
    
    ----字符串转为大写
    function _M.upper(str)
        if ''==str or nil==str then
            return str;
        end
        return string.upper(str);
    end
    
    ----字符串转为小写
    function _M.lower(str)
        if ''==str or nil==str then
            return str;
        end
        return string.lower(str);
    end
    
    return _M;


    测试用例:

    ----test.lua
    package.path= '/opt/lualib/?.lua;d:/opt/lapi/inc/?.lua;';
    local func= require "function";
    local str = func.curl("https://www.baidu.com/s");
    ngx.say(str);
    local str = '{    "a": 123,    "c": "d",    "e": "f"}';
    local res = func.json_decode(str);
    func.dump(res);
    local t = {a="b",c="de", f="ji"};
    local res = func.json_encode(t);
    func.dump(res);
    local newtab = func.implode('|', t);
    func.dump(newtab);
    func.dump((func.explode('|', newtab)));
    local result = func.preg_match("([a-z]+)", "abc 1234, hello");
    if result then
    func.dump(result);
    else
    ngx.say("no matched.");
    end
    local result = func.preg_match_all("(?<numbers>[\\d]+)", "abc 1234, hello, 78d90");
    if result then
    func.dump(result);
    else
    ngx.say("no matched.");
    end




评论