/lualib/socket.lua

local skynet = require "skynet"
local c = require "socket.c"

local socket = {}
local fd
local object
local data = {}

local function presend()
	if next(data) then
		local tmp = data
		data = {}
		for _,v in ipairs(tmp) do
			socket.write(fd, v)
		end
	end
end

function socket.connect(addr)
	local ip, port = string.match(addr,"([^:]+):(.+)")
	port = tonumber(port)
	socket.close()
	fd = c.open(ip,port)
	if fd == nil then
		return true
	end
	skynet.send(".connection", "text", "ADD", fd , skynet.address(skynet.self()))
	object = c.new()
	presend()
end

function socket.stdin()
	skynet.send(".connection", "text", "ADD", 1 , skynet.address(skynet.self()))
	object = c.new()
end

function socket.push(msg,sz)
	if msg then
		c.push(object, msg, sz)
	end
end

function socket.read(bytes)
	return c.read(object, bytes)
end

function socket.readline(sep)
	return c.readline(object, sep)
end

function socket.readblock(...)
	return c.readblock(object,...)
end

function socket.write(...)
	local str = c.write(fd, ...)
	if str then
		socket.close()
		table.insert(data, str)
	end
end

function socket.writeblock(...)
	local str = c.writeblock(fd, ...)
	if str then
		socket.close()
		table.insert(data, str)
	end
end

function socket.close()
	if fd then
		c.close(fd)
		skynet.send(".connection","text", "DEL", fd)
		fd = nil
	end
end

return socket