root/lu/SocketWriter.lua

--------------------------------------------------------------------------------
-- Title:               SocketWriter.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 os = require( "os" )

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

-- constant(s)
local Timeout = 60

-- initialization method
function self:init( aSocket )
        self = super.init( self )

        self._socket = aSocket
        self._status = "timeout"
        
        return self
end

-- method to access this reader socket
function self:socket()
        return self._socket
end

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

        return self:socket():close()
end

-- method to access this writer status
function self:status()
        return self._status
end

 -- method to write a value
function self:write( aValue, aTimeout )
        local aTime = os.time()
        local aStatus = nil
        local anIndex = 0
        local aSize = 0

        aTimeout = aTimeout or Timeout 

        while true do                
                aSize, aStatus, anIndex = self:socket():send( aValue, anIndex + 1 )
                
                self._status = aStatus
                
                if aSize ~= nil or aStatus ~= "timeout" then
                        return aSize, aStatus 
                end
                
                if aStatus == "timeout" then
                        if os.difftime( os.time(), aTime ) >= aTimeout then
                                return aSize, aStatus
                        end
                else
                        aTime = os.time()
                end

                LUTask:yield( "writer" )
        end
        
        return aSize, aStatus
end

 -- method to write a value, followed by a new line
function self:writeln( aValue )
        if aValue ~= nil then
                aValue = aValue .. self:eol()
        else
                aValue = self:eol()
        end

        return self:write( aValue )
end

-- method to access this writer task
function self:task()
        return self._task
end

-- class method to access the EOF marker
function self:eol()
        return "\r\n"
end

return self