root/lu/SocketRunLoop.lua

--------------------------------------------------------------------------------
-- Title:               SocketRunLoop.lua
-- Description:         Like a square peg in a round hole
-- Author:              Raphaël Szwarc http://alt.textdrive.com/lua/
-- Creation Date:       February 1, 2005
-- Legal:               Copyright (C) 2005 Raphaël Szwarc
--------------------------------------------------------------------------------

-- import dependencies
local LURunLoop = require( "LURunLoop" )
local LUMap = require( "LUMap" )
local SocketTask = require( "SocketTask" )
local socket = require( "socket" )

-- define the class
local super = LURunLoop
local self = super()

-- constant(s)
local Timeout = 0.1

 -- method to add a task to this run loop
function self:addTask( aTask )
        super.addTask( self, aTask )
        
        self:runTask( aTask )
        
        return self
end

function self:sockets()
        if self._sockets == nil then
                self._sockets = LUMap( nil, "kv" )
        end
        
        return self._sockets
end

function self:readers()
        if self._readers == nil then
                self._readers = LUMap( nil, "kv" )
        end
        
        return self._readers
end

function self:writers()
        if self._writers == nil then
                self._writers = LUMap( nil, "kv" )
        end
        
        return self._writers
end

-- method to access this run loop selected tasks
function self:selectedTasks()
        local someTasks = LUMap()
        local someSockets = self:sockets()
        local someReaders = self:readers():values():content()
        local someWriters = self:writers():values():content()
        
        someReaders, someWriters = socket.select( someReaders, someWriters, Timeout )
        
        for _, aSocket in ipairs( someReaders ) do
                local aTask = someSockets:get( aSocket )
                
                someTasks:put( aTask, aSocket )
        end
        
        for _, aSocket in ipairs( someWriters ) do
                local aTask = someSockets:get( aSocket )
                
                someTasks:put( aTask, aSocket )
        end
        
        return someTasks
end

-- method to 'run' the given task
function self:runTask( aTask )
        local someTasks = self:tasks()
        local aStatus = aTask:status()
        
        if aStatus == "dead" then
                someTasks:remove( aTask )
        elseif aStatus == "suspended" then
                local aStatus, aType = aTask:resume()

                if aStatus == true then
                        local aSocket = aTask:target():socket()

                        self:sockets():put( aSocket, aTask )
                        
                        if aType == "reader" then
                                self:readers():put( aTask, aSocket )
                        else
                                self:writers():put( aTask, aSocket )
                        end
                else
                        someTasks:remove( aTask )
                end
        end
        
        return self
end

-- method to 'run' through all this run loop tasks
function self:run()
        for aTask, _ in self:selectedTasks():iterator() do
                self:runTask( aTask )
        end

        for aTask, _ in self:tasks():iterator() do
                self:runTask( aTask )
        end

        return self
end

return self