client.lua

local zmq = require 'lzmq'
local zloop = require 'lzmq.loop'
local Connection = require 'somata.connection'

local Client = {}
Client.__index = Client

function Client.create(loop, registry_host)
    local client = {}
    setmetatable(client, Client)

    client.ctx = zmq.context()

    if loop == nil then
        client.loop = zloop.new(1, client.ctx)
    else
        client.loop = loop
    end

    if registry_host == nil then
        client.registry_host = "localhost"
    else
        client.registry_host = registry_host
    end

    client.service_connections = {}
    client.registry_connection = Connection.create(client.ctx, client.loop, "tcp://" .. client.registry_host .. ":8420", 'registry')

    return client
end

function Client:getConnection(service_name, cb)
    if self.service_connections[service_name] then
        cb(nil, self.service_connections[service_name])
    else
        self.registry_connection:sendMethod("getService", {service_name}, function(err, service)
            if service ~= nil then
                local service_connection = Connection.create(self.ctx, self.loop, "tcp://" .. self.registry_host .. ":" .. service.port)
                self.service_connections[service_name] = service_connection
                cb(nil, service_connection)
            else
                cb("No such service")
            end
        end)
    end
end

function Client:remote(service_name, method, args, cb)
    self:getConnection(service_name, function (err, service_connection)
        if service_connection ~= nil then
            service_connection:sendMethod(method, args, cb)
        else
            print("[remote] Can't get service connection:", err)
            cb("Can't get service connection")
        end
    end)
end

return Client