dpt.info.lua

-- Info package
-- This package provides information tables that are built up and maintained over the course of the dissection.

-- Package header
local master = diffusion or {}
if master.info ~= nil then
	return master.info
end

local int_to_string = master.utilities.int_to_string


-- -----------------------------------
-- The Alias Table

local AliasTable = {}

function AliasTable:new()
	local result = {}
	setmetatable( result, self )
	self.__index = self
	return result
end

function AliasTable:setAlias( tcpStream, alias, topicName )
	-- Get the table for the tcpStream, or create a new one
	local conversation = self[tcpStream] or {}
	conversation[alias] = topicName
	self[tcpStream] = conversation
end

function AliasTable:getAlias( tcpStream, alias )
	local conversation = self[tcpStream]
	if conversation == nil then
		return nil
	end
	return conversation[alias]
end

local aliasTable = AliasTable:new()

-- ------------------------------------
-- The EndpointTable Table
local EndpointTable = {}
function EndpointTable:new()
	local result = {}
	setmetatable( result, self )
	self.__index = self
	return result
end
function EndpointTable:add( host, port, server )
	local machine = self[host] or {}
	machine[port] = server
	self[host] = machine
end
function EndpointTable:get( host, port )
	return self[host][port]
end

local clientTable = EndpointTable:new()
local serverTable = EndpointTable:new()

-- -----------------------------------
-- Create and register a listener for TCP connections

local tcpConnections = {}
function tcpConnections:len()
	local result = 0
	local i,v
	for i,v in pairs( self ) do result = result +1 end
	return result
end


-- -----------------------------------
-- Stores information about specific service requests
local ServiceMessageTable = {}
function ServiceMessageTable:new()
	local result = {}
	setmetatable( result, self )
	self.__index = self
	return result
end
-- Add information about a service request
function ServiceMessageTable:addRequest( tcpStream, requestSrc, conversation, time )
	local serviceConversationStream = self[tcpStream] or {}
	local serviceConversation = serviceConversationStream[requestSrc] or {}
	serviceConversation[conversation] = {}
	serviceConversation[conversation].time = time
	serviceConversationStream[requestSrc] = serviceConversation
	self[tcpStream] = serviceConversationStream
end
-- Get the time of a service request
function ServiceMessageTable:getRequestTime( tcpStream, requestSrc, conversation )
	local res = self[tcpStream][requestSrc][conversation]
	return res.time
end

local serviceMessageTable = ServiceMessageTable:new()

local DescriptionsTable = {
	count = 0
}
function DescriptionsTable:new()
	local result = {}
	setmetatable( result, DescriptionsTable )
	DescriptionsTable.__index = DescriptionsTable
	return result
end
function DescriptionsTable:addDescription( description )
	local pos = self.count
	self[pos] = description
	self.count = pos + 1
end
function DescriptionsTable:summarise()
	if self.count == 1 then
		return self[0]
	end

	local desc = string.format( "%d messages", self.count )
	for i = 0, self.count - 1 do
		desc = desc .. " [" .. self[i] .. "]"
		if i < self.count - 1 then
			desc = desc .. ","
		end
	end
	return desc
end

-- -----------------------------------
-- The Topic Info Table

local TopicInfoTable = {}

function TopicInfoTable:new()
	local result = {}
	setmetatable( result, self )
	self.__index = self
	return result
end

function TopicInfoTable:setInfo( tcpStream, id, topicPath, topicDetails )
	-- Get the table for the tcpStream, or create a new one
	local stream = self[tcpStream] or {}
	local info = stream[id] or {}
	info.topicPath = topicPath
	info.alias = string.format ( "!%s", int_to_string( id, 36 ) )
	info.topicDetails = topicDetails
	stream[id] = info
	stream[info.alias] = info
	self[tcpStream] = stream
end

function TopicInfoTable:getTopicPath( tcpStream, id )
	if self[tcpStream] == nil then
		return nil
	end
	if self[tcpStream][id] == nil then
		return nil
	end
	return self[tcpStream][id].topicPath
end

function TopicInfoTable:getTopicDetails( tcpStream, id )
	if self[tcpStream] == nil then
		return nil
	end
	if self[tcpStream][id] == nil then
		return nil
	end
	return self[tcpStream][id].topicDetails
end

local topicInfoTable = TopicInfoTable:new()

local UpdateStreamInfo = {}
function UpdateStreamInfo:new()
	local result = {}
	setmetatable( result, self )
	self.__index = self
	return result
end
function UpdateStreamInfo:recordCreateRequest( tcpStream, conversation, path )
	local updateStream = {
		path = path
	}
	if self[tcpStream] ~= nil then
		self[tcpStream].conversations[conversation] = updateStream
	else
		local conversations = {}
		conversations[conversation] = updateStream
		self[tcpStream] = {
			conversations = conversations,
		}
	end
end
function UpdateStreamInfo:recordCreateResponse( tcpStream, conversation, updateStreamId )
	if self.streams == nil then
		self.streams = {}
	end

	local conn = self[tcpStream]
	if conn ~= nil then
		local updateStream = conn.conversations[conversation]
		if updateStream ~= nil then
			-- Found the update stream creation request

			-- Find the stream information for the partition
			local partition = self.streams[updateStreamId.partition.int]
			if partition == nil then
				partition = {}
				self.streams[updateStreamId.partition.int] = partition
			end

			-- Find the stream information for the generation
			local generation = partition[updateStreamId.generation.int]
			if generation == nil then
				generation = {}
				partition[updateStreamId.generation.int] = generation
			end

			-- Find the stream information for the topic Id
			local topicInfo = generation[updateStreamId.topicId.int]
			if topicInfo == nil then
				topicInfo = {}
				generation[updateStreamId.topicId.int] = topicInfo
			end

			-- Update the stream information for the instance
			topicInfo[updateStreamId.instance.int] = updateStream

			-- Return stream information
			return updateStream
		end
	end
end
function UpdateStreamInfo:getUpdateStream( updateStreamId )
	if self.streams == nil then
		return nil
	end

	-- Find the stream information for the partition
	local partition = self.streams[updateStreamId.partition.int]
	if partition == nil then
		return nil
	end

	-- Find the stream information for the generation
	local generation = partition[updateStreamId.generation.int]
	if generation == nil then
		return nil
	end

	-- Find the stream information for the topic Id
	local topicInfo = generation[updateStreamId.topicId.int]
	if topicInfo == nil then
		return nil
	end

	-- Return stream information
	return topicInfo[updateStreamId.instance.int]
end
local updateStreamTable = UpdateStreamInfo:new()

-- Package footer
master.info = {
	aliasTable = aliasTable,
	tcpConnections = tcpConnections,
	topicInfoTable = topicInfoTable,
	clientTable = clientTable,
	serverTable = serverTable,
	serviceMessageTable = serviceMessageTable,
	DescriptionsTable = DescriptionsTable,
	updateStreamTable = updateStreamTable
}
diffusion = master
return master.info