root/lu/SocketServer.lua

--------------------------------------------------------------------------------
-- Title:               SocketServer.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 LUObject = require( "LUObject" )
local LUTask = require( "LUTask" )
local LULog = require( "LULog" )
local SocketRunLoop = require( "SocketRunLoop" )
local SocketTask = require( "SocketTask" )
local socket = require( "socket" )

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

-- initialization method
function self:init( anAddress, aPort, aDelegate )
        self = super.init( self )

        self._address = anAddress
        self._port = aPort
        self._delegate = aDelegate
        
        return self
end
        
-- method to access this server address
function self:address()
        if self._socket ~= nil then
                local aSocketAddress, aSocketPort = self._socket:getsockname()
                
                if aSocketAddress ~= nil then
                        return aSocketAddress
                end
        end

        if self._address == nil then
                self._address = "*"
        end

        return self._address
end

 -- method to close this socket
function self:close()
        self._status = "closed"

        return self:socket():close()
end

-- method to access this server port
function self:port()
        if self._socket ~= nil then
                local aSocketAddress, aSocketPort = self._socket:getsockname()
                
                if aSocketPort ~= nil then
                        return aSocketPort
                end
        end

        if self._port == nil then
                self._port = 0
        end

        return self._port
end

-- method to access this server delegate
function self:delegate()
        return self._delegate
end

-- method to access this server socket
function self:socket()
        if self._socket == nil then
                local anAddress = self:address()
                local aPort = self:port()
                local aSocket, aStatus = socket.bind( anAddress, aPort )
                
                if aSocket ~= nil then
                        aSocket:settimeout( 0 )
                        aSocket:setoption( "linger", { on = false, timeout = 0 } )
                        aSocket:setoption( "tcp-nodelay", true )
                        
                        self._socket = aSocket
                else
                        LULog:warning( self .. " '" .. aStatus .. "'" )
                
                        return nil, aStatus
                end
        end

        return self._socket
end

-- method to start this server task
function self:start()
        if self._task == nil then
                self:socket()
                self:task():run()
        end
        
        return self
end

-- method to access this server task
function self:task()
        if self._task == nil then
                self._task = LUTask( self, SocketRunLoop:default() )
        end
        
        return self._task
end

-- method to run this server
function self:run()
        local aServerSocket = self:socket()
        local aDelegate = self:delegate()
        
        while true do
                local aSocket, aStatus = aServerSocket:accept()
                
                if aSocket ~= nil then
                        local aSocketTask = SocketTask( self, aSocket, aDelegate )

                        aSocket:settimeout( 0 )
                        aSocket:setoption( "linger", { on = false, timeout = 0 } )
                        aSocket:setoption( "tcp-nodelay", true )
                         
                        aSocketTask:start()
                end
                
                if aStatus ~= nil and aStatus ~= "timeout" then
                        LULog:warning( self .. " '" .. aStatus .. "'" )
                end

                LUTask:yield( "reader" )
       end
end

-- method for string representation
function self:toString()
        return "SocketServer@" .. self:address() .. ":" .. self:port()
end

return self