异步IO框架

以下代码由PHP站长网 52php.cn收集自互联网

现在PHP站长网小编把它分享给大家，仅供参考

local socket = require "levent.socket"
local levent = require "levent.levent"
local seri = require "levent.tpseri"
local dns    = require "levent.dns"

local mt = {}
mt.__index = mt

BUFLEN = 4096
mt.get_base_header = function(self)
    local start_pos,end_pos
    while true do
        self.client_buffer = self.client_buffer .. self.client:recv(BUFLEN)
        start_pos,end_pos = self.client_buffer:find("n")
        if start_pos then
            break
        end
    end
    base_header = self.client_buffer:sub(1,start_pos)
    local result = {}
    for i in string.gmatch(base_header,"%S+") do
        table.insert(result,i)
    end
    self.client_buffer = self.client_buffer:sub(start_pos + 1,-1)
    return table.unpack(result)
end

mt.connect_target = function(self,host)
    local i = host:find(":")
    local port
    if not i then
        port = 80
    else
        host = host:sub(1,i)
        port = tonumber(host:sub(i,-1))
    end
    local ips,err = dns.resolve(host)
    local ip = ips[1]
    self.target,err = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
    self.target:connect(ip,port)
end

mt.method_CONNECT = function(self)
    print("enter method connect") 
    self.path = self.path:sub(8,-1)
    local i = self.path:find("/")
    local host = self.path:sub(1,i - 1)
    local path = self.path:sub(i,-1)
    self:connect_target(host)
    self.target:sendall("HTTP/1.1 200 Connection establishedn Proxy-agent: 0.0.1nn")
    self.client_buffer = ""
    self:flush_two_socket()
end

mt.keep_request = function(self)
    print("keep request")
    while true do
        local req,err = self.client:recv(BUFLEN)
        if req and #req > 0 then
            self.target:sendall(req)
        else
            break
        end
    end
end

mt.keep_resp = function(self)
    print("keep resp")
    while true do
        local resp,err = self.target:recv(BUFLEN)
        if resp and #resp > 0 then
            self.client:sendall(resp)
        else
            break
        end
    end
end

mt.flush_two_socket = function(self)
    levent.spawn(self.keep_resp,self)
    levent.spawn(self.keep_request,self)
    levent.sleep(1)
end

mt.method_others = function(self)
    self.path = self.path:sub(8,-1)
    self:connect_target(host)
    local first_data = string.format("%s %s %srn",self.method,path,self.protocol)..self.client_buffer
    print(string.format("%q",first_data))
    self.target:sendall(first_data)
    self:flush_two_socket()
end

local other_method = {
    OPTIONS=1,GET = 1,HEAD = 1,POST=1,PUT=1,DELETE=1,TRACE=1,}

mt.handle = function(self,csock)
    self.client = csock
    self.client_buffer = ""
    self.timeout = timeout
    self.method,self.path,self.protocol = self:get_base_header()
    print(self.method,self.protocol)
    if self.method == "CONNECT" then
        self:method_CONNECT()
    elseif other_method[self.method] then
        self:method_others()
    end
    self.client:close()
    self.target:close()
    print("close all socket")
end

local sock,socket.SOCK_STREAM)
assert(sock,errmsg)
print("bind:",sock:bind("0.0.0.0",8858))
print("listen:",sock:listen())

function start_server()
    while true do
        local csock,err = sock:accept()
        if not csock then
            print("accept failed:",err)
            break
        end
        print("new conn from:",csock:getpeername())
        local a = {}
        setmetatable(a,mt)
        levent.spawn(a.handle,a,csock)
    end
end

levent.start(start_server)

以上内容由PHP站长网【52php.cn】收集整理供大家参考研究

如果以上内容对您有帮助，欢迎收藏、点赞、推荐、分享。

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!