Initial version

Author: Mons

.gitmodules

@@ -0,0 +1,3 @@
+[submodule "test-run"]
+	path = test-run
+	url = https://github.com/tarantool/test-run.git

package.reload-scm-1.rockspec

@@ -0,0 +1,22 @@
+package = 'package.reload'
+version = 'scm-1'
+source  = {
+    url    = 'git://github.com/Mons/tnt-package-reload.git',
+    branch = 'master',
+}
+description = {
+    summary  = "Module for unloading previously loaded modules",
+    homepage = 'https://github.com/Mons/tnt-package-reload.git',
+    license  = 'BSD',
+}
+dependencies = {
+    'lua >= 5.1'
+}
+build = {
+    type = 'builtin',
+    modules = {
+        ['package.reload'] = 'package/reload.lua'
+    }
+}
+
+-- vim: syntax=lua

package/reload.lua

@@ -0,0 +1,131 @@
+--[[
+
+-- usage --
+
+package.reload:register(object) -> object:destroy()
+package.reload:register(cb)     -> cb()
+
+package.reload:register(cb,...) -> cb(...)
+package.reload:register(ref,cb) -> cb(ref)
+
+]]
+
+local M
+local N = ...
+local fiber = require 'fiber'
+local log = require 'log'
+
+local function numstr( n )
+	if n == 1 then return '1st'
+	elseif n == 2 then return '2nd'
+	elseif n == 3 then return '3rd'
+	else return tostring(n)..'th'
+	end
+end
+
+package.loaded[N] = nil
+
+if not package.reload then
+	M = setmetatable({
+		O      = {};
+		F      = {};
+		C      = {};
+		loaded = {};
+		count  = 1;
+		script = arg[0];
+	},{
+		__tostring = function () return 'package.reload{}' end;
+		__call = function(m,...)
+			dofile(M.script)
+		end;
+	})
+	local loaded = {}
+	for m in pairs(package.loaded) do
+		M.loaded[m] = package.loaded[m]
+		table.insert(loaded,m)
+	end
+	log.info("1st load. loaded: %s",table.concat(loaded, ", "))
+	package.reload = M
+else
+	M = package.reload
+	M:_reload()
+end
+
+function M:_reload()
+	M.count = M.count + 1
+	local unload = {}
+	for m in pairs(package.loaded) do
+		if not M.loaded[m] then			
+			table.insert(unload,m)
+			package.loaded[m] = nil
+		end
+	end
+	log.info("%s load. Unloading {%s}",numstr(M.count),table.concat(unload, ", "))
+	M:cleanup()
+end
+
+function M:cleanup()
+	log.info("%s:cleanup...",N)
+	if self.main then
+		package.loaded[self.main] = nil
+	end
+	collectgarbage()
+	for ref,cb in pairs(self.O) do
+		cb(ref)
+		self.O[ref] = nil
+	end
+	for f in pairs(self.F) do
+		f()
+		self.F[f] = nil
+	end
+	for t in pairs(self.C) do
+		local cb = t[1]
+		table.remove(t,1)
+		cb(unpack(t))
+		self.C[t] = nil
+	end
+	log.info("%s:cleanup finished",N)
+	collectgarbage()
+end
+
+M.deregister = M.cleanup
+
+function M:register(...)
+	assert( self == box.reload, "Static call" )
+	if select('#',...) == 1 then
+		local arg = ...
+		print("one arg ",type(arg))
+		if type(arg) == 'table' then
+			if arg.destroy then
+				self.O[ arg ] = arg.destroy
+			else
+				error("One arg call with object, but have no destroy method")
+			end
+		elseif type(arg) == 'function' then
+			self.F[arg] = arg
+		else
+			error("One arg call with unsupported type: ",type(arg))
+		end
+	else
+		local arg1 = ...
+		if type(arg1) == 'function' then
+			local t = {...}
+			self.C[t] = t
+		elseif select('#',...) == 2 then
+			local ref,cb = ...
+			if type(cb) == 'function' then
+				self.O[ref] = cb
+			else
+				error("Bad arguments")
+			end
+		else
+			error("Bad arguments: ", ...)
+		end
+	end
+end
+fiber.create(function()
+	fiber.sleep(0)
+	package.loaded[N] = nil
+	return
+end)
+return M